Require test db to be empty for creation

PennChopMicrobiomeProgram · Nov 21, 2024 · a5f471a · a5f471a
1 parent 0a71ab9
commit a5f471a
Showing 1 changed file with 4 additions and 8 deletions.
diff --git a/app/metadatalib/src/metadatalib/db.py b/app/metadatalib/src/metadatalib/db.py
@@ -14,15 +14,11 @@ def create_test_db(session: Session = None):
         session = imported_session
         Base.metadata.create_all(engine)
 
-    if "sqlite" not in SQLALCHEMY_DATABASE_URI:
-        print("Not a SQLite database, skipping test database creation.")
-        sys.exit(1)
-
     if session.query(Project).count():
-        session.execute(delete(Project))
-        session.execute(delete(Submission))
-        session.execute(delete(Sample))
-        session.execute(delete(Annotation))
+        sys.stderr.write(
+            "Database already contains data, skipping test data creation\n"
+        )
+        sys.exit(1)
 
     p1 = Project(
         project_id=1,