Get started with user tests

tests/config/database.py

@@ -2,6 +2,7 @@ from sqlalchemy import create_engine
 from sqlalchemy.orm import sessionmaker
 
 from src.database import Base
+from src.models import CourseProgress, HighScore, User
 
 SQLALCHEMY_DATABASE_URL = "postgresql://admin:WeSign123!@localhost/wesigntest"
 
@@ -12,6 +13,15 @@ TestSessionLocal = sessionmaker(autocommit=False, autoflush=False, bind=engine)
 Base.metadata.create_all(bind=engine)
 
 
+def clear_db():
+    db = TestSessionLocal()
+
+    db.query(HighScore).delete()
+    db.query(CourseProgress).delete()
+    db.query(User).delete()
+    db.commit()
+
+
 def override_get_db():
     try:
         db = TestSessionLocal()

tests/config/setup.py

@@ -1,26 +0,0 @@
-import sys
-
-import pytest
-from sqlalchemy import create_engine
-
-sys.path.append("..")
-
-from src.database import Base as ProductionBase
-from tests.config.database import (SQLALCHEMY_DATABASE_URL, TestBase,
-                                   TestSessionLocal)
-
-
-@pytest.fixture(scope="function")
-def db_session():
-    engine = create_engine(SQLALCHEMY_DATABASE_URL)
-    print(SQLALCHEMY_DATABASE_URL)
-    ProductionBase.metadata.create_all(bind=engine)
-    TestBase.metadata.create_all(bind=engine)
-    session = TestSessionLocal(bind=engine)
-    try:
-        yield session
-    finally:
-        session.rollback()
-        TestBase.metadata.drop_all(bind=engine)
-        ProductionBase.metadata.drop_all(bind=engine)
-        session.close()