| Index: tools/telemetry/telemetry/page_set_archive_info_unittest.py
|
| ===================================================================
|
| --- tools/telemetry/telemetry/page_set_archive_info_unittest.py (revision 180130)
|
| +++ tools/telemetry/telemetry/page_set_archive_info_unittest.py (working copy)
|
| @@ -1,101 +0,0 @@
|
| -# 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')
|
| - f = open(self.page_set_archive_info_file, 'w')
|
| - f.write(archive_info_contents)
|
| - f.close()
|
| -
|
| - # Write the existing .wpr files.
|
| - for i in [1, 2]:
|
| - f = open(os.path.join(self.tmp_dir, ('data_00%d.wpr' % i)), 'w')
|
| - f.write(archive_info_contents)
|
| - f.close()
|
| -
|
| - # 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')
|
| - f = open(new_temp_recording, 'w')
|
| - f.write('wpr data')
|
| - f.close()
|
| -
|
| - 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'
|
| - f = open(new_temp_recording, 'w')
|
| - f.write('wpr data')
|
| - f.close()
|
| -
|
| - 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)))
|
|
|