fix flaky test

artoonie · Jul 27, 2024 · 5db72ff · 5db72ff
1 parent b25b3c7
commit 5db72ff
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/electionpage/tests.py b/electionpage/tests.py
@@ -118,8 +118,8 @@ def test_election_pages(self):
         self.assertEqual(bargraphButton.get_attribute('class'), 'btn btn-primary')
 
         # And the height was correctly set via PostMessages
-        self.assertEqual(bargraphIframeWrapper.find_element(
-            By.TAG_NAME, 'iframe').get_attribute('height'), '328px')
+        self._ensure_eventually_asserts(lambda: self.assertEqual(bargraphIframeWrapper.find_element(
+            By.TAG_NAME, 'iframe').get_attribute('height'), '328px'))
 
     @Mocker()
     def test_scrape_all(self, requestMock):