1711c89eee
Add nodes/scene_validator_node.py: SceneValidatorNode, a thin adapter that
delegates to validate_scene (MASTER_SPEC §20).
- Required input scene (SCENE model, serialized via to_json); optional config
(COMPILER_CONFIG) defaulting to Config() when unconnected.
- Outputs (scene, warnings, errors): the validated SCENE plus warnings/errors
passed through unchanged. No compiler logic in the node.
- Registered as SceneCompilerValidator ("Scene Validator").
Tests: node metadata/inputs, valid scene pass-through, delegation + message
surfacing + config threading (monkeypatched), default config when absent, and
ComfyUI registration.
Co-authored-by: kunail0804 <kunail0804@gmail.com>
Co-authored-by: Claude Opus 4.8 <noreply@anthropic.com>
89 lines
3.0 KiB
Python
89 lines
3.0 KiB
Python
"""Tests for the Scene Validator ComfyUI node (issue #20)."""
|
|
|
|
from __future__ import annotations
|
|
|
|
import nodes
|
|
import nodes.scene_validator_node as node_module
|
|
from compiler.common.config import Config
|
|
from compiler.common.message_codes import message
|
|
from compiler.common.result import CompilerResult
|
|
from nodes.scene_validator_node import SceneValidatorNode
|
|
from schemas.models import Scene
|
|
|
|
|
|
def sample_scene() -> Scene:
|
|
return Scene.from_json(
|
|
{
|
|
"characters": [
|
|
{
|
|
"id": 0,
|
|
"identity": ["female"],
|
|
"appearance": [],
|
|
"clothing": ["dress"],
|
|
"accessories": [],
|
|
"pose": [],
|
|
"expression": ["smile"],
|
|
"actions": [],
|
|
}
|
|
],
|
|
"interactions": [],
|
|
"objects": [],
|
|
"environment": ["classroom"],
|
|
"camera": [],
|
|
"lighting": [],
|
|
"metadata": {"schema_version": "1.0"},
|
|
}
|
|
)
|
|
|
|
|
|
def test_node_metadata() -> None:
|
|
assert SceneValidatorNode.RETURN_TYPES == ("SCENE", "STRING", "STRING")
|
|
assert SceneValidatorNode.RETURN_NAMES == ("scene", "warnings", "errors")
|
|
assert SceneValidatorNode.CATEGORY == "Scene Compiler"
|
|
inputs = SceneValidatorNode.INPUT_TYPES()
|
|
assert "scene" in inputs["required"]
|
|
assert "config" in inputs["optional"]
|
|
|
|
|
|
def test_valid_scene_passes_through() -> None:
|
|
scene, warnings, errors = SceneValidatorNode().run(sample_scene())
|
|
assert isinstance(scene, Scene)
|
|
assert (warnings, errors) == ("", "")
|
|
|
|
|
|
def test_delegates_and_surfaces_messages(monkeypatch) -> None:
|
|
captured = {}
|
|
|
|
def fake_validate(scene_json, config):
|
|
captured["allow_unknown"] = config.validator.allow_unknown_fields
|
|
return (
|
|
CompilerResult(data="VALIDATED")
|
|
.add_warning(message("SC0015", "unexpected field removed"))
|
|
.add_error(message("SC0002", "bad"))
|
|
)
|
|
|
|
monkeypatch.setattr(node_module, "validate_scene", fake_validate)
|
|
config = Config.from_json({"validator": {"allow_unknown_fields": True}})
|
|
scene, warnings, errors = SceneValidatorNode().run(sample_scene(), config)
|
|
assert scene == "VALIDATED"
|
|
assert captured["allow_unknown"] is True
|
|
assert "SC0015: unexpected field removed" in warnings
|
|
assert "SC0002: bad" in errors
|
|
|
|
|
|
def test_default_config_used_when_absent(monkeypatch) -> None:
|
|
captured = {}
|
|
|
|
def fake_validate(scene_json, config):
|
|
captured["config"] = config
|
|
return CompilerResult(data="V")
|
|
|
|
monkeypatch.setattr(node_module, "validate_scene", fake_validate)
|
|
SceneValidatorNode().run(sample_scene())
|
|
assert isinstance(captured["config"], Config)
|
|
|
|
|
|
def test_registered_for_comfyui() -> None:
|
|
assert nodes.NODE_CLASS_MAPPINGS["SceneCompilerValidator"] is SceneValidatorNode
|
|
assert nodes.NODE_DISPLAY_NAME_MAPPINGS["SceneCompilerValidator"] == "Scene Validator"
|