def testMissingArchiveDataFile(self): story_set = test_stories.DummyStorySet(['story']) with self.assertRaises(story_runner.ArchiveError): story_runner._UpdateAndCheckArchives(story_set.archive_data_file, story_set.wpr_archive_info, story_set.stories, self._mock_story_filter)
def testArchiveDataFileDoesNotExist(self): story_set = test_stories.DummyStorySet( ['story'], archive_data_file='does_not_exist.json') with self.assertRaises(story_runner.ArchiveError): story_runner._UpdateAndCheckArchives( story_set.archive_data_file, story_set.wpr_archive_info, story_set.stories, self._mock_story_filter)
def testArchiveWithMissingStory(self): # This test file has a recording for a 'http://www.testurl.com' story only. archive_data_file = os.path.join( util.GetUnittestDataDir(), 'archive_files', 'test.json') story_set = test_stories.DummyStorySet( ['http://www.testurl.com', 'http://www.google.com'], archive_data_file=archive_data_file) with self.assertRaises(story_runner.ArchiveError): story_runner._UpdateAndCheckArchives( story_set.archive_data_file, story_set.wpr_archive_info, story_set.stories, self._mock_story_filter)
def testArchiveWithMissingWprFile(self): # This test file claims to have recordings for both # 'http://www.testurl.com' and 'http://www.google.com'; but the file with # the wpr recording for the later story is actually missing. archive_data_file = os.path.join( util.GetUnittestDataDir(), 'archive_files', 'test_missing_wpr_file.json') story_set = test_stories.DummyStorySet( ['http://www.testurl.com', 'http://www.google.com'], archive_data_file=archive_data_file) with self.assertRaises(story_runner.ArchiveError): story_runner._UpdateAndCheckArchives( story_set.archive_data_file, story_set.wpr_archive_info, story_set.stories, self._mock_story_filter)
def testMissingArchiveDataFileWithSkippedStory(self): story_set = test_stories.DummyStorySet(['story']) self._mock_story_filter.ShouldSkip.return_value = True success = story_runner._UpdateAndCheckArchives( story_set.archive_data_file, story_set.wpr_archive_info, story_set.stories, self._mock_story_filter) self.assertTrue(success)
def testUpdateAndCheckArchivesSuccess(self): # This test file has a recording for a 'http://www.testurl.com' story only. archive_data_file = os.path.join(util.GetUnittestDataDir(), 'archive_files', 'test.json') story_set = test_stories.DummyStorySet( ['http://www.testurl.com'], archive_data_file=archive_data_file) success = story_runner._UpdateAndCheckArchives( story_set.archive_data_file, story_set.wpr_archive_info, story_set.stories, self._mock_story_filter) self.assertTrue(success)
def testUpdateAndCheckArchives(self): usr_stub = system_stub.Override(story_runner, ['cloud_storage']) wpr_stub = system_stub.Override(archive_info, ['cloud_storage']) try: uss = user_story_set.UserStorySet() uss.AddUserStory(page_module.Page( 'http://www.testurl.com', uss, uss.base_dir)) # Page set missing archive_data_file. self.assertRaises( story_runner.ArchiveError, story_runner._UpdateAndCheckArchives, uss.archive_data_file, uss.wpr_archive_info, uss.user_stories) uss = user_story_set.UserStorySet( archive_data_file='missing_archive_data_file.json') uss.AddUserStory(page_module.Page( 'http://www.testurl.com', uss, uss.base_dir)) # Page set missing json file specified in archive_data_file. self.assertRaises( story_runner.ArchiveError, story_runner._UpdateAndCheckArchives, uss.archive_data_file, uss.wpr_archive_info, uss.user_stories) uss = user_story_set.UserStorySet( archive_data_file='../../unittest_data/archive_files/test.json', cloud_storage_bucket=cloud_storage.PUBLIC_BUCKET) uss.AddUserStory(page_module.Page( 'http://www.testurl.com', uss, uss.base_dir)) # Page set with valid archive_data_file. self.assertTrue(story_runner._UpdateAndCheckArchives( uss.archive_data_file, uss.wpr_archive_info, uss.user_stories)) uss.AddUserStory(page_module.Page( 'http://www.google.com', uss, uss.base_dir)) # Page set with an archive_data_file which exists but is missing a page. self.assertRaises( story_runner.ArchiveError, story_runner._UpdateAndCheckArchives, uss.archive_data_file, uss.wpr_archive_info, uss.user_stories) uss = user_story_set.UserStorySet( archive_data_file='../../unittest_data/test_missing_wpr_file.json', cloud_storage_bucket=cloud_storage.PUBLIC_BUCKET) uss.AddUserStory(page_module.Page( 'http://www.testurl.com', uss, uss.base_dir)) uss.AddUserStory(page_module.Page( 'http://www.google.com', uss, uss.base_dir)) # Page set with an archive_data_file which exists and contains all pages # but fails to find a wpr file. self.assertRaises( story_runner.ArchiveError, story_runner._UpdateAndCheckArchives, uss.archive_data_file, uss.wpr_archive_info, uss.user_stories) finally: usr_stub.Restore() wpr_stub.Restore()
def testUpdateAndCheckArchives(self): usr_stub = system_stub.Override(story_runner, ['cloud_storage']) wpr_stub = system_stub.Override(archive_info, ['cloud_storage']) archive_data_dir = os.path.join( util.GetTelemetryDir(), 'telemetry', 'internal', 'testing', 'archive_files') try: story_set = story_module.StorySet() story_set.AddStory(page_module.Page( 'http://www.testurl.com', story_set, story_set.base_dir)) # Page set missing archive_data_file. self.assertRaises( story_runner.ArchiveError, story_runner._UpdateAndCheckArchives, story_set.archive_data_file, story_set.wpr_archive_info, story_set.stories) story_set = story_module.StorySet( archive_data_file='missing_archive_data_file.json') story_set.AddStory(page_module.Page( 'http://www.testurl.com', story_set, story_set.base_dir)) # Page set missing json file specified in archive_data_file. self.assertRaises( story_runner.ArchiveError, story_runner._UpdateAndCheckArchives, story_set.archive_data_file, story_set.wpr_archive_info, story_set.stories) story_set = story_module.StorySet( archive_data_file=os.path.join(archive_data_dir, 'test.json'), cloud_storage_bucket=cloud_storage.PUBLIC_BUCKET) story_set.AddStory(page_module.Page( 'http://www.testurl.com', story_set, story_set.base_dir)) # Page set with valid archive_data_file. self.assertTrue(story_runner._UpdateAndCheckArchives( story_set.archive_data_file, story_set.wpr_archive_info, story_set.stories)) story_set.AddStory(page_module.Page( 'http://www.google.com', story_set, story_set.base_dir)) # Page set with an archive_data_file which exists but is missing a page. self.assertRaises( story_runner.ArchiveError, story_runner._UpdateAndCheckArchives, story_set.archive_data_file, story_set.wpr_archive_info, story_set.stories) story_set = story_module.StorySet( archive_data_file= os.path.join(archive_data_dir, 'test_missing_wpr_file.json'), cloud_storage_bucket=cloud_storage.PUBLIC_BUCKET) story_set.AddStory(page_module.Page( 'http://www.testurl.com', story_set, story_set.base_dir)) story_set.AddStory(page_module.Page( 'http://www.google.com', story_set, story_set.base_dir)) # Page set with an archive_data_file which exists and contains all pages # but fails to find a wpr file. self.assertRaises( story_runner.ArchiveError, story_runner._UpdateAndCheckArchives, story_set.archive_data_file, story_set.wpr_archive_info, story_set.stories) finally: usr_stub.Restore() wpr_stub.Restore()
def testUpdateAndCheckArchives(self): usr_stub = system_stub.Override(story_runner, ['cloud_storage']) wpr_stub = system_stub.Override(archive_info, ['cloud_storage']) archive_data_dir = os.path.join(util.GetTelemetryDir(), 'telemetry', 'internal', 'testing', 'archive_files') try: story_set = story_module.StorySet() story_set.AddStory( page_module.Page('http://www.testurl.com', story_set, story_set.base_dir)) # Page set missing archive_data_file. self.assertRaises(story_runner.ArchiveError, story_runner._UpdateAndCheckArchives, story_set.archive_data_file, story_set.wpr_archive_info, story_set.stories) story_set = story_module.StorySet( archive_data_file='missing_archive_data_file.json') story_set.AddStory( page_module.Page('http://www.testurl.com', story_set, story_set.base_dir)) # Page set missing json file specified in archive_data_file. self.assertRaises(story_runner.ArchiveError, story_runner._UpdateAndCheckArchives, story_set.archive_data_file, story_set.wpr_archive_info, story_set.stories) story_set = story_module.StorySet( archive_data_file=os.path.join(archive_data_dir, 'test.json'), cloud_storage_bucket=cloud_storage.PUBLIC_BUCKET) story_set.AddStory( page_module.Page('http://www.testurl.com', story_set, story_set.base_dir)) # Page set with valid archive_data_file. self.assertTrue( story_runner._UpdateAndCheckArchives( story_set.archive_data_file, story_set.wpr_archive_info, story_set.stories)) story_set.AddStory( page_module.Page('http://www.google.com', story_set, story_set.base_dir)) # Page set with an archive_data_file which exists but is missing a page. self.assertRaises(story_runner.ArchiveError, story_runner._UpdateAndCheckArchives, story_set.archive_data_file, story_set.wpr_archive_info, story_set.stories) story_set = story_module.StorySet( archive_data_file=os.path.join(archive_data_dir, 'test_missing_wpr_file.json'), cloud_storage_bucket=cloud_storage.PUBLIC_BUCKET) story_set.AddStory( page_module.Page('http://www.testurl.com', story_set, story_set.base_dir)) story_set.AddStory( page_module.Page('http://www.google.com', story_set, story_set.base_dir)) # Page set with an archive_data_file which exists and contains all pages # but fails to find a wpr file. self.assertRaises(story_runner.ArchiveError, story_runner._UpdateAndCheckArchives, story_set.archive_data_file, story_set.wpr_archive_info, story_set.stories) finally: usr_stub.Restore() wpr_stub.Restore()