Improve functional test.

mzdaniel · Sep 6, 2016 · ef4d04d · ef4d04d
1 parent b4cae4c
commit ef4d04d
Showing 1 changed file with 6 additions and 5 deletions.
diff --git a/tests/test_sphinxserve.py b/tests/test_sphinxserve.py
@@ -106,11 +106,12 @@ def append(line, filename):
             fh.write(line + '\n')
 
     def get_page(url, timestamp):
-        for x in range(10):  # give up to 3 sec to do the test
-            r = get(url)
-            if timestamp != r.headers['last-modified']:
-                break
-            sleep(0.3)
+        with Timeout(5, False):
+            while True:
+                r = get(url)
+                if timestamp != r.headers['last-modified']:
+                    break
+                sleep(0.3)
         return Ret(r.text, timestamp=r.headers['last-modified'])
 
     # Test main header in rendered page