Index: tools/telemetry/telemetry/unittest/page_test_test_case.py |
diff --git a/tools/telemetry/telemetry/unittest/page_test_test_case.py b/tools/telemetry/telemetry/unittest/page_test_test_case.py |
index 2fb3ee41a290a24e1395d9b36918ce787ec088c3..c89f31f48a6cba85b258d2104987ec4f43775e5f 100644 |
--- a/tools/telemetry/telemetry/unittest/page_test_test_case.py |
+++ b/tools/telemetry/telemetry/unittest/page_test_test_case.py |
@@ -85,20 +85,21 @@ class PageTestTestCase(unittest.TestCase): |
def __init__(self, *args, **kwargs): |
measurement_class.__init__(self, *args, **kwargs) |
- # Inject fake tracing methods to browser |
+ # Inject fake tracing methods to tracing_controller |
def TabForPage(self, page, browser): |
- ActualStartTracing = browser.StartTracing |
+ ActualStartTracing = browser.platform.tracing_controller.Start |
def FakeStartTracing(*args, **kwargs): |
ActualStartTracing(*args, **kwargs) |
start_tracing_called[0] = True |
raise exceptions.IntentionalException |
browser.StartTracing = FakeStartTracing |
- ActualStopTracing = browser.StopTracing |
+ ActualStopTracing = browser.platform.tracing_controller.Stop |
def FakeStopTracing(*args, **kwargs): |
- ActualStopTracing(*args, **kwargs) |
+ result = ActualStopTracing(*args, **kwargs) |
stop_tracing_called[0] = True |
- browser.StopTracing = FakeStopTracing |
+ return result |
+ browser.platform.tracing_controller.Stop = FakeStopTracing |
return measurement_class.TabForPage(self, page, browser) |