Index: tools/telemetry/telemetry/page/record_wpr_unittest.py |
diff --git a/tools/telemetry/telemetry/page/record_wpr_unittest.py b/tools/telemetry/telemetry/page/record_wpr_unittest.py |
index a434fa9622837dd38f4f3f3a1743a36327237598..5a1710df5d07c07756f7989982b6642eb45b7e98 100644 |
--- a/tools/telemetry/telemetry/page/record_wpr_unittest.py |
+++ b/tools/telemetry/telemetry/page/record_wpr_unittest.py |
@@ -148,7 +148,7 @@ class RecordWprUnitTests(tab_test_case.TabTestCase): |
flags = [] |
mock_page_set = MockPageSet(url=self._url) |
wpr_recorder = record_wpr.WprRecorder(self._test_data_dir, |
- mock_page_set, flags) |
+ mock_page_set, 'run', flags) |
results = wpr_recorder.CreateResults() |
wpr_recorder.Record(results) |
self.assertEqual(set(mock_page_set.pages), results.pages_that_succeeded) |
@@ -157,7 +157,7 @@ class RecordWprUnitTests(tab_test_case.TabTestCase): |
flags = ['--mock-benchmark-url', self._url] |
mock_benchmark = MockBenchmark() |
wpr_recorder = record_wpr.WprRecorder(self._test_data_dir, mock_benchmark, |
- flags) |
+ 'run', flags) |
results = wpr_recorder.CreateResults() |
wpr_recorder.Record(results) |
self.assertEqual(set(mock_benchmark.mock_page_set.pages), |
@@ -170,7 +170,7 @@ class RecordWprUnitTests(tab_test_case.TabTestCase): |
'--mock-page-test-option', |
] |
wpr_recorder = record_wpr.WprRecorder(self._test_data_dir, MockBenchmark(), |
- flags) |
+ 'run', flags) |
# page_runner command-line args |
self.assertEquals(2, wpr_recorder.options.page_repeat) |
# benchmark command-line args |
@@ -183,7 +183,7 @@ class RecordWprUnitTests(tab_test_case.TabTestCase): |
def testRecordingEnabled(self): |
flags = ['--mock-benchmark-url', self._url] |
wpr_recorder = record_wpr.WprRecorder(self._test_data_dir, MockBenchmark(), |
- flags) |
+ 'run', flags) |
self.assertEqual(wpr_modes.WPR_RECORD, |
wpr_recorder.options.browser_options.wpr_mode) |