Update To 11.40.268.0

[platform/framework/web/crosswalk.git] / src / tools / telemetry / telemetry / page / record_wpr_unittest.py

diff --git a/src/tools/telemetry/telemetry/page/record_wpr_unittest.py b/src/tools/telemetry/telemetry/page/record_wpr_unittest.py
index 75e79b2..0197451 100644 (file)
--- a/src/tools/telemetry/telemetry/page/record_wpr_unittest.py
+++ b/src/tools/telemetry/telemetry/page/record_wpr_unittest.py
@@ -145,7 +145,7 @@ class RecordWprUnitTests(tab_test_case.TabTestCase):
 
   @decorators.Disabled('chromeos') # crbug.com/404868.
   def testWprRecorderWithPageSet(self):
-    flags = []
+    flags = ['--browser', self._browser.browser_type]
     mock_page_set = MockPageSet(url=self._url)
     wpr_recorder = record_wpr.WprRecorder(self._test_data_dir,
                                           mock_page_set, flags)
@@ -154,7 +154,8 @@ class RecordWprUnitTests(tab_test_case.TabTestCase):
     self.assertEqual(set(mock_page_set.pages), results.pages_that_succeeded)
 
   def testWprRecorderWithBenchmark(self):
-    flags = ['--mock-benchmark-url', self._url]
+    flags = ['--mock-benchmark-url', self._url,
+             '--browser', self._browser.browser_type]
     mock_benchmark = MockBenchmark()
     wpr_recorder = record_wpr.WprRecorder(self._test_data_dir, mock_benchmark,
                                           flags)
@@ -163,6 +164,20 @@ class RecordWprUnitTests(tab_test_case.TabTestCase):
     self.assertEqual(set(mock_benchmark.mock_page_set.pages),
                      results.pages_that_succeeded)
 
+  def testPageSetBaseDirFlag(self):
+    flags = [
+       '--page-set-base-dir', self._test_data_dir,
+       '--mock-benchmark-url', self._url,
+       '--browser', self._browser.browser_type,
+    ]
+    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',