[BattOr][Telemetry] Changes needed to run BattOr tests on telemetry.

telemetry/telemetry/value/trace_unittest.py

 # 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
+from tracing_build import html2trace
 
 
 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))
@@ skipping 72 matching lines @@
       # pylint: enable=no-member
       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 testFindTraceParts(self):
+    raw_data = {
+      'powerTraceAsString': 'Battor Data',
+      'traceEvents': 'Chrome Data',
+      'tabIds': 'Tab Data',
+    }
+    data = trace_data.TraceData(raw_data)
+    v = trace.TraceValue(None, data)
+    tempdir = tempfile.mkdtemp()
+    temp_path = os.path.join(tempdir, 'test.json')
+    battor_seen = False
+    chrome_seen = False
+    tabs_seen = False
+    try:
+      trace_files = html2trace.CopyTraceDataFromHTMLFilePath(v.filename,
+                                                             temp_path)
+      for f in trace_files:
+        with open(f, 'r') as trace_file:
+          d = trace_file.read()
+          if d == raw_data['powerTraceAsString']:
+            self.assertFalse(battor_seen)
+            battor_seen = True
+          elif d == raw_data['traceEvents']:
+            self.assertFalse(chrome_seen)
+            chrome_seen = True
+          elif d == raw_data['tabIds']:
+            self.assertFalse(tabs_seen)
+            tabs_seen = True
+      self.assertTrue(battor_seen)
+      self.assertTrue(chrome_seen)
+      self.assertTrue(tabs_seen)
+    finally:
+      shutil.rmtree(tempdir)
+      os.remove(v.filename)
+
 
 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