38 lines
1.0 KiB
Python
38 lines
1.0 KiB
Python
|
from __future__ import annotations
|
||
|
|
|
||
|
|
import pytest
|
||
|
|
|
||
|
|
from schemas.agent.consumer_registry import AgentConsumerBinding, ConsumerRegistry
|
||
|
|
from schemas.agent.pipeline_spec import (
|
||
|
|
ContextPolicy,
|
||
|
|
ExecutorKind,
|
||
|
|
PipelineSpec,
|
||
|
|
StageSpec,
|
||
|
|
)
|
||
|
|
|
||
|
|
|
||
|
|
def test_consumer_registry_rejects_duplicate_bits() -> None:
|
||
|
|
with pytest.raises(ValueError, match="duplicate visibility bit"):
|
||
|
|
ConsumerRegistry(
|
||
|
|
bindings=[
|
||
|
|
AgentConsumerBinding(agent_type="router", bit=16),
|
||
|
|
AgentConsumerBinding(agent_type="worker", bit=16),
|
||
|
|
]
|
||
|
|
)
|
||
|
|
|
||
|
|
|
||
|
|
def test_pipeline_spec_requires_non_empty_stages() -> None:
|
||
|
|
with pytest.raises(ValueError, match="at least 1 item"):
|
||
|
|
PipelineSpec(mode="worker", stages=[])
|
||
|
|
|
||
|
|
|
||
|
|
def test_stage_spec_normalizes_stage_name() -> None:
|
||
|
|
spec = StageSpec(
|
||
|
|
stage_name=" Worker ",
|
||
|
|
executor_kind=ExecutorKind.REACT,
|
||
|
|
default_visibility_mask=1,
|
||
|
|
context_policy=ContextPolicy(consumer_agent_type="worker", count=20),
|
||
|
|
)
|
||
|
|
|
||
|
|
assert spec.stage_name == "worker"
|