Attempted fix for flaky test

openedx · Feb 10, 2016 · df2fc66 · df2fc66
1 parent 37b79f9
commit df2fc66
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/tests/integration/test_sizing.py b/tests/integration/test_sizing.py
@@ -3,6 +3,7 @@
 from collections import namedtuple
 import os.path
 from selenium.webdriver.common.keys import Keys
+from selenium.webdriver.support.ui import WebDriverWait
 
 from xblockutils.resources import ResourceLoader
 
@@ -109,6 +110,8 @@ def test_square_image_desktop(self):
 
     def _size_for_mobile(self):
         self.browser.set_window_size(375, 627)  # iPhone 6 viewport size
+        wait = WebDriverWait(self.browser, 2)
+        wait.until(lambda browser: browser.get_window_size()["width"] == 375)
 
     def test_wide_image_mobile(self):
         """ Test the upper, larger, wide image in a mobile-sized window """