Smoke test for tracing infrastructure in PyAuto

chrome/test/functional/tracing/tracing_smoke_test.py

+#!/usr/bin/env python
+# Copyright (c) 2012 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 pyauto_tracing
+import pyauto
+import tracer
+
+
+class TracingSmokeTest(pyauto.PyUITest):
+  """Test basic functionality of the tracing API."""
+  def setUp(self):
+    super(TracingSmokeTest, self).setUp()
+    self._tracer_factory = tracer.TracerFactory(self)
+
+  def testGetData(self):
+    """Check that we can find a CrBrowserMain thread."""
+    tracer = self._tracer_factory.Produce()
+    tracer.BeginTracing()
+    model = tracer.EndTracing()
+    self.assertEqual(1, len(model.FindAllThreadsNamed('CrBrowserMain')))
+
+  def testMultipleTraces(self):
+    """Check that we can run multiple traces on the same tracer."""
+    tracer = self._tracer_factory.Produce()
+    tracer.BeginTracing()
+    model1 = tracer.EndTracing()
+    tracer.BeginTracing()
+    model2 = tracer.EndTracing()
+    del tracer
+    self.assertEqual(1, len(model1.FindAllThreadsNamed('CrBrowserMain')))
+    self.assertEqual(1, len(model2.FindAllThreadsNamed('CrBrowserMain')))
+
+  def testMultipleTracers(self):
+    """Check that we can run multiple traces with multiple tracers."""
+    tracer1 = self._tracer_factory.Produce()
+    tracer2 = self._tracer_factory.Produce()
+    del self._tracer_factory
+    # Nested calls to beginTracing is untested and probably won't work.
+    tracer1.BeginTracing()
+    model1 = tracer1.EndTracing()
+    del tracer1
+    tracer2.BeginTracing()
+    model2 = tracer2.EndTracing()
+    del tracer2
+    self.assertEqual(1, len(model1.FindAllThreadsNamed('CrBrowserMain')))
+    del model1
+    self.assertEqual(1, len(model2.FindAllThreadsNamed('CrBrowserMain')))
+    del model2
+
+
+if __name__ == '__main__':
+  pyauto_tracing.Main()