mirror of
https://github.com/comfyanonymous/ComfyUI.git
synced 2025-07-29 09:16:48 +00:00
[tests] Handle null device index in schema validation tests
ComfyUI returns null for CPU device index, but OpenAPI/JSON Schema doesn't have good support for nullable fields. Remove null index fields before validation as a workaround.
This commit is contained in:
parent
d74a189e1c
commit
cb66a887af
@ -181,6 +181,13 @@ def test_response_schema_validation(
|
|||||||
pytest.skip(f"Endpoint {endpoint_path} did not return valid JSON")
|
pytest.skip(f"Endpoint {endpoint_path} did not return valid JSON")
|
||||||
return
|
return
|
||||||
|
|
||||||
|
# Special handling for system_stats endpoint
|
||||||
|
if endpoint_path == '/system_stats' and isinstance(response_data, dict):
|
||||||
|
# Remove null index fields before validation
|
||||||
|
for device in response_data.get('devices', []):
|
||||||
|
if 'index' in device and device['index'] is None:
|
||||||
|
del device['index']
|
||||||
|
|
||||||
# Validate the response
|
# Validate the response
|
||||||
validation_result = validate_response(
|
validation_result = validate_response(
|
||||||
response_data,
|
response_data,
|
||||||
@ -242,6 +249,12 @@ def test_system_stats_response(require_server, api_client, api_spec: Dict[str, A
|
|||||||
assert 'vram_total' in device, "Device missing 'vram_total' field"
|
assert 'vram_total' in device, "Device missing 'vram_total' field"
|
||||||
assert 'vram_free' in device, "Device missing 'vram_free' field"
|
assert 'vram_free' in device, "Device missing 'vram_free' field"
|
||||||
|
|
||||||
|
# Remove null index fields before validation
|
||||||
|
# This is needed because ComfyUI returns null for CPU device index
|
||||||
|
for device in stats.get('devices', []):
|
||||||
|
if 'index' in device and device['index'] is None:
|
||||||
|
del device['index']
|
||||||
|
|
||||||
# Perform schema validation
|
# Perform schema validation
|
||||||
validation_result = validate_response(
|
validation_result = validate_response(
|
||||||
stats,
|
stats,
|
||||||
|
Loading…
x
Reference in New Issue
Block a user