Revert of telemetry: Remove command line args from page test

tools/telemetry/telemetry/page/record_wpr_unittest.py

Index: tools/telemetry/telemetry/page/record_wpr_unittest.py
diff --git a/tools/telemetry/telemetry/page/record_wpr_unittest.py b/tools/telemetry/telemetry/page/record_wpr_unittest.py
index b88bfd89c94fa432d3b9c53df72deb5acd593664..49cf7dcab74822c90ea903db38567e2d7db2a4c2 100644
--- a/tools/telemetry/telemetry/page/record_wpr_unittest.py
+++ b/tools/telemetry/telemetry/page/record_wpr_unittest.py
@@ -49,6 +49,10 @@
     self._action_name_to_run = "RunBaz"
     self.func_calls = []
 
+  @classmethod
+  def AddCommandLineArgs(cls, parser):
+    parser.add_option('--mock-page-test-option', action="store_true")
+
   def WillNavigateToPage(self, page, tab):
     self.func_calls.append('WillNavigateToPage')
 
@@ -75,7 +79,7 @@
   mock_page_set = None
 
   @classmethod
-  def AddBenchmarkCommandLineArgs(cls, group):
+  def AddTestCommandLineArgs(cls, group):
     group.add_option('', '--mock-benchmark-url', action='store', type='string')
 
   def CreatePageSet(self, options):
@@ -178,6 +182,7 @@
     flags = [
         '--page-repeat', '2',
         '--mock-benchmark-url', self._url,
+        '--mock-page-test-option',
     ]
     wpr_recorder = record_wpr.WprRecorder(self._test_data_dir, MockBenchmark(),
                                           flags)
@@ -185,6 +190,8 @@
     self.assertEquals(2, wpr_recorder.options.page_repeat)
     # benchmark command-line args
     self.assertEquals(self._url, wpr_recorder.options.mock_benchmark_url)
+    # benchmark's page_test command-line args
+    self.assertTrue(wpr_recorder.options.mock_page_test_option)
     # invalid command-line args
     self.assertFalse(hasattr(wpr_recorder.options, 'not_a_real_option'))