| Index: tools/telemetry/telemetry/value/trace_unittest.py
|
| diff --git a/tools/telemetry/telemetry/value/trace_unittest.py b/tools/telemetry/telemetry/value/trace_unittest.py
|
| deleted file mode 100644
|
| index d3481461adafc4770557b2d008d0bf488e41f449..0000000000000000000000000000000000000000
|
| --- a/tools/telemetry/telemetry/value/trace_unittest.py
|
| +++ /dev/null
|
| @@ -1,129 +0,0 @@
|
| -# Copyright 2014 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 story
|
| -from telemetry import page as page_module
|
| -from telemetry.testing import system_stub
|
| -from telemetry.timeline import trace_data
|
| -from telemetry.value import trace
|
| -
|
| -
|
| -class TestBase(unittest.TestCase):
|
| -
|
| - def setUp(self):
|
| - story_set = story.StorySet(base_dir=os.path.dirname(__file__))
|
| - story_set.AddStory(
|
| - page_module.Page('http://www.bar.com/', story_set, story_set.base_dir))
|
| - story_set.AddStory(
|
| - page_module.Page('http://www.baz.com/', story_set, story_set.base_dir))
|
| - story_set.AddStory(
|
| - page_module.Page('http://www.foo.com/', story_set, story_set.base_dir))
|
| - self.story_set = story_set
|
| -
|
| - self._cloud_storage_stub = system_stub.Override(trace, ['cloud_storage'])
|
| -
|
| - def tearDown(self):
|
| - if self._cloud_storage_stub:
|
| - self._cloud_storage_stub.Restore()
|
| - self._cloud_storage_stub = None
|
| -
|
| - @property
|
| - def pages(self):
|
| - return self.story_set.stories
|
| -
|
| -
|
| -class TestSet(object):
|
| - """ A test set that represents a set that contains any key. """
|
| -
|
| - def __contains__(self, key):
|
| - return True
|
| -
|
| -
|
| -class TestDefaultDict(object):
|
| - """ A test default dict that represents a dictionary that contains any key
|
| - with value |default_value|. """
|
| -
|
| - def __init__(self, default_value):
|
| - self._default_value = default_value
|
| - self._test_set = TestSet()
|
| -
|
| - def __contains__(self, key):
|
| - return key in self._test_set
|
| -
|
| - def __getitem__(self, key):
|
| - return self._default_value
|
| -
|
| - def keys(self):
|
| - return self._test_set
|
| -
|
| -
|
| -class ValueTest(TestBase):
|
| - def testRepr(self):
|
| - v = trace.TraceValue(self.pages[0], trace_data.TraceData({'test': 1}),
|
| - important=True, description='desc')
|
| -
|
| - self.assertEquals('TraceValue(http://www.bar.com/, trace)', str(v))
|
| -
|
| - def testAsDictWhenTraceSerializedAndUploaded(self):
|
| - tempdir = tempfile.mkdtemp()
|
| - try:
|
| - v = trace.TraceValue(None, trace_data.TraceData({'test': 1}))
|
| - fh = v.Serialize(tempdir)
|
| - trace.cloud_storage.SetCalculatedHashesForTesting(
|
| - {fh.GetAbsPath(): 123})
|
| - bucket = trace.cloud_storage.PUBLIC_BUCKET
|
| - cloud_url = v.UploadToCloud(bucket)
|
| - d = v.AsDict()
|
| - self.assertEqual(d['file_id'], fh.id)
|
| - self.assertEqual(d['cloud_url'], cloud_url)
|
| - finally:
|
| - shutil.rmtree(tempdir)
|
| -
|
| - def testAsDictWhenTraceIsNotSerializedAndUploaded(self):
|
| - test_temp_file = tempfile.NamedTemporaryFile(delete=False)
|
| - try:
|
| - v = trace.TraceValue(None, trace_data.TraceData({'test': 1}))
|
| - trace.cloud_storage.SetCalculatedHashesForTesting(
|
| - TestDefaultDict(123))
|
| - bucket = trace.cloud_storage.PUBLIC_BUCKET
|
| - cloud_url = v.UploadToCloud(bucket)
|
| - d = v.AsDict()
|
| - self.assertEqual(d['cloud_url'], cloud_url)
|
| - finally:
|
| - if os.path.exists(test_temp_file.name):
|
| - test_temp_file.close()
|
| - os.remove(test_temp_file.name)
|
| -
|
| -
|
| -def _IsEmptyDir(path):
|
| - return os.path.exists(path) and not os.listdir(path)
|
| -
|
| -
|
| -class NoLeakedTempfilesTests(TestBase):
|
| -
|
| - def setUp(self):
|
| - super(NoLeakedTempfilesTests, self).setUp()
|
| - self.temp_test_dir = tempfile.mkdtemp()
|
| - self.actual_tempdir = trace.tempfile.tempdir
|
| - trace.tempfile.tempdir = self.temp_test_dir
|
| -
|
| - def testNoLeakedTempFileOnImplicitCleanUp(self):
|
| - with trace.TraceValue(None, trace_data.TraceData({'test': 1})):
|
| - pass
|
| - self.assertTrue(_IsEmptyDir(self.temp_test_dir))
|
| -
|
| - def testNoLeakedTempFileWhenUploadingTrace(self):
|
| - v = trace.TraceValue(None, trace_data.TraceData({'test': 1}))
|
| - v.CleanUp()
|
| - self.assertTrue(_IsEmptyDir(self.temp_test_dir))
|
| -
|
| - def tearDown(self):
|
| - super(NoLeakedTempfilesTests, self).tearDown()
|
| - shutil.rmtree(self.temp_test_dir)
|
| - trace.tempfile.tempdir = self.actual_tempdir
|
|
|