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 735205065c1a5a60f9cc0ce66a4cfc4adf54df40..ede2520e19bb560261b5a70c0b2bcf756f8501dc 100644 |
--- a/tools/telemetry/telemetry/core/tab_unittest.py |
+++ b/tools/telemetry/telemetry/core/tab_unittest.py |
@@ -13,6 +13,26 @@ def _IsDocumentVisible(tab): |
return not tab.EvaluateJavaScript('document.hidden || document.webkitHidden') |
+class FakePlatform(object): |
+ def __init__(self): |
+ self._is_video_capture_running = False |
+ |
+ #pylint: disable=W0613 |
+ def StartVideoCapture(self, min_bitrate_mbps): |
+ self._is_video_capture_running = True |
+ |
+ def StopVideoCapture(self): |
+ self._is_video_capture_running = False |
+ return [] |
+ |
+ def SetFullPerformanceModeEnabled(self, enabled): |
+ pass |
+ |
+ @property |
+ def is_video_capture_running(self): |
+ return self._is_video_capture_running |
+ |
+ |
class TabTest(tab_test_case.TabTestCase): |
def testNavigateAndWaitToForCompleteState(self): |
self._browser.SetHTTPServerDirectories(util.GetUnittestDataDir()) |
@@ -46,6 +66,27 @@ class TabTest(tab_test_case.TabTestCase): |
util.WaitFor(lambda: _IsDocumentVisible(self._tab), timeout=5) |
self.assertFalse(_IsDocumentVisible(new_tab)) |
+ def testIsTimelineRecordingRunningTab(self): |
+ self.assertFalse(self._tab.is_timeline_recording_running) |
+ self._tab.StartTimelineRecording() |
+ self.assertTrue(self._tab.is_timeline_recording_running) |
+ self._tab.StopTimelineRecording() |
+ self.assertFalse(self._tab.is_timeline_recording_running) |
+ |
+ #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) |
+ try: |
+ self._tab.StopVideoCapture().next() |
+ except Exception: |
+ pass |
+ self.assertFalse(self._tab.is_video_capture_running) |
+ self._tab.browser._platform = original_platform |
+ |
class GpuTabTest(tab_test_case.TabTestCase): |
def setUp(self): |