Decorate pytest fixtures correctly with beartype

Author: adamtheturtle

tests/conftest.py

@@ -8,6 +8,7 @@
 import uuid
 
 import pytest
+from beartype import beartype
 from vws import VWS, CloudRecoService
 
 from mock_vws.database import VuforiaDatabase
@@ -20,6 +21,7 @@
 ]
 
 
+@beartype
 @pytest.fixture(name="vws_client")
 def fixture_vws_client(vuforia_database: VuforiaDatabase) -> VWS:
     """
@@ -31,6 +33,7 @@ def fixture_vws_client(vuforia_database: VuforiaDatabase) -> VWS:
     )
 
 
+@beartype
 @pytest.fixture
 def cloud_reco_client(vuforia_database: VuforiaDatabase) -> CloudRecoService:
     """
@@ -42,6 +45,7 @@ def cloud_reco_client(vuforia_database: VuforiaDatabase) -> CloudRecoService:
     )
 
 
+@beartype
 @pytest.fixture(name="inactive_vws_client")
 def fixture_inactive_vws_client(inactive_database: VuforiaDatabase) -> VWS:
     """
@@ -53,6 +57,7 @@ def fixture_inactive_vws_client(inactive_database: VuforiaDatabase) -> VWS:
     )
 
 
+@beartype
 @pytest.fixture
 def inactive_cloud_reco_client(
     inactive_database: VuforiaDatabase,
@@ -66,6 +71,7 @@ def inactive_cloud_reco_client(
     )
 
 
+@beartype
 @pytest.fixture
 def target_id(
     image_file_success_state_low_rating: io.BytesIO,
@@ -85,6 +91,7 @@ def target_id(
     )
 
 
+@beartype
 @pytest.fixture(
     params=[
         "add_target",
@@ -106,6 +113,7 @@ def endpoint(request: pytest.FixtureRequest) -> Endpoint:
     return endpoint_fixture
 
 
+@beartype
 @pytest.fixture(
     params=[
         pytest.param(
@@ -140,6 +148,7 @@ def not_base64_encoded_processable(request: pytest.FixtureRequest) -> str:
     return not_base64_encoded_string
 
 
+@beartype
 @pytest.fixture(
     params=[
         pytest.param(

tests/mock_vws/fixtures/credentials.py

@@ -38,8 +38,8 @@ class _InactiveVuforiaDatabaseSettings(_VuforiaDatabaseSettings):
     )
 
 
-@pytest.fixture
 @beartype
+@pytest.fixture
 def vuforia_database() -> VuforiaDatabase:
     """
     Return VWS credentials from environment variables.
@@ -55,6 +55,7 @@ def vuforia_database() -> VuforiaDatabase:
     )
 
 
+@beartype
 @pytest.fixture
 def inactive_database() -> VuforiaDatabase:
     """

tests/mock_vws/fixtures/prepared_requests.py

@@ -38,6 +38,7 @@ def _wait_for_target_processed(vws_client: VWS, target_id: str) -> None:
     vws_client.wait_for_target_processed(target_id=target_id)
 
 
+@beartype
 @pytest.fixture
 def add_target(
     vuforia_database: VuforiaDatabase,
@@ -96,6 +97,7 @@ def add_target(
     )
 
 
+@beartype
 @pytest.fixture
 def delete_target(
     vuforia_database: VuforiaDatabase,
@@ -145,6 +147,7 @@ def delete_target(
     )
 
 
+@beartype
 @pytest.fixture
 def database_summary(vuforia_database: VuforiaDatabase) -> Endpoint:
     """
@@ -191,8 +194,8 @@ def database_summary(vuforia_database: VuforiaDatabase) -> Endpoint:
     )
 
 
-@pytest.fixture
 @beartype
+@pytest.fixture
 def get_duplicates(
     vuforia_database: VuforiaDatabase,
     target_id: str,
@@ -244,8 +247,8 @@ def get_duplicates(
     )
 
 
-@pytest.fixture
 @beartype
+@pytest.fixture
 def get_target(
     vuforia_database: VuforiaDatabase,
     target_id: str,
@@ -296,6 +299,7 @@ def get_target(
     )
 
 
+@beartype
 @pytest.fixture
 def target_list(vuforia_database: VuforiaDatabase) -> Endpoint:
     """
@@ -342,6 +346,7 @@ def target_list(vuforia_database: VuforiaDatabase) -> Endpoint:
     )
 
 
+@beartype
 @pytest.fixture
 def target_summary(
     vuforia_database: VuforiaDatabase,
@@ -393,6 +398,7 @@ def target_summary(
     )
 
 
+@beartype
 @pytest.fixture
 def update_target(
     vuforia_database: VuforiaDatabase,
@@ -447,6 +453,7 @@ def update_target(
     )
 
 
+@beartype
 @pytest.fixture
 def query(
     vuforia_database: VuforiaDatabase,

tests/mock_vws/fixtures/vuforia_backends.py

@@ -30,7 +30,6 @@
 
 
 @RETRY_ON_TOO_MANY_REQUESTS
-@beartype
 def _delete_all_targets(*, database_keys: VuforiaDatabase) -> None:
     """
     Delete all targets.
@@ -229,6 +228,7 @@ def pytest_collection_modifyitems(
                 item.add_marker(skip_docker_build_tests_marker)
 
 
+@beartype
 @pytest.fixture(
     params=list(VuforiaBackend),
     ids=[backend.value for backend in list(VuforiaBackend)],
@@ -268,6 +268,7 @@ def verify_mock_vuforia(
     )
 
 
+@beartype
 @pytest.fixture(
     params=[item for item in VuforiaBackend if item != VuforiaBackend.REAL],
     ids=[
@@ -277,7 +278,6 @@ def verify_mock_vuforia(
         ]
     ],
 )
-@beartype
 def mock_only_vuforia(
     request: pytest.FixtureRequest,
     vuforia_database: VuforiaDatabase,

tests/mock_vws/test_docker.py

@@ -51,8 +51,8 @@ def wait_for_health_check(container: Container) -> None:
         raise ValueError(error_message)
 
 
-@pytest.fixture(name="custom_bridge_network")
 @beartype
+@pytest.fixture(name="custom_bridge_network")
 def fixture_custom_bridge_network() -> Iterator[Network]:
     """
     Yield a custom bridge network which containers can connect to.

tests/mock_vws/test_flask_app_usage.py

@@ -10,6 +10,7 @@
 import pytest
 import requests
 import responses
+from beartype import beartype
 from PIL import Image
 from requests_mock_flask import add_flask_app_to_mock
 from vws import VWS, CloudRecoService
@@ -25,6 +26,7 @@
 _EXAMPLE_URL_FOR_TARGET_MANAGER = "http://" + uuid.uuid4().hex + ".com"
 
 
+@beartype
 @pytest.fixture(autouse=True)
 def _(monkeypatch: pytest.MonkeyPatch) -> Iterator[None]:
     """