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)
Beispiel #3
0
    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)
Beispiel #5
0
    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)
Beispiel #9
0
    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)