51 lines
2.1 KiB
Python
51 lines
2.1 KiB
Python
"""Real-framework acceptance; opt in with COMFYUI_SOURCE_ROOT."""
|
|
import json
|
|
import os
|
|
from pathlib import Path
|
|
import subprocess
|
|
import sys
|
|
|
|
import pytest
|
|
|
|
|
|
def test_native_v3_runtime():
|
|
comfy_root = os.environ.get('COMFYUI_SOURCE_ROOT')
|
|
if not comfy_root:
|
|
pytest.skip('Set COMFYUI_SOURCE_ROOT to run native ComfyUI V3 acceptance')
|
|
helper = Path(__file__).parent / 'helpers' / 'v3_runtime_check.py'
|
|
result = subprocess.run(
|
|
[sys.executable, str(helper), str(Path(comfy_root).resolve())],
|
|
capture_output=True, text=True, timeout=120,
|
|
env={**os.environ, 'COMFYUI_IS_WORKER': '1'},
|
|
)
|
|
assert result.returncode == 0, result.stdout + '\n' + result.stderr
|
|
assert 'V3_ACCEPTANCE_OK' in result.stdout
|
|
|
|
|
|
def test_image_list_network_runtime():
|
|
comfy_root = os.environ.get('COMFYUI_SOURCE_ROOT')
|
|
if not comfy_root:
|
|
pytest.skip('Set COMFYUI_SOURCE_ROOT for model-free multi-process HTTP acceptance')
|
|
helper = Path(__file__).parent / 'helpers' / 'image_list_network_check.py'
|
|
result = subprocess.run(
|
|
[sys.executable, str(helper), str(Path(comfy_root).resolve())],
|
|
capture_output=True, text=True, timeout=180,
|
|
env={**os.environ, 'COMFYUI_IS_WORKER': '1'},
|
|
)
|
|
assert result.returncode == 0, result.stdout + '\n' + result.stderr
|
|
assert 'IMAGE_LIST_NETWORK_OK' in result.stdout
|
|
assert 'IMAGE_LIST_CANCEL_OK' in result.stdout
|
|
assert 'IMAGE_LIST_MASTER_OK' in result.stdout
|
|
assert 'IMAGE_LIST_PREFLIGHT_OK' in result.stdout
|
|
assert 'IMAGE_LIST_SEQUENTIAL_OK' in result.stdout
|
|
assert 'IMAGE_LIST_FAILURE_OK' in result.stdout
|
|
reports = [json.loads(line.removeprefix('IMAGE_LIST_TARGETS_OK '))
|
|
for line in result.stdout.splitlines() if line.startswith('IMAGE_LIST_TARGETS_OK ')]
|
|
assert len(reports) == 1, result.stdout
|
|
assert reports[0] == {
|
|
'modes': ['delegate-only', 'mixed', 'master-only'],
|
|
'outer_targets': ['4', '5'], 'worker_targets': ['4'],
|
|
'required_utility_once_per_item': True, 'preview_discarded': True,
|
|
'saved_prompt_unchanged': True, 'capture_once_per_outer_prompt': True,
|
|
}
|