diff options
Diffstat (limited to 'tests/mobly/controller_manager_test.py')
-rwxr-xr-x | tests/mobly/controller_manager_test.py | 99 |
1 files changed, 57 insertions, 42 deletions
diff --git a/tests/mobly/controller_manager_test.py b/tests/mobly/controller_manager_test.py index 53e4cfa..46aa273 100755 --- a/tests/mobly/controller_manager_test.py +++ b/tests/mobly/controller_manager_test.py @@ -49,14 +49,16 @@ class ControllerManagerTest(unittest.TestCase): def test_register_controller_no_config(self): c_manager = controller_manager.ControllerManager('SomeClass', {}) - with self.assertRaisesRegex(signals.ControllerError, - 'No corresponding config found for'): + with self.assertRaisesRegex( + signals.ControllerError, 'No corresponding config found for' + ): c_manager.register_controller(mock_controller) def test_register_controller_no_config_for_not_required(self): c_manager = controller_manager.ControllerManager('SomeClass', {}) self.assertIsNone( - c_manager.register_controller(mock_controller, required=False)) + c_manager.register_controller(mock_controller, required=False) + ) def test_register_controller_dup_register(self): """Verifies correctness of registration, internal tally of controllers @@ -65,8 +67,9 @@ class ControllerManagerTest(unittest.TestCase): """ mock_ctrlr_config_name = mock_controller.MOBLY_CONTROLLER_CONFIG_NAME controller_configs = {mock_ctrlr_config_name: ['magic1', 'magic2']} - c_manager = controller_manager.ControllerManager('SomeClass', - controller_configs) + c_manager = controller_manager.ControllerManager( + 'SomeClass', controller_configs + ) c_manager.register_controller(mock_controller) registered_name = 'mock_controller' self.assertTrue(registered_name in c_manager._controller_objects) @@ -81,8 +84,9 @@ class ControllerManagerTest(unittest.TestCase): def test_register_controller_return_value(self): mock_ctrlr_config_name = mock_controller.MOBLY_CONTROLLER_CONFIG_NAME controller_configs = {mock_ctrlr_config_name: ['magic1', 'magic2']} - c_manager = controller_manager.ControllerManager('SomeClass', - controller_configs) + c_manager = controller_manager.ControllerManager( + 'SomeClass', controller_configs + ) magic_devices = c_manager.register_controller(mock_controller) self.assertEqual(magic_devices[0].magic, 'magic1') self.assertEqual(magic_devices[1].magic, 'magic2') @@ -90,8 +94,9 @@ class ControllerManagerTest(unittest.TestCase): def test_register_controller_change_return_value(self): mock_ctrlr_config_name = mock_controller.MOBLY_CONTROLLER_CONFIG_NAME controller_configs = {mock_ctrlr_config_name: ['magic1', 'magic2']} - c_manager = controller_manager.ControllerManager('SomeClass', - controller_configs) + c_manager = controller_manager.ControllerManager( + 'SomeClass', controller_configs + ) magic_devices = c_manager.register_controller(mock_controller) magic1 = magic_devices.pop(0) self.assertIs(magic1, c_manager._controller_objects['mock_controller'][0]) @@ -100,8 +105,9 @@ class ControllerManagerTest(unittest.TestCase): def test_register_controller_less_than_min_number(self): mock_ctrlr_config_name = mock_controller.MOBLY_CONTROLLER_CONFIG_NAME controller_configs = {mock_ctrlr_config_name: ['magic1', 'magic2']} - c_manager = controller_manager.ControllerManager('SomeClass', - controller_configs) + c_manager = controller_manager.ControllerManager( + 'SomeClass', controller_configs + ) expected_msg = 'Expected to get at least 3 controller objects, got 2.' with self.assertRaisesRegex(signals.ControllerError, expected_msg): c_manager.register_controller(mock_controller, min_number=3) @@ -110,19 +116,22 @@ class ControllerManagerTest(unittest.TestCase): def test_get_controller_info_record_not_serializable(self, _): mock_ctrlr_config_name = mock_controller.MOBLY_CONTROLLER_CONFIG_NAME controller_configs = {mock_ctrlr_config_name: ['magic1', 'magic2']} - c_manager = controller_manager.ControllerManager('SomeClass', - controller_configs) + c_manager = controller_manager.ControllerManager( + 'SomeClass', controller_configs + ) c_manager.register_controller(mock_controller) record = c_manager.get_controller_info_records()[0] actual_controller_info = record.controller_info - self.assertEqual(actual_controller_info, - "[{'MyMagic': 'magic1'}, {'MyMagic': 'magic2'}]") + self.assertEqual( + actual_controller_info, "[{'MyMagic': 'magic1'}, {'MyMagic': 'magic2'}]" + ) def test_controller_record_exists_without_get_info(self): mock_ctrlr_config_name = mock_controller.MOBLY_CONTROLLER_CONFIG_NAME controller_configs = {mock_ctrlr_config_name: ['magic1', 'magic2']} - c_manager = controller_manager.ControllerManager('SomeClass', - controller_configs) + c_manager = controller_manager.ControllerManager( + 'SomeClass', controller_configs + ) get_info = getattr(mock_controller, 'get_info') delattr(mock_controller, 'get_info') try: @@ -139,8 +148,9 @@ class ControllerManagerTest(unittest.TestCase): mock_get_info_func.return_value = None mock_ctrlr_config_name = mock_controller.MOBLY_CONTROLLER_CONFIG_NAME controller_configs = {mock_ctrlr_config_name: ['magic1', 'magic2']} - c_manager = controller_manager.ControllerManager('SomeClass', - controller_configs) + c_manager = controller_manager.ControllerManager( + 'SomeClass', controller_configs + ) c_manager.register_controller(mock_controller) record = c_manager.get_controller_info_records()[0] self.assertIsNone(record.controller_info) @@ -149,13 +159,15 @@ class ControllerManagerTest(unittest.TestCase): @mock.patch('mobly.expects._ExpectErrorRecorder.add_error') @mock.patch('tests.lib.mock_controller.get_info') - def test_get_controller_info_records_error(self, mock_get_info_func, - mock_add_error): + def test_get_controller_info_records_error( + self, mock_get_info_func, mock_add_error + ): mock_get_info_func.side_effect = Exception('Record info failed.') mock_ctrlr_config_name = mock_controller.MOBLY_CONTROLLER_CONFIG_NAME controller_configs = {mock_ctrlr_config_name: ['magic1', 'magic2']} - c_manager = controller_manager.ControllerManager('SomeClass', - controller_configs) + c_manager = controller_manager.ControllerManager( + 'SomeClass', controller_configs + ) c_manager.register_controller(mock_controller) self.assertFalse(c_manager.get_controller_info_records()) mock_add_error.assert_called_once() @@ -165,36 +177,37 @@ class ControllerManagerTest(unittest.TestCase): def test_get_controller_info_records(self): mock_ctrlr_config_name = mock_controller.MOBLY_CONTROLLER_CONFIG_NAME controller_configs = {mock_ctrlr_config_name: ['magic1', 'magic2']} - c_manager = controller_manager.ControllerManager('SomeClass', - controller_configs) + c_manager = controller_manager.ControllerManager( + 'SomeClass', controller_configs + ) c_manager.register_controller(mock_controller) record = c_manager.get_controller_info_records()[0] record_dict = record.to_dict() record_dict.pop('Timestamp') self.assertEqual( - record_dict, { - 'Controller Info': [{ - 'MyMagic': 'magic1' - }, { - 'MyMagic': 'magic2' - }], + record_dict, + { + 'Controller Info': [{'MyMagic': 'magic1'}, {'MyMagic': 'magic2'}], 'Controller Name': 'MagicDevice', - 'Test Class': 'SomeClass' - }) + 'Test Class': 'SomeClass', + }, + ) def test_get_controller_info_without_registration(self): mock_ctrlr_config_name = mock_controller.MOBLY_CONTROLLER_CONFIG_NAME controller_configs = {mock_ctrlr_config_name: ['magic1', 'magic2']} - c_manager = controller_manager.ControllerManager('SomeClass', - controller_configs) + c_manager = controller_manager.ControllerManager( + 'SomeClass', controller_configs + ) self.assertFalse(c_manager.get_controller_info_records()) @mock.patch('tests.lib.mock_controller.destroy') def test_unregister_controller(self, mock_destroy_func): mock_ctrlr_config_name = mock_controller.MOBLY_CONTROLLER_CONFIG_NAME controller_configs = {mock_ctrlr_config_name: ['magic1', 'magic2']} - c_manager = controller_manager.ControllerManager('SomeClass', - controller_configs) + c_manager = controller_manager.ControllerManager( + 'SomeClass', controller_configs + ) objects = c_manager.register_controller(mock_controller) c_manager.unregister_controllers() mock_destroy_func.assert_called_once_with(objects) @@ -206,8 +219,9 @@ class ControllerManagerTest(unittest.TestCase): def test_unregister_controller_error(self, mock_destroy_func, mock_add_error): mock_ctrlr_config_name = mock_controller.MOBLY_CONTROLLER_CONFIG_NAME controller_configs = {mock_ctrlr_config_name: ['magic1', 'magic2']} - c_manager = controller_manager.ControllerManager('SomeClass', - controller_configs) + c_manager = controller_manager.ControllerManager( + 'SomeClass', controller_configs + ) c_manager.register_controller(mock_controller) mock_destroy_func.side_effect = Exception('Failed in destroy.') c_manager.unregister_controllers() @@ -221,13 +235,14 @@ class ControllerManagerTest(unittest.TestCase): def test_unregister_controller_without_registration(self, mock_destroy_func): mock_ctrlr_config_name = mock_controller.MOBLY_CONTROLLER_CONFIG_NAME controller_configs = {mock_ctrlr_config_name: ['magic1', 'magic2']} - c_manager = controller_manager.ControllerManager('SomeClass', - controller_configs) + c_manager = controller_manager.ControllerManager( + 'SomeClass', controller_configs + ) c_manager.unregister_controllers() mock_destroy_func.assert_not_called() self.assertFalse(c_manager._controller_objects) self.assertFalse(c_manager._controller_modules) -if __name__ == "__main__": +if __name__ == '__main__': unittest.main() |