diff --git a/tests/clients/test_sqlalchemy.py b/tests/clients/test_sqlalchemy.py index eb931a72..20e1fadc 100644 --- a/tests/clients/test_sqlalchemy.py +++ b/tests/clients/test_sqlalchemy.py @@ -15,7 +15,7 @@ from tests.helpers import testenv engine = create_engine( - f"postgresql://{testenv['postgresql_user']}:{testenv['postgresql_pw']}@{testenv['postgresql_host']}:{testenv['postgresql_port']}/{testenv['postgresql_db']}" + f"postgresql+psycopg2://{testenv['postgresql_user']}:{testenv['postgresql_pw']}@{testenv['postgresql_host']}:{testenv['postgresql_port']}/{testenv['postgresql_db']}" ) Session = sessionmaker(bind=engine) @@ -54,7 +54,7 @@ def db_setup() -> None: password="3X}vP66ADoCFT2g?HPvoem2eJh,zWXgd36Rb/{aRq/>7EYy6@EEH4BP(oeXac@mR", ) -sqlalchemy_url = f"postgresql://{testenv['postgresql_host']}:{testenv['postgresql_port']}/{testenv['postgresql_db']}" +sqlalchemy_url = f"postgresql+psycopg2://{testenv['postgresql_host']}:{testenv['postgresql_port']}/{testenv['postgresql_db']}" @pytest.mark.usefixtures("db_setup") @@ -271,10 +271,10 @@ def test_error_before_tracing(self) -> None: current_span = get_current_span() assert not current_span.is_recording() - invalid_connection_url = "postgresql://user1:pwd1@localhost:9999/mydb1" + invalid_connection_url = "postgresql+psycopg2://user1:pwd1@localhost:9999/mydb1" with pytest.raises( OperationalError, - match=r"^(\(psycopg2\.OperationalError\)).*", + match=r"^\(psycopg2?\.OperationalError\).*", ) as context_manager: engine = create_engine(invalid_connection_url) with engine.connect() as connection: diff --git a/tests/requirements-pre315.txt b/tests/requirements-pre315.txt index 83aafcee..6a6bcd98 100644 --- a/tests/requirements-pre315.txt +++ b/tests/requirements-pre315.txt @@ -24,7 +24,7 @@ mock>=4.0.3 moto>=4.1.2 mysqlclient>=2.0.3 PyMySQL[rsa]>=1.0.2 -psycopg2-binary>=2.8.6 +psycopg2-binary>=2.9.12 pika>=1.2.0 # protobuf<=6.30.2 pymongo>=3.11.4 diff --git a/tests/requirements.txt b/tests/requirements.txt index 73a19092..0df51830 100644 --- a/tests/requirements.txt +++ b/tests/requirements.txt @@ -20,7 +20,7 @@ moto>=4.1.2 mysqlclient>=2.0.3 pymssql>=2.2.0 PyMySQL[rsa]>=1.0.2 -psycopg2-binary>=2.8.6 +psycopg2-binary>=2.9.12 pika>=1.2.0 protobuf<=6.33.4 pymongo>=3.11.4