Index: tools/telemetry/telemetry/user_story/user_story_runner_unittest.py |
diff --git a/tools/telemetry/telemetry/user_story/user_story_runner_unittest.py b/tools/telemetry/telemetry/user_story/user_story_runner_unittest.py |
index f628798379f88cdfb8c471d271406b5287710437..c381f465f7ae3dca1b8d0e9c60c2cacf3a57dee7 100644 |
--- a/tools/telemetry/telemetry/user_story/user_story_runner_unittest.py |
+++ b/tools/telemetry/telemetry/user_story/user_story_runner_unittest.py |
@@ -9,9 +9,11 @@ import sys |
from telemetry import benchmark |
from telemetry import user_story |
from telemetry.core import exceptions |
+from telemetry.page import page_set |
from telemetry.page import page_test |
from telemetry.page import test_expectations |
from telemetry.results import results_options |
+from unittest_data import test_simple_one_page_set |
from telemetry.unittest_util import options_for_unittests |
from telemetry.unittest_util import system_stub |
from telemetry.user_story import shared_user_story_state |
@@ -340,3 +342,36 @@ class UserStoryRunnerTest(unittest.TestCase): |
self.assertIn('*RESULT metric: metric= [1,2,3,4] unit', contents) |
finally: |
sys.stdout = real_stdout |
+ |
+ def testCheckArchives(self): |
+ ps = page_set.PageSet() |
nednguyen
2014/12/03 18:27:43
If this test use real page_set, it shouldn't be in
aiolos (Not reviewing)
2014/12/04 00:29:31
Is this something I should be considering still? I
nednguyen
2014/12/04 05:17:31
Yes, my bad. Please ignore the original comment.
|
+ # Empty page set. |
+ self.assertFalse(user_story_runner._CheckArchives(ps, ps.pages)) |
+ ps.AddPageWithDefaultRunNavigate('http://127.0.0.1:62350/blank.html') |
nednguyen
2014/12/03 19:35:59
Is there any reason why you choose 'http://127.0.0
aiolos (Not reviewing)
2014/12/04 00:29:31
I pulled it from an existing test page set. I don'
|
+ # Page set missing archive_data_file. |
+ self.assertFalse(user_story_runner._CheckArchives(ps, ps.pages)) |
+ |
+ ps = page_set.PageSet(archive_data_file='missing_archive_data_file.json') |
+ ps.AddPageWithDefaultRunNavigate('http://127.0.0.1:62350/blank.html') |
+ # Page set missing json file specified in archive_data_file. |
+ self.assertFalse(user_story_runner._CheckArchives(ps, ps.pages)) |
+ |
+ ps = page_set.PageSet(archive_data_file='../../unittest_data/test.json', |
+ bucket=page_set.PUBLIC_BUCKET) |
+ self.assertFalse(user_story_runner._CheckArchives(ps, ps.pages)) |
+ # Empty page set with an archive_data_file which exists. |
+ ps.AddPageWithDefaultRunNavigate('http://127.0.0.1:62350/blank.html') |
+ # Page set with valid archive_data_file. |
+ self.assertTrue(user_story_runner._CheckArchives(ps, ps.pages)) |
+ ps.AddPageWithDefaultRunNavigate('http://www.google.com') |
+ # Page set with an archive_data_file which exists but is missing a page. |
+ self.assertFalse(user_story_runner._CheckArchives(ps, ps.pages)) |
+ |
+ ps = page_set.PageSet( |
+ archive_data_file='../../unittest_data/test_missing_wpr_file.json', |
+ bucket=page_set.PUBLIC_BUCKET) |
+ ps.AddPageWithDefaultRunNavigate('http://127.0.0.1:62350/blank.html') |
+ ps.AddPageWithDefaultRunNavigate('http://www.google.com') |
+ # Page set with an archive_data_file which exists and contains all pages |
+ # but fails to find a wpr file. |
+ self.assertFalse(user_story_runner._CheckArchives(ps, ps.pages)) |