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 |
index 6c9bda41584725dcd1301c29489ba97f22a43be2..f2f991ba7ad11b848a708c362ea484bd7f6d63fd 100644 |
--- a/tools/telemetry/telemetry/value/trace_unittest.py |
+++ b/tools/telemetry/telemetry/value/trace_unittest.py |
@@ -7,6 +7,7 @@ import unittest |
from telemetry.page import page_set |
from telemetry.timeline import tracing_timeline_data |
+from telemetry.unittest_util import system_stub |
from telemetry.value import trace |
class TestBase(unittest.TestCase): |
@@ -16,6 +17,13 @@ class TestBase(unittest.TestCase): |
self.page_set.AddPageWithDefaultRunNavigate("http://www.baz.com/") |
self.page_set.AddPageWithDefaultRunNavigate("http://www.foo.com/") |
+ 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.page_set.pages |
@@ -24,8 +32,11 @@ class ValueTest(TestBase): |
def testAsDict(self): |
v = trace.TraceValue( |
None, tracing_timeline_data.TracingTimelineData({'test' : 1})) |
- fh_id = v.GetAssociatedFileHandle().id |
- |
+ fh = v.GetAssociatedFileHandle() |
+ 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['file_id'], fh.id) |
+ self.assertEqual(d['cloud_url'], cloud_url) |