From 607928922367b6e1e484aa2eef0f1405ac17a913 Mon Sep 17 00:00:00 2001 From: Robert Sparks Date: Mon, 15 Jul 2024 09:43:22 -0500 Subject: [PATCH] fix: Improve ballot clearing tests --- ietf/doc/tests_status_change.py | 12 ++++++++++-- 1 file changed, 10 insertions(+), 2 deletions(-) diff --git a/ietf/doc/tests_status_change.py b/ietf/doc/tests_status_change.py index c69abd4bfe..bec48ed4ef 100644 --- a/ietf/doc/tests_status_change.py +++ b/ietf/doc/tests_status_change.py @@ -495,10 +495,14 @@ def test_clear_ballot(self): doc.relateddocument_set.create(target=Document.objects.get(name='rfc9998'),relationship_id='tohist') create_ballot_if_not_open(None, doc, Person.objects.get(user__username="secretary"), "statchg") doc.set_state(State.objects.get(slug='iesgeval',type='statchg')) + old_ballot = doc.ballot_open("statchg") + self.assertIsNotNone(old_ballot) r = self.client.post(url, dict()) self.assertEqual(r.status_code,302) - self.assertTrue(doc.ballot_open("statchg")) + new_ballot = doc.ballot_open("statchg") + self.assertIsNotNone(new_ballot) + self.assertNotEqual(new_ballot, old_ballot) self.assertEqual(doc.get_state_slug("statchg"),"iesgeval") def test_clear_deferred_ballot(self): @@ -511,10 +515,14 @@ def test_clear_deferred_ballot(self): doc.relateddocument_set.create(target=Document.objects.get(name='rfc9998'),relationship_id='tohist') create_ballot_if_not_open(None, doc, Person.objects.get(user__username="secretary"), "statchg") doc.set_state(State.objects.get(slug='defer',type='statchg')) + old_ballot = doc.ballot_open("statchg") + self.assertIsNotNone(old_ballot) r = self.client.post(url, dict()) self.assertEqual(r.status_code,302) - self.assertTrue(doc.ballot_open("statchg")) + new_ballot = doc.ballot_open("statchg") + self.assertIsNotNone(new_ballot) + self.assertNotEqual(new_ballot, old_ballot) self.assertEqual(doc.get_state_slug("statchg"),"iesgeval") def setUp(self):