Enable some profile_chrome unit tests on Trybots

systrace/profile_chrome/fake_agent_1.py

-# Copyright 2014 The Chromium Authors. All rights reserved.
+# Copyright 2016 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 optparse
 import tempfile
-import unittest
-import zipfile
 
-from profile_chrome import profiler
-from profile_chrome import ui
 from systrace import trace_result
+from systrace import tracing_agents
 
 
 class FakeAgent(object):
   def __init__(self, contents='fake-contents'):
     self.contents = contents
     self.stopped = False
     self.filename = None
     self.config = None
     self.timeout = None
 
@@ skipping 24 matching lines @@
 
   # pylint: disable=unused-argument, no-self-use
   def RecordClockSyncMarker(self, sync_id, did_record_sync_marker_callback):
     print ('Clock sync marker cannot be recorded since explicit clock sync '
            'is not supported.')
 
   def __repr__(self):
     return 'faketrace'
 
 
-class ProfilerTest(unittest.TestCase):
-  def setUp(self):
-    ui.EnableTestMode()
+class FakeConfig(tracing_agents.TracingConfig):
+  def __init__(self):
+    tracing_agents.TracingConfig.__init__(self)
 
-  def testCaptureBasicProfile(self):
-    agent = FakeAgent()
-    result = profiler.CaptureProfile(None, [agent], 1)
 
-    try:
-      self.assertTrue(agent.stopped)
-      self.assertTrue(os.path.exists(result))
-      self.assertTrue(result.endswith('.html'))
-    finally:
-      if os.path.exists(result):
-        os.remove(result)
+# pylint: disable=unused-argument
+def try_create_agent(config):
+  return FakeAgent()
 
-  def testCaptureJsonProfile(self):
-    agent = FakeAgent()
-    result = profiler.CaptureProfile(None, [agent], 1, write_json=True)
+def add_options(parser):
+  options = optparse.OptionGroup(parser, 'Fake options.')
+  return options
 
-    try:
-      self.assertFalse(result.endswith('.html'))
-      with open(result) as f:
-        self.assertEquals(f.read(), agent.contents)
-    finally:
-      if os.path.exists(result):
-        os.remove(result)
-
-  def testCaptureMultipleProfiles(self):
-    agents = [FakeAgent('c1'), FakeAgent('c2')]
-    result = profiler.CaptureProfile(None, agents, 1, write_json=True)
-
-    try:
-      self.assertTrue(result.endswith('.zip'))
-      self.assertTrue(zipfile.is_zipfile(result))
-    finally:
-      if os.path.exists(result):
-        os.remove(result)
+# pylint: disable=unused-argument
+def get_config(options):
+  return FakeConfig()