def test_main_dead_device(self):
# Collect calls to 'subprocess.Popen' and simulate an unresponsive device.
def mocked_popen_calls(args, **kwargs):
if (args[0] == spawn_device_monitor._RUN_PY):
# ts_mon was called, so collect args and return
send_ts_mon_call.extend(args)
return None
elif (args[0] == '/some/adb/path'):
# adb was called, so collect args and throw exception
adb_calls.extend(args)
raise subprocess.CalledProcessError(1, None, None)
else:
self.fail('Unexpected Popen call: %s' % (' '.join(args)))
self.mock(spawn_device_monitor.subprocess, 'Popen', mocked_popen_calls)
try:
send_ts_mon_call = []
adb_calls = []
spawn_device_monitor.main(
['/some/adb/path', '["device_serial_1"]'])
except SimulatedSigterm:
pass
# Should build args to send_ts_mon_values without any metrics.
expected_cmd = [
spawn_device_monitor._RUN_PY,
'infra.tools.send_ts_mon_values',
'--ts-mon-device-role',
'temperature_monitor',
]
self.assertEquals(expected_cmd, send_ts_mon_call)
def test_blacklist_file_scan(self):
# Collect calls to 'subprocess.Popen'
def mocked_popen_calls(args, **kwargs):
if (args[0] == spawn_device_monitor._RUN_PY):
# ts_mon was called, so collect args and return
send_ts_mon_call.extend(args)
return None
elif (args[0] == '/some/adb/path'):
# adb was called, have it return nothing cause we're just testing
# the blacklist scanning
return mocked_Popen(stdout='', stderr=None)
else:
self.fail('Unexpected Popen call: %s' % (' '.join(args)))
self.mock(
spawn_device_monitor.subprocess,
'Popen',
mocked_popen_calls)
# Mock open and feed it a dummy blacklist file
bl_file_contents = ('{"bad_serial1": {"timestamp": 1445554107.38759, '
'"reason": "oom"}, "bad_serial2": {"timestamp": '
'1445554107.387428, "reason": "no_juice"}}')
m = mock.mock_open(read_data=bl_file_contents)
with mock.patch('os.path.exists', return_value=True):
with mock.patch('__builtin__.open', m, create=True):
try:
send_ts_mon_call = []
spawn_device_monitor.main([
'/some/adb/path',
'["good_serial1", "bad_serial1"]',
'some_master_name',
'some_builder_name',
'--blacklist-file',
'/some/blacklist/file/path'])
except SimulatedSigterm:
pass
expected_cmd = [spawn_device_monitor._RUN_PY,
'infra.tools.send_ts_mon_values',
'--ts-mon-device-role',
'temperature_monitor',
'--string',
'{"builder": "some_builder_name", "master": "some_master_name", '
'"name": "dev/status", '
'"value": "good", "device_id": "good_serial1"}',
'--string',
'{"builder": "some_builder_name", "master": "some_master_name", '
'"name": "dev/status", '
'"value": "oom", "device_id": "bad_serial1"}',
'--string',
'{"builder": "some_builder_name", "master": "some_master_name", '
'"name": "dev/status", '
'"value": "no_juice", "device_id": "bad_serial2"}',
]
self.assertItemsEqual(expected_cmd, send_ts_mon_call)
def test_main_hung_device(self):
# Collect calls to 'subprocess.Popen' and
# simulate a device that hangs on dumpsys.
def mocked_popen_calls(args, **kwargs):
if (args[0] == spawn_device_monitor._RUN_PY):
# ts_mon was called, so collect args and return
send_ts_mon_call.extend(args)
return None
elif (args[0] == '/some/adb/path'):
# adb was called, so collect args and define the output,
# but simulate a timeout when dumpsys is called
adb_calls.append(args)
if args[4].startswith('grep'):
return mocked_Popen(stdout='some_thermal_file_name',
stderr=None)
elif args[4].startswith('cat'):
return mocked_Popen(stdout='12', stderr=None)
elif args[4].startswith('dumpsys'):
return mocked_Popen(
stdout=None,
stderr=spawn_device_monitor.AdbDeviceTimeout)
else:
self.fail('Unexpected adb command: %s' % (' '.join(args)))
else:
self.fail('Unexpected Popen call: %s' % (' '.join(args)))
self.mock(spawn_device_monitor.subprocess, 'Popen', mocked_popen_calls)
# Collect calls to os.kill so that no process is actually killed
def mocked_kill(*args):
pass
self.mock(spawn_device_monitor.os, 'kill', mocked_kill)
try:
send_ts_mon_call = []
adb_calls = []
spawn_device_monitor.main(
['/some/adb/path', '["device_serial_1"]'])
except SimulatedSigterm:
pass
expected_cmd = [
spawn_device_monitor._RUN_PY, 'infra.tools.send_ts_mon_values',
'--ts-mon-device-role', 'temperature_monitor', '--float',
'{"name": "dev/mobile/cpu/temperature", '
'"value": 12, "device_id": "device_serial_1"}'
]
self.assertEquals(expected_cmd, send_ts_mon_call)
def test_main_dead_device(self):
# Collect calls to 'subprocess.Popen' and simulate an unresponsive device.
def mocked_popen_calls(args, **kwargs):
if (args[0] == spawn_device_monitor._RUN_PY):
# ts_mon was called, so collect args and return
send_ts_mon_call.extend(args)
return None
elif (args[0] == '/some/adb/path'):
# adb was called, so collect args and throw exception
adb_calls.extend(args)
raise subprocess.CalledProcessError(1, None, None)
else:
self.fail('Unexpected Popen call: %s' % (' '.join(args)))
self.mock(
spawn_device_monitor.subprocess,
'Popen',
mocked_popen_calls)
try:
send_ts_mon_call = []
adb_calls = []
spawn_device_monitor.main([
'/some/adb/path',
'["device_serial_1"]',
'some_master_name',
'some_builder_name'])
except SimulatedSigterm:
pass
# Should build args to send_ts_mon_values without any metrics.
expected_cmd = [spawn_device_monitor._RUN_PY,
'infra.tools.send_ts_mon_values',
'--ts-mon-device-role',
'temperature_monitor',
]
self.assertEquals(expected_cmd, send_ts_mon_call)
def test_main_responsive_device(self):
# Collect calls to 'subprocess.Popen' and simulate a responsive device.
def mocked_popen_calls(args, **kwargs):
if (args[0] == spawn_device_monitor._RUN_PY):
# ts_mon was called, so collect args and return
send_ts_mon_call.extend(args)
return None
elif (args[0] == '/some/adb/path'):
# adb was called, so collect args and define the output
adb_calls.append(args)
if (args[2] == 'device_serial_1'):
if args[4].startswith('grep'):
return mocked_Popen(stdout='some_thermal_file_name', stderr=None)
elif args[4].startswith('cat'):
return mocked_Popen(stdout='12', stderr=None)
elif args[4].startswith('dumpsys'):
return mocked_Popen(stdout='temperature: 456\nlevel: 96',
stderr=None)
else:
self.fail('Unexpected adb command: %s' % (' '.join(args)))
elif (args[2] == 'device_serial_2'):
if args[4].startswith('grep'):
return mocked_Popen(stdout='some_thermal_file_name', stderr=None)
elif args[4].startswith('cat'):
return mocked_Popen(stdout='56789', stderr=None)
elif args[4].startswith('dumpsys'):
return mocked_Popen(stdout='level: 11\ntemperature: 987',
stderr=None)
else:
self.fail('Unexpected adb command: %s' % (' '.join(args)))
else:
self.fail('Unexpected device serial: %s' % (' '.join(args)))
else:
self.fail('Unexpected Popen call: %s' % (' '.join(args)))
self.mock(
spawn_device_monitor.subprocess,
'Popen',
mocked_popen_calls)
try:
send_ts_mon_call = []
adb_calls = []
spawn_device_monitor.main([
'/some/adb/path',
'["device_serial_1", "device_serial_2"]',
'some_master_name',
'some_builder_name'])
except SimulatedSigterm:
pass
# Should build args to send_ts_mon_values correctly.
expected_cmd = [spawn_device_monitor._RUN_PY,
'infra.tools.send_ts_mon_values',
'--ts-mon-device-role',
'temperature_monitor',
'--float',
'{"builder": "some_builder_name", "master": "some_master_name", '
'"name": "dev/cpu/temperature", '
'"value": 12, "device_id": "device_serial_1"}',
'--float',
'{"builder": "some_builder_name", "master": "some_master_name", '
'"name": "dev/battery/temperature", '
'"value": 456, "device_id": "device_serial_1"}',
'--float',
'{"builder": "some_builder_name", "master": "some_master_name", '
'"name": "dev/battery/charge", '
'"value": 96, "device_id": "device_serial_1"}',
'--float',
'{"builder": "some_builder_name", '
'"master": "some_master_name", "name": "dev/cpu/temperature", '
'"value": 56, "device_id": "device_serial_2"}',
'--float',
'{"builder": "some_builder_name", "master": "some_master_name", '
'"name": "dev/battery/temperature", '
'"value": 987, "device_id": "device_serial_2"}',
'--float',
'{"builder": "some_builder_name", "master": "some_master_name", '
'"name": "dev/battery/charge", '
'"value": 11, "device_id": "device_serial_2"}',
]
self.assertEquals(expected_cmd, send_ts_mon_call)
def test_main_hung_device(self):
# Collect calls to 'subprocess.Popen' and
# simulate a device that hangs on dumpsys.
def mocked_popen_calls(args, **kwargs):
if (args[0] == spawn_device_monitor._RUN_PY):
# ts_mon was called, so collect args and return
send_ts_mon_call.extend(args)
return None
elif (args[0] == '/some/adb/path'):
# adb was called, so collect args and define the output,
# but simulate a timeout when dumpsys is called
adb_calls.append(args)
if args[4].startswith('grep'):
return mocked_Popen(stdout='some_thermal_file_name', stderr=None)
elif args[4].startswith('cat'):
return mocked_Popen(stdout='12', stderr=None)
elif args[4].startswith('dumpsys'):
return mocked_Popen(
stdout=None,
stderr=spawn_device_monitor.AdbDeviceTimeout
)
else:
self.fail('Unexpected adb command: %s' % (' '.join(args)))
else:
self.fail('Unexpected Popen call: %s' % (' '.join(args)))
self.mock(
spawn_device_monitor.subprocess,
'Popen',
mocked_popen_calls)
# Collect calls to os.kill so that no process is actually killed
def mocked_kill(*args):
pass
self.mock(
spawn_device_monitor.os,
'kill',
mocked_kill)
try:
send_ts_mon_call = []
adb_calls = []
spawn_device_monitor.main([
'/some/adb/path',
'["device_serial_1"]',
'some_master_name',
'some_builder_name'])
except SimulatedSigterm:
pass
expected_cmd = [spawn_device_monitor._RUN_PY,
'infra.tools.send_ts_mon_values',
'--ts-mon-device-role',
'temperature_monitor',
'--float',
'{"builder": "some_builder_name", "master": "some_master_name", '
'"name": "dev/cpu/temperature", '
'"value": 12, "device_id": "device_serial_1"}'
]
self.assertEquals(expected_cmd, send_ts_mon_call)