plexus-python-pulse-common 1.1.75__py3-none-any.whl

plexus/pulse/common/utils/testutils.py

@@ -0,0 +1,172 @@
+import contextlib
+import uuid as py_uuid
+from collections.abc import Callable, Generator, Iterator
+from typing import Protocol
+
+import pytest
+import sqlalchemy as sa
+import sqlalchemy.orm as sa_orm
+import textcase
+from fastapi import FastAPI
+from plexus.common.utils.dbutils import ConnectionMaker, Dialects, Drivers
+from plexus.common.utils.dbutils import make_scheme
+from plexus.common.utils.jsonutils import JsonType
+from plexus.common.utils.jsonutils import json_difference, json_reformat
+from sqlmodel import SQLModel
+
+from plexus.pulse.common.utils.datautils import compute_vin_code_check_digit
+
+__all__ = [
+    "generate_dummy_uuid_str",
+    "generate_dummy_uuid",
+    "generate_dummy_vin_code",
+    "case_insensitive_json_compare",
+    "patched_value_factory",
+    "patched_postgresql_session_maker",
+]
+
+
+def generate_dummy_uuid_str(*nums: int) -> str:
+    if len(nums) > 8:
+        raise ValueError("a maximum of 8 integers can be provided")
+    if not all(0 <= num <= 0xFFFF for num in nums):
+        raise ValueError("all integers must be in the range 0 to 65535 (0xFFFF)")
+    i0, i1, i2, i3, i4, i5, i6, i7 = list(nums) + [0] * (8 - len(nums))
+    return f"{i0:04x}{i1:04x}-{i2:04x}-{i3:04x}-{i4:04x}-{i5:04x}{i6:04x}{i7:04x}"
+
+
+def generate_dummy_uuid(*nums: int) -> py_uuid.UUID:
+    return py_uuid.UUID(generate_dummy_uuid_str(*nums))
+
+
+def generate_dummy_vin_code(*nums: int) -> str:
+    if len(nums) > 4:
+        raise ValueError("a maximum of 4 integers can be provided")
+    if not all(0 <= num <= 9999 for num in nums):
+        raise ValueError("all integers must be in the range 0 to 9999")
+    i0, i1, i2, i3 = list(nums) + [0] * (4 - len(nums))
+    unchecked_vin_code = f"{i0:04d}{i1:04d}0{i2:04d}{i3:04d}"
+    check_digit = compute_vin_code_check_digit(unchecked_vin_code)
+    return unchecked_vin_code[:8] + check_digit + unchecked_vin_code[-8:]
+
+
+def case_insensitive_json_compare(a: JsonType, b: JsonType, *, print_diff_messages: bool = True) -> bool:
+    identical = True
+    for node_path, diff_message in json_difference(json_reformat(a, key_formatter=textcase.camel),
+                                                   json_reformat(b, key_formatter=textcase.camel),
+                                                   []):
+        if print_diff_messages:
+            print(node_path, diff_message)
+        identical = False
+    return identical
+
+
+class PatcherProtocol[T](Protocol):
+
+    def value_factory(self) -> Generator[T, None, None]:
+        ...
+
+    @contextlib.contextmanager
+    def use(self, value: T) -> Generator[T, None, None]:
+        ...
+
+
+def patched_value_factory[T](
+    app: FastAPI,
+    value_factory: Callable[[], Generator[T, None, None]],
+) -> Callable[[], Generator[PatcherProtocol[T], None, None]]:
+    """
+    Create a pytest fixture that allows patching a FastAPI dependency value factory.
+
+    :param app: FastAPI application instance.
+    :param value_factory: Original value factory to be overridden.
+    """
+
+    class Patcher(object):
+        value: T | None = None
+
+        def value_factory(self) -> Generator[T, None, None]:
+            if self.value is not None:
+                yield self.value
+            else:
+                yield from value_factory()
+
+        @contextlib.contextmanager
+        def use(self, value: T) -> Generator[T, None, None]:
+            self.value = value
+            try:
+                yield self.value
+            finally:
+                self.value = None
+
+    @pytest.fixture(scope="function")
+    def fixture() -> Generator[PatcherProtocol[T], None, None]:
+        patcher = Patcher()
+        app.dependency_overrides[value_factory] = patcher.value_factory
+        try:
+            yield patcher
+        finally:
+            app.dependency_overrides[value_factory] = value_factory
+
+    return fixture
+
+
+def patched_postgresql_session_maker(
+    postgresql_test_fixture_name: str,
+    base_model: type[SQLModel],
+    app: FastAPI,
+    session_maker: Callable[[], Generator[sa_orm.Session]],
+    **kwargs,
+) -> Callable[[pytest.FixtureRequest], Generator[Callable[[], Iterator[sa_orm.Session]]]]:
+    """
+    Create a pytest fixture that provides a patched session maker for PostgreSQL tests.
+
+    :param postgresql_test_fixture_name: Name of the PostgreSQL test fixture to use.
+    :param base_model: The base SQLModel model containing metadata.
+    :param app: FastAPI application instance.
+    :param session_maker: Original session maker to be overridden.
+    :return: A pytest fixture function.
+    """
+
+    @pytest.fixture(scope="function")
+    def fixture(request: pytest.FixtureRequest):
+        postgresql_test_fixture = request.getfixturevalue(postgresql_test_fixture_name)
+
+        cm = ConnectionMaker.create(
+            scheme=make_scheme(Dialects.postgresql, Drivers.psycopg),
+            host=postgresql_test_fixture.info.host,
+            port=postgresql_test_fixture.info.port,
+            username=postgresql_test_fixture.info.user,
+            password="",
+            database=postgresql_test_fixture.info.dbname,
+        )
+
+        def make_session_maker() -> Iterator[sa_orm.Session]:
+            with cm.make_session(**kwargs) as db:
+                try:
+                    # language=postgresql
+                    db.execute(sa.sql.text("SET TIMEZONE TO 'UTC'"))
+                    yield db
+                finally:
+                    db.close()
+
+        app.dependency_overrides[session_maker] = make_session_maker
+
+        with cm.make_session() as session:
+            for table in base_model.metadata.tables.values():
+                session.execute(sa.schema.CreateSchema(table.schema, if_not_exists=True))
+            session.commit()
+        base_model.metadata.create_all(cm.engine)
+
+        try:
+            yield make_session_maker
+        finally:
+            base_model.metadata.drop_all(cm.engine)
+            with cm.make_session() as session:
+                for table in base_model.metadata.tables.values():
+                    session.execute(sa.schema.DropSchema(table.schema, if_exists=True))
+                session.commit()
+
+            app.dependency_overrides[session_maker] = session_maker
+
+    return fixture

