def testWprRecorderWithPageSet(self):
flags = self.GetBrowserDeviceFlags()
mock_story_set = MockStorySet(url=self._url)
wpr_recorder = record_wpr.WprRecorder(self._test_data_dir,
mock_story_set, flags)
results = wpr_recorder.CreateResults()
wpr_recorder.Record(results)
self.assertEqual(set(mock_story_set.stories), results.pages_that_succeeded)
def testWprRecorderWithPageSet(self):
flags = self.GetBrowserDeviceFlags()
mock_story_set = MockStorySet(url=self._url)
with record_wpr.WprRecorder(self._test_data_dir, mock_story_set,
flags) as wpr_recorder:
results = wpr_recorder.CreateResults()
wpr_recorder.Record(results)
self.assertItemsEqual(mock_story_set.stories,
_SuccessfulStories(results))
def testWprRecorderWithPageSet(self):
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)
results = wpr_recorder.CreateResults()
wpr_recorder.Record(results)
self.assertEqual(set(mock_page_set.pages), results.pages_that_succeeded)
def testWprRecorderWithTimelineBasedMeasurementBenchmark(self):
flags = self.GetBrowserDeviceFlags()
flags.extend(['--mock-benchmark-url', self._url])
mock_benchmark = MockTimelineBasedMeasurementBenchmark()
with record_wpr.WprRecorder(self._test_data_dir, mock_benchmark,
flags) as wpr_recorder:
results = wpr_recorder.CreateResults()
wpr_recorder.Record(results)
self.assertItemsEqual(mock_benchmark.mock_story_set.stories,
_SuccessfulStories(results))
def testWprRecorderWithTimelineBasedMeasurementBenchmark(self):
flags = self.GetBrowserDeviceFlags()
flags.extend(['--mock-benchmark-url', self._url])
mock_benchmark = MockTimelineBasedMeasurementBenchmark()
wpr_recorder = record_wpr.WprRecorder(self._test_data_dir,
mock_benchmark, flags)
results = wpr_recorder.CreateResults()
wpr_recorder.Record(results)
self.assertEqual(set(mock_benchmark.mock_story_set.stories),
results.pages_that_succeeded)
def testWprRecorderWithBenchmark(self):
flags = ['--mock-benchmark-url', self._url,
'--browser', self._browser.browser_type,
'--device', self._device]
mock_benchmark = MockBenchmark()
wpr_recorder = record_wpr.WprRecorder(self._test_data_dir, mock_benchmark,
flags)
results = wpr_recorder.CreateResults()
wpr_recorder.Record(results)
self.assertEqual(set(mock_benchmark.mock_page_set.pages),
results.pages_that_succeeded)
def testPageSetBaseDirFlag(self):
flags = self.GetBrowserDeviceFlags()
flags.extend(['--page-set-base-dir', self._test_data_dir,
'--mock-benchmark-url', self._url])
mock_benchmark = MockBenchmark()
wpr_recorder = record_wpr.WprRecorder(
'non-existent-dummy-dir', mock_benchmark, flags)
results = wpr_recorder.CreateResults()
wpr_recorder.Record(results)
self.assertEqual(set(mock_benchmark.mock_story_set.stories),
results.pages_that_succeeded)
def testWprRecorderWithBenchmark(self):
flags = self.GetBrowserDeviceFlags()
flags.extend(['--mock-benchmark-url', self._url])
mock_benchmark = MockBenchmark()
wpr_recorder = record_wpr.WprRecorder(self._test_data_dir,
mock_benchmark, flags)
wpr_recorder._options.output_dir = None
results = wpr_recorder.CreateResults()
wpr_recorder.Record(results)
self.assertItemsEqual(mock_benchmark.mock_story_set.stories,
_SuccessfulStories(results))
def testRecorderPageTest_BrowserMethods(self):
flags = ['--mock-benchmark-url', self._url]
page_test = MockBenchmark().test()
record_page_test = record_wpr.RecorderPageTest(page_test)
with record_wpr.WprRecorder(self._test_data_dir, MockBenchmark(),
flags) as wpr_recorder:
record_page_test.CustomizeBrowserOptions(wpr_recorder.options)
record_page_test.WillStartBrowser(self._tab.browser.platform)
record_page_test.DidStartBrowser(self._tab.browser)
self.assertTrue('CustomizeBrowserOptions' in page_test.func_calls)
self.assertTrue('WillStartBrowser' in page_test.func_calls)
self.assertTrue('DidStartBrowser' in page_test.func_calls)
def testPageSetBaseDirFlag(self):
flags = self.GetBrowserDeviceFlags()
flags.extend([
'--page-set-base-dir', self._test_data_dir, '--mock-benchmark-url',
self._url
])
mock_benchmark = MockBenchmark()
with record_wpr.WprRecorder('non-existent-dummy-dir', mock_benchmark,
flags) as wpr_recorder:
results = wpr_recorder.CreateResults()
wpr_recorder.Record(results)
self.assertItemsEqual(mock_benchmark.mock_story_set.stories,
_SuccessfulStories(results))
def testPageSetBaseDirFlag(self):
flags = [
'--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('non-existent-dummy-dir',
mock_benchmark, flags)
results = wpr_recorder.CreateResults()
wpr_recorder.Record(results)
self.assertEqual(set(mock_benchmark.mock_page_set.pages),
results.pages_that_succeeded)
def testCommandLineFlags(self):
flags = [
'--page-repeat', '2',
'--mock-benchmark-url', self._url,
'--upload',
]
wpr_recorder = record_wpr.WprRecorder(self._test_data_dir, MockBenchmark(),
flags)
# page_runner command-line args
self.assertEquals(2, wpr_recorder.options.page_repeat)
# benchmark command-line args
self.assertEquals(self._url, wpr_recorder.options.mock_benchmark_url)
# record_wpr command-line arg to upload to cloud-storage.
self.assertTrue(wpr_recorder.options.upload)
# invalid command-line args
self.assertFalse(hasattr(wpr_recorder.options, 'not_a_real_option'))
def testUseLiveSitesUnsupported(self):
flags = ['--use-live-sites']
with self.assertRaises(SystemExit):
record_wpr.WprRecorder(self._test_data_dir, MockBenchmark(), flags)
def testRecordingEnabled(self):
flags = ['--mock-benchmark-url', self._url]
with record_wpr.WprRecorder(self._test_data_dir, MockBenchmark(),
flags) as wpr_recorder:
self.assertEqual(wpr_modes.WPR_RECORD,
wpr_recorder.options.browser_options.wpr_mode)
