31 lines
1.5 KiB
Python
31 lines
1.5 KiB
Python
|
import datetime
|
||
|
|
from uuid import UUID
|
||
|
|
|
||
|
|
from teal.db import UniqueViolation
|
||
|
|
|
||
|
|
|
||
|
|
def test_unique_violation():
|
||
|
|
class IntegrityErrorMock:
|
||
|
|
def __init__(self) -> None:
|
||
|
|
self.params = {
|
||
|
|
'uuid': UUID('f5efd26e-8754-46bc-87bf-fbccc39d60d9'),
|
||
|
|
'version': '11.0',
|
||
|
|
'software': 'Workbench', 'elapsed': datetime.timedelta(0, 4),
|
||
|
|
'expected_events': None,
|
||
|
|
'id': UUID('dbdef3d8-2cac-48cb-adb8-419bc3e59687')
|
||
|
|
}
|
||
|
|
|
||
|
|
def __str__(self):
|
||
|
|
return """(psycopg2.IntegrityError) duplicate key value violates unique constraint "snapshot_uuid_key"
|
||
|
|
DETAIL: Key (uuid)=(f5efd26e-8754-46bc-87bf-fbccc39d60d9) already exists.
|
||
|
|
[SQL: 'INSERT INTO snapshot (uuid, version, software, elapsed, expected_events, id)
|
||
|
|
VALUES (%(uuid)s, %(version)s, %(software)s, %(elapsed)s, CAST(%(expected_events)s
|
||
|
|
AS snapshotexpectedevents[]), %(id)s)'] [parameters: {'uuid': UUID('f5efd26e-8754-46bc-87bf-fbccc39d60d9'),
|
||
|
|
'version': '11.0', 'software': 'Workbench', 'elapsed': datetime.timedelta(0, 4), 'expected_events': None,
|
||
|
|
'id': UUID('dbdef3d8-2cac-48cb-adb8-419bc3e59687')}] (Background on this error at: http://sqlalche.me/e/gkpj)"""
|
||
|
|
|
||
|
|
u = UniqueViolation(IntegrityErrorMock())
|
||
|
|
assert u.constraint == 'snapshot_uuid_key'
|
||
|
|
assert u.field_name == 'uuid'
|
||
|
|
assert u.field_value == UUID('f5efd26e-8754-46bc-87bf-fbccc39d60d9')
|