plexus_python_pulse_common-1.1.75.dist-info/METADATA

@@ -0,0 +1,42 @@
+Metadata-Version: 2.4
+Name: plexus-python-pulse-common
+Version: 1.1.75
+Classifier: Programming Language :: Python :: 3
+Classifier: Programming Language :: Python :: 3.12
+Classifier: Programming Language :: Python :: 3.13
+Classifier: Programming Language :: Python :: 3.14
+Requires-Python: <3.15,>=3.12
+Requires-Dist: aiobotocore<2.27,>=2.26
+Requires-Dist: boto3<1.42,>=1.41
+Requires-Dist: boto3-stubs[s3]<1.42,>=1.41
+Requires-Dist: docker>=7.1
+Requires-Dist: fsspec>=2025.10
+Requires-Dist: jinja2>=3.0
+Requires-Dist: jsonschema>=4.0
+Requires-Dist: lxml>=6.0
+Requires-Dist: markdown-it-py>=4.0
+Requires-Dist: numpy>=2.3
+Requires-Dist: pydantic>=2.8
+Requires-Dist: pygeodesy>=24.10
+Requires-Dist: pyparsing>=3.2
+Requires-Dist: pyproj>=3.6
+Requires-Dist: pyquaternion>=0.9
+Requires-Dist: pyyaml>=6.0
+Requires-Dist: sqlalchemy>=2.0
+Requires-Dist: s3fs>=2025.10
+Requires-Dist: rich>=13.9
+Requires-Dist: textcase>=0.4
+Requires-Dist: ujson>=5.9
+Requires-Dist: plexus-python-common[all]>=1.1
+Provides-Extra: all
+Requires-Dist: plexus-python-pulse-common[api]; extra == "all"
+Provides-Extra: api
+Requires-Dist: fastapi>=0.119.0; extra == "api"
+Requires-Dist: sqlmodel>=0.0.25; extra == "api"
+Provides-Extra: test
+Requires-Dist: ddt>=1.7; extra == "test"
+Requires-Dist: moto[s3,server]>=5.1; extra == "test"
+Requires-Dist: pytest-cov>=5.0; extra == "test"
+Requires-Dist: pytest-order>=1.3; extra == "test"
+Requires-Dist: pytest-postgresql>=6.1; extra == "test"
+Requires-Dist: pytest>=8.3; extra == "test"

plexus_python_pulse_common-1.1.75.dist-info/RECORD

