diff --git a/tally_ho/apps/tally/tests/views/test_super_admin.py b/tally_ho/apps/tally/tests/views/test_super_admin.py index d0fd9fa16..b862fdf7f 100644 --- a/tally_ho/apps/tally/tests/views/test_super_admin.py +++ b/tally_ho/apps/tally/tests/views/test_super_admin.py @@ -1160,10 +1160,10 @@ def test_get_result_form_with_duplicate_results(self): # create duplicate final results create_result(result_form_4, result.candidate, self.user, votes) ballot_1_duplicates = views.get_result_form_with_duplicate_results( - ballot=ballot_1.pk, + ballot=ballot_1.number, tally_id=tally.pk) ballot_2_duplicates = views.get_result_form_with_duplicate_results( - ballot=ballot_2.pk, + ballot=ballot_2.number, tally_id=tally.pk) all_duplicates = views.get_result_form_with_duplicate_results( tally_id=tally.pk) @@ -1229,11 +1229,11 @@ def test_duplicate_result_form_view_get(self): request, tally_id=tally.pk, barcode=barcode, - ballot_id=ballot.pk) + ballot_id=ballot.number) response.render() self.assertIn(("{}{}").format( "Duplicate result forms list for ballot: ", - ballot.pk), str(response.content)) + ballot.number), str(response.content)) self.assertIn(("{}{}").format("Result form barcode: ", barcode), str(response.content)) self.assertIn("Send to clearance", str(response.content)) @@ -1277,7 +1277,7 @@ def test_duplicate_result_form_view_duplicate_reviewed_post(self): request = self.factory.post('/', data=data) request.user = self.user configure_messages(request) - response = view(request, tally_id=tally.pk, ballot_id=ballot.pk) + response = view(request, tally_id=tally.pk, ballot_id=ballot.number) result_form_1.reload() result_form_2.reload() @@ -1310,7 +1310,7 @@ def test_duplicate_result_form_view_send_clearance_post(self): request.user = self.user configure_messages(request) request.session = {'result_form': result_form.pk} - response = view(request, tally_id=tally.pk, ballot_id=ballot.pk) + response = view(request, tally_id=tally.pk, ballot_id=ballot.number) result_form.reload() self.assertEqual(response.status_code, 302) @@ -1339,7 +1339,7 @@ def test_duplicate_result_form_view_send_clearance_post(self): request.user = self.user configure_messages(request) request.session = {'result_form': result_form_2.pk} - response = view(request, tally_id=tally.pk, ballot_id=ballot.pk) + response = view(request, tally_id=tally.pk, ballot_id=ballot.number) result_form_2.reload() self.assertNotEqual(result_form_2.form_state, FormState.CLEARANCE) @@ -1384,7 +1384,7 @@ def test_duplicate_result_form_view_send_all_clearance_post(self): request = self.factory.post('/', data=data) request.user = self.user configure_messages(request) - response = view(request, tally_id=tally.pk, ballot_id=ballot.pk) + response = view(request, tally_id=tally.pk, ballot_id=ballot.number) result_form_1.reload() result_form_2.reload() @@ -1432,7 +1432,7 @@ def test_duplicate_archived_result_forms_send_all_clearance_post(self): request = self.factory.post('/', data=data) request.user = self.user configure_messages(request) - response = view(request, tally_id=tally.pk, ballot_id=ballot.pk) + response = view(request, tally_id=tally.pk, ballot_id=ballot.number) result_form_1.reload() result_form_2.reload()