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
|
#!/usr/bin/env python
# Copyright 2017 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.
"""
Unit tests for some APIs with conditional logic in adb_wrapper.py
"""
import unittest
from devil import devil_env
from devil.android import device_errors
from devil.android.sdk import adb_wrapper
with devil_env.SysPath(devil_env.PYMOCK_PATH):
import mock # pylint: disable=import-error
class AdbWrapperTest(unittest.TestCase):
def setUp(self):
self.device_serial = 'ABC12345678'
self.adb = adb_wrapper.AdbWrapper(self.device_serial)
def _MockRunDeviceAdbCmd(self, return_value):
return mock.patch.object(
self.adb, '_RunDeviceAdbCmd',
mock.Mock(side_effect=None, return_value=return_value))
def testDisableVerityWhenDisabled(self):
with self._MockRunDeviceAdbCmd('Verity already disabled on /system'):
self.adb.DisableVerity()
def testDisableVerityWhenEnabled(self):
with self._MockRunDeviceAdbCmd(
'Verity disabled on /system\nNow reboot your device for settings to '
'take effect'):
self.adb.DisableVerity()
def testEnableVerityWhenEnabled(self):
with self._MockRunDeviceAdbCmd('Verity already enabled on /system'):
self.adb.EnableVerity()
def testEnableVerityWhenDisabled(self):
with self._MockRunDeviceAdbCmd(
'Verity enabled on /system\nNow reboot your device for settings to '
'take effect'):
self.adb.EnableVerity()
def testFailEnableVerity(self):
with self._MockRunDeviceAdbCmd('error: closed'):
self.assertRaises(device_errors.AdbCommandFailedError,
self.adb.EnableVerity)
def testFailDisableVerity(self):
with self._MockRunDeviceAdbCmd('error: closed'):
self.assertRaises(device_errors.AdbCommandFailedError,
self.adb.DisableVerity)
@mock.patch('devil.utils.cmd_helper.GetCmdStatusAndOutputWithTimeout')
def testDeviceUnreachable(self, get_cmd_mock):
get_cmd_mock.return_value = (
1, "error: device '%s' not found" % self.device_serial)
self.assertRaises(device_errors.DeviceUnreachableError, self.adb.Shell,
'/bin/true')
@mock.patch('devil.utils.cmd_helper.GetCmdStatusAndOutputWithTimeout')
def testWaitingForDevice(self, get_cmd_mock):
get_cmd_mock.return_value = (1, '- waiting for device - ')
self.assertRaises(device_errors.DeviceUnreachableError, self.adb.Shell,
'/bin/true')
|