Created test user

src/data_connection.py

@@ -29,3 +29,9 @@ def get_single_product(barcode: int) -> product_classes.Product:
         
 def create_db_and_tables() -> None:
     db_connection.create_db_and_tables()
+
+def get_db_engine():
+    return db_connection.get_engine()
+
+def delete_database():
+    db_connection.delete_database()

src/main.py

@@ -3,6 +3,7 @@ from fastapi.middleware.cors import CORSMiddleware
 from fastapi.responses import FileResponse
 import os
 from contextlib import asynccontextmanager
+from sqlmodel import Session
 
 
 from src.classes import product_classes, cash_classes, user_classes
@@ -11,10 +12,23 @@ from src.modules import price_to_cash_calculator
 from shop_validators import image_validator
 from src import data_connection
 
+async def get_session():
+    with Session(data_connection.get_db_engine()) as session:
+        yield session
+
+def create_test_data():
+    user_1 = user_classes.UserInDB(username="yarninator", email="yarn@inator.com",full_name="Yarn Inator", disabled=False, hashed_password="abcdefghijklmnop")
+    with Session(data_connection.get_db_engine()) as session:
+        session.add(user_1)
+        session.commit()
+
+
 @asynccontextmanager
 async def lifespan(app: FastAPI):
     # Startup
+    data_connection.delete_database()
     data_connection.create_db_and_tables()
+    create_test_data()
     yield
     # Shutdown
 

src/modules/db_connection.py

@@ -12,3 +12,6 @@ def get_engine():
 
 def create_db_and_tables() -> None:
     SQLModel.metadata.create_all(engine)
+
+def delete_database():
+    SQLModel.metadata.drop_all(engine)