blob: 29c36e7e79cfb9a168387e19ee6b87293e08f586 (
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
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
|
# 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 logging
import platform
import os
import sys
import time
import unittest
if __name__ == '__main__':
sys.path.append(
os.path.join(os.path.dirname(__file__), '..'))
from battor import battor_wrapper
from devil.utils import battor_device_mapping
from devil.utils import find_usb_devices
import py_utils
from py_utils import cloud_storage
_SUPPORTED_CQ_PLATFORMS = ['win', 'linux', 'mac']
class BattOrWrapperDeviceTest(unittest.TestCase):
def setUp(self):
self._platform = py_utils.GetHostOsName()
self._battor_list = None
if self._platform == 'linux':
device_tree = find_usb_devices.GetBusNumberToDeviceTreeMap()
self._battor_list = battor_device_mapping.GetBattOrList(device_tree)
if not battor_wrapper.IsBattOrConnected(self._platform):
self._battor_list = []
def testFullRun(self):
# If battor_list is an empty list, a BattOr was expected but not found.
if self._battor_list is not None and not self._battor_list:
logging.critical('No BattOrs attached. Cannot run tests.')
return
if self._platform not in _SUPPORTED_CQ_PLATFORMS:
logging.critical('Platform %s is not supported on CQ.' % self._platform)
return
battor_path = (None if not self._battor_list
else '/dev/%s' % self._battor_list[0])
battor = battor_wrapper.BattOrWrapper(
self._platform, battor_path=battor_path,
serial_log_bucket=cloud_storage.TELEMETRY_OUTPUT)
try:
battor.StartShell()
self.assertTrue(isinstance(battor.GetFirmwareGitHash(), basestring))
# We expect the git hash to be a valid 6 character hexstring. This will
# throw a ValueError exception otherwise.
int(battor.GetFirmwareGitHash(), 16)
self.assertTrue(len(battor.GetFirmwareGitHash()) == 7)
battor.StopShell()
battor.StartShell()
battor.StartTracing()
# TODO(rnephew): This sleep is required for now because crbug.com/602266
# causes the BattOr to crash when the trace time is too short. Once that
# bug is fixed, we should remove this delay.
time.sleep(1)
battor.RecordClockSyncMarker('abc')
# Sleep here because clock sync marker will be flaky if not.
time.sleep(1)
battor.StopTracing()
# Below is a work around for crbug.com/603309. On this short of a trace, 5
# seconds is enough to ensure that the trace will finish flushing to the
# file. The process is then killed so that BattOrWrapper knows that the
# process has been closed after tracing stops.
if self._platform == 'win':
time.sleep(5)
battor._battor_shell.kill()
results = battor.CollectTraceData().splitlines()
except:
if battor._battor_shell is not None:
battor._battor_shell.kill()
battor._battor_shell = None
raise
self.assertTrue('# BattOr' in results[0])
self.assertTrue('# voltage_range' in results[1])
self.assertTrue('# current_range' in results[2])
self.assertTrue('# sample_rate' in results[3])
# First line with results. Should be 3 'words'.
self.assertTrue(len(results[4].split()) == 3)
clock_sync_found = False
for entry in results:
if '<abc>' in entry:
clock_sync_found = True
break
self.assertTrue(clock_sync_found, 'BattOr Data:%s\n' % repr(results))
if __name__ == '__main__':
logging.getLogger().setLevel(logging.DEBUG)
unittest.main(verbosity=2)
|
