pytest-databases 0.15.0__tar.gz → 0.16.0__tar.gz

{pytest_databases-0.15.0 → pytest_databases-0.16.0}/.github/workflows/cd.yaml

@@ -13,7 +13,7 @@ jobs:
       release_body: ${{ steps.git-cliff.outputs.content }}
     steps:
       - name: Checkout
-        uses: actions/checkout@v5
+        uses: actions/checkout@v6
         with:
           fetch-depth: 0
 

{pytest_databases-0.15.0 → pytest_databases-0.16.0}/.github/workflows/ci.yaml

@@ -26,7 +26,7 @@ jobs:
         cdist-group: [1, 2, 3]
 
     steps:
-      - uses: actions/checkout@v5
+      - uses: actions/checkout@v6
 
       - name: Install additional dependencies
         run: sudo apt-get update && sudo ACCEPT_EULA=Y apt-get install -y msodbcsql18 libmariadb-dev && sudo apt-get autoremove -y  && sudo apt-get clean -y
@@ -37,10 +37,10 @@ jobs:
           python-version: ${{ matrix.python-version }}
 
       - name: Install uv
-        uses: astral-sh/setup-uv@v6
+        uses: astral-sh/setup-uv@v7
 
       - name: Intall dependencies
-        run: uv sync --frozen
+        run: uv sync --frozen --all-extras
 
       - if: matrix.python-version == '3.12'
         name: Run tests with coverage tracking
@@ -51,14 +51,14 @@ jobs:
         run: uv run pytest --cdist-group=${{ matrix.cdist-group }}/3 -k "not elasticsearch"
 
       - if: matrix.python-version == '3.12'
-        uses: actions/upload-artifact@v4
+        uses: actions/upload-artifact@v6
         with:
           name: coverage-xml
           path: coverage.xml
 
       - if: matrix.python-version == '3.12'
         name: Upload coverage reports to Codecov
-        uses: codecov/codecov-action@v5.5.1
+        uses: codecov/codecov-action@v5.5.2
         with:
           token: ${{ secrets.CODECOV_TOKEN }}
           slug: litestar-org/pytest-databases
@@ -70,7 +70,7 @@ jobs:
       - name: Install additional dependencies
         run: sudo ACCEPT_EULA=Y apt-get install -y msodbcsql18 libmariadb-dev
 
-      - uses: actions/checkout@v5
+      - uses: actions/checkout@v6
 
       - name: Set up Python
         uses: actions/setup-python@v6
@@ -78,10 +78,10 @@ jobs:
           python-version: 3.9
 
       - name: Install uv
-        uses: astral-sh/setup-uv@v6
+        uses: astral-sh/setup-uv@v7
 
       - name: Intall dependencies
-        run: uv sync --frozen
+        run: uv sync --frozen --all-extras
 
       - name: Run tests with coverage tracking
         run: uv run pytest -k elasticsearch
@@ -94,7 +94,7 @@ jobs:
   #   runs-on: ubuntu-latest
   #   steps:
   #     - name: Check out repository
-  #       uses: actions/checkout@v5
+  #       uses: actions/checkout@v6
   #       with:
   #         fetch-depth: 0 # Shallow clones should be disabled for a better relevancy of analysis
   #     - name: Download Artifacts

{pytest_databases-0.15.0 → pytest_databases-0.16.0}/.github/workflows/docs.yml

@@ -20,7 +20,7 @@ jobs:
       url: ${{ steps.deployment.outputs.page_url }}
     runs-on: ubuntu-latest
     steps:
-      - uses: actions/checkout@v5
+      - uses: actions/checkout@v6
 
       - name: Install additional dependencies
         run: sudo apt-get update && sudo ACCEPT_EULA=Y apt-get install -y msodbcsql18 libmariadb-dev && sudo apt-get autoremove -y  && sudo apt-get clean -y
@@ -31,7 +31,7 @@ jobs:
           python-version: 3.12
 
       - name: Install uv
-        uses: astral-sh/setup-uv@v6
+        uses: astral-sh/setup-uv@v7
 
       - name: Set up Python
         run: uv python install 3.12

{pytest_databases-0.15.0 → pytest_databases-0.16.0}/.github/workflows/release.yaml

@@ -16,10 +16,10 @@ jobs:
         run: sudo ACCEPT_EULA=Y apt-get install -y msodbcsql18 libmariadb-dev
 
       - name: Check out repository
-        uses: actions/checkout@v5
+        uses: actions/checkout@v6
 
       - name: Install uv
