Index: tools/telemetry/telemetry/testing/fakes/__init__.py |
diff --git a/tools/telemetry/telemetry/testing/fakes/__init__.py b/tools/telemetry/telemetry/testing/fakes/__init__.py |
index 2cbfc385fb70f7acfac0ec1150c17d46f4b45d62..a95088a5e50f0604408cfe65da3b2c135e9beabc 100644 |
--- a/tools/telemetry/telemetry/testing/fakes/__init__.py |
+++ b/tools/telemetry/telemetry/testing/fakes/__init__.py |
@@ -23,13 +23,18 @@ from telemetry.testing.internal import fake_gpu_info |
# fakes API. |
class FakePlatform(object): |
+ def __init__(self): |
+ self._network_controller = None |
+ |
@property |
def is_host_platform(self): |
raise NotImplementedError |
@property |
def network_controller(self): |
- return _FakeNetworkController() |
+ if self._network_controller is None: |
+ self._network_controller = _FakeNetworkController() |
+ return self._network_controller |
@property |
def tracing_controller(self): |
@@ -250,12 +255,21 @@ class _FakeCredentials(object): |
class _FakeNetworkController(object): |
- def SetReplayArgs(self, *args, **kwargs): |
+ def __init__(self): |
+ self.wpr_mode = None |
+ |
+ def Open(self, wpr_mode, *args, **kwargs): |
+ self.wpr_mode = wpr_mode |
+ |
+ def StartReplay(self, *args, **kwargs): |
pass |
- def UpdateReplayForExistingBrowser(self): |
+ def StopReplay(self): |
pass |
+ def Close(self): |
+ self.wpr_mode = None |
+ |
class _FakeTab(object): |
def __init__(self, browser, tab_id): |