fix: failing baseten test

flowaicom · Oct 28, 2024 · 1fd7e6a · 1fd7e6a
1 parent 27413b4
commit 1fd7e6a
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/tests/unit/models/test_baseten.py b/tests/unit/models/test_baseten.py
@@ -361,6 +361,14 @@ async def test_baseten_init_valid(
 
         # Test asynchronous initialization without webhook secret
         monkeypatch.delenv("BASETEN_WEBHOOK_SECRET", raising=False)
+    with (
+        patch("flow_judge.models.baseten.get_deployed_model_id", return_value="mock_model_id"),
+        patch("flow_judge.models.baseten.ensure_model_deployment", return_value=True),
+        patch(
+            "flow_judge.models.baseten.ensure_baseten_webhook_secret",
+            return_value=False,
+        ),
+    ):
         with pytest.raises(
             BasetenError,
             match=(