Index: tools/telemetry/telemetry/page/page_set_archive_info_unittest.py |
diff --git a/tools/telemetry/telemetry/page/page_set_archive_info_unittest.py b/tools/telemetry/telemetry/page/page_set_archive_info_unittest.py |
index 1ee68a19aa15d2d1d8340b0bf2a887efdaa78b59..bcb48370192f42e70ee3c5e9acc022ccd47c650e 100644 |
--- a/tools/telemetry/telemetry/page/page_set_archive_info_unittest.py |
+++ b/tools/telemetry/telemetry/page/page_set_archive_info_unittest.py |
@@ -48,7 +48,7 @@ class TestPageSetArchiveInfo(unittest.TestCase): |
# Create the PageSetArchiveInfo object to be tested. |
self.archive_info = page_set_archive_info.PageSetArchiveInfo.FromFile( |
- self.page_set_archive_info_file) |
+ self.page_set_archive_info_file, cloud_storage.PUBLIC_BUCKET) |
def tearDown(self): |
shutil.rmtree(self.tmp_dir) |
@@ -157,7 +157,7 @@ class TestPageSetArchiveInfo(unittest.TestCase): |
# Create the PageSetArchiveInfo object to be tested. |
self.archive_info = page_set_archive_info.PageSetArchiveInfo.FromFile( |
- self.page_set_archive_info_file) |
+ self.page_set_archive_info_file, cloud_storage.PUBLIC_BUCKET) |
# Add a recording for all the pages. |
new_temp_recording = os.path.join(self.tmp_dir, 'recording.wpr') |
@@ -181,6 +181,6 @@ class TestPageSetArchiveInfo(unittest.TestCase): |
# Check that the archive info was written correctly. |
self.assertTrue(os.path.exists(self.page_set_archive_info_file)) |
read_archive_info = page_set_archive_info.PageSetArchiveInfo.FromFile( |
- self.page_set_archive_info_file) |
+ self.page_set_archive_info_file, cloud_storage.PUBLIC_BUCKET) |
self.assertEquals(new_recording, |
read_archive_info.WprFilePathForPage(page1)) |