fixes tests

Author: pcrespov

packages/postgres-database/src/simcore_postgres_database/errors.py

@@ -1,25 +1,29 @@
-""" aiopg errors
+"""aiopg errors
 
-    StandardError
-    |__ Warning
-    |__ Error
-        |__ InterfaceError
-        |__ DatabaseError
-            |__ DataError
-            |__ OperationalError
-            |__ IntegrityError
-            |__ InternalError
-            |__ ProgrammingError
-            |__ NotSupportedError
+StandardError
+|__ Warning
+|__ Error
+    |__ InterfaceError
+    |__ DatabaseError
+        |__ DataError
+        |__ OperationalError
+        |__ IntegrityError
+        |__ InternalError
+        |__ ProgrammingError
+        |__ NotSupportedError
 
-    - aiopg reuses DBAPI exceptions
-        SEE https://aiopg.readthedocs.io/en/stable/core.html?highlight=Exception#exceptions
-        SEE http://initd.org/psycopg/docs/module.html#dbapi-exceptions
-        SEE https://www.postgresql.org/docs/current/errcodes-appendix.html
+- aiopg reuses DBAPI exceptions
+    SEE https://aiopg.readthedocs.io/en/stable/core.html?highlight=Exception#exceptions
+    SEE http://initd.org/psycopg/docs/module.html#dbapi-exceptions
+    SEE https://www.postgresql.org/docs/current/errcodes-appendix.html
 """
+
 # NOTE: psycopg2.errors are created dynamically
 # pylint: disable=no-name-in-module
-from psycopg2 import DatabaseError, DataError
+from psycopg2 import (
+    DatabaseError,
+    DataError,
+)
 from psycopg2 import Error as DBAPIError
 from psycopg2 import (
     IntegrityError,

packages/postgres-database/tests/test_models_products_prices.py

@@ -8,9 +8,9 @@
 
 import pytest
 import sqlalchemy as sa
+from asyncpg.exceptions import CheckViolationError, ForeignKeyViolationError
 from faker import Faker
 from pytest_simcore.helpers.faker_factories import random_product
-from simcore_postgres_database.errors import CheckViolation, ForeignKeyViolation
 from simcore_postgres_database.models.products import products
 from simcore_postgres_database.models.products_prices import products_prices
 from simcore_postgres_database.utils_products_prices import (
@@ -62,7 +62,7 @@ async def test_non_negative_price_not_allowed(
     connection: AsyncConnection, fake_product: Row, faker: Faker
 ):
     # negative price not allowed
-    with pytest.raises(CheckViolation) as exc_info:
+    with pytest.raises(CheckViolationError) as exc_info:
         await connection.execute(
             products_prices.insert().values(
                 product_name=fake_product.name,
@@ -102,7 +102,7 @@ async def test_delete_price_constraints(
     )
 
     # should not be able to delete a product w/o deleting price first
-    with pytest.raises(ForeignKeyViolation) as exc_info:
+    with pytest.raises(ForeignKeyViolationError) as exc_info:
         await connection.execute(products.delete())
 
     assert exc_info.match("delete")