diff options
Diffstat (limited to 'tests/mobly/controllers/android_device_lib/services/snippet_management_service_test.py')
| -rwxr-xr-x | tests/mobly/controllers/android_device_lib/services/snippet_management_service_test.py | 70 |
1 files changed, 47 insertions, 23 deletions
diff --git a/tests/mobly/controllers/android_device_lib/services/snippet_management_service_test.py b/tests/mobly/controllers/android_device_lib/services/snippet_management_service_test.py index 16a30aa..b6431d5 100755 --- a/tests/mobly/controllers/android_device_lib/services/snippet_management_service_test.py +++ b/tests/mobly/controllers/android_device_lib/services/snippet_management_service_test.py @@ -19,7 +19,9 @@ from mobly.controllers.android_device_lib import snippet_client_v2 from mobly.controllers.android_device_lib.services import snippet_management_service MOCK_PACKAGE = 'com.mock.package' -SNIPPET_CLIENT_V2_CLASS_PATH = 'mobly.controllers.android_device_lib.snippet_client_v2.SnippetClientV2' +SNIPPET_CLIENT_V2_CLASS_PATH = ( + 'mobly.controllers.android_device_lib.snippet_client_v2.SnippetClientV2' +) class SnippetManagementServiceTest(unittest.TestCase): @@ -27,7 +29,8 @@ class SnippetManagementServiceTest(unittest.TestCase): def test_empty_manager_start_stop(self): manager = snippet_management_service.SnippetManagementService( - mock.MagicMock()) + mock.MagicMock() + ) manager.start() # When no client is registered, manager is never alive. self.assertFalse(manager.is_alive) @@ -38,21 +41,24 @@ class SnippetManagementServiceTest(unittest.TestCase): def test_get_snippet_client(self, mock_class): mock_client = mock_class.return_value manager = snippet_management_service.SnippetManagementService( - mock.MagicMock()) + mock.MagicMock() + ) manager.add_snippet_client('foo', MOCK_PACKAGE) self.assertEqual(manager.get_snippet_client('foo'), mock_client) @mock.patch(SNIPPET_CLIENT_V2_CLASS_PATH) def test_get_snippet_client_fail(self, _): manager = snippet_management_service.SnippetManagementService( - mock.MagicMock()) + mock.MagicMock() + ) self.assertIsNone(manager.get_snippet_client('foo')) @mock.patch(SNIPPET_CLIENT_V2_CLASS_PATH) def test_stop_with_live_client(self, mock_class): mock_client = mock_class.return_value manager = snippet_management_service.SnippetManagementService( - mock.MagicMock()) + mock.MagicMock() + ) manager.add_snippet_client('foo', MOCK_PACKAGE) mock_client.initialize.assert_called_once_with() manager.stop() @@ -67,28 +73,35 @@ class SnippetManagementServiceTest(unittest.TestCase): def test_add_snippet_client_without_config(self, mock_class): mock_client = mock_class.return_value manager = snippet_management_service.SnippetManagementService( - mock.MagicMock()) + mock.MagicMock() + ) manager.add_snippet_client('foo', MOCK_PACKAGE) mock_class.assert_called_once_with( - package=mock.ANY, ad=mock.ANY, config=None) + package=mock.ANY, ad=mock.ANY, config=None + ) @mock.patch(SNIPPET_CLIENT_V2_CLASS_PATH) def test_add_snippet_client_with_config(self, mock_class): mock_client = mock_class.return_value manager = snippet_management_service.SnippetManagementService( - mock.MagicMock()) + mock.MagicMock() + ) snippet_config = snippet_client_v2.Config() manager.add_snippet_client('foo', MOCK_PACKAGE, snippet_config) mock_class.assert_called_once_with( - package=mock.ANY, ad=mock.ANY, config=snippet_config) + package=mock.ANY, ad=mock.ANY, config=snippet_config + ) @mock.patch(SNIPPET_CLIENT_V2_CLASS_PATH) def test_add_snippet_client_dup_name(self, _): manager = snippet_management_service.SnippetManagementService( - mock.MagicMock()) + mock.MagicMock() + ) manager.add_snippet_client('foo', MOCK_PACKAGE) - msg = ('.* Name "foo" is already registered with package ".*", it ' - 'cannot be used again.') + msg = ( + '.* Name "foo" is already registered with package ".*", it ' + 'cannot be used again.' + ) with self.assertRaisesRegex(snippet_management_service.Error, msg): manager.add_snippet_client('foo', MOCK_PACKAGE + 'ha') @@ -97,10 +110,13 @@ class SnippetManagementServiceTest(unittest.TestCase): mock_client = mock_class.return_value mock_client.package = MOCK_PACKAGE manager = snippet_management_service.SnippetManagementService( - mock.MagicMock()) + mock.MagicMock() + ) manager.add_snippet_client('foo', MOCK_PACKAGE) - msg = ('Snippet package "com.mock.package" has already been loaded ' - 'under name "foo".') + msg = ( + 'Snippet package "com.mock.package" has already been loaded ' + 'under name "foo".' + ) with self.assertRaisesRegex(snippet_management_service.Error, msg): manager.add_snippet_client('bar', MOCK_PACKAGE) @@ -109,7 +125,8 @@ class SnippetManagementServiceTest(unittest.TestCase): mock_client = mock.MagicMock() mock_class.return_value = mock_client manager = snippet_management_service.SnippetManagementService( - mock.MagicMock()) + mock.MagicMock() + ) manager.add_snippet_client('foo', MOCK_PACKAGE) manager.remove_snippet_client('foo') msg = 'No snippet client is registered with name "foo".' @@ -121,17 +138,20 @@ class SnippetManagementServiceTest(unittest.TestCase): mock_client = mock.MagicMock() mock_class.return_value = mock_client manager = snippet_management_service.SnippetManagementService( - mock.MagicMock()) + mock.MagicMock() + ) with self.assertRaisesRegex( snippet_management_service.Error, - 'No snippet client is registered with name "foo".'): + 'No snippet client is registered with name "foo".', + ): manager.remove_snippet_client('foo') @mock.patch(SNIPPET_CLIENT_V2_CLASS_PATH) def test_start_with_live_service(self, mock_class): mock_client = mock_class.return_value manager = snippet_management_service.SnippetManagementService( - mock.MagicMock()) + mock.MagicMock() + ) manager.add_snippet_client('foo', MOCK_PACKAGE) mock_client.initialize.reset_mock() mock_client.is_alive = True @@ -146,7 +166,8 @@ class SnippetManagementServiceTest(unittest.TestCase): def test_pause(self, mock_class): mock_client = mock_class.return_value manager = snippet_management_service.SnippetManagementService( - mock.MagicMock()) + mock.MagicMock() + ) manager.add_snippet_client('foo', MOCK_PACKAGE) manager.pause() mock_client.close_connection.assert_called_once_with() @@ -155,7 +176,8 @@ class SnippetManagementServiceTest(unittest.TestCase): def test_resume_positive_case(self, mock_class): mock_client = mock_class.return_value manager = snippet_management_service.SnippetManagementService( - mock.MagicMock()) + mock.MagicMock() + ) manager.add_snippet_client('foo', MOCK_PACKAGE) mock_client.is_alive = False manager.resume() @@ -165,7 +187,8 @@ class SnippetManagementServiceTest(unittest.TestCase): def test_resume_negative_case(self, mock_class): mock_client = mock_class.return_value manager = snippet_management_service.SnippetManagementService( - mock.MagicMock()) + mock.MagicMock() + ) manager.add_snippet_client('foo', MOCK_PACKAGE) mock_client.is_alive = True manager.resume() @@ -176,7 +199,8 @@ class SnippetManagementServiceTest(unittest.TestCase): mock_client = mock.MagicMock() mock_class.return_value = mock_client manager = snippet_management_service.SnippetManagementService( - mock.MagicMock()) + mock.MagicMock() + ) manager.add_snippet_client('foo', MOCK_PACKAGE) manager.foo.ha('param') mock_client.ha.assert_called_once_with('param') |