Index: systrace/profile_chrome/perf_tracing_agent_unittest.py |
diff --git a/systrace/profile_chrome/perf_tracing_agent_unittest.py b/systrace/profile_chrome/perf_tracing_agent_unittest.py |
index 1367e9b759f3aacb0eb1fbdc56cf6eb7f9a3bb75..cd90b3691744ee7cd8982fc8279fe830d689f252 100644 |
--- a/systrace/profile_chrome/perf_tracing_agent_unittest.py |
+++ b/systrace/profile_chrome/perf_tracing_agent_unittest.py |
@@ -21,12 +21,11 @@ class PerfProfilerAgentTest(agents_unittest.BaseAgentTest): |
if not perf_tracing_agent.PerfProfilerAgent.IsSupported(): |
return |
ui.EnableTestMode() |
- categories = ['cycles'] |
- agent = perf_tracing_agent.PerfProfilerAgent(self.device, |
- categories) |
+ categories = 'cycles' |
+ agent = perf_tracing_agent.PerfProfilerAgent(self.device) |
try: |
- agent.StartAgentTracing(None, None) |
+ agent.StartAgentTracing(perf_tracing_agent.PerfConfig(categories)) |
finally: |
agent.StopAgentTracing() |