Merge pull request #59 from PennChopMicrobiomeProgram/58-safeguard-against-test-db-overwrites

Require test db to be empty for creation

Author: Ulthran

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,