dagster-postgres 0.26.8__tar.gz → 0.26.11__tar.gz

{dagster-postgres-0.26.8 → dagster-postgres-0.26.11}/MANIFEST.in

@@ -3,3 +3,5 @@ graft dagster_postgres/alembic
 global-exclude __pycache__
 global-exclude *.py[co]
 include dagster_postgres/py.typed
+exclude conftest.py
+include dagster_postgres/test_fixtures/docker-compose.yml

{dagster-postgres-0.26.8/dagster_postgres.egg-info → dagster-postgres-0.26.11}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: dagster-postgres
-Version: 0.26.8
+Version: 0.26.11
 Summary: A Dagster integration for postgres
 Home-page: https://github.com/dagster-io/dagster/tree/master/python_modules/libraries/dagster-postgres
 Author: Dagster Labs

{dagster-postgres-0.26.8 → dagster-postgres-0.26.11}/dagster_postgres/event_log/event_log.py

@@ -228,7 +228,7 @@ class PostgresEventLogStorage(SqlEventLogStorage, ConfigurableClass):
         insert_event_statement = self.prepare_insert_event_batch(events)
         with self._connect() as conn:
             result = conn.execute(insert_event_statement.returning(SqlEventLogStorageTable.c.id))
-            event_ids = [cast(int, row[0]) for row in result.fetchall()]
+            event_ids = [cast("int", row[0]) for row in result.fetchall()]
 
         # We only update the asset table with the last event
         self.store_asset_event(events[-1], event_ids[-1])

dagster-postgres-0.26.11/dagster_postgres/test_fixtures/__init__.py

@@ -0,0 +1,75 @@
+import tempfile
+from contextlib import contextmanager
+from pathlib import Path
+
+import pytest
+from dagster._core.test_utils import instance_for_test
+from dagster._utils.merger import merge_dicts
+from dagster_test.fixtures import docker_compose_cm, network_name_from_yml
+
+from dagster_postgres.utils import get_conn_string, wait_for_connection
+
+compose_file = Path(__file__).parent / "docker-compose.yml"
+
+
+@pytest.fixture(scope="session")
+def postgres_network():
+    yield network_name_from_yml(compose_file)
+
+
+@pytest.fixture(scope="session")
+def postgres_hostname():
+    with docker_compose_cm(docker_compose_yml=compose_file) as hostnames:
+        yield hostnames["postgres"]
+
+
+@pytest.fixture(scope="session")
+def postgres_conn_str(postgres_hostname):
+    conn_str = get_conn_string(
+        username="test",
+        password="test",
+        hostname=postgres_hostname,
+        db_name="test",
+        params=dict(connect_timeout=5),
+    )
+    wait_for_connection(
+        conn_str,
+        retry_limit=10,
+        retry_wait=3,
+    )
+
+    yield conn_str
+
+
+@pytest.fixture
+def postgres_instance(postgres_conn_str):
+    @contextmanager
+    def _instance(overrides=None):
+        with tempfile.TemporaryDirectory() as temp_dir:
+            with instance_for_test(
+                temp_dir=temp_dir,
+                overrides=merge_dicts(
+                    {
+                        "run_storage": {
+                            "module": "dagster_postgres.run_storage.run_storage",
+                            "class": "PostgresRunStorage",
+                            "config": {"postgres_url": postgres_conn_str},
+                        },
+                        "event_log_storage": {
+                            "module": "dagster_postgres.event_log.event_log",
+                            "class": "PostgresEventLogStorage",
+                            "config": {"postgres_url": postgres_conn_str},
+                        },
+                        "schedule_storage": {
+                            "module": "dagster_postgres.schedule_storage.schedule_storage",
+                            "class": "PostgresScheduleStorage",
+                            "config": {"postgres_url": postgres_conn_str},
+                        },
+                    },
+                    overrides if overrides else {},
+                ),
+            ) as instance:
+                instance.wipe()
+                yield instance
+
+    return _instance

dagster-postgres-0.26.11/dagster_postgres/test_fixtures/docker-compose.yml

@@ -0,0 +1,10 @@
+services:
+  postgres:
+    image: postgres:16
+    container_name: postgres
+    ports:
+      - "5432:5432"
+    environment:
+      POSTGRES_PASSWORD: "test"
+      POSTGRES_USER: "test"
+      POSTGRES_DB: "test"

dagster-postgres-0.26.11/dagster_postgres/version.py

@@ -0,0 +1 @@
+__version__ = "0.26.11"

{dagster-postgres-0.26.8 → dagster-postgres-0.26.11/dagster_postgres.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: dagster-postgres
-Version: 0.26.8
+Version: 0.26.11
 Summary: A Dagster integration for postgres
 Home-page: https://github.com/dagster-io/dagster/tree/master/python_modules/libraries/dagster-postgres
 Author: Dagster Labs

{dagster-postgres-0.26.8 → dagster-postgres-0.26.11}/dagster_postgres.egg-info/SOURCES.txt

@@ -20,4 +20,6 @@ dagster_postgres/event_log/event_log.py
 dagster_postgres/run_storage/__init__.py
 dagster_postgres/run_storage/run_storage.py
 dagster_postgres/schedule_storage/__init__.py
-dagster_postgres/schedule_storage/schedule_storage.py
+dagster_postgres/schedule_storage/schedule_storage.py
+dagster_postgres/test_fixtures/__init__.py
+dagster_postgres/test_fixtures/docker-compose.yml

dagster-postgres-0.26.11/dagster_postgres.egg-info/requires.txt

@@ -0,0 +1,2 @@
+dagster==1.10.11
+psycopg2-binary

{dagster-postgres-0.26.8 → dagster-postgres-0.26.11}/setup.py

@@ -38,6 +38,6 @@ setup(
     },
     include_package_data=True,
     python_requires=">=3.9,<3.13",
-    install_requires=["dagster==1.10.8", "psycopg2-binary"],
+    install_requires=["dagster==1.10.11", "psycopg2-binary"],
     zip_safe=False,
 )

dagster-postgres-0.26.8/dagster_postgres/version.py

@@ -1 +0,0 @@
-__version__ = "0.26.8"

dagster-postgres-0.26.8/dagster_postgres.egg-info/requires.txt

@@ -1,2 +0,0 @@
-dagster==1.10.8
-psycopg2-binary