def test_load_device(self, mock_get_instances, mock_list_adb_devices):
mock_list_adb_devices.return_value = ['1234', '4312']
mock_device = mock.MagicMock(spec=android_device.AndroidDevice)
mock_get_instances.return_value = [mock_device]
json_shell = jsonrpc_shell_base.JsonRpcShellBase()
json_shell.load_device(serial='1234')
self.assertEqual(json_shell._ad, mock_device)
def test_load_device_when_device_not_found(self, mock_list_adb_devices):
mock_list_adb_devices.return_value = ['4321']
json_shell = jsonrpc_shell_base.JsonRpcShellBase()
with self.assertRaisesRegex(jsonrpc_shell_base.Error,
'Device "1234" is not found by adb.'):
json_shell.load_device(serial='1234')
def test_load_device_when_unspecified_device(self, mock_list_adb_devices):
mock_list_adb_devices.return_value = ['1234', '4321']
json_shell = jsonrpc_shell_base.JsonRpcShellBase()
with self.assertRaisesRegex(jsonrpc_shell_base.Error,
'Expected one phone.*'):
json_shell.load_device()
def test_load_device_when_no_devices(self, mock_list_adb_devices):
mock_list_adb_devices.return_value = []
json_shell = jsonrpc_shell_base.JsonRpcShellBase()
with self.assertRaisesRegex(jsonrpc_shell_base.Error,
'No adb device found!'):
json_shell.load_device()
