v1.1.0b1

packages/bthlabs-jsonrpc-core/tests/registry/test_MethodRedistry.py

@@ -13,7 +13,7 @@ def test_init():
 
 
 @mock.patch.object(registry.MethodRegistry, '__init__')
-def test_shared_registry(mock_init):
+def test_shared_registry(mock_init: mock.Mock):
     # Given
     mock_init.return_value = None
 
@@ -32,7 +32,8 @@ def test_shared_registry(mock_init):
 
 
 @mock.patch.object(registry.MethodRegistry, '__init__')
-def test_shared_registry_with_instance(mock_init, fake_method_registry):
+def test_shared_registry_with_instance(mock_init: mock.Mock,
+                                       fake_method_registry: mock.Mock):
     # Given
     mock_init.return_value = None
 
@@ -52,7 +53,7 @@ def test_shared_registry_with_instance(mock_init, fake_method_registry):
     registry.MethodRegistry.INSTANCE = None
 
 
-def test_register_method(fake_handler):
+def test_register_method(fake_handler: mock.Mock):
     # Given
     the_registry = registry.MethodRegistry()
 
@@ -64,7 +65,7 @@ def test_register_method(fake_handler):
     assert the_registry.registry['testing'] == expected_namespace
 
 
-def test_register_method_existing_namespace(fake_handler):
+def test_register_method_existing_namespace(fake_handler: mock.Mock):
     # Given
     spam_handler = mock.Mock()
 
@@ -93,7 +94,7 @@ def test_get_methods():
     assert set(result) == expected_methods
 
 
-def test_get_handler(fake_handler):
+def test_get_handler(fake_handler: mock.Mock):
     # Given
     spam_handler = mock.Mock()