[Telemetry] Reorganize everything.

tools/telemetry/telemetry/page_set_archive_info_unittest.py

-# Copyright (c) 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 os
-import shutil
-import tempfile
-import unittest
-
-from telemetry import page_set_archive_info
-
-class MockPage(object):
-  def __init__(self, url):
-    self.url = url
-
-url1 = 'http://www.foo.com/'
-url2 = 'http://www.bar.com/'
-url3 = 'http://www.baz.com/'
-recording1 = 'data_001.wpr'
-recording2 = 'data_002.wpr'
-archive_info_contents = ("""
-{
-"archives": {
-  "%s": ["%s", "%s"],
-  "%s": ["%s"]
-}
-}
-""" % (recording1, url1, url2, recording2, url3))
-page1 = MockPage(url1)
-page2 = MockPage(url2)
-page3 = MockPage(url3)
-
-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, '/tmp/pageset.json')
-
-  def tearDown(self):
-    shutil.rmtree(self.tmp_dir)
-
-  def testReadingArchiveInfo(self):
-    self.assertEquals(recording1, os.path.basename(
-        self.archive_info.WprFilePathForPage(page1)))
-    self.assertEquals(recording1, os.path.basename(
-        self.archive_info.WprFilePathForPage(page2)))
-    self.assertEquals(recording2, os.path.basename(
-        self.archive_info.WprFilePathForPage(page3)))
-
-  def testModifications(self):
-    new_recording1 = '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.WprFilePathForPage(page1))
-    self.assertEquals(new_temp_recording,
-                      self.archive_info.WprFilePathForPage(page2))
-    self.assertEquals(new_temp_recording,
-                      self.archive_info.WprFilePathForPage(page3))
-
-    self.archive_info.AddRecordedPages([page2])
-
-    self.assertTrue(os.path.exists(os.path.join(self.tmp_dir, new_recording1)))
-    self.assertFalse(os.path.exists(
-        os.path.join(self.tmp_dir, new_temp_recording)))
-
-    self.assertTrue(os.path.exists(os.path.join(self.tmp_dir, recording1)))
-    self.assertTrue(os.path.exists(os.path.join(self.tmp_dir, recording2)))
-
-    new_recording2 = '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(os.path.join(self.tmp_dir, new_recording2)))
-    self.assertFalse(os.path.exists(
-        os.path.join(self.tmp_dir, new_temp_recording)))
-
-    self.assertTrue(os.path.exists(os.path.join(self.tmp_dir, recording1)))
-    # recording2 is no longer needed, so it was deleted.
-    self.assertFalse(os.path.exists(os.path.join(self.tmp_dir, recording2)))
-
-  def testCreatingNewArchiveInfo(self):
-    # Write only the page set without the corresponding metadata file.
-    page_set_contents = ("""
-    {
-        archive_data_file": "new-metadata.json",
-        "pages": [
-            {
-                "url": "%s",
-            }
-        ]
-    }""" % url1)
-
-    page_set_file = os.path.join(self.tmp_dir, 'new.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-metadata.json')
-
-    # Create the PageSetArchiveInfo object to be tested.
-    self.archive_info = page_set_archive_info.PageSetArchiveInfo.FromFile(
-        self.page_set_archive_info_file, page_set_file)
-
-    # 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.WprFilePathForPage(page1))
-
-    self.archive_info.AddRecordedPages([page1])
-
-    # Expected name for the recording (decided by PageSetArchiveInfo).
-    new_recording = os.path.join(self.tmp_dir, 'new_000.wpr')
-
-    self.assertTrue(os.path.exists(os.path.join(self.tmp_dir, new_recording)))
-    self.assertFalse(os.path.exists(
-        os.path.join(self.tmp_dir, new_temp_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, '/tmp/pageset.json')
-    self.assertEquals(new_recording,
-                      read_archive_info.WprFilePathForPage(page1))