Smoke test for tracing infrastructure in PyAuto

chrome/test/functional/tracing/tracing_smoke_test.py

+#!/usr/bin/env python
+# Copyright (c) 2011 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 unittest
+
+import pyauto_tracing
+import pyauto
+import tracer
+
+class TracingSmokeTest(pyauto.PyUITest):
+  def setUp(self):
+    super(TracingSmokeTest, self).setUp()
+    self._tracer_factory = tracer.TracerFactory(self)
+
+  def testGetData(self):
+    tracer = self._tracer_factory.produce()
+    tracer.beginTracing()
+    model = tracer.endTracing()
+    self.assertEqual(1, len(model.findAllThreadsNamed("CrBrowserMain")))
+
+  def testMultipleTraces(self):
+    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):
+    tracer1 = self._tracer_factory.produce()
+    tracer2 = self._tracer_factory.produce()
+    del self._tracer_factory
+    # I don't know what will happen if you try to nest calls to beginTracing.
+    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()