adb_profile_chrome: Refactor into multiple modules and add tests

build/android/chrome_profiler/systrace_controller_unittest.py

Index: build/android/chrome_profiler/systrace_controller_unittest.py
diff --git a/build/android/chrome_profiler/systrace_controller_unittest.py b/build/android/chrome_profiler/systrace_controller_unittest.py
new file mode 100644
index 0000000000000000000000000000000000000000..448f3283945b8555d0c1e48ed2dd75ef51f59dd7
--- /dev/null
+++ b/build/android/chrome_profiler/systrace_controller_unittest.py
@@ -0,0 +1,36 @@
+# 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
+
+from chrome_profiler import controllers_unittest
+from chrome_profiler import systrace_controller
+
+
+class SystraceControllerTest(controllers_unittest.BaseControllerTest):
+  def testGetCategories(self):
+    categories = \
+        systrace_controller.SystraceController.GetCategories(self.device)
+    self.assertTrue(categories)
+    assert 'gfx' in ' '.join(categories)
+
+  def testTracing(self):
+    categories = ['gfx', 'input', 'view']
+    ring_buffer = False
+    controller = systrace_controller.SystraceController(self.device,
+                                                        categories,
+                                                        ring_buffer)
+
+    interval = 1
+    try:
+      controller.StartTracing(interval)
+    finally:
+      controller.StopTracing()
+
+    result = controller.PullTrace()
+    try:
+      with open(result) as f:
+        self.assertTrue('CPU#' in f.read())
+    finally:
+      os.remove(result)