fix database test

Pythagora-io · Dec 31, 2023 · 4dbabc9 · 4dbabc9
1 parent adc4529
commit 4dbabc9
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/pilot/test_main_e2e.py b/pilot/test_main_e2e.py
@@ -1,3 +1,4 @@
+import os
 import builtins
 import pytest
 from unittest.mock import patch
@@ -12,6 +13,8 @@
 
 def test_init():
     # When
+    os.environ['DB_NAME'] = ':memory:'
+
     args = init()
 
     # Then
@@ -58,4 +61,3 @@ def test_end_to_end(endpoint, model, monkeypatch):
     # When
     with patch('utils.questionary.questionary', mock_questionary):
         project.start()
-