From d6655676f3a094b19e7b3fa9a2eedd45b04ae21a Mon Sep 17 00:00:00 2001
From: Yariv Menachem <yariv220895@gmail.com>
Date: Wed, 11 Dec 2024 17:40:40 +0200
Subject: [PATCH] added defualt value for creating athe repo

---
 src/jobspy/db/job_repository.py | 19 ++++++++++++-------
 src/tests/test_db.py            |  6 +++++-
 2 files changed, 17 insertions(+), 8 deletions(-)

diff --git a/src/jobspy/db/job_repository.py b/src/jobspy/db/job_repository.py
index a10c7f2..45bfd3e 100644
--- a/src/jobspy/db/job_repository.py
+++ b/src/jobspy/db/job_repository.py
@@ -1,25 +1,30 @@
 import os
 from typing import List
+from dotenv import load_dotenv
 from pymongo import MongoClient, UpdateOne
 
 from jobspy.jobs import JobPost
 
+load_dotenv()
+
 
 class JobRepository:
 
-    def __init__(self):
+    def __init__(self, database_name: str = None):
         self.mongoUri = os.getenv("MONGO_URI")
-        # Connect to MongoDB server
+        if not self.mongoUri:
+            raise ValueError("MONGO_URI environment variable is not set")
         self.client = MongoClient(self.mongoUri)
-        # Access a database (it will be created automatically if it doesn't exist)
-        self.db = self.client["jobs_database"]
-        # Access a collection
+        if database_name is None:
+            database_name = os.getenv("MONGO_DB_NAME")
+            if not database_name:
+                raise ValueError(
+                    "MONGO_DB_NAME environment variable is not set")
+        self.db = self.client[database_name]
         self.collection = self.db["jobs"]
 
     def insert_job(self, job: JobPost):
-        # Convert JobPost to dictionary
         job_dict = job.model_dump(exclude={"date_posted"})
-        # If it doesn't exist, insert a new job with the current `created_at` and `updated_at`
         self.collection.insert_one(job_dict)
         print(f"Inserted new job with title {job.title}.")
 
diff --git a/src/tests/test_db.py b/src/tests/test_db.py
index 27ebda1..dbd296a 100644
--- a/src/tests/test_db.py
+++ b/src/tests/test_db.py
@@ -1,6 +1,10 @@
+import os
+from dotenv import load_dotenv
 from jobspy.db.job_repository import JobRepository
 from tests.test_util import createMockJob
 
+load_dotenv()
+
 
 class DatabaseTests:
 
@@ -9,7 +13,7 @@ class DatabaseTests:
         This block ensures that the script runs the test only when executed directly,
         not when imported as a module.
         """
-        self.jobRepository = JobRepository()
+        self.jobRepository = JobRepository(os.getenv("MONGO_DB_NAME") + "dev")
 
     def insert_job(self):
         """