Move t.p.page_set_archive_info.PageSetArchiveInfo to t.wpr.archive_info.WprArchiveInfo.

tools/telemetry/telemetry/page/page_set_archive_info_unittest.py

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
deleted file mode 100644
index 4856700fdd1e0df41fc1b3f1a23db2b64bbb6a7f..0000000000000000000000000000000000000000
--- a/tools/telemetry/telemetry/page/page_set_archive_info_unittest.py
+++ /dev/null
@@ -1,186 +0,0 @@
-# Copyright 2012 The Chromium Authors. All rights reserved.
-# Use of this source code is governed by a BSD-style license that can be
-# found in the LICENSE file.
-import json
-import os
-import shutil
-import tempfile
-import unittest
-
-from telemetry.page import page
-from telemetry.page import page_set_archive_info
-from telemetry.util import cloud_storage
-
-
-class MockPage(page.Page):
-  def __init__(self, url, name=None):
-    super(MockPage, self).__init__(url, None, name=name)
-
-
-page1 = MockPage('http://www.foo.com/', 'Foo')
-page2 = MockPage('http://www.bar.com/', 'Bar')
-page3 = MockPage('http://www.baz.com/')
-recording1 = 'data_001.wpr'
-recording2 = 'data_002.wpr'
-archive_info_contents = ("""
-{
-"archives": {
-  "%s": ["%s", "%s"],
-  "%s": ["%s"]
-}
-}
-""" % (recording1, page1.display_name, page2.display_name, recording2,
-       page3.display_name))
-
-
-class TestPageSetArchiveInfo(unittest.TestCase):
-  def setUp(self):
-    self.tmp_dir = tempfile.mkdtemp()
-    # Write the metadata.
-    self.page_set_archive_info_file = os.path.join(self.tmp_dir, 'info.json')
-    with open(self.page_set_archive_info_file, 'w') as f:
-      f.write(archive_info_contents)
-
-    # Write the existing .wpr files.
-    for i in [1, 2]:
-      with open(os.path.join(self.tmp_dir, ('data_00%d.wpr' % i)), 'w') as f:
-        f.write(archive_info_contents)
-
-    # Create the PageSetArchiveInfo object to be tested.
-    self.archive_info = page_set_archive_info.PageSetArchiveInfo.FromFile(
-        self.page_set_archive_info_file, cloud_storage.PUBLIC_BUCKET)
-
-  def tearDown(self):
-    shutil.rmtree(self.tmp_dir)
-
-  def assertCorrectHashFile(self, file_path):
-    self.assertTrue(os.path.exists(file_path + '.sha1'))
-    with open(file_path + '.sha1', 'rb') as f:
-      self.assertEquals(cloud_storage.CalculateHash(file_path), f.read())
-
-  def testReadingArchiveInfo(self):
-    self.assertIsNotNone(self.archive_info.WprFilePathForUserStory(page1))
-    self.assertEquals(recording1, os.path.basename(
-        self.archive_info.WprFilePathForUserStory(page1)))
-
-    self.assertIsNotNone(self.archive_info.WprFilePathForUserStory(page2))
-    self.assertEquals(recording1, os.path.basename(
-        self.archive_info.WprFilePathForUserStory(page2)))
-
-    self.assertIsNotNone(self.archive_info.WprFilePathForUserStory(page3))
-    self.assertEquals(recording2, os.path.basename(
-        self.archive_info.WprFilePathForUserStory(page3)))
-
-  def testArchiveInfoFileGetsUpdated(self):
-    """Ensures that the archive info file is updated correctly."""
-
-    expected_archive_file_contents = {
-        u'description': (u'Describes the Web Page Replay archives for a page'
-                         u' set. Don\'t edit by hand! Use record_wpr for'
-                         u' updating.'),
-        u'archives': {
-            u'data_003.wpr': [u'Bar', u'http://www.baz.com/'],
-            u'data_001.wpr': [u'Foo']
-        }
-    }
-
-    new_temp_recording = os.path.join(self.tmp_dir, 'recording.wpr')
-    with open(new_temp_recording, 'w') as f:
-      f.write('wpr data')
-    self.archive_info.AddNewTemporaryRecording(new_temp_recording)
-    self.archive_info.AddRecordedPages([page2, page3])
-
-    with open(self.page_set_archive_info_file, 'r') as f:
-      archive_file_contents = json.load(f)
-      self.assertEquals(expected_archive_file_contents, archive_file_contents)
-
-  def testModifications(self):
-    recording1_path = os.path.join(self.tmp_dir, recording1)
-    recording2_path = os.path.join(self.tmp_dir, recording2)
-
-    new_recording1 = os.path.join(self.tmp_dir, 'data_003.wpr')
-    new_temp_recording = os.path.join(self.tmp_dir, 'recording.wpr')
-    with open(new_temp_recording, 'w') as f:
-      f.write('wpr data')
-
-    self.archive_info.AddNewTemporaryRecording(new_temp_recording)
-
-    self.assertEquals(new_temp_recording,
-                      self.archive_info.WprFilePathForUserStory(page1))
-    self.assertEquals(new_temp_recording,
-                      self.archive_info.WprFilePathForUserStory(page2))
-    self.assertEquals(new_temp_recording,
-                      self.archive_info.WprFilePathForUserStory(page3))
-
-    self.archive_info.AddRecordedPages([page2])
-
-    self.assertTrue(os.path.exists(new_recording1))
-    self.assertFalse(os.path.exists(new_temp_recording))
-
-    self.assertTrue(os.path.exists(recording1_path))
-    self.assertTrue(os.path.exists(recording2_path))
-    self.assertCorrectHashFile(new_recording1)
-
-    new_recording2 = os.path.join(self.tmp_dir, 'data_004.wpr')
-    with open(new_temp_recording, 'w') as f:
-      f.write('wpr data')
-
-    self.archive_info.AddNewTemporaryRecording(new_temp_recording)
-    self.archive_info.AddRecordedPages([page3])
-
-    self.assertTrue(os.path.exists(new_recording2))
-    self.assertCorrectHashFile(new_recording2)
-    self.assertFalse(os.path.exists(new_temp_recording))
-
-    self.assertTrue(os.path.exists(recording1_path))
-    # recording2 is no longer needed, so it was deleted.
-    self.assertFalse(os.path.exists(recording2_path))
-
-  def testCreatingNewArchiveInfo(self):
-    # Write only the page set without the corresponding metadata file.
-    page_set_contents = ("""
-    {
-        archive_data_file": "new_archive_info.json",
-        "pages": [
-            {
-                "url": "%s",
-            }
-        ]
-    }""" % page1.url)
-
-    page_set_file = os.path.join(self.tmp_dir, 'new_page_set.json')
-    with open(page_set_file, 'w') as f:
-      f.write(page_set_contents)
-
-    self.page_set_archive_info_file = os.path.join(self.tmp_dir,
-                                                   'new_archive_info.json')
-
-    # Create the PageSetArchiveInfo object to be tested.
-    self.archive_info = page_set_archive_info.PageSetArchiveInfo.FromFile(
-        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')
-    with open(new_temp_recording, 'w') as f:
-      f.write('wpr data')
-
-    self.archive_info.AddNewTemporaryRecording(new_temp_recording)
-
-    self.assertEquals(new_temp_recording,
-                      self.archive_info.WprFilePathForUserStory(page1))
-
-    self.archive_info.AddRecordedPages([page1])
-
-    # Expected name for the recording (decided by PageSetArchiveInfo).
-    new_recording = os.path.join(self.tmp_dir, 'new_archive_info_000.wpr')
-
-    self.assertTrue(os.path.exists(new_recording))
-    self.assertFalse(os.path.exists(new_temp_recording))
-    self.assertCorrectHashFile(new_recording)
-
-    # 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, cloud_storage.PUBLIC_BUCKET)
-    self.assertEquals(new_recording,
-                      read_archive_info.WprFilePathForUserStory(page1))