| OLD | NEW |
| 1 # Copyright 2013 The Chromium Authors. All rights reserved. | 1 # Copyright 2013 The Chromium Authors. All rights reserved. |
| 2 # Use of this source code is governed by a BSD-style license that can be | 2 # Use of this source code is governed by a BSD-style license that can be |
| 3 # found in the LICENSE file. | 3 # found in the LICENSE file. |
| 4 | 4 |
| 5 import json | 5 import json |
| 6 import os | 6 import os |
| 7 | 7 |
| 8 from telemetry import benchmark | 8 from telemetry import benchmark |
| 9 from telemetry.core import exceptions | 9 from telemetry.core import exceptions |
| 10 from telemetry.core import wpr_modes | 10 from telemetry.core import wpr_modes |
| (...skipping 90 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 101 } | 101 } |
| 102 } | 102 } |
| 103 """) | 103 """) |
| 104 try: | 104 try: |
| 105 ps = page_set.PageSet() | 105 ps = page_set.PageSet() |
| 106 measurement = PageTestForReplay() | 106 measurement = PageTestForReplay() |
| 107 | 107 |
| 108 # First record an archive with only www.google.com. | 108 # First record an archive with only www.google.com. |
| 109 self._options.browser_options.wpr_mode = wpr_modes.WPR_RECORD | 109 self._options.browser_options.wpr_mode = wpr_modes.WPR_RECORD |
| 110 | 110 |
| 111 ps.wpr_archive_info = page_set_archive_info.PageSetArchiveInfo( | 111 # pylint: disable=protected-access |
| 112 ps._wpr_archive_info = page_set_archive_info.PageSetArchiveInfo( |
| 112 '', '', ps.bucket, json.loads(archive_info_template % | 113 '', '', ps.bucket, json.loads(archive_info_template % |
| 113 (test_archive, google_url))) | 114 (test_archive, google_url))) |
| 114 ps.pages = [page_module.Page(google_url, ps)] | 115 ps.pages = [page_module.Page(google_url, ps)] |
| 115 all_results = self.RunMeasurement(measurement, ps, options=self._options) | 116 all_results = self.RunMeasurement(measurement, ps, options=self._options) |
| 116 self.assertEquals(0, len(all_results.failures)) | 117 self.assertEquals(0, len(all_results.failures)) |
| 117 | 118 |
| 118 # Now replay it and verify that google.com is found but foo.com is not. | 119 # Now replay it and verify that google.com is found but foo.com is not. |
| 119 self._options.browser_options.wpr_mode = wpr_modes.WPR_REPLAY | 120 self._options.browser_options.wpr_mode = wpr_modes.WPR_REPLAY |
| 120 | 121 |
| 121 ps.wpr_archive_info = page_set_archive_info.PageSetArchiveInfo( | 122 # pylint: disable=protected-access |
| 123 ps._wpr_archive_info = page_set_archive_info.PageSetArchiveInfo( |
| 122 '', '', ps.bucket, json.loads(archive_info_template % | 124 '', '', ps.bucket, json.loads(archive_info_template % |
| 123 (test_archive, foo_url))) | 125 (test_archive, foo_url))) |
| 124 ps.pages = [page_module.Page(foo_url, ps)] | 126 ps.pages = [page_module.Page(foo_url, ps)] |
| 125 all_results = self.RunMeasurement(measurement, ps, options=self._options) | 127 all_results = self.RunMeasurement(measurement, ps, options=self._options) |
| 126 self.assertEquals(1, len(all_results.failures)) | 128 self.assertEquals(1, len(all_results.failures)) |
| 127 | 129 |
| 128 ps.wpr_archive_info = page_set_archive_info.PageSetArchiveInfo( | 130 # pylint: disable=protected-access |
| 131 ps._wpr_archive_info = page_set_archive_info.PageSetArchiveInfo( |
| 129 '', '', ps.bucket, json.loads(archive_info_template % | 132 '', '', ps.bucket, json.loads(archive_info_template % |
| 130 (test_archive, google_url))) | 133 (test_archive, google_url))) |
| 131 ps.pages = [page_module.Page(google_url, ps)] | 134 ps.pages = [page_module.Page(google_url, ps)] |
| 132 all_results = self.RunMeasurement(measurement, ps, options=self._options) | 135 all_results = self.RunMeasurement(measurement, ps, options=self._options) |
| 133 self.assertEquals(0, len(all_results.failures)) | 136 self.assertEquals(0, len(all_results.failures)) |
| 134 | 137 |
| 135 self.assertTrue(os.path.isfile(test_archive)) | 138 self.assertTrue(os.path.isfile(test_archive)) |
| 136 | 139 |
| 137 finally: | 140 finally: |
| 138 if os.path.isfile(test_archive): | 141 if os.path.isfile(test_archive): |
| 139 os.remove(test_archive) | 142 os.remove(test_archive) |
| 140 | 143 |
| 141 def testRunActions(self): | 144 def testRunActions(self): |
| 142 ps = self.CreateEmptyPageSet() | 145 ps = self.CreateEmptyPageSet() |
| 143 page = PageWithAction('file://blank.html', ps) | 146 page = PageWithAction('file://blank.html', ps) |
| 144 ps.AddPage(page) | 147 ps.AddPage(page) |
| 145 measurement = PageTestWithAction() | 148 measurement = PageTestWithAction() |
| 146 self.RunMeasurement(measurement, ps, options=self._options) | 149 self.RunMeasurement(measurement, ps, options=self._options) |
| 147 self.assertTrue(page.run_test_action_called) | 150 self.assertTrue(page.run_test_action_called) |
| OLD | NEW |