summaryrefslogtreecommitdiffstats
path: root/tools/profile_chrome/systrace_controller_unittest.py
blob: 30b22c137530c30f0e0b3c80f4ed5f81aa838da4 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
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 profile_chrome import controllers_unittest
from profile_chrome 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)