test: github env setting check

FLYAI4 · Jan 24, 2024 · 3e49dbd · 3e49dbd
1 parent 7b1f6d1
commit 3e49dbd
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 1 deletion.
diff --git a/.coverage b/.coverage
diff --git a/src/libs/db_manager.py b/src/libs/db_manager.py
@@ -12,8 +12,11 @@ def __init__(
         self.db = os.environ.get('DB_NAME')
 
     def get_session(self) -> pymongo.MongoClient:
+        if not self.url:
+            raise Exception("No url")
+
         try:
             self.client = pymongo.MongoClient(self.url)
             return self.client[self.db]
         except Exception as e:
-            DBError(**DBErrorCode.DBConnectionError.value, err=e)
+            raise DBError(**DBErrorCode.DBConnectionError.value, err=e)