diff options
Diffstat (limited to 'tests/mobly/controllers/android_device_lib/service_manager_test.py')
| -rwxr-xr-x | tests/mobly/controllers/android_device_lib/service_manager_test.py | 56 |
1 files changed, 34 insertions, 22 deletions
diff --git a/tests/mobly/controllers/android_device_lib/service_manager_test.py b/tests/mobly/controllers/android_device_lib/service_manager_test.py index 1eae045..471c83f 100755 --- a/tests/mobly/controllers/android_device_lib/service_manager_test.py +++ b/tests/mobly/controllers/android_device_lib/service_manager_test.py @@ -62,7 +62,7 @@ class ServiceManagerTest(unittest.TestCase): def assert_recorded_one_error(self, message): self.assertEqual(expects.recorder.error_count, 1) - for _, error in (expects.DEFAULT_TEST_RESULT_RECORD.extra_errors.items()): + for _, error in expects.DEFAULT_TEST_RESULT_RECORD.extra_errors.items(): self.assertIn(message, error.details) def test_service_manager_instantiation(self): @@ -102,13 +102,13 @@ class ServiceManagerTest(unittest.TestCase): manager.register('mock_service', base_service) def test_register_wrong_subclass_type(self): - class MyClass: pass manager = service_manager.ServiceManager(mock.MagicMock()) - with self.assertRaisesRegex(service_manager.Error, - '.* is not a subclass of BaseService!'): + with self.assertRaisesRegex( + service_manager.Error, '.* is not a subclass of BaseService!' + ): manager.register('mock_service', MyClass) def test_register_dup_alias(self): @@ -139,7 +139,8 @@ class ServiceManagerTest(unittest.TestCase): service1.ha = mock.MagicMock() service2.ha = mock.MagicMock() manager.for_each( - lambda service: manager._service_objects.pop(service.alias)) + lambda service: manager._service_objects.pop(service.alias) + ) self.assertFalse(manager._service_objects) def test_for_each_one_fail(self): @@ -169,14 +170,16 @@ class ServiceManagerTest(unittest.TestCase): service3.create_output_excerpts = mock.MagicMock() service1.create_output_excerpts.return_value = ['path/to/1.txt'] service2.create_output_excerpts.return_value = [ - 'path/to/2-1.txt', 'path/to/2-2.txt' + 'path/to/2-1.txt', + 'path/to/2-2.txt', ] service3.create_output_excerpts.return_value = [] mock_test_info = mock.MagicMock(output_path='path/to') result = manager.create_output_excerpts_all(mock_test_info) self.assertEqual(result['mock_service1'], ['path/to/1.txt']) - self.assertEqual(result['mock_service2'], - ['path/to/2-1.txt', 'path/to/2-2.txt']) + self.assertEqual( + result['mock_service2'], ['path/to/2-1.txt', 'path/to/2-2.txt'] + ) self.assertEqual(result['mock_service3'], []) def test_unregister(self): @@ -201,7 +204,8 @@ class ServiceManagerTest(unittest.TestCase): manager = service_manager.ServiceManager(mock.MagicMock()) with self.assertRaisesRegex( service_manager.Error, - '.* No service is registered with alias "mock_service"'): + '.* No service is registered with alias "mock_service"', + ): manager.unregister('mock_service') def test_unregister_handle_error_from_stop(self): @@ -211,7 +215,8 @@ class ServiceManagerTest(unittest.TestCase): service.stop_func.side_effect = Exception('Something failed in stop.') manager.unregister('mock_service') self.assert_recorded_one_error( - 'Failed to stop service instance "mock_service".') + 'Failed to stop service instance "mock_service".' + ) def test_unregister_all(self): manager = service_manager.ServiceManager(mock.MagicMock()) @@ -238,7 +243,8 @@ class ServiceManagerTest(unittest.TestCase): self.assertTrue(service1.is_alive) self.assertFalse(service2.is_alive) self.assert_recorded_one_error( - 'Failed to stop service instance "mock_service1".') + 'Failed to stop service instance "mock_service1".' + ) def test_start_all(self): manager = service_manager.ServiceManager(mock.MagicMock()) @@ -256,7 +262,8 @@ class ServiceManagerTest(unittest.TestCase): self.assertEqual(service2.start_func.call_count, 1) self.assertEqual( mock_call_tracker.mock_calls, - [mock.call.start1(None), mock.call.start2(None)]) + [mock.call.start1(None), mock.call.start2(None)], + ) def test_start_all_with_already_started_services(self): manager = service_manager.ServiceManager(mock.MagicMock()) @@ -295,8 +302,9 @@ class ServiceManagerTest(unittest.TestCase): manager.stop_all() self.assertFalse(service1.is_alive) self.assertFalse(service2.is_alive) - self.assertEqual(mock_call_tracker.mock_calls, - [mock.call.stop2(), mock.call.stop1()]) + self.assertEqual( + mock_call_tracker.mock_calls, [mock.call.stop2(), mock.call.stop1()] + ) self.assertEqual(service1.start_func.call_count, 1) self.assertEqual(service2.start_func.call_count, 1) self.assertEqual(service1.stop_func.call_count, 1) @@ -359,8 +367,8 @@ class ServiceManagerTest(unittest.TestCase): mock_call_tracker.pause2 = service2.pause_func manager.pause_all() self.assertEqual( - mock_call_tracker.mock_calls, - [mock.call.pause2(), mock.call.pause1()]) + mock_call_tracker.mock_calls, [mock.call.pause2(), mock.call.pause1()] + ) self.assertEqual(service1.pause_func.call_count, 1) self.assertEqual(service2.pause_func.call_count, 1) self.assertEqual(service1.resume_func.call_count, 0) @@ -392,8 +400,8 @@ class ServiceManagerTest(unittest.TestCase): manager.pause_all() manager.resume_all() self.assertEqual( - mock_call_tracker.mock_calls, - [mock.call.resume1(), mock.call.resume2()]) + mock_call_tracker.mock_calls, [mock.call.resume1(), mock.call.resume2()] + ) self.assertEqual(service1.pause_func.call_count, 1) self.assertEqual(service2.pause_func.call_count, 1) self.assertEqual(service1.resume_func.call_count, 1) @@ -434,8 +442,10 @@ class ServiceManagerTest(unittest.TestCase): def test_start_services_non_existent(self): manager = service_manager.ServiceManager(mock.MagicMock()) - msg = ('.* No service is registered under the name "mock_service", ' - 'cannot start.') + msg = ( + '.* No service is registered under the name "mock_service", ' + 'cannot start.' + ) with self.assertRaisesRegex(service_manager.Error, msg): manager.start_services(['mock_service']) @@ -452,8 +462,10 @@ class ServiceManagerTest(unittest.TestCase): def test_resume_services_non_existent(self): manager = service_manager.ServiceManager(mock.MagicMock()) - msg = ('.* No service is registered under the name "mock_service", ' - 'cannot resume.') + msg = ( + '.* No service is registered under the name "mock_service", ' + 'cannot resume.' + ) with self.assertRaisesRegex(service_manager.Error, msg): manager.resume_services(['mock_service']) |