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 75ae0160b37f397bc6fafa8522b45f924dd74057..3da7384a5477a9bc5e860fb918f2720ba1717f83 100644 |
--- a/tools/telemetry/telemetry/page/record_wpr_unittest.py |
+++ b/tools/telemetry/telemetry/page/record_wpr_unittest.py |
@@ -144,7 +144,8 @@ class RecordWprUnitTests(tab_test_case.TabTestCase): |
mock_page_set = MockPageSet(url=self._url) |
wpr_recorder = record_wpr.WprRecorder(self._test_data_dir, |
mock_page_set, flags) |
- results = wpr_recorder.Record() |
+ results = wpr_recorder.CreateResults() |
+ wpr_recorder.Record(results) |
self.assertEqual(set(mock_page_set.pages), results.pages_that_succeeded) |
def testWprRecorderWithBenchmark(self): |
@@ -152,7 +153,8 @@ class RecordWprUnitTests(tab_test_case.TabTestCase): |
mock_benchmark = MockBenchmark() |
wpr_recorder = record_wpr.WprRecorder(self._test_data_dir, mock_benchmark, |
flags) |
- results = wpr_recorder.Record() |
+ results = wpr_recorder.CreateResults() |
+ wpr_recorder.Record(results) |
self.assertEqual(set(mock_benchmark.mock_page_set.pages), |
results.pages_that_succeeded) |