Index: tools/telemetry/telemetry/value/profiler_unittest.py |
diff --git a/tools/telemetry/telemetry/value/trace_unittest.py b/tools/telemetry/telemetry/value/profiler_unittest.py |
similarity index 53% |
copy from tools/telemetry/telemetry/value/trace_unittest.py |
copy to tools/telemetry/telemetry/value/profiler_unittest.py |
index 6c9bda41584725dcd1301c29489ba97f22a43be2..06788fa137661b5b27dc0e4423e2316a26cc234a 100644 |
--- a/tools/telemetry/telemetry/value/trace_unittest.py |
+++ b/tools/telemetry/telemetry/value/profiler_unittest.py |
@@ -3,11 +3,13 @@ |
# found in the LICENSE file. |
import os |
+import tempfile |
import unittest |
from telemetry.page import page_set |
-from telemetry.timeline import tracing_timeline_data |
-from telemetry.value import trace |
+from telemetry.util import cloud_storage |
+from telemetry.util import file_handle |
+from telemetry.value import profiler |
class TestBase(unittest.TestCase): |
def setUp(self): |
@@ -22,10 +24,19 @@ class TestBase(unittest.TestCase): |
class ValueTest(TestBase): |
def testAsDict(self): |
- v = trace.TraceValue( |
- None, tracing_timeline_data.TracingTimelineData({'test' : 1})) |
- fh_id = v.GetAssociatedFileHandle().id |
- |
+ v = profiler.ProfilerValue(None, 'trace', None) |
d = v.AsDict() |
- self.assertEqual(d['file_id'], fh_id) |
+ self.assertEqual(d['profiler_name'], 'trace') |
+ |
+ def testUploadToCloud(self): |
+ try: |
+ tf = tempfile.NamedTemporaryFile() |
+ tf_handle = file_handle.FromTempFile(tf) |
+ v = profiler.ProfilerValue(None, 'trace', tf_handle) |
+ self.assertEqual(v.GetAssociatedFileHandle(), tf_handle) |
+ bucket = cloud_storage.PUBLIC_BUCKET |
+ v.UploadToCloud(bucket) |
+ self.assertTrue(v.cloud_url) |
+ finally: |
+ tf.close() |