From 3e49dbd1a72624f8f6a2ef3c35c551547bdde63f Mon Sep 17 00:00:00 2001 From: robert-min Date: Wed, 24 Jan 2024 12:17:23 +0900 Subject: [PATCH] test: github env setting check --- .coverage | Bin 53248 -> 53248 bytes src/libs/db_manager.py | 5 ++++- 2 files changed, 4 insertions(+), 1 deletion(-) diff --git a/.coverage b/.coverage index b5ea4783566ea1a8601043c51344052e5d90df6a..a932903dc9f6d536440a20d32acd26c15f6b13b5 100644 GIT binary patch delta 22 dcmZozz}&Eac|%=4Bje=y{%FRU&A0ni9ROzO2>}2A delta 22 dcmZozz}&Eac|%=4BlG0?{%A)2&A0ni9ROr=2$lc< diff --git a/src/libs/db_manager.py b/src/libs/db_manager.py index 57e62f2..619c304 100644 --- 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)