zlei6/ComfyUI
1
0
Fork 0
mirror of https://github.com/Comfy-Org/ComfyUI.git synced 2026-08-24 10:03:55 +08:00
Code Issues Packages Projects Releases Wiki Activity

Revert "Include workflow_id in all execution WebSocket messages (CORE-198) (#…" (#13901)

Browse Source 
This reverts commit 4f6018982d.
This commit is contained in:
comfyanonymous
2026-05-14 15:35:42 -07:00
committed by GitHub
parent 4f6018982d
commit 616cab4f97
7 changed files with 21 additions and 398 deletions
Download Patch File Download Diff File Expand all files Collapse all files

35
tests/execution/test_jobs.py
View File

@@ -10,44 +10,9 @@ from comfy_execution.jobs import (
get_outputs_summary,
apply_sorting,
has_3d_extension,
extract_workflow_id,
)
class TestExtractWorkflowId:
"""Unit tests for extract_workflow_id()."""
def test_returns_id_from_extra_pnginfo(self):
assert extract_workflow_id({'extra_pnginfo': {'workflow': {'id': 'wf-123'}}}) == 'wf-123'
def test_missing_extra_data_returns_none(self):
assert extract_workflow_id(None) is None
def test_non_dict_extra_data_returns_none(self):
assert extract_workflow_id('not-a-dict') is None
def test_missing_extra_pnginfo_returns_none(self):
assert extract_workflow_id({}) is None
def test_missing_workflow_returns_none(self):
assert extract_workflow_id({'extra_pnginfo': {}}) is None
def test_missing_id_returns_none(self):
assert extract_workflow_id({'extra_pnginfo': {'workflow': {}}}) is None
def test_empty_string_id_returns_none(self):
assert extract_workflow_id({'extra_pnginfo': {'workflow': {'id': ''}}}) is None
def test_non_string_id_returns_none(self):
assert extract_workflow_id({'extra_pnginfo': {'workflow': {'id': 42}}}) is None
def test_non_dict_workflow_returns_none(self):
assert extract_workflow_id({'extra_pnginfo': {'workflow': 'not-a-dict'}}) is None
def test_non_dict_extra_pnginfo_returns_none(self):
assert extract_workflow_id({'extra_pnginfo': 'not-a-dict'}) is None
class TestJobStatus:
"""Test JobStatus constants."""