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()