Delete testing user afer unit test session

database/crud.py

@@ -2,6 +2,7 @@ from datetime import datetime
 
 from fastapi import HTTPException
 from passlib.context import CryptContext
+from sqlalchemy.inspection import inspect
 
 from app.schemas import *
 from constants import SHA1_SALT
@@ -31,11 +32,12 @@ def insert_data(model, data, db):
     return item
 
 
-# FIXME db.id has to be replaced with the table's UID
 def delete_data(model, data, db):
-    item = instantiate_model(model=model, data=data)
-    result = db.query(item).filter(item.email == data.email).delete()
-    return result
+    table = eval(model)
+    primary_key_tuple = inspect(table).primary_key
+    primary_key = primary_key_tuple[0]
+    db.query(table).filter(primary_key == data).delete()
+    db.commit()
 
 
 def fetch_user_by_key(data, db):