Index: tools/telemetry/telemetry/core/tab_unittest.py |
diff --git a/tools/telemetry/telemetry/core/tab_unittest.py b/tools/telemetry/telemetry/core/tab_unittest.py |
index 1df26236334baf972b0b75a28c312b5fa31955e8..bf9250edc897154a5d3e150c6574c4ce6a583ddc 100644 |
--- a/tools/telemetry/telemetry/core/tab_unittest.py |
+++ b/tools/telemetry/telemetry/core/tab_unittest.py |
@@ -19,6 +19,10 @@ def _IsDocumentVisible(tab): |
return not tab.EvaluateJavaScript('document.hidden || document.webkitHidden') |
+class FakePlatformBackend(object): |
+ def __init__(self): |
+ self.platform = FakePlatform() |
+ |
class FakePlatform(object): |
def __init__(self): |
self._is_video_capture_running = False |
@@ -83,14 +87,16 @@ class TabTest(tab_test_case.TabTestCase): |
#pylint: disable=W0212 |
def testIsVideoCaptureRunning(self): |
- original_platform = self._tab.browser._platform |
- self._tab.browser._platform = FakePlatform() |
- self.assertFalse(self._tab.is_video_capture_running) |
- self._tab.StartVideoCapture(min_bitrate_mbps=2) |
- self.assertTrue(self._tab.is_video_capture_running) |
- self.assertIsNotNone(self._tab.StopVideoCapture()) |
- self.assertFalse(self._tab.is_video_capture_running) |
- self._tab.browser._platform = original_platform |
+ original_platform_backend = self._tab.browser._platform_backend |
+ try: |
+ self._tab.browser._platform_backend = FakePlatformBackend() |
+ self.assertFalse(self._tab.is_video_capture_running) |
+ self._tab.StartVideoCapture(min_bitrate_mbps=2) |
+ self.assertTrue(self._tab.is_video_capture_running) |
+ self.assertIsNotNone(self._tab.StopVideoCapture()) |
+ self.assertFalse(self._tab.is_video_capture_running) |
+ finally: |
+ self._tab.browser._platform_backend = original_platform_backend |
def testHighlight(self): |
self.assertEquals(self._tab.url, 'about:blank') |