Make return value depend on database response

Author: PGijsbers

src/routers/openml/flows.py

@@ -25,7 +25,7 @@ def flow_exists(
             status_code=http.client.NOT_FOUND,
             detail="Flow not found.",
         )
-    return {"flow_id": 1}
+    return {"flow_id": flow.id}
 
 
 @router.get("/{flow_id}")

tests/routers/openml/flows_test.py

@@ -1,9 +1,27 @@
 import http.client
 
 import deepdiff.diff
+from pytest_mock import MockerFixture
 from starlette.testclient import TestClient
 
 
+def test_flow_exists_api(py_api: TestClient, mocker: MockerFixture) -> None:
+    # Mock database.flows.get_by_namne(...) returns the flow with id one.
+    fake_flow = mocker.MagicMock(id=2)
+    mocked_db = mocker.patch(
+        "database.flows.get_by_name",
+        return_value=fake_flow,
+    )
+    response = py_api.get("/flows/exists/weka.ZeroR/Weka_3.9.0_12024")
+    # assert mocked_db.assert_called_once_with(
+    #     name="weka.ZeroR",
+    #     external_version="Weka_3.9.0_12024",
+    # )
+    mocked_db.assert_called()
+    assert response.status_code == http.client.OK
+    assert response.json() == {"flow_id": fake_flow.id}
+
+
 def test_flow_exists(py_api: TestClient) -> None:
     response = py_api.get("/flows/exists/weka.ZeroR/Weka_3.9.0_12024")
     assert response.status_code == http.client.OK