-        uses: astral-sh/setup-uv@v6
+        uses: astral-sh/setup-uv@v7
 
       - name: Set up Python
         run: uv python install 3.12

{pytest_databases-0.15.0 → pytest_databases-0.16.0}/.pre-commit-config.yaml

@@ -19,7 +19,7 @@ repos:
 
   # Ruff replaces black, flake8, autoflake and isort
   - repo: https://github.com/charliermarsh/ruff-pre-commit
-    rev: "v0.13.3" # make sure this is always consistent with hatch configs
+    rev: "v0.14.13" # make sure this is always consistent with hatch configs
     hooks:
       - id: ruff
         args: [--config, ./pyproject.toml]

{pytest_databases-0.15.0 → pytest_databases-0.16.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: pytest-databases
-Version: 0.15.0
+Version: 0.16.0
 Summary: Reusable database fixtures for any and all databases.
 Project-URL: Documentation, https://github.com/litestar-org/pytest-databases#readme
 Project-URL: Issues, https://github.com/litestar-org/pytest-databases/issues
@@ -8,7 +8,7 @@ Project-URL: Source, https://github.com/litestar-org/pytest-databases
 Author-email: Cody Fincher <cody@litestar.dev>
 License-Expression: MIT
 License-File: LICENSE
-Keywords: alloydb,alloydbomni,azure,bigquery,cockroachdb,database,dragonflydb,duckdb,elasticsearch,migration,mssql,mysql,oracle,postgres,redis,spanner,valkey
+Keywords: alloydb,alloydbomni,azure,bigquery,cockroachdb,database,dragonflydb,duckdb,elasticsearch,flightsql,gizmosql,migration,mongodb,mssql,mysql,oracle,postgres,redis,spanner,valkey
 Classifier: Development Status :: 4 - Beta
 Classifier: Programming Language :: Python
 Classifier: Programming Language :: Python :: 3.9
@@ -34,12 +34,17 @@ Provides-Extra: elasticsearch7
 Requires-Dist: elasticsearch7; extra == 'elasticsearch7'
 Provides-Extra: elasticsearch8
 Requires-Dist: elasticsearch8; extra == 'elasticsearch8'
+Provides-Extra: gizmosql
+Requires-Dist: adbc-driver-flightsql; extra == 'gizmosql'
+Requires-Dist: pyarrow; extra == 'gizmosql'
 Provides-Extra: keydb
 Requires-Dist: redis; extra == 'keydb'
 Provides-Extra: mariadb
 Requires-Dist: mariadb; extra == 'mariadb'
 Provides-Extra: minio
 Requires-Dist: minio; extra == 'minio'
+Provides-Extra: mongodb
+Requires-Dist: pymongo; extra == 'mongodb'
 Provides-Extra: mssql
 Requires-Dist: pymssql; extra == 'mssql'
 Provides-Extra: mysql
@@ -54,6 +59,8 @@ Provides-Extra: spanner
 Requires-Dist: google-cloud-spanner; extra == 'spanner'
 Provides-Extra: valkey
 Requires-Dist: valkey; extra == 'valkey'
+Provides-Extra: yugabyte
+Requires-Dist: psycopg2-yugabytedb; extra == 'yugabyte'
 Description-Content-Type: text/markdown
 
 <!-- markdownlint-disable no-inline-html -->

{pytest_databases-0.15.0 → pytest_databases-0.16.0}/docs/conf.py

@@ -63,6 +63,9 @@ autodoc_default_options = {
     "exclude-members": "__weakref__",
 }
 
+# Mock imports for modules not needed during doc generation
+autodoc_mock_imports = ["OpenSSL"]
+
 # Intersphinx settings
 intersphinx_mapping = {
     "python": ("https://docs.python.org/3", None),

pytest_databases-0.16.0/docs/supported-databases/gizmosql.rst

@@ -0,0 +1,102 @@
+GizmoSQL
+========
+
+Integration with `GizmoSQL <https://github.com/gizmodata/gizmosql>`_, a high-performance SQL server built on Apache Arrow Flight SQL with DuckDB/SQLite backends.
+
+.. note::
+
+   GizmoSQL always runs with TLS enabled using auto-generated self-signed certificates.
+   The connection fixtures automatically skip certificate verification for testing purposes.
+
+Installation
+------------
+
+.. code-block:: bash
+
+   pip install pytest-databases[gizmosql]
+
+Usage Example
+-------------
+
+Using the service fixture:
+
+.. code-block:: python
+
+    import pytest
+    from adbc_driver_flightsql import dbapi as flightsql
+    from pytest_databases.docker.gizmosql import GizmoSQLService, _make_connection_kwargs
+
+    pytest_plugins = ["pytest_databases.docker.gizmosql"]
+
+    def test(gizmosql_service: GizmoSQLService) -> None:
+        db_kwargs = _make_connection_kwargs(
+            gizmosql_service.username,
+            gizmosql_service.password,
+        )
+        with flightsql.connect(
+            uri=gizmosql_service.uri,
+            db_kwargs=db_kwargs,
+            autocommit=True,
+        ) as conn:
+            with conn.cursor() as cur:
+                cur.execute("SELECT 1")
+                result = cur.fetchone()
+                assert result is not None and result[0] == 1
+
+Using the connection fixture:
+
+.. code-block:: python
+
+    import pytest
+
+    pytest_plugins = ["pytest_databases.docker.gizmosql"]
+
+    def test(gizmosql_connection) -> None:
+        with gizmosql_connection.cursor() as cur:
+            cur.execute("""
+                CREATE TABLE test_table (id INTEGER, name VARCHAR);
+                INSERT INTO test_table VALUES (1, 'test');
+            """)
+
+        with gizmosql_connection.cursor() as cur:
+            cur.execute("SELECT * FROM test_table")
+            result = cur.fetchone()
+            assert result is not None
+            assert result[0] == 1
+
+.. note::
+
+   Due to Flight SQL semantics, DDL and DML statements should be combined in a single
+   ``execute()`` call for immediate visibility across cursor operations.
+
+Available Fixtures
+------------------
+
+* ``gizmosql_image``: The Docker image to use for GizmoSQL (default: ``gizmodata/gizmosql:latest``).
+* ``gizmosql_username``: The username for authentication.
+* ``gizmosql_password``: The password for authentication.
+* ``gizmosql_service``: A fixture that provides a GizmoSQL service container.
+* ``gizmosql_connection``: A fixture that provides an ADBC Flight SQL connection.
+* ``xdist_gizmosql_isolation_level``: Xdist isolation level (default: ``server``).
+
+Parallel Testing (xdist)
+------------------------
+
+GizmoSQL only supports ``server`` isolation level for pytest-xdist parallel testing.
+This means each xdist worker gets its own dedicated container. Database-level isolation
+is not supported because DuckDB (the default backend) doesn't support multiple databases
+per instance.
+
+.. code-block:: python
+
+    @pytest.fixture(scope="session")
+    def xdist_gizmosql_isolation_level():
+        return "server"  # This is the only supported value
+
+Service API
+-----------
+
+.. automodule:: pytest_databases.docker.gizmosql
+   :members:
+   :undoc-members:
+   :show-inheritance:

{pytest_databases-0.15.0 → pytest_databases-0.16.0}/docs/supported-databases/index.rst

@@ -14,6 +14,9 @@ This section provides detailed information on the supported databases, including
    spanner
    bigquery
    cockroachdb
+   yugabyte
+   mongodb
+   gizmosql
    redis
    valkey
    elasticsearch

pytest_databases-0.16.0/docs/supported-databases/mongodb.rst

@@ -0,0 +1,76 @@
+MongoDB
+=======
+
+Integration with `MongoDB <https://www.mongodb.com/>`_, a NoSQL document-oriented database.
+
+This integration uses the official `PyMongo <https://pymongo.readthedocs.io/>`_ driver to interact with MongoDB.
+
+Installation
+------------
+
+.. code-block:: bash
+
+   pip install pytest-databases[mongodb]
+
+
+Usage Example
+-------------
+
+.. code-block:: python
+
+    import pytest
+    import pymongo
+    from pytest_databases.docker.mongodb import MongoDBService
+
+    pytest_plugins = ["pytest_databases.docker.mongodb"]
+
+    def test_mongodb_service(mongodb_service: MongoDBService) -> None:
+        client = pymongo.MongoClient(
+            host=mongodb_service.host,
+            port=mongodb_service.port,
+            username=mongodb_service.username,
+            password=mongodb_service.password,
+        )
+        # Ping the server to ensure connection
+        client.admin.command("ping")
+        client.close()
+
+    def test_mongodb_connection(mongodb_connection: pymongo.MongoClient) -> None:
+        # mongodb_connection is an instance of pymongo.MongoClient
+        # You can use it to interact with the database
+        db = mongodb_connection["mydatabase"]
+        collection = db["mycollection"]
+        collection.insert_one({"name": "test_document", "value": 1})
+        result = collection.find_one({"name": "test_document"})
+        assert result is not None
+        assert result["value"] == 1
+        # Clean up (optional, depending on your test needs)
+        collection.delete_one({"name": "test_document"})
+        mongodb_connection.close()
+
+    def test_mongodb_database(mongodb_database: pymongo.database.Database) -> None:
+        # mongodb_database is an instance of pymongo.database.Database
+        # This fixture provides a database that is unique per test function if xdist is used
+        # and xdist_mongodb_isolation_level is "database" (the default).
+        collection = mongodb_database["mycollection"]
+        collection.insert_one({"name": "another_document", "value": 2})
+        result = collection.find_one({"name": "another_document"})
+        assert result is not None
+        assert result["value"] == 2
+        # No need to close the database object explicitly, the connection is managed by mongodb_connection
+
+Available Fixtures
+------------------
+
+* ``mongodb_service``: A fixture that provides a MongoDB service, giving access to connection details like host, port, username, and password.
+* ``mongodb_connection``: A fixture that provides a ``pymongo.MongoClient`` instance connected to the MongoDB service.
+* ``mongodb_database``: A fixture that provides a ``pymongo.database.Database`` instance.
+* ``mongodb_image``: A fixture that returns the Docker image name used for the MongoDB service (default: "mongo:latest"). You can override this fixture to use a different MongoDB version.
+
+Service API
+-----------
+
+.. automodule:: pytest_databases.docker.mongodb
+   :members: MongoDBService, _provide_mongodb_service
+   :undoc-members:
+   :show-inheritance:

pytest_databases-0.16.0/docs/supported-databases/yugabyte.rst

@@ -0,0 +1,53 @@
+Yugabyte
+========
+
+Integration with `Yugabyte DB <https://www.yugabyte.com/>`_
+
+Installation
+------------
+
+.. code-block:: bash
+
+   pip install pytest-databases[yugabyte]
+
+Usage Example
+-------------
+
+.. code-block:: python
+
+    import pytest
+    import psycopg
+    from pytest_databases.docker.yugabyte import YugabyteService
+
+    pytest_plugins = ["pytest_databases.docker.yugabyte"]
+
+    @pytest.fixture(scope="session")
+    def yugabyte_uri(yugabyte_service: YugabyteService) -> str:
+        opts = "&".join(f"{k}={v}" for k, v in yugabyte_service.driver_opts.items())
+        return f"postgresql://yugabyte:yugabyte@{yugabyte_service.host}:{yugabyte_service.port}/{yugabyte_service.database}?{opts}"
+
+    def test_yugabyte_service(yugabyte_uri: str) -> None:
+        with psycopg.connect(yugabyte_uri) as conn:
+            db_open = conn.execute("SELECT 1").fetchone()
+            assert db_open is not None and db_open[0] == 1
+
+    def test_yugabyte_connection(yugabyte_connection: psycopg.Connection) -> None:
+        yugabyte_connection.execute("CREATE TABLE if not exists simple_table as SELECT 1")
+        result = yugabyte_connection.execute("select * from simple_table").fetchone()
+        assert result is not None and result[0] == 1
+
+Available Fixtures
+------------------
+
+* ``yugabyte_image``: The Docker image to use for Yugabyte DB.
+* ``yugabyte_service``: A fixture that provides a Yugabyte DB service.
+* ``yugabyte_connection``: A fixture that provides a Yugabyte DB connection.
+* ``yugabyte_driver_opts``: A fixture that provides driver options for Yugabyte DB.
+
+Service API
+-----------
+
+.. automodule:: pytest_databases.docker.yugabyte
+   :members:
+   :undoc-members:
+   :show-inheritance:

{pytest_databases-0.15.0 → pytest_databases-0.16.0}/pyproject.toml

@@ -12,7 +12,7 @@ license = "MIT"
 name = "pytest-databases"
 readme = "README.md"
 requires-python = ">=3.9"
-version = "0.15.0"
+version = "0.16.0"
 #
 authors = [{ name = "Cody Fincher", email = "cody@litestar.dev" }]
 keywords = [
@@ -33,6 +33,9 @@ keywords = [
   "azure",
   "valkey",
   "dragonflydb",
+  "mongodb",
+  "gizmosql",
+  "flightsql",
 ]
 # options under https://pypi.org/classifiers/
 classifiers = [
@@ -61,9 +64,11 @@ cockroachdb = ["psycopg"]
 dragonfly = ["redis"]
 elasticsearch7 = ["elasticsearch7"]
 elasticsearch8 = ["elasticsearch8"]
+gizmosql = ["adbc-driver-flightsql", "pyarrow"]
 keydb = ["redis"]
 mariadb = ["mariadb"]
 minio = ["minio"]
+mongodb = ["pymongo"]
 mssql = ["pymssql"]
 mysql = ["mysql-connector-python"]
 oracle = ["oracledb"]
@@ -71,34 +76,18 @@ postgres = ["psycopg>=3"]
 redis = ["redis"]
 spanner = ["google-cloud-spanner"]
 valkey = ["valkey"]
+yugabyte = ["psycopg2-yugabytedb"]
 
 [dependency-groups]
+build = ["bump-my-version"]
 dev = [
-  # tests
-  "bump-my-version",
-  "pytest-databases[azure-storage,bigquery,cockroachdb,dragonfly,elasticsearch7,elasticsearch8,keydb,mssql,mysql,mariadb,oracle,postgres,redis,spanner,minio,valkey]",
-  "coverage[toml]>=6.2",
-  "pytest",
-  "pytest-cov",
-  "pytest-cdist>=0.2",
-  "pytest-mock",
-  "pytest-click",
-  "pytest-xdist",
-  "slotscheck",
-  "psycopg-binary",                                                                                                                                                    # This fixes tests failing on M series CPUs.
-  # lint
-  "mypy",
-  "ruff",
-  "pyright",
-  "pre-commit",
-  "types-click",
-  "types-six",
-  "types-decorator",
-  "types-pyyaml",
-  "types-docutils",
-  "types-redis",
-  "types-pymysql",
-  # docs
+  { include-group = "extra" },
+  { include-group = "lint" },
+  { include-group = "doc" },
+  { include-group = "test" },
+  { include-group = "build" },
+]
+doc = [
   "auto-pytabs[sphinx]>=0.5.0",
   "shibuya",
   "sphinx>=7.0.0; python_version <= \"3.9\"",
@@ -115,6 +104,31 @@ dev = [
   "sphinx-autodoc-typehints",
   "sphinx-rtd-theme",
 ]
+extra = ["psycopg2-binary", "psycopg[binary,pool]", "psycopg2-yugabytedb-binary"]
+lint = [
+  "mypy",
+  "ruff",
+  "pyright",
+  "pre-commit",
+  "types-click",
+  "types-six",
+  "types-decorator",
+  "types-pyyaml",
+  "types-docutils",
+  "types-redis",
+  "types-pymysql",
+  "slotscheck",
+]
+test = [
+  "coverage[toml]>=6.2",
+  "pytest",
+  "pytest-cov",
+  "pytest-cdist>=0.2",
+  "pytest-mock",
+  "pytest-click",
+  "pytest-xdist",
+  "pytest-sugar",
+]
 
 ##################
 # External Tools #
@@ -124,7 +138,7 @@ dev = [
 allow_dirty = true
 commit = false
 commit_args = "--no-verify"
-current_version = "0.15.0"
+current_version = "0.16.0"
 ignore_missing_files = false
 ignore_missing_version = false
 message = "chore(release): bump to `v{new_version}`"
@@ -297,6 +311,8 @@ lint.ignore = [
   "DOC502",  # Raised exception missing from docstring
   "A005",    # module shadows builtin
   "S608",
+  "RUF067",  # __init__ module should only contain docstrings and re-exports
+  "B901",    # Using `yield` and `return {value}` in a generator function can lead to confusing behavior
 ]
 lint.select = ["ALL"]
 # Allow unused variables when underscore-prefixed.
@@ -384,6 +400,7 @@ markers = [
   "duckdb: DuckDB Tests",
   "mssql: Microsoft SQL Server Tests",
   "elasticsearch: Elasticsearch Tests",
+  "gizmosql: GizmoSQL Tests",
 ]
 testpaths = ["tests"]
 

{pytest_databases-0.15.0 → pytest_databases-0.16.0}/src/pytest_databases/docker/azure_blob.py

@@ -48,7 +48,7 @@ def azure_blob_service(
     azurite_in_memory: bool,
     azure_blob_xdist_isolation_level: XdistIsolationLevel,
 ) -> Generator[ServiceContainer, None, None]:
-    command = "azurite-blob --blobHost 0.0.0.0 --blobPort 10000"
+    command = "azurite-blob --blobHost 0.0.0.0 --blobPort 10000 --skipApiVersionCheck"
     if azurite_in_memory:
         command += " --inMemoryPersistence"