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)