| OLD | NEW |
| (Empty) |
| 1 # Copyright 2014 The Chromium Authors. All rights reserved. | |
| 2 # Use of this source code is governed by a BSD-style license that can be | |
| 3 # found in the LICENSE file. | |
| 4 | |
| 5 import os | |
| 6 import json | |
| 7 | |
| 8 from adb_profile_chrome import chrome_controller | |
| 9 from adb_profile_chrome import controllers_unittest | |
| 10 | |
| 11 | |
| 12 class ChromeControllerTest(controllers_unittest.BaseControllerTest): | |
| 13 def testGetCategories(self): | |
| 14 # Not supported on stable yet. | |
| 15 # TODO(skyostil): Remove this once category queries roll into stable. | |
| 16 if self.browser == 'stable': | |
| 17 return | |
| 18 | |
| 19 categories = \ | |
| 20 chrome_controller.ChromeTracingController.GetCategories( | |
| 21 self.device, self.package_info) | |
| 22 | |
| 23 self.assertEquals(len(categories), 2) | |
| 24 self.assertTrue(categories[0]) | |
| 25 self.assertTrue(categories[1]) | |
| 26 | |
| 27 def testTracing(self): | |
| 28 categories = '*' | |
| 29 ring_buffer = False | |
| 30 controller = chrome_controller.ChromeTracingController(self.device, | |
| 31 self.package_info, | |
| 32 categories, | |
| 33 ring_buffer) | |
| 34 | |
| 35 interval = 1 | |
| 36 try: | |
| 37 controller.StartTracing(interval) | |
| 38 finally: | |
| 39 controller.StopTracing() | |
| 40 | |
| 41 result = controller.PullTrace() | |
| 42 try: | |
| 43 with open(result) as f: | |
| 44 json.loads(f.read()) | |
| 45 finally: | |
| 46 os.remove(result) | |
| OLD | NEW |