opensource/dify
3
0
Fork 0
mirror of https://github.com/langgenius/dify.git synced 2026-10-16 15:35:32 +08:00
Code Issues Packages Projects Releases Wiki Activity

test(tracing): reject invalid MLflow app identity before node queries

Browse Source 
Cover missing, empty and non-string app identities in workflow trace metadata. Assert that rejected identities raise before the exporter queries workflow node internals.
This commit is contained in:
-LAN-
2026-10-16 07:34:07 +08:00
parent 9bd393db55
commit 802f7abfeb
1 changed files with 7 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files
api/providers/trace/trace-mlflow/tests/unit_tests/mlflow_trace/test_mlflow_trace.py
+7
View File
@@ -863,6 +863,13 @@ class TestGetWorkflowNodes:
result = trace_instance._get_workflow_nodes(_make_workflow_trace_info())
assert result == ["n1", "n2"]
@pytest.mark.parametrize("metadata", [{}, {"app_id": ""}, {"app_id": 123}], ids=["missing", "empty", "non-string"])
def test_rejects_invalid_app_id_before_querying_nodes(self, metadata, trace_instance, mock_db):
with pytest.raises(ValueError, match="No app_id found in workflow trace metadata"):
trace_instance._get_workflow_nodes(_make_workflow_trace_info(metadata=metadata))
mock_db.session.scalars.assert_not_called()
# ── _get_node_span_type ─────────────────────────────────────────────────────