catch different exc in tests for different backends

ormar/models/metaclass.py

@@ -248,7 +248,10 @@ def populate_meta_sqlalchemy_table_if_required(
 ) -> Type["Model"]:
     if not hasattr(new_model.Meta, "table"):
         new_model.Meta.table = sqlalchemy.Table(
-            new_model.Meta.tablename, new_model.Meta.metadata, *new_model.Meta.columns, *new_model.Meta.constraints
+            new_model.Meta.tablename,
+            new_model.Meta.metadata,
+            *new_model.Meta.columns,
+            *new_model.Meta.constraints,
         )
     return new_model
 
@@ -306,7 +309,7 @@ class ModelMetaclass(pydantic.main.ModelMetaclass):
         )
 
         if hasattr(new_model, "Meta"):
-            if not hasattr(new_model.Meta, 'constraints'):
+            if not hasattr(new_model.Meta, "constraints"):
                 new_model.Meta.constraints = []
             new_model = populate_meta_orm_model_fields(attrs, new_model)
             new_model = populate_meta_tablename_columns_and_pk(name, new_model)

tests/test_unique_constraints.py

@@ -1,10 +1,11 @@
 import asyncio
 import sqlite3
 
+import asyncpg
 import databases
+import pymysql
 import pytest
 import sqlalchemy
-from sqlalchemy.exc import IntegrityError
 
 import ormar
 from tests.settings import DATABASE_URL
@@ -18,7 +19,7 @@ class Product(ormar.Model):
         tablename = "products"
         metadata = metadata
         database = database
-        constraints = [ormar.UniqueColumns('name', 'company')]
+        constraints = [ormar.UniqueColumns("name", "company")]
 
     id: ormar.Integer(primary_key=True)
     name: ormar.String(max_length=100)
@@ -45,9 +46,15 @@ async def create_test_database():
 async def test_unique_columns():
     async with database:
         async with database.transaction(force_rollback=True):
-            await Product.objects.create(name='Cookies', company='Nestle')
-            await Product.objects.create(name='Mars', company='Mars')
-            await Product.objects.create(name='Mars', company='Nestle')
+            await Product.objects.create(name="Cookies", company="Nestle")
+            await Product.objects.create(name="Mars", company="Mars")
+            await Product.objects.create(name="Mars", company="Nestle")
 
-            with pytest.raises((IntegrityError, sqlite3.IntegrityError)):
-                await Product.objects.create(name='Mars', company='Mars')
+            with pytest.raises(
+                (
+                    sqlite3.IntegrityError,
+                    pymysql.IntegrityError,
+                    asyncpg.exceptions.UniqueViolationError,
+                )
+            ):
+                await Product.objects.create(name="Mars", company="Mars")