Fix failing tests

onaio · Aug 26, 2020 · 6b79642 · 6b79642
1 parent ed5249d
commit 6b79642
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/tally_ho/apps/tally/tests/views/reports/test_administrative_areas_reports.py b/tally_ho/apps/tally/tests/views/reports/test_administrative_areas_reports.py
@@ -450,7 +450,7 @@ def test_centers_list_redirect_for_regions_under_investigation(self):
         self.assertEquals(response.status_code, 302)
         self.assertIn(f'/data/center-list/{tally_id}/{region_id}/',
                       response['location'])
-        self.assertEquals(request.session, {'station_ids': [tally_id]})
+        self.assertEquals(request.session, {'station_ids': [self.station.pk]})
 
     def test_centers_list_redirect_regions_exclude_after_investigation(self):
         """
@@ -483,7 +483,7 @@ def test_centers_list_redirect_regions_exclude_after_investigation(self):
         self.assertEquals(response.status_code, 302)
         self.assertIn(f'/data/center-list/{tally_id}/{region_id}/',
                       response['location'])
-        self.assertEquals(request.session, {'station_ids': [tally_id]})
+        self.assertEquals(request.session, {'station_ids': [self.station.pk]})
 
     def test_candidates_list_redirect_regions_votes_summary_report(self):
         """
@@ -818,4 +818,4 @@ def test_sub_constituency_turn_out_and_votes_summary_reports(self):
             'Sub Constituency votes per candidate')
         self.assertContains(
             response,
-            'Sub Constituency candidates list by ballot order')
+            'Sub Constituency candidates list by ballot order')