Index: tools/chrome_proxy/integration_tests/chrome_proxy_benchmark.py |
diff --git a/tools/chrome_proxy/integration_tests/chrome_proxy_benchmark.py b/tools/chrome_proxy/integration_tests/chrome_proxy_benchmark.py |
index e1fcb0a10c15af08dc15ed78b8fad1db6bdc42cf..df51b4f48666f73e5c34eb9e34744503a01ff48d 100644 |
--- a/tools/chrome_proxy/integration_tests/chrome_proxy_benchmark.py |
+++ b/tools/chrome_proxy/integration_tests/chrome_proxy_benchmark.py |
@@ -89,7 +89,12 @@ class ChromeProxyBlockOnce(ChromeProxyBenchmark): |
class ChromeProxySafeBrowsingOn(ChromeProxyBenchmark): |
tag = 'safebrowsing_on' |
test = measurements.ChromeProxySafebrowsingOn |
- page_set = pagesets.SafebrowsingPageSet |
+ |
+ # Override CreateUserStorySet so that we can instantiate SafebrowsingPageSet |
+ # with a non default param. |
+ def CreateUserStorySet(self, options): |
+ del options # unused |
+ return pagesets.SafebrowsingPageSet(expect_timeout=True) |
@classmethod |
def Name(cls): |