[polymer] Merge of master into polymer10-migration

telemetry/telemetry/internal/platform/tracing_agent/cpu_tracing_agent_unittest.py

+# 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 json
+import sys
+import time
+import unittest
+
+from telemetry import decorators
+from telemetry.internal.platform.tracing_agent import cpu_tracing_agent
+from telemetry.internal.platform import tracing_agent
+from telemetry.internal.platform import linux_platform_backend
+from telemetry.internal.platform import mac_platform_backend
+from telemetry.internal.platform import win_platform_backend
+from telemetry.timeline import trace_data
+from telemetry.timeline import tracing_config
+
+SNAPSHOT_KEYS = ['pid', 'command', 'pCpu', 'pMem']
+TRACE_EVENT_KEYS = ['name', 'tid', 'pid', 'ph', 'args', 'local', 'id', 'ts']
+
+
+class FakeAndroidPlatformBackend(object):
+  def __init__(self):
+    self.device = 'fake_device'
+
+  def GetOSName(self):
+    return 'android'
+
+
+class CpuTracingAgentTest(unittest.TestCase):
+
+  def setUp(self):
+    self._config = tracing_config.TracingConfig()
+    self._config.enable_cpu_trace = True
+    if sys.platform.startswith('win'):
+      self._desktop_backend = win_platform_backend.WinPlatformBackend()
+    elif sys.platform.startswith('darwin'):
+      self._desktop_backend = mac_platform_backend.MacPlatformBackend()
+    else:
+      self._desktop_backend = linux_platform_backend.LinuxPlatformBackend()
+    self._agent = cpu_tracing_agent.CpuTracingAgent(self._desktop_backend)
+
+  @decorators.Enabled('linux', 'mac', 'win')
+  def testInit(self):
+    self.assertTrue(isinstance(self._agent,
+                               tracing_agent.TracingAgent))
+    self.assertFalse(self._agent._snapshots)
+    self.assertFalse(self._agent._snapshot_ongoing)
+
+  @decorators.Enabled('linux', 'mac', 'win')
+  def testIsSupported(self):
+    self.assertTrue(cpu_tracing_agent.CpuTracingAgent.IsSupported(
+      self._desktop_backend))
+    self.assertFalse(cpu_tracing_agent.CpuTracingAgent.IsSupported(
+      FakeAndroidPlatformBackend()))
+
+  @decorators.Enabled('linux', 'mac', 'win')
+  def testStartAgentTracing(self):
+    self.assertFalse(self._agent._snapshot_ongoing)
+    self.assertFalse(self._agent._snapshots)
+    self.assertTrue(self._agent.StartAgentTracing(self._config, 0))
+    self.assertTrue(self._agent._snapshot_ongoing)
+    time.sleep(2)
+    self.assertTrue(self._agent._snapshots)
+    self._agent.StopAgentTracing()
+
+  @decorators.Enabled('linux', 'mac', 'win')
+  def testStartAgentTracingNotEnabled(self):
+    self._config.enable_cpu_trace = False
+    self.assertFalse(self._agent._snapshot_ongoing)
+    self.assertFalse(self._agent.StartAgentTracing(self._config, 0))
+    self.assertFalse(self._agent._snapshot_ongoing)
+    self.assertFalse(self._agent._snapshots)
+    time.sleep(2)
+    self.assertFalse(self._agent._snapshots)
+
+  @decorators.Enabled('linux', 'mac', 'win')
+  def testStopAgentTracingBeforeStart(self):
+    self.assertRaises(AssertionError, self._agent.StopAgentTracing)
+
+  @decorators.Enabled('linux', 'mac', 'win')
+  def testStopAgentTracing(self):
+    self._agent.StartAgentTracing(self._config, 0)
+    self._agent.StopAgentTracing()
+    self.assertFalse(self._agent._snapshot_ongoing)
+
+  @decorators.Enabled('linux', 'mac', 'win')
+  def testCollectAgentTraceDataBeforeStop(self):
+    self._agent.StartAgentTracing(self._config, 0)
+    self.assertRaises(AssertionError, self._agent.CollectAgentTraceData,
+        trace_data.TraceDataBuilder())
+    self._agent.StopAgentTracing()
+
+  @decorators.Enabled('linux', 'mac', 'win')
+  def testCollectAgentTraceData(self):
+    builder = trace_data.TraceDataBuilder()
+    self._agent.StartAgentTracing(self._config, 0)
+    self._agent.StopAgentTracing()
+    self._agent.CollectAgentTraceData(builder)
+    self.assertFalse(self._agent._snapshot_ongoing)
+    builder = builder.AsData()
+    self.assertTrue(builder.HasTraceFor(trace_data.CPU_TRACE_DATA))
+
+  @decorators.Enabled('linux', 'mac', 'win')
+  def testCollectAgentTraceDataFormat(self):
+    builder = trace_data.TraceDataBuilder()
+    self._agent.StartAgentTracing(self._config, 0)
+    time.sleep(2)
+    self._agent.StopAgentTracing()
+    self._agent.CollectAgentTraceData(builder)
+    builder = builder.AsData()
+    data = json.loads(builder.GetTraceFor(trace_data.CPU_TRACE_DATA))
+    self.assertTrue(data)
+    self.assertEquals(set(data[0].keys()), set(TRACE_EVENT_KEYS))
+    self.assertEquals(set(data[0]['args'].keys()), set(['processes']))
+    self.assertTrue(data[0]['args']['processes'])
+    self.assertEquals(set(data[0]['args']['processes'][0].keys()),
+                      set(SNAPSHOT_KEYS))
+
+  @decorators.Enabled('linux', 'mac', 'win')
+  def testMinimumCpuThreshold(self):
+    builder = trace_data.TraceDataBuilder()
+    self._agent.StartAgentTracing(self._config, 0)
+    time.sleep(2)
+    self._agent.StopAgentTracing()
+    self._agent.CollectAgentTraceData(builder)
+    builder = builder.AsData()
+    data = json.loads(builder.GetTraceFor(trace_data.CPU_TRACE_DATA))
+    self.assertTrue(data)
+    for snapshot in data:
+      for process in snapshot['args']['processes']:
+        self.assertTrue(process['pCpu'] >= cpu_tracing_agent.DEFAULT_MIN_PCPU)
+
+  @decorators.Enabled('linux', 'mac')
+  def testParseLine(self):
+    collector = self._agent._collector
+    invalid_inputs = ['', '1000 chrome', '1000 chrome 1.0 1.0 1.0']
+    for invalid_input in invalid_inputs:
+      self.assertFalse(collector._ParseLine(invalid_input))
+    valid_input = '1000 chrome 20.0 10.0 '
+    output = collector._ParseLine(valid_input)
+    self.assertTrue(output['pCpu'] == '20.0' and
+                    output['pMem'] == '10.0'and
+                    output['pid'] == '1000'and
+                    output['command'] == 'chrome')