test(focoos): update mock_remote_model fixture and clear API key for …

…initialization test
FocoosAI · Jan 8, 2025 · b964357 · b964357
1 parent e1685c4
commit b964357
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/tests/test_focoos.py b/tests/test_focoos.py
@@ -70,7 +70,7 @@ def mock_list_models_as_base_models(mock_list_models):
 
 
 @pytest.fixture
-def mock_remote_model() -> RemoteModel:
+def mock_remote_model():
     return MagicMock(spec=RemoteModel, model_ref="ref1")
 
 
@@ -85,6 +85,7 @@ def test_focoos_initialization_no_api_key(focoos_instance: Focoos):
             status_code=200, json=lambda: {"email": "[email protected]"}
         )
     )
+    FOCOOS_CONFIG.focoos_api_key = ""
     with pytest.raises(ValueError):
         Focoos(host_url="http://mock-host-url.com")