Add missing time import

Author: lenacohen

tests/selenium/content_filtering_test.py

@@ -6,6 +6,7 @@
 
 import pbtest
 
+import time
 
 class ContentFilteringTest(pbtest.PBSeleniumTest):
     """Content filtering tests."""
@@ -52,14 +53,17 @@ def test_blocking(self):
         self.assert_block()
 
     # TODO sporadic "Cookie fixture should have set a cookie" failures
-    @pytest.mark.flaky(reruns=5)
+    @pytest.mark.flaky(reruns=7)
     def test_cookieblocking_stops_saving(self):
         self.load_url(f"https://{self.COOKIE_DOMAIN}/")
+        time.sleep(1)
         assert not self.driver.get_cookies(), (
             "Visiting the domain directly does not set a cookie")
 
         self.load_url(self.COOKIE_FIXTURE_URL)
+        time.sleep(1)
         self.load_url(f"https://{self.COOKIE_DOMAIN}/")
+        time.sleep(1)
         assert len(self.driver.get_cookies()) == 1, (
             "Cookie fixture should have set a cookie")
 
@@ -72,15 +76,17 @@ def test_cookieblocking_stops_saving(self):
             "Cookie fixture should have been blocked from setting a cookie")
 
     # TODO sporadic "We should have sent a cookie at this point" failures
-    @pytest.mark.flaky(reruns=5)
+    @pytest.mark.flaky(reruns=7)
     def test_cookieblocking_stops_sending(self):
         self.load_url(self.COOKIE_FIXTURE_URL)
+        time.sleep(1)
         self.wait_for_and_switch_to_frame("iframe[src]", timeout=1)
         self.wait_for_any_text('body')
         assert self.find_el_by_css('body').text == "cookies=0", (
             "No cookies should've been sent to start with")
 
         self.load_url(self.COOKIE_FIXTURE_URL)
+        time.sleep(1)
         self.wait_for_and_switch_to_frame("iframe[src]", timeout=1)
         self.wait_for_any_text('body')
         assert self.find_el_by_css('body').text == "cookies=1", (
@@ -89,6 +95,7 @@ def test_cookieblocking_stops_sending(self):
         self.cookieblock_domain(self.COOKIE_DOMAIN)
 
         self.load_url(self.COOKIE_FIXTURE_URL)
+        time.sleep(1)
         self.wait_for_and_switch_to_frame("iframe[src]", timeout=1)
         self.wait_for_any_text('body')
         assert self.find_el_by_css('body').text == "cookies=0", (
@@ -110,7 +117,7 @@ def test_cookieblocking_base_overwrites_subdomain_block(self):
         self.assert_load()
 
     # @pytest.mark.flaky(reruns=3, condition=pbtest.shim.browser_type == "edge")
-    @pytest.mark.flaky(reruns=5)
+    @pytest.mark.flaky(reruns=7)
     def test_blocking_fp_script_served_from_cookieblocked_cdn(self):
         """Since we have a surrogate script for FingerprintJS served from
         cdn.jsdelivr.net, we need to test surrogation rather than blocking."""

tests/selenium/dnt_test.py

@@ -2,6 +2,7 @@
 
 import json
 import unittest
+import time
 
 import pytest