@@ -0,0 +1,29 @@
+plexus/pulse/common/__init__.py,sha256=Kxyx_3ekA19__tyo3XvEeagUU3tzQMQtLIah07a4y3M,183
+plexus/pulse/common/carto/OSMFile.py,sha256=JQ9sq9B5hk82k16z1jDLshTV1FemJJYoKTcu-S3WAWg,8170
+plexus/pulse/common/carto/OSMNode.py,sha256=-G8ij4Rqq-uQxDH_F_s1F2INn2egd4wErLxg0OiOxmk,617
+plexus/pulse/common/carto/OSMTags.py,sha256=9sbs_nD2RIvGnkyZS1PAdgQ8KBCCMaciPUzgFc9-718,3452
+plexus/pulse/common/carto/OSMWay.py,sha256=y36uLV1MpjpPaF6qAfq8Hljdh-mtWv6n4aAr6fLS_7M,570
+plexus/pulse/common/carto/__init__.py,sha256=rL3wircaMKsMA8gYpeMlVH9q_EdBvHBdfMMT045ZxjI,288
+plexus/pulse/common/resources/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+plexus/pulse/common/resources/tags/__init__.py,sha256=1BmOq9aIT2Fr4YTzDwTrYySDhiyRpQ8Bih1gqzlt5v8,1135
+plexus/pulse/common/resources/tags/unittest-1.0.0.tagset.yaml,sha256=8zWjsYp3OxGfBJSiS6-r08q61Jhulol65iV5-pvF1-M,5338
+plexus/pulse/common/resources/tags/universal-1.0.0.tagset.yaml,sha256=PRxyS5xwWcNwUjAbYqN-SafLZVviE-QSzU-DXyXNffw,80736
+plexus/pulse/common/utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+plexus/pulse/common/utils/apiutils.py,sha256=8H1wuahkPseOjZHQcrT70EUUDTJ-j7c9lw8kx8SoG0Y,1000
+plexus/pulse/common/utils/bagutils.py,sha256=5JGnMgR8Q9OciW5dE7_2U7n1FGVEXMZXv1SrmI5jznw,13175
+plexus/pulse/common/utils/config.py,sha256=YPtFK5QHTVsXwcFBGV0rlZad0GFtsgwysQNyKYjdzvk,2409
+plexus/pulse/common/utils/datautils.py,sha256=OWlnyFSa6UaPbSSG3HzCJHxdiHyfHAE-FUVHaLfNZoM,9335
+plexus/pulse/common/utils/dockerutils.py,sha256=7DF9yFGz8uHwpYk115xBWqOjs5etz0UE7yh66eDvBzs,5793
+plexus/pulse/common/utils/gisutils.py,sha256=UX_IHPeAx9mbA9p_SsbPFwdsdp7bGeaQhWaCTCzvrBM,11466
+plexus/pulse/common/utils/jsonutils.py,sha256=zSbW94klN2jrkMdBECu4RNJX6zzNaSxauQ3FuihuB6E,3342
+plexus/pulse/common/utils/ormutils.py,sha256=MxSUIyxzGFY9wc12Y6QwaWZkbsYivSF5hNTyGmXFQYU,60683
+plexus/pulse/common/utils/pathutils.py,sha256=S5xk2WEmxP8MiuSKIU3cAEtk4Nk5Lz7IqC5FCd8BEFI,9585
+plexus/pulse/common/utils/s3utils.py,sha256=RArl95rWoyBEFyWU9X0c8WgoQBjP_jOK3OkPoKxIMnA,36025
+plexus/pulse/common/utils/sqlutils.py,sha256=D6kTBjhO5YlNRt3uFlPt6z3uH61m9ajEzPYmsI6NoFc,231
+plexus/pulse/common/utils/strutils.py,sha256=PkmMHCcHN9vALCy_8fJTf00Dg7I2N4QTCXdBmB3t7rk,16440
+plexus/pulse/common/utils/tagutils.py,sha256=smM79dyRCiswKTTcI-At7FLxDH54VN4Id3aFHbDxzYo,61696
+plexus/pulse/common/utils/testutils.py,sha256=hKPRvoN4mg8QGQJH121l59a0uo8jN5rntGQAVYplGBw,6166
+plexus_python_pulse_common-1.1.75.dist-info/METADATA,sha256=-FgPWbRw1LBvHkqaaX0tHVW34Nk8j6y_GulobNLZ9Os,1495
+plexus_python_pulse_common-1.1.75.dist-info/WHEEL,sha256=aeYiig01lYGDzBgS8HxWXOg3uV61G9ijOsup-k9o1sk,91
+plexus_python_pulse_common-1.1.75.dist-info/top_level.txt,sha256=ug_g7CVwaMQuas5UzAXbHUrQvKGCn8ezc6ZNvvRlJOE,7
+plexus_python_pulse_common-1.1.75.dist-info/RECORD,,

plexus_python_pulse_common-1.1.75.dist-info/WHEEL

@@ -0,0 +1,5 @@
+Wheel-Version: 1.0
+Generator: setuptools (82.0.1)
+Root-Is-Purelib: true
+Tag: py3-none-any
+

plexus_python_pulse_common-1.1.75.dist-info/top_level.txt

@@ -0,0 +1 @@
+plexus