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 42467aefd40ad1ae5c65bdde0f74ffe804cc03ce..fcafadb2026a22b40329d7b9c35b35a06ee26e38 100644 |
--- a/tools/telemetry/telemetry/page/record_wpr_unittest.py |
+++ b/tools/telemetry/telemetry/page/record_wpr_unittest.py |
@@ -126,7 +126,8 @@ class RecordWprUnitTests(tab_test_case.TabTestCase): |
@decorators.Disabled('chromeos') # crbug.com/404868. |
def testWprRecorderWithPageSet(self): |
- flags = ['--browser', self._browser.browser_type] |
+ flags = ['--browser', self._browser.browser_type, |
+ '--device', self._device] |
mock_page_set = MockPageSet(url=self._url) |
wpr_recorder = record_wpr.WprRecorder(self._test_data_dir, |
mock_page_set, flags) |
@@ -136,7 +137,8 @@ class RecordWprUnitTests(tab_test_case.TabTestCase): |
def testWprRecorderWithBenchmark(self): |
flags = ['--mock-benchmark-url', self._url, |
- '--browser', self._browser.browser_type] |
+ '--browser', self._browser.browser_type, |
+ '--device', self._device] |
mock_benchmark = MockBenchmark() |
wpr_recorder = record_wpr.WprRecorder(self._test_data_dir, mock_benchmark, |
flags) |
@@ -150,6 +152,7 @@ class RecordWprUnitTests(tab_test_case.TabTestCase): |
'--page-set-base-dir', self._test_data_dir, |
'--mock-benchmark-url', self._url, |
'--browser', self._browser.browser_type, |
+ '--device', self._device |
] |
mock_benchmark = MockBenchmark() |
wpr_recorder = record_wpr.WprRecorder( |