Forráskód Böngészése

Type the mock provider

Annotate MockAWSCloudProvider (it subclasses AWSCloudProvider) and add
cloudbridge.providers.mock.* to the pragmatic mypy tier alongside aws.
Nuwan Goonasekera 1 napja
szülő
commit
b9ff7716f1
2 módosított fájl, 6 hozzáadás és 4 törlés
  1. 5 3
      cloudbridge/providers/mock/provider.py
  2. 1 1
      pyproject.toml

+ 5 - 3
cloudbridge/providers/mock/provider.py

@@ -6,6 +6,8 @@
     boto being hijacked, which will cause AWS to malfunction.
     See notes below.
 """
+from typing import Any
+
 from moto import mock_aws
 
 from ..aws import AWSCloudProvider
@@ -22,18 +24,18 @@ class MockAWSCloudProvider(AWSCloudProvider, TestMockHelperMixin):
     """
     PROVIDER_ID = 'mock'
 
-    def __init__(self, config):
+    def __init__(self, config: dict[str, Any]) -> None:
         self.setUpMock()
         super(MockAWSCloudProvider, self).__init__(config)
 
-    def setUpMock(self):
+    def setUpMock(self) -> None:
         """
         Let Moto take over all socket communications
         """
         self.mock_aws = mock_aws()
         self.mock_aws.start()
 
-    def tearDownMock(self):
+    def tearDownMock(self) -> None:
         """
         Stop Moto intercepting all socket communications
         """

+ 1 - 1
pyproject.toml

@@ -148,7 +148,7 @@ ignore_errors = true
 # through cast()/ignore adds noise without value. A more-specific module pattern
 # here wins over the broad providers.* exemption above.
 [[tool.mypy.overrides]]
-module = ["cloudbridge.providers.aws.*"]
+module = ["cloudbridge.providers.aws.*", "cloudbridge.providers.mock.*"]
 ignore_errors = false
 warn_return_any = false
 disallow_untyped_calls = false