|
13 | 13 |
|
14 | 14 | from ietf.person.models import Person, Email |
15 | 15 | from ietf.group.models import Group, Role |
16 | | -from ietf.doc.models import Document, BallotPositionDocEvent |
| 16 | +from ietf.doc.models import Document, BallotDocEvent, BallotPositionDocEvent |
17 | 17 | from ietf.submit.models import IdSubmissionDetail |
18 | 18 |
|
19 | 19 | class SubmitTestCase(django.test.TestCase): |
@@ -159,6 +159,7 @@ def test_submit_existing(self): |
159 | 159 |
|
160 | 160 | # make a discuss to see if the AD gets an email |
161 | 161 | ballot_position = BallotPositionDocEvent() |
| 162 | + ballot_position.ballot = draft.latest_event(BallotDocEvent, type="created_ballot") |
162 | 163 | ballot_position.pos_id = "discuss" |
163 | 164 | ballot_position.type = "changed_ballot_position" |
164 | 165 | ballot_position.doc = draft |
@@ -229,7 +230,7 @@ def test_submit_existing(self): |
229 | 230 | self.assertTrue(name in unicode(outbox[-2])) |
230 | 231 | self.assertTrue("mars" in unicode(outbox[-2])) |
231 | 232 | self.assertTrue(draft.ad.role_email("ad").address in unicode(outbox[-2])) |
232 | | - self.assertTrue(ballot_position.ad.email_address() in unicode(outbox[-2])) |
| 233 | + self.assertTrue(ballot_position.ad.role_email("ad").address in unicode(outbox[-2])) |
233 | 234 | self.assertTrue("New Version Notification" in outbox[-1]["Subject"]) |
234 | 235 | self.assertTrue(name in unicode(outbox[-1])) |
235 | 236 | self.assertTrue("mars" in unicode(outbox[-1])) |
|
0 commit comments