Use more beartype checks

Author: adamtheturtle

src/mock_vws/_database_matchers.py

@@ -4,11 +4,13 @@
 
 from collections.abc import Iterable, Mapping
 
+from beartype import beartype
 from vws_auth_tools import authorization_header
 
 from mock_vws.database import VuforiaDatabase
 
 
+@beartype
 def get_database_matching_client_keys(
     *,
     request_headers: Mapping[str, str],
@@ -54,6 +56,7 @@ def get_database_matching_client_keys(
     raise ValueError
 
 
+@beartype
 def get_database_matching_server_keys(
     *,
     request_headers: dict[str, str],

src/mock_vws/_flask_server/target_manager.py

@@ -10,6 +10,7 @@
 from http import HTTPMethod, HTTPStatus
 from zoneinfo import ZoneInfo
 
+from beartype import beartype
 from flask import Flask, Response, request
 from pydantic_settings import BaseSettings
 
@@ -78,6 +79,7 @@ def delete_database(database_name: str) -> Response:
 
 
 @TARGET_MANAGER_FLASK_APP.route("/databases", methods=[HTTPMethod.GET])
+@beartype
 def get_databases() -> Response:
     """
     Return a list of all databases.

src/mock_vws/_flask_server/vwq.py

@@ -10,6 +10,7 @@
 from http import HTTPMethod, HTTPStatus
 
 import requests
+from beartype import beartype
 from flask import Flask, Response, request
 from pydantic_settings import BaseSettings
 
@@ -58,6 +59,7 @@ class VWQSettings(BaseSettings):
     )
 
 
+@beartype
 def get_all_databases() -> set[VuforiaDatabase]:
     """
     Get all database objects from the target manager back-end.

src/mock_vws/_flask_server/vws.py

@@ -74,6 +74,7 @@ class VWSSettings(BaseSettings):
     )
 
 
+@beartype
 def get_all_databases() -> set[VuforiaDatabase]:
     """
     Get all database objects from the task manager back-end.
@@ -222,6 +223,7 @@ def add_target() -> Response:
 
 
 @VWS_FLASK_APP.route("/targets/<string:target_id>", methods=[HTTPMethod.GET])
+@beartype
 def get_target(target_id: str) -> Response:
     """
     Get details of a target.
@@ -439,6 +441,7 @@ def target_summary(target_id: str) -> Response:
     "/duplicates/<string:target_id>",
     methods=[HTTPMethod.GET],
 )
+@beartype
 def get_duplicates(target_id: str) -> Response:
     """
     Get targets which may be considered duplicates of a given target.

src/mock_vws/_requests_mock_server/mock_web_services_api.py

@@ -5,17 +5,19 @@
 https://developer.vuforia.com/library/web-api/cloud-targets-web-services-api
 """
 
-from __future__ import annotations
-
 import base64
 import dataclasses
 import datetime
 import email.utils
 import uuid
+from collections.abc import Callable
 from http import HTTPMethod, HTTPStatus
-from typing import TYPE_CHECKING
 from zoneinfo import ZoneInfo
 
+from beartype import beartype
+from requests_mock.request import Request
+from requests_mock.response import Context
+
 from mock_vws._constants import ResultCodes, TargetStatuses
 from mock_vws._database_matchers import get_database_matching_server_keys
 from mock_vws._mock_common import Route, json_dump
@@ -26,17 +28,10 @@
     TargetStatusProcessingError,
     ValidatorError,
 )
+from mock_vws.image_matchers import ImageMatcher
 from mock_vws.target import Target
-
-if TYPE_CHECKING:
-    from collections.abc import Callable
-
-    from requests_mock.request import Request
-    from requests_mock.response import Context
-
-    from mock_vws.image_matchers import ImageMatcher
-    from mock_vws.target_manager import TargetManager
-    from mock_vws.target_raters import TargetTrackingRater
+from mock_vws.target_manager import TargetManager
+from mock_vws.target_raters import TargetTrackingRater
 
 _TARGET_ID_PATTERN = "[A-Za-z0-9]+"
 
@@ -403,6 +398,7 @@ def target_list(self, request: Request, context: Context) -> str:
         path_pattern=f"/targets/{_TARGET_ID_PATTERN}",
         http_methods={HTTPMethod.GET},
     )
+    @beartype
     def get_target(self, request: Request, context: Context) -> str:
         """
         Get details of a target.
@@ -471,6 +467,7 @@ def get_target(self, request: Request, context: Context) -> str:
         path_pattern=f"/duplicates/{_TARGET_ID_PATTERN}",
         http_methods={HTTPMethod.GET},
     )
+    @beartype
     def get_duplicates(self, request: Request, context: Context) -> str:
         """
         Get targets which may be considered duplicates of a given target.

src/mock_vws/database.py

@@ -93,6 +93,7 @@ def to_dict(self) -> DatabaseDict:
             "targets": targets,
         }
 
+    @beartype
     def get_target(self, target_id: str) -> Target:
         """
         Return a target from the database with the given ID.

tests/mock_vws/fixtures/prepared_requests.py

@@ -12,6 +12,7 @@
 
 import pytest
 import requests
+from beartype import beartype
 from urllib3.filepost import encode_multipart_formdata
 from vws_auth_tools import authorization_header, rfc_1123_date
 
@@ -197,6 +198,7 @@ def database_summary(vuforia_database: VuforiaDatabase) -> Endpoint:
 
 
 @pytest.fixture
+@beartype
 def get_duplicates(
     vuforia_database: VuforiaDatabase,
     target_id: str,
@@ -249,6 +251,7 @@ def get_duplicates(
 
 
 @pytest.fixture
+@beartype
 def get_target(
     vuforia_database: VuforiaDatabase,
     target_id: str,