Add migration test for flow_exists

Author: PGijsbers

tests/conftest.py

@@ -87,3 +87,22 @@ def flow(expdb_test: Connection) -> Flow:
     )
     (flow_id,) = expdb_test.execute(text("""SELECT LAST_INSERT_ID();""")).one()
     return Flow(id=flow_id, name="name", external_version="external_version")
+
+
+@pytest.fixture()
+def persisted_flow(flow: Flow, expdb_test: Connection) -> Iterator[Flow]:
+    expdb_test.commit()
+    yield flow
+    # We want to ensure the commit below does not accidentally persist new
+    # data to the database.
+    expdb_test.rollback()
+    expdb_test.execute(
+        text(
+            """
+            DELETE FROM implementation
+            WHERE id = :flow_id
+            """,
+        ),
+        parameters={"flow_id": flow.id},
+    )
+    expdb_test.commit()

tests/routers/openml/migration/flows_migration_test.py

@@ -9,10 +9,17 @@
 )
 from starlette.testclient import TestClient
 
+from tests.conftest import Flow
 
+
+@pytest.mark.mut()
 @pytest.mark.php()
-def test_flow_exists(py_api: TestClient, php_api: TestClient) -> None:
-    path = "exists/weka.ZeroR/Weka_3.9.0_12024"
+def test_flow_exists(
+    persisted_flow: Flow,
+    py_api: TestClient,
+    php_api: TestClient,
+) -> None:
+    path = f"exists/{persisted_flow.name}/{persisted_flow.external_version}"
     py_response = py_api.get(f"/flows/{path}")
     php_response = php_api.get(f"/flow/{path}")