make pipeline pass

kris524 · Oct 28, 2023 · 879625f · 879625f
1 parent 853cbde
commit 879625f
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 3 deletions.
diff --git a/notion_to_google_task_sync.py b/notion_to_google_task_sync.py
@@ -22,12 +22,15 @@
 
 load_dotenv()
 NOTION_ID = os.getenv("NOTION_KEY")
-# if NOTION_ID is None:
-#     raise KeyError("Missing NOTION ID environment variable")
+if NOTION_ID is None:
+    raise KeyError("Missing NOTION ID environment variable")
 SCOPES = ["https://www.googleapis.com/auth/tasks"]
 
 r = redis.Redis(host="localhost", port=6379, decode_responses=True)
 
+class MockClient:
+    def __init__(self) -> None:
+        client = Client()
 
 # -----------------
 # NOTION FUNCTIONS

diff --git a/test_main.py b/test_main.py
@@ -25,7 +25,8 @@
 
 
 class TestNotionFunctions(unittest.TestCase):
-    @patch("notion_to_google_task_sync.Client")
+
+    @patch("notion_to_google_task_sync.MockClient")
     def test_get_all_pages(self, mock_client):
         """Test correct ids are collected"""
         mock_search = Mock()