def testCTBenchmarks_missingDataFile(self): for benchmark in self.ct_benchmarks: parser = OptionParser() parser.user_agent = 'mobile' parser.urls_list = self.urls_list parser.use_live_sites = False benchmark.AddBenchmarkCommandLineArgs(parser) # Should fail due to missing archive_data_file. try: benchmark.ProcessCommandLineArgs(None, parser) self.fail('Expected AttributeError') except AttributeError as e: if six.PY2: expected_error = ( "OptionParser instance has no attribute 'archive_data_file'" ) actual_error = e.message else: expected_error = ( "'OptionParser' object has no attribute 'archive_data_file'" ) actual_error = str(e) self.assertEquals(actual_error, expected_error) # Now add an empty archive_data_file. parser.archive_data_file = '' benchmark.ProcessCommandLineArgs(self.mock_parser, parser) self.assertEquals('Please specify --archive-data-file.', self.mock_parser.err_msg)
def testCTBenchmarks_missingUrlsList(self): for benchmark in self.ct_benchmarks: parser = OptionParser() parser.user_agent = 'mobile' parser.archive_data_file = self.archive_data_file benchmark.AddBenchmarkCommandLineArgs(parser) # Should fail due to missing urls_list. try: benchmark.ProcessCommandLineArgs(None, parser) self.fail('Expected AttributeError') except AttributeError as e: if six.PY2: self.assertEquals( "OptionParser instance has no attribute 'urls_list'", str(e)) else: self.assertEquals( "'OptionParser' object has no attribute 'urls_list'", str(e)) # Now add an empty urls_list. parser.urls_list = '' benchmark.ProcessCommandLineArgs(self.mock_parser, parser) self.assertEquals('Please specify --urls-list.', self.mock_parser.err_msg)
def testCTBenchmarks_missingDataFileUseLiveSites(self): for benchmark in self.ct_benchmarks: parser = OptionParser() parser.user_agent = 'mobile' parser.urls_list = self.urls_list parser.use_live_sites = True parser.archive_data_file = None benchmark.AddBenchmarkCommandLineArgs(parser) # Should pass. benchmark.ProcessCommandLineArgs(self.mock_parser, parser) self.assertIsNone(self.mock_parser.err_msg)
def testCTBenchmarks_wrongAgent(self): for benchmark in self.ct_benchmarks: parser = OptionParser() parser.user_agent = 'mobileeeeee' parser.archive_data_file = self.archive_data_file parser.urls_list = self.urls_list benchmark.AddBenchmarkCommandLineArgs(parser) benchmark.ProcessCommandLineArgs(None, parser) try: benchmark.CreateStorySet(parser) self.fail('Expected ValueError') except ValueError, e: self.assertEquals('user_agent mobileeeeee is unrecognized', e.message)
def testCTBenchmarks_wrongAgent(self): for benchmark in self.ct_benchmarks: parser = OptionParser() parser.user_agent = 'mobileeeeee' parser.archive_data_file = self.archive_data_file parser.urls_list = self.urls_list benchmark.AddBenchmarkCommandLineArgs(parser) benchmark.ProcessCommandLineArgs(None, parser) try: benchmark.CreateStorySet(parser) self.fail('Expected ValueError') except ValueError as e: self.assertEquals('user_agent mobileeeeee is unrecognized', e.message)
def testCTBenchmarks_missingUrlsList(self): for benchmark in self.ct_benchmarks: parser = OptionParser() parser.user_agent = 'mobile' parser.archive_data_file = self.archive_data_file benchmark.AddBenchmarkCommandLineArgs(parser) # Should fail due to missing urls_list. try: benchmark.ProcessCommandLineArgs(None, parser) self.fail('Expected AttributeError') except AttributeError, e: self.assertEquals( 'OptionParser instance has no attribute \'urls_list\'', e.message) # Now add an empty urls_list. parser.urls_list = '' benchmark.ProcessCommandLineArgs(self.mock_parser, parser) self.assertEquals('Please specify --urls_list.', self.mock_parser.err_msg)
def testCTBenchmarks_missingDataFile(self): for benchmark in self.ct_benchmarks: parser = OptionParser() parser.user_agent = 'mobile' parser.urls_list = self.urls_list benchmark.AddBenchmarkCommandLineArgs(parser) # Should fail due to missing archive_data_file. try: benchmark.ProcessCommandLineArgs(None, parser) self.fail('Expected AttributeError') except AttributeError, e: self.assertEquals( 'OptionParser instance has no attribute \'archive_data_file\'', e.message) # Now add an empty archive_data_file. parser.archive_data_file = '' benchmark.ProcessCommandLineArgs(self.mock_parser, parser) self.assertEquals('Please specify --archive_data_file.', self.mock_parser.err_msg)
def testCTBenchmarks(self): for benchmark in self.ct_benchmarks: parser = OptionParser() parser.user_agent = 'mobile' parser.archive_data_file = self.archive_data_file parser.urls_list = self.urls_list benchmark.AddBenchmarkCommandLineArgs(parser) benchmark.ProcessCommandLineArgs(None, parser) ct_page_set = benchmark.CreateStorySet(parser) self.assertEquals( len(self.urls_list.split(',')), len(ct_page_set.stories)) self.assertEquals( self.archive_data_file, ct_page_set.archive_data_file) for i in range(len(self.urls_list.split(','))): url = self.urls_list.split(',')[i] story = ct_page_set.stories[i] self.assertEquals(url, story.url) self.assertEquals( self.shared_page_state_class, story.shared_state_class) self.assertEquals(self.archive_data_file, story.archive_data_file)
def testCTBenchmarks(self): for benchmark in self.ct_benchmarks: parser = OptionParser() parser.user_agent = 'mobile' parser.archive_data_file = self.archive_data_file parser.urls_list = self.urls_list benchmark.AddBenchmarkCommandLineArgs(parser) benchmark.ProcessCommandLineArgs(None, parser) ct_page_set = benchmark.CreateStorySet(parser) self.assertEquals(len(self.urls_list.split(',')), len(ct_page_set.stories)) self.assertEquals(self.archive_data_file, ct_page_set.archive_data_file) for i in range(len(self.urls_list.split(','))): url = self.urls_list.split(',')[i] story = ct_page_set.stories[i] self.assertEquals(url, story.url) self.assertEquals(self.shared_page_state_class, story.shared_state_class) self.assertEquals(self.archive_data_file, story.archive_data_file)