add set of unit tests

This commit is contained in:
grillazz
2021-03-28 17:12:35 +02:00
parent 786bb23eab
commit c3fc4e3146
5 changed files with 102 additions and 9 deletions

View File

@@ -3,6 +3,8 @@ from typing import Any
from sqlalchemy.exc import SQLAlchemyError
from sqlalchemy.ext.asyncio import AsyncSession
from sqlalchemy.ext.declarative import as_declarative, declared_attr
from fastapi import HTTPException, status
from icecream import ic
@as_declarative()
@@ -19,5 +21,6 @@ class Base:
db_session.add(self)
return await db_session.commit()
except SQLAlchemyError as ex:
print(f"Have to rollback, save failed: {ex}")
raise
ic("Have to rollback, save failed:")
ic(ex)
raise HTTPException(status_code=status.HTTP_422_UNPROCESSABLE_ENTITY, detail=ex.__str__())

View File

@@ -12,7 +12,7 @@ class Stuff(Base):
__tablename__ = "stuff"
id = Column(UUID(as_uuid=True), unique=True, default=uuid.uuid4, autoincrement=True)
name = Column(String, nullable=False, primary_key=True, unique=True)
description = Column(String, nullable=False, unique=True)
description = Column(String, nullable=False)
def __init__(self, name: str, description: str):
self.name = name