Merge "Move GetDeviceTest from adb's test_device.py to adb package."
This commit is contained in:
commit
93e5cd85d1
1 changed files with 0 additions and 56 deletions
|
|
@ -34,11 +34,8 @@ import tempfile
|
||||||
import time
|
import time
|
||||||
import unittest
|
import unittest
|
||||||
|
|
||||||
import mock
|
|
||||||
|
|
||||||
import adb
|
import adb
|
||||||
|
|
||||||
|
|
||||||
def requires_root(func):
|
def requires_root(func):
|
||||||
def wrapper(self, *args):
|
def wrapper(self, *args):
|
||||||
if self.device.get_prop('ro.debuggable') != '1':
|
if self.device.get_prop('ro.debuggable') != '1':
|
||||||
|
|
@ -76,59 +73,6 @@ def requires_non_root(func):
|
||||||
return wrapper
|
return wrapper
|
||||||
|
|
||||||
|
|
||||||
class GetDeviceTest(unittest.TestCase):
|
|
||||||
def setUp(self):
|
|
||||||
self.android_serial = os.getenv('ANDROID_SERIAL')
|
|
||||||
if 'ANDROID_SERIAL' in os.environ:
|
|
||||||
del os.environ['ANDROID_SERIAL']
|
|
||||||
|
|
||||||
def tearDown(self):
|
|
||||||
if self.android_serial is not None:
|
|
||||||
os.environ['ANDROID_SERIAL'] = self.android_serial
|
|
||||||
else:
|
|
||||||
if 'ANDROID_SERIAL' in os.environ:
|
|
||||||
del os.environ['ANDROID_SERIAL']
|
|
||||||
|
|
||||||
@mock.patch('adb.device.get_devices')
|
|
||||||
def test_explicit(self, mock_get_devices):
|
|
||||||
mock_get_devices.return_value = ['foo', 'bar']
|
|
||||||
device = adb.get_device('foo')
|
|
||||||
self.assertEqual(device.serial, 'foo')
|
|
||||||
|
|
||||||
@mock.patch('adb.device.get_devices')
|
|
||||||
def test_from_env(self, mock_get_devices):
|
|
||||||
mock_get_devices.return_value = ['foo', 'bar']
|
|
||||||
os.environ['ANDROID_SERIAL'] = 'foo'
|
|
||||||
device = adb.get_device()
|
|
||||||
self.assertEqual(device.serial, 'foo')
|
|
||||||
|
|
||||||
@mock.patch('adb.device.get_devices')
|
|
||||||
def test_arg_beats_env(self, mock_get_devices):
|
|
||||||
mock_get_devices.return_value = ['foo', 'bar']
|
|
||||||
os.environ['ANDROID_SERIAL'] = 'bar'
|
|
||||||
device = adb.get_device('foo')
|
|
||||||
self.assertEqual(device.serial, 'foo')
|
|
||||||
|
|
||||||
@mock.patch('adb.device.get_devices')
|
|
||||||
def test_no_such_device(self, mock_get_devices):
|
|
||||||
mock_get_devices.return_value = ['foo', 'bar']
|
|
||||||
self.assertRaises(adb.DeviceNotFoundError, adb.get_device, ['baz'])
|
|
||||||
|
|
||||||
os.environ['ANDROID_SERIAL'] = 'baz'
|
|
||||||
self.assertRaises(adb.DeviceNotFoundError, adb.get_device)
|
|
||||||
|
|
||||||
@mock.patch('adb.device.get_devices')
|
|
||||||
def test_unique_device(self, mock_get_devices):
|
|
||||||
mock_get_devices.return_value = ['foo']
|
|
||||||
device = adb.get_device()
|
|
||||||
self.assertEqual(device.serial, 'foo')
|
|
||||||
|
|
||||||
@mock.patch('adb.device.get_devices')
|
|
||||||
def test_no_unique_device(self, mock_get_devices):
|
|
||||||
mock_get_devices.return_value = ['foo', 'bar']
|
|
||||||
self.assertRaises(adb.NoUniqueDeviceError, adb.get_device)
|
|
||||||
|
|
||||||
|
|
||||||
class DeviceTest(unittest.TestCase):
|
class DeviceTest(unittest.TestCase):
|
||||||
def setUp(self):
|
def setUp(self):
|
||||||
self.device = adb.get_device()
|
self.device = adb.get_device()
|
||||||
|
|
|
||||||
Loading…
Add table
Reference in a new issue