LineageOS/android_development
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Fix tests.

Browse Source 
Bug: None
Test: pytest
Change-Id: Ibe4259750a8d6f1800b5f659a80b301226d6929d
This commit is contained in:
Dan Albert
2022-12-16 13:59:48 -08:00
parent 91e796d227
commit 808c90cf8d
1 changed files with 13 additions and 13 deletions
Download Patch File Download Diff File Expand all files Collapse all files

26
python-packages/adb/tests/test_adb.py
View File

@@ -15,7 +15,7 @@
# #
import os import os
import unittest import unittest
import mock from unittest.mock import Mock, patch
import adb import adb
@@ -32,42 +32,42 @@ class GetDeviceTest(unittest.TestCase):
if 'ANDROID_SERIAL' in os.environ: if 'ANDROID_SERIAL' in os.environ:
del os.environ['ANDROID_SERIAL'] del os.environ['ANDROID_SERIAL']
@mock.patch('adb.device.get_devices') @patch('adb.get_devices')
def test_explicit(self, mock_get_devices: mock.Mock) -> None: def test_explicit(self, mock_get_devices: Mock) -> None:
mock_get_devices.return_value = ['foo', 'bar'] mock_get_devices.return_value = ['foo', 'bar']
device = adb.get_device('foo') device = adb.get_device('foo')
self.assertEqual(device.serial, 'foo') self.assertEqual(device.serial, 'foo')
@mock.patch('adb.device.get_devices') @patch('adb.get_devices')
def test_from_env(self, mock_get_devices: mock.Mock) -> None: def test_from_env(self, mock_get_devices: Mock) -> None:
mock_get_devices.return_value = ['foo', 'bar'] mock_get_devices.return_value = ['foo', 'bar']
os.environ['ANDROID_SERIAL'] = 'foo' os.environ['ANDROID_SERIAL'] = 'foo'
device = adb.get_device() device = adb.get_device()
self.assertEqual(device.serial, 'foo') self.assertEqual(device.serial, 'foo')
@mock.patch('adb.device.get_devices') @patch('adb.get_devices')
def test_arg_beats_env(self, mock_get_devices: mock.Mock) -> None: def test_arg_beats_env(self, mock_get_devices: Mock) -> None:
mock_get_devices.return_value = ['foo', 'bar'] mock_get_devices.return_value = ['foo', 'bar']
os.environ['ANDROID_SERIAL'] = 'bar' os.environ['ANDROID_SERIAL'] = 'bar'
device = adb.get_device('foo') device = adb.get_device('foo')
self.assertEqual(device.serial, 'foo') self.assertEqual(device.serial, 'foo')
@mock.patch('adb.device.get_devices') @patch('adb.get_devices')
def test_no_such_device(self, mock_get_devices: mock.Mock) -> None: def test_no_such_device(self, mock_get_devices: Mock) -> None:
mock_get_devices.return_value = ['foo', 'bar'] mock_get_devices.return_value = ['foo', 'bar']
self.assertRaises(adb.DeviceNotFoundError, adb.get_device, ['baz']) self.assertRaises(adb.DeviceNotFoundError, adb.get_device, ['baz'])
os.environ['ANDROID_SERIAL'] = 'baz' os.environ['ANDROID_SERIAL'] = 'baz'
self.assertRaises(adb.DeviceNotFoundError, adb.get_device) self.assertRaises(adb.DeviceNotFoundError, adb.get_device)
@mock.patch('adb.device.get_devices') @patch('adb.get_devices')
def test_unique_device(self, mock_get_devices: mock.Mock) -> None: def test_unique_device(self, mock_get_devices: Mock) -> None:
mock_get_devices.return_value = ['foo'] mock_get_devices.return_value = ['foo']
device = adb.get_device() device = adb.get_device()
self.assertEqual(device.serial, 'foo') self.assertEqual(device.serial, 'foo')
@mock.patch('adb.device.get_devices') @patch('adb.get_devices')
def test_no_unique_device(self, mock_get_devices: mock.Mock) -> None: def test_no_unique_device(self, mock_get_devices: Mock) -> None:
mock_get_devices.return_value = ['foo', 'bar'] mock_get_devices.return_value = ['foo', 'bar']
self.assertRaises(adb.NoUniqueDeviceError, adb.get_device) self.assertRaises(adb.NoUniqueDeviceError, adb.get_device)