apache-airflow-providers-postgres 6.7.0__tar.gz → 6.7.1__tar.gz

{apache_airflow_providers_postgres-6.7.0 → apache_airflow_providers_postgres-6.7.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: apache-airflow-providers-postgres
-Version: 6.7.0
+Version: 6.7.1
 Summary: Provider package apache-airflow-providers-postgres for Apache Airflow
 Keywords: airflow-provider,postgres,airflow,integration
 Author-email: Apache Software Foundation <dev@airflow.apache.org>
@@ -40,8 +40,8 @@ Requires-Dist: psycopg[binary]>=3.2.9 ; extra == "psycopg" and ( python_version
 Requires-Dist: psycopg[binary]>=3.3.3 ; extra == "psycopg" and ( python_version >= '3.14')
 Requires-Dist: sqlalchemy>=1.4.54 ; extra == "sqlalchemy"
 Project-URL: Bug Tracker, https://github.com/apache/airflow/issues
-Project-URL: Changelog, https://airflow.apache.org/docs/apache-airflow-providers-postgres/6.7.0/changelog.html
-Project-URL: Documentation, https://airflow.apache.org/docs/apache-airflow-providers-postgres/6.7.0
+Project-URL: Changelog, https://airflow.apache.org/docs/apache-airflow-providers-postgres/6.7.1/changelog.html
+Project-URL: Documentation, https://airflow.apache.org/docs/apache-airflow-providers-postgres/6.7.1
 Project-URL: Mastodon, https://fosstodon.org/@airflow
 Project-URL: Slack Chat, https://s.apache.org/airflow-slack
 Project-URL: Source Code, https://github.com/apache/airflow
@@ -79,7 +79,7 @@ Provides-Extra: sqlalchemy
 
 Package ``apache-airflow-providers-postgres``
 
-Release: ``6.7.0``
+Release: ``6.7.1``
 
 
 `PostgreSQL <https://www.postgresql.org/>`__
@@ -92,7 +92,7 @@ This is a provider package for ``postgres`` provider. All classes for this provi
 are in ``airflow.providers.postgres`` python package.
 
 You can find package information and changelog for the provider
-in the `documentation <https://airflow.apache.org/docs/apache-airflow-providers-postgres/6.7.0/>`_.
+in the `documentation <https://airflow.apache.org/docs/apache-airflow-providers-postgres/6.7.1/>`_.
 
 Installation
 ------------
@@ -156,5 +156,5 @@ Extra                Dependencies
 ===================  ============================================================================================================================================================
 
 The changelog for the provider package can be found in the
-`changelog <https://airflow.apache.org/docs/apache-airflow-providers-postgres/6.7.0/changelog.html>`_.
+`changelog <https://airflow.apache.org/docs/apache-airflow-providers-postgres/6.7.1/changelog.html>`_.
 

{apache_airflow_providers_postgres-6.7.0 → apache_airflow_providers_postgres-6.7.1}/README.rst

@@ -23,7 +23,7 @@
 
 Package ``apache-airflow-providers-postgres``
 
-Release: ``6.7.0``
+Release: ``6.7.1``
 
 
 `PostgreSQL <https://www.postgresql.org/>`__
@@ -36,7 +36,7 @@ This is a provider package for ``postgres`` provider. All classes for this provi
 are in ``airflow.providers.postgres`` python package.
 
 You can find package information and changelog for the provider
-in the `documentation <https://airflow.apache.org/docs/apache-airflow-providers-postgres/6.7.0/>`_.
+in the `documentation <https://airflow.apache.org/docs/apache-airflow-providers-postgres/6.7.1/>`_.
 
 Installation
 ------------
@@ -100,4 +100,4 @@ Extra                Dependencies
 ===================  ============================================================================================================================================================
 
 The changelog for the provider package can be found in the
-`changelog <https://airflow.apache.org/docs/apache-airflow-providers-postgres/6.7.0/changelog.html>`_.
+`changelog <https://airflow.apache.org/docs/apache-airflow-providers-postgres/6.7.1/changelog.html>`_.

{apache_airflow_providers_postgres-6.7.0 → apache_airflow_providers_postgres-6.7.1}/docs/changelog.rst

@@ -27,6 +27,18 @@
 Changelog
 ---------
 
+6.7.1
+.....
+
+Misc
+~~~~
+
+* ``Refactor PostgresHook and associated runtime tests (#66893)``
+
+.. Below changes are excluded from the changelog. Move them to
+   appropriate section above if needed. Do not delete the lines(!):
+
+
 6.7.0
 .....
 

{apache_airflow_providers_postgres-6.7.0 → apache_airflow_providers_postgres-6.7.1}/docs/index.rst

@@ -78,7 +78,7 @@ apache-airflow-providers-postgres package
 `PostgreSQL <https://www.postgresql.org/>`__
 
 
-Release: 6.7.0
+Release: 6.7.1
 
 Provider package
 ----------------
@@ -138,5 +138,5 @@ Downloading official packages
 You can download officially released packages and verify their checksums and signatures from the
 `Official Apache Download site <https://downloads.apache.org/airflow/providers/>`_
 
-* `The apache-airflow-providers-postgres 6.7.0 sdist package <https://downloads.apache.org/airflow/providers/apache_airflow_providers_postgres-6.7.0.tar.gz>`_ (`asc <https://downloads.apache.org/airflow/providers/apache_airflow_providers_postgres-6.7.0.tar.gz.asc>`__, `sha512 <https://downloads.apache.org/airflow/providers/apache_airflow_providers_postgres-6.7.0.tar.gz.sha512>`__)
-* `The apache-airflow-providers-postgres 6.7.0 wheel package <https://downloads.apache.org/airflow/providers/apache_airflow_providers_postgres-6.7.0-py3-none-any.whl>`_ (`asc <https://downloads.apache.org/airflow/providers/apache_airflow_providers_postgres-6.7.0-py3-none-any.whl.asc>`__, `sha512 <https://downloads.apache.org/airflow/providers/apache_airflow_providers_postgres-6.7.0-py3-none-any.whl.sha512>`__)
+* `The apache-airflow-providers-postgres 6.7.1 sdist package <https://downloads.apache.org/airflow/providers/apache_airflow_providers_postgres-6.7.1.tar.gz>`_ (`asc <https://downloads.apache.org/airflow/providers/apache_airflow_providers_postgres-6.7.1.tar.gz.asc>`__, `sha512 <https://downloads.apache.org/airflow/providers/apache_airflow_providers_postgres-6.7.1.tar.gz.sha512>`__)
+* `The apache-airflow-providers-postgres 6.7.1 wheel package <https://downloads.apache.org/airflow/providers/apache_airflow_providers_postgres-6.7.1-py3-none-any.whl>`_ (`asc <https://downloads.apache.org/airflow/providers/apache_airflow_providers_postgres-6.7.1-py3-none-any.whl.asc>`__, `sha512 <https://downloads.apache.org/airflow/providers/apache_airflow_providers_postgres-6.7.1-py3-none-any.whl.sha512>`__)

{apache_airflow_providers_postgres-6.7.0 → apache_airflow_providers_postgres-6.7.1}/provider.yaml

@@ -23,12 +23,13 @@ description: |
 
 state: ready
 lifecycle: production
-source-date-epoch: 1779138287
+source-date-epoch: 1780426378
 # Note that those versions are maintained by release manager - do not update them manually
 # with the exception of case where other provider in sources has >= new provider version.
 # In such case adding >= NEW_VERSION and bumping to NEW_VERSION in a provider have
 # to be done in the same PR
 versions:
+  - 6.7.1
   - 6.7.0
   - 6.6.3
   - 6.6.2

{apache_airflow_providers_postgres-6.7.0 → apache_airflow_providers_postgres-6.7.1}/pyproject.toml

@@ -25,7 +25,7 @@ build-backend = "flit_core.buildapi"
 
 [project]
 name = "apache-airflow-providers-postgres"
-version = "6.7.0"
+version = "6.7.1"
 description = "Provider package apache-airflow-providers-postgres for Apache Airflow"
 readme = "README.rst"
 license = "Apache-2.0"
@@ -137,8 +137,8 @@ apache-airflow-providers-common-sql = {workspace = true}
 apache-airflow-providers-standard = {workspace = true}
 
 [project.urls]
-"Documentation" = "https://airflow.apache.org/docs/apache-airflow-providers-postgres/6.7.0"
-"Changelog" = "https://airflow.apache.org/docs/apache-airflow-providers-postgres/6.7.0/changelog.html"
+"Documentation" = "https://airflow.apache.org/docs/apache-airflow-providers-postgres/6.7.1"
+"Changelog" = "https://airflow.apache.org/docs/apache-airflow-providers-postgres/6.7.1/changelog.html"
 "Bug Tracker" = "https://github.com/apache/airflow/issues"
 "Source Code" = "https://github.com/apache/airflow"
 "Slack Chat" = "https://s.apache.org/airflow-slack"

{apache_airflow_providers_postgres-6.7.0 → apache_airflow_providers_postgres-6.7.1}/src/airflow/providers/postgres/__init__.py

@@ -29,7 +29,7 @@ from airflow import __version__ as airflow_version
 
 __all__ = ["__version__"]
 
-__version__ = "6.7.0"
+__version__ = "6.7.1"
 
 if packaging.version.parse(packaging.version.parse(airflow_version).base_version) < packaging.version.parse(
     "2.11.0"

{apache_airflow_providers_postgres-6.7.0 → apache_airflow_providers_postgres-6.7.1}/src/airflow/providers/postgres/hooks/postgres.py

@@ -230,6 +230,29 @@ class PostgresHook(DbApiHook):
         valid_cursors = ", ".join(cursor_types.keys())
         raise ValueError(f"Invalid cursor passed {_cursor}. Valid options are: {valid_cursors}")
 
+    def _get_cursor_config(self, raw_cursor: str) -> tuple[str, Any]:
+        cursor = self._get_cursor(raw_cursor)
+
+        if USE_PSYCOPG3:
+            return "row_factory", cursor
+
+        return "cursor_factory", cursor
+
+    def _create_connection(self, conn_args: dict[str, Any]) -> CompatConnection:
+        if USE_PSYCOPG3:
+            from psycopg.connection import Connection as pgConnection
+
+            connection = pgConnection.connect(**cast("Any", conn_args))
+
+            register_default_adapters(connection)
+
+            if self.enable_log_db_messages and hasattr(connection, "add_notice_handler"):
+                connection.add_notice_handler(self._notice_handler)
+
+            return connection
+
+        return ppg2_connect(**conn_args)
+
     def _generate_cursor_name(self):
         """Generate a unique name for server-side cursor."""
         import uuid
@@ -262,30 +285,13 @@ class PostgresHook(DbApiHook):
             if arg_name not in self.ignored_extra_options:
                 conn_args[arg_name] = arg_val
 
-        if USE_PSYCOPG3:
-            from psycopg.connection import Connection as pgConnection
-
-            raw_cursor = conn.extra_dejson.get("cursor")
-            if raw_cursor:
-                conn_args["row_factory"] = self._get_cursor(raw_cursor)
-
-            # Use Any type for the connection args to avoid type conflicts
-            connection = pgConnection.connect(**cast("Any", conn_args))
-            self.conn = cast("CompatConnection", connection)
-
-            # Register JSON handlers for both json and jsonb types
-            # This ensures JSON data is properly decoded from bytes to Python objects
-            register_default_adapters(connection)
+        raw_cursor = conn.extra_dejson.get("cursor")
 
-            # Add the notice handler AFTER the connection is established
-            if self.enable_log_db_messages and hasattr(self.conn, "add_notice_handler"):
-                self.conn.add_notice_handler(self._notice_handler)
-        else:  # psycopg2
-            raw_cursor = conn.extra_dejson.get("cursor", False)
-            if raw_cursor:
-                conn_args["cursor_factory"] = self._get_cursor(raw_cursor)
+        if raw_cursor:
+            key, value = self._get_cursor_config(raw_cursor)
+            conn_args[key] = value
 
-            self.conn = cast("CompatConnection", ppg2_connect(**conn_args))
+        self.conn = self._create_connection(conn_args)
 
         return self.conn
 

{apache_airflow_providers_postgres-6.7.0 → apache_airflow_providers_postgres-6.7.1}/tests/unit/postgres/hooks/test_postgres.py

@@ -19,7 +19,6 @@ from __future__ import annotations
 
 import json
 import os
-from types import SimpleNamespace
 from unittest import mock
 
 import pandas as pd
@@ -30,7 +29,7 @@ import sqlalchemy
 from airflow.models import Connection
 from airflow.providers.common.compat.sdk import AirflowException, AirflowOptionalProviderFeatureException
 from airflow.providers.postgres.dialects.postgres import PostgresDialect
-from airflow.providers.postgres.hooks.postgres import CompatConnection, PostgresHook
+from airflow.providers.postgres.hooks.postgres import PostgresHook
 
 from tests_common.test_utils.common_sql import mock_db_hook
 from tests_common.test_utils.version_compat import NOTSET
@@ -59,36 +58,6 @@ else:
     import psycopg2.extras
 
 
-@pytest.fixture
-def postgres_hook_setup():
-    """Set up mock PostgresHook for testing."""
-    table = "test_postgres_hook_table"
-    cur = mock.MagicMock(rowcount=0)
-    conn = mock.MagicMock(spec=CompatConnection)
-    conn.cursor.return_value = cur
-
-    class UnitTestPostgresHook(PostgresHook):
-        conn_name_attr = "test_conn_id"
-
-        def get_conn(self):
-            return conn
-
-    db_hook = UnitTestPostgresHook()
-
-    # Return a namespace with all the objects
-    setup = SimpleNamespace(table=table, cur=cur, conn=conn, db_hook=db_hook)
-
-    yield setup
-
-    # Teardown - only for real database tests
-    try:
-        with PostgresHook().get_conn() as real_conn:
-            with real_conn.cursor() as real_cur:
-                real_cur.execute(f"DROP TABLE IF EXISTS {table}")
-    except Exception:
-        pass  # Ignore cleanup errors for unit tests
-
-
 @pytest.fixture
 def mock_connect(mocker):
     """Mock the connection object according to the correct psycopg version."""
@@ -816,10 +785,8 @@ class TestPostgresHook:
         ) == INSERT_SQL_STATEMENT.format('"schema"')
 
 
-@pytest.mark.backend("postgres")
-@pytest.mark.skipif(USE_PSYCOPG3, reason="psycopg v3 is available")
-class TestPostgresHookPPG2:
-    """PostgresHook tests that are specific to psycopg2."""
+class _BasePostgresHookRuntimeTests:
+    """Shared runtime tests for psycopg2 and psycopg3."""
 
     table = "test_postgres_hook_table"
 
@@ -841,6 +808,121 @@ class TestPostgresHookPPG2:
             with conn.cursor() as cur:
                 cur.execute(f"DROP TABLE IF EXISTS {self.table}")
 
+    def test_insert_rows(self):
+        table = "table"
+        rows = [("hello",), ("world",)]
+
+        self.db_hook.insert_rows(table, rows)
+
+        assert self.conn.close.call_count == 1
+        assert self.cur.close.call_count == 1
+        assert self.conn.commit.call_count == 2
+
+        sql = f"INSERT INTO {table}  VALUES (%s)"
+        self.cur.executemany.assert_any_call(sql, rows)
+
+    def test_insert_rows_replace(self):
+        table = "table"
+        rows = [
+            (1, "hello"),
+            (2, "world"),
+        ]
+        fields = ("id", "value")
+
+        self.db_hook.insert_rows(
+            table,
+            rows,
+            fields,
+            replace=True,
+            replace_index=fields[0],
+        )
+
+        assert self.conn.close.call_count == 1
+        assert self.cur.close.call_count == 1
+        assert self.conn.commit.call_count == 2
+
+        sql = (
+            f"INSERT INTO {table} ({fields[0]}, {fields[1]}) VALUES (%s,%s) "
+            f"ON CONFLICT ({fields[0]}) DO UPDATE SET {fields[1]} = excluded.{fields[1]}"
+        )
+        self.cur.executemany.assert_any_call(sql, rows)
+
+    def test_insert_rows_replace_missing_target_field_arg(self):
+        table = "table"
+        rows = [
+            (1, "hello"),
+            (2, "world"),
+        ]
+        fields = ("id", "value")
+
+        with pytest.raises(
+            ValueError,
+            match="PostgreSQL ON CONFLICT upsert syntax requires column names",
+        ):
+            self.db_hook.insert_rows(
+                table,
+                rows,
+                replace=True,
+                replace_index=fields[0],
+            )
+
+    def test_insert_rows_replace_missing_replace_index_arg(self):
+        table = "table"
+        rows = [
+            (1, "hello"),
+            (2, "world"),
+        ]
+        fields = ("id", "value")
+
+        with pytest.raises(
+            ValueError,
+            match="PostgreSQL ON CONFLICT upsert syntax requires an unique index",
+        ):
+            self.db_hook.insert_rows(
+                table,
+                rows,
+                fields,
+                replace=True,
+            )
+
+    def test_insert_rows_replace_all_index(self):
+        table = "table"
+        rows = [
+            (1, "hello"),
+            (2, "world"),
+        ]
+        fields = ("id", "value")
+
+        self.db_hook.insert_rows(
+            table,
+            rows,
+            fields,
+            replace=True,
+            replace_index=fields,
+        )
+
+        assert self.conn.close.call_count == 1
+        assert self.cur.close.call_count == 1
+        assert self.conn.commit.call_count == 2
+
+        sql = (
+            f"INSERT INTO {table} ({', '.join(fields)}) VALUES (%s,%s) "
+            f"ON CONFLICT ({', '.join(fields)}) DO NOTHING"
+        )
+        self.cur.executemany.assert_any_call(sql, rows)
+
+    def test_dialect_name(self):
+        assert self.db_hook.dialect_name == "postgresql"
+
+    def test_dialect(self):
+        assert isinstance(self.db_hook.dialect, PostgresDialect)
+
+
+@pytest.mark.backend("postgres")
+@pytest.mark.skipif(USE_PSYCOPG3, reason="psycopg v3 is available")
+class TestPostgresHookPPG2(_BasePostgresHookRuntimeTests):
+    """PostgresHook tests that are specific to psycopg2."""
+
     def test_copy_expert(self, mocker):
         open_mock = mocker.mock_open(read_data='{"some": "json"}')
         mocker.patch("airflow.providers.postgres.hooks.postgres.open", open_mock)
@@ -915,169 +997,59 @@ class TestPostgresHookPPG2:
         assert call_kw["sql"] == sql
         assert call_kw["sql_parameters"] == parameters
 
-    def test_insert_rows(self, postgres_hook_setup):
-        setup = postgres_hook_setup
-        table = "table"
-        rows = [("hello",), ("world",)]
-
-        setup.db_hook.insert_rows(table, rows)
-
-        assert setup.conn.close.call_count == 1
-        assert setup.cur.close.call_count == 1
-
-        commit_count = 2  # The first and last commit
-        assert commit_count == setup.conn.commit.call_count
-
-        sql = f"INSERT INTO {table}  VALUES (%s)"
-        setup.cur.executemany.assert_any_call(sql, rows)
-
     @mock.patch("airflow.providers.common.sql.hooks.sql.send_sql_hook_lineage")
-    def test_insert_rows_hook_lineage(self, mock_send_lineage, postgres_hook_setup):
-        setup = postgres_hook_setup
+    def test_insert_rows_hook_lineage(self, mock_send_lineage):
         table = "table"
         rows = [("hello",), ("world",)]
 
-        setup.db_hook.insert_rows(table, rows)
+        self.db_hook.insert_rows(table, rows)
 
         mock_send_lineage.assert_called_once()
+
         call_kw = mock_send_lineage.call_args.kwargs
-        assert call_kw["context"] is setup.db_hook
+
+        assert call_kw["context"] is self.db_hook
         assert call_kw["sql"] == f"INSERT INTO {table}  VALUES (%s)"
         assert call_kw["row_count"] == 2
 
     @mock.patch("airflow.providers.postgres.hooks.postgres.execute_batch")
-    def test_insert_rows_fast_executemany(self, mock_execute_batch, postgres_hook_setup):
-        setup = postgres_hook_setup
+    def test_insert_rows_fast_executemany(self, mock_execute_batch):
         table = "table"
         rows = [("hello",), ("world",)]
 
-        setup.db_hook.insert_rows(table, rows, fast_executemany=True)
+        self.db_hook.insert_rows(table, rows, fast_executemany=True)
 
-        assert setup.conn.close.call_count == 1
-        assert setup.cur.close.call_count == 1
+        assert self.conn.close.call_count == 1
+        assert self.cur.close.call_count == 1
 
         commit_count = 2  # The first and last commit
-        assert setup.conn.commit.call_count == commit_count
+        assert self.conn.commit.call_count == commit_count
 
         mock_execute_batch.assert_called_once_with(
-            setup.cur,
+            self.cur,
             f"INSERT INTO {table}  VALUES (%s)",  # expected SQL
             [("hello",), ("world",)],  # expected values
             page_size=1000,
         )
 
         # executemany should NOT be called in this mode
-        setup.cur.executemany.assert_not_called()
+        self.cur.executemany.assert_not_called()
 
     @mock.patch("airflow.providers.postgres.hooks.postgres.send_sql_hook_lineage")
     @mock.patch("airflow.providers.postgres.hooks.postgres.execute_batch")
-    def test_insert_rows_fast_executemany_hook_lineage(
-        self, mock_execute_batch, mock_send_lineage, postgres_hook_setup
-    ):
-        setup = postgres_hook_setup
+    def test_insert_rows_fast_executemany_hook_lineage(self, mock_execute_batch, mock_send_lineage):
+
         table = "table"
         rows = [("hello",), ("world",)]
 
-        setup.db_hook.insert_rows(table, rows, fast_executemany=True)
+        self.db_hook.insert_rows(table, rows, fast_executemany=True)
 
         mock_send_lineage.assert_called_once()
         call_kw = mock_send_lineage.call_args.kwargs
-        assert call_kw["context"] is setup.db_hook
+        assert call_kw["context"] is self.db_hook
         assert call_kw["sql"] == f"INSERT INTO {table}  VALUES (%s)"
         assert call_kw["row_count"] == 2
 
-    def test_insert_rows_replace(self, postgres_hook_setup):
-        setup = postgres_hook_setup
-        table = "table"
-        rows = [
-            (
-                1,
-                "hello",
-            ),
-            (
-                2,
-                "world",
-            ),
-        ]
-        fields = ("id", "value")
-
-        setup.db_hook.insert_rows(table, rows, fields, replace=True, replace_index=fields[0])
-
-        assert setup.conn.close.call_count == 1
-        assert setup.cur.close.call_count == 1
-
-        commit_count = 2  # The first and last commit
-        assert commit_count == setup.conn.commit.call_count
-
-        sql = (
-            f"INSERT INTO {table} ({fields[0]}, {fields[1]}) VALUES (%s,%s) "
-            f"ON CONFLICT ({fields[0]}) DO UPDATE SET {fields[1]} = excluded.{fields[1]}"
-        )
-        setup.cur.executemany.assert_any_call(sql, rows)
-
-    def test_insert_rows_replace_missing_target_field_arg(self, postgres_hook_setup):
-        setup = postgres_hook_setup
-        table = "table"
-        rows = [
-            (
-                1,
-                "hello",
-            ),
-            (
-                2,
-                "world",
-            ),
-        ]
-        fields = ("id", "value")
-        with pytest.raises(ValueError, match="PostgreSQL ON CONFLICT upsert syntax requires column names"):
-            setup.db_hook.insert_rows(table, rows, replace=True, replace_index=fields[0])
-
-    def test_insert_rows_replace_missing_replace_index_arg(self, postgres_hook_setup):
-        setup = postgres_hook_setup
-        table = "table"
-        rows = [
-            (
-                1,
-                "hello",
-            ),
-            (
-                2,
-                "world",
-            ),
-        ]
-        fields = ("id", "value")
-        with pytest.raises(ValueError, match="PostgreSQL ON CONFLICT upsert syntax requires an unique index"):
-            setup.db_hook.insert_rows(table, rows, fields, replace=True)
-
-    def test_insert_rows_replace_all_index(self, postgres_hook_setup):
-        setup = postgres_hook_setup
-        table = "table"
-        rows = [
-            (
-                1,
-                "hello",
-            ),
-            (
-                2,
-                "world",
-            ),
-        ]
-        fields = ("id", "value")
-
-        setup.db_hook.insert_rows(table, rows, fields, replace=True, replace_index=fields)
-
-        assert setup.conn.close.call_count == 1
-        assert setup.cur.close.call_count == 1
-
-        commit_count = 2  # The first and last commit
-        assert commit_count == setup.conn.commit.call_count
-
-        sql = (
-            f"INSERT INTO {table} ({', '.join(fields)}) VALUES (%s,%s) "
-            f"ON CONFLICT ({', '.join(fields)}) DO NOTHING"
-        )
-        setup.cur.executemany.assert_any_call(sql, rows)
-
     @pytest.mark.usefixtures("reset_logging_config")
     def test_get_all_db_log_messages(self, mocker):
         messages = ["a", "b", "c"]
@@ -1120,40 +1092,12 @@ class TestPostgresHookPPG2:
         finally:
             hook.run(sql=f"DROP PROCEDURE {proc_name} (s text)")
 
-    def test_dialect_name(self, postgres_hook_setup):
-        setup = postgres_hook_setup
-        assert setup.db_hook.dialect_name == "postgresql"
-
-    def test_dialect(self, postgres_hook_setup):
-        setup = postgres_hook_setup
-        assert isinstance(setup.db_hook.dialect, PostgresDialect)
-
 
 @pytest.mark.backend("postgres")
 @pytest.mark.skipif(not USE_PSYCOPG3, reason="psycopg v3 or sqlalchemy v2 are not available")
-class TestPostgresHookPPG3:
+class TestPostgresHookPPG3(_BasePostgresHookRuntimeTests):
     """PostgresHook tests that are specific to psycopg3."""
 
-    table = "test_postgres_hook_table"
-
-    def setup_method(self):
-        self.cur = mock.MagicMock(rowcount=0)
-        self.conn = conn = mock.MagicMock()
-        self.conn.cursor.return_value = self.cur
-
-        class UnitTestPostgresHook(PostgresHook):
-            conn_name_attr = "test_conn_id"
-
-            def get_conn(self):
-                return conn
-
-        self.db_hook = UnitTestPostgresHook()
-
-    def teardown_method(self):
-        with PostgresHook().get_conn() as conn:
-            with conn.cursor() as cur:
-                cur.execute(f"DROP TABLE IF EXISTS {self.table}")
-
     def test_copy_expert_from(self, mocker):
         """Tests copy_expert with a 'COPY FROM STDIN' operation."""
         statement = "COPY test_table FROM STDIN"
@@ -1235,109 +1179,6 @@ class TestPostgresHookPPG3:
         )
         self.conn.commit.assert_called_once()
 
-    def test_insert_rows(self):
-        table = "table"
-        rows = [("hello",), ("world",)]
-
-        self.db_hook.insert_rows(table, rows)
-
-        assert self.conn.close.call_count == 1
-        assert self.cur.close.call_count == 1
-
-        commit_count = 2  # The first and last commit
-        assert commit_count == self.conn.commit.call_count
-
-        sql = f"INSERT INTO {table}  VALUES (%s)"
-        self.cur.executemany.assert_any_call(sql, rows)
-
-    def test_insert_rows_replace(self):
-        table = "table"
-        rows = [
-            (
-                1,
-                "hello",
-            ),
-            (
-                2,
-                "world",
-            ),
-        ]
-        fields = ("id", "value")
-
-        self.db_hook.insert_rows(table, rows, fields, replace=True, replace_index=fields[0])
-
-        assert self.conn.close.call_count == 1
-        assert self.cur.close.call_count == 1
-
-        commit_count = 2  # The first and last commit
-        assert commit_count == self.conn.commit.call_count
-
-        sql = (
-            f"INSERT INTO {table} ({fields[0]}, {fields[1]}) VALUES (%s,%s) "
-            f"ON CONFLICT ({fields[0]}) DO UPDATE SET {fields[1]} = excluded.{fields[1]}"
-        )
-        self.cur.executemany.assert_any_call(sql, rows)
-
-    def test_insert_rows_replace_missing_target_field_arg(self):
-        table = "table"
-        rows = [
-            (
-                1,
-                "hello",
-            ),
-            (
-                2,
-                "world",
-            ),
-        ]
-        fields = ("id", "value")
-        with pytest.raises(ValueError, match="PostgreSQL ON CONFLICT upsert syntax requires column names"):
-            self.db_hook.insert_rows(table, rows, replace=True, replace_index=fields[0])
-
-    def test_insert_rows_replace_missing_replace_index_arg(self):
-        table = "table"
-        rows = [
-            (
-                1,
-                "hello",
-            ),
-            (
-                2,
-                "world",
-            ),
-        ]
-        fields = ("id", "value")
-        with pytest.raises(ValueError, match="PostgreSQL ON CONFLICT upsert syntax requires an unique index"):
-            self.db_hook.insert_rows(table, rows, fields, replace=True)
-
-    def test_insert_rows_replace_all_index(self):
-        table = "table"
-        rows = [
-            (
-                1,
-                "hello",
-            ),
-            (
-                2,
-                "world",
-            ),
-        ]
-        fields = ("id", "value")
-
-        self.db_hook.insert_rows(table, rows, fields, replace=True, replace_index=fields)
-
-        assert self.conn.close.call_count == 1
-        assert self.cur.close.call_count == 1
-
-        commit_count = 2  # The first and last commit
-        assert commit_count == self.conn.commit.call_count
-
-        sql = (
-            f"INSERT INTO {table} ({', '.join(fields)}) VALUES (%s,%s) "
-            f"ON CONFLICT ({', '.join(fields)}) DO NOTHING"
-        )
-        self.cur.executemany.assert_any_call(sql, rows)
-
     @pytest.mark.skip(reason="Notice handling is callback-based in psycopg3 and cannot be tested this way.")
     def test_get_all_db_log_messages(self, mocker):
         pass
@@ -1366,9 +1207,3 @@ class TestPostgresHookPPG3:
             mock_logger.info.assert_any_call("Message from db: 42")
         finally:
             hook.run(sql=f"DROP PROCEDURE {proc_name} (s text)")
-
-    def test_dialect_name(self):
-        assert self.db_hook.dialect_name == "postgresql"
-
-    def test_dialect(self):
-        assert isinstance(self.db_hook.dialect, PostgresDialect)