*

Author: kartikbehl99

ssh_manager_backend/db/database.py

@@ -2,7 +2,9 @@
 from sqlalchemy.ext.declarative import declarative_base
 from sqlalchemy.orm import scoped_session, sessionmaker
 
-DB_URI = "postgresql+psycopg2://postgres:ssh_manager@localhost/ssh_manager_dev"
+DB_URI = (
+    "postgresql+psycopg2://postgres:vP28ObNJLhb5qFDe@35.222.241.198/ssh_manager_test"
+)
 engine = create_engine(DB_URI, echo=True)
 db_session = scoped_session(
     sessionmaker(autocommit=False, autoflush=False, bind=engine)

tests/test_ssh_manager_backend.py

@@ -1,5 +1,17 @@
 from ssh_manager_backend import __version__
+from ssh_manager_backend.db.database import db_session
+from ssh_manager_backend.db.schema import AccessControl, Key, KeyMapping, User
 
 
 def test_version():
     assert __version__ == "0.1.0"
+
+
+def test_delete():
+    session = db_session()
+    session.query(User).delete()
+    session.query(Key).delete()
+    session.query(KeyMapping).delete()
+    session.query(AccessControl).delete()
+    session.commit()
+    db_session.remove()