OpeItcLoc03/meeting-room
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity
Files
49e0ce5290436a3b2c7fe44caaa341bad71917f7
meeting-room/tests/test_models.py
vitya d292d32afa feat: add Pydantic models and EventEmitter for v2 core 
- models.py: ProviderConfig, RoleConfig, DiscussionConfig, ScenarioConfig,
  Message, ToolCallRecord, Session — compatible with config.yaml format
- events.py: Event constants, Event dataclass, EventEmitter with on/off/emit
- 67 tests passing (37 models + 30 events)

Co-Authored-By: Claude Opus 4.7 <noreply@anthropic.com>
2026-05-04 00:00:21 +03:00

396 lines
13 KiB
Python
Raw Blame History

"""Tests for meeting_room.models — Pydantic v2 data models."""
import copy
from datetime import datetime, timezone
import pytest
import yaml
from meeting_room.models import (
DefaultsConfig,
DiscussionConfig,
Message,
ProviderConfig,
RoleConfig,
ScenarioConfig,
Session,
ToolCallRecord,
)
# ---------------------------------------------------------------------------
# Fixtures — sample data matching config.yaml structure
# ---------------------------------------------------------------------------
SAMPLE_YAML = """\
providers:
routerai:
base_url: "https://routerai.ru/api/v1"
api_key: "sk-test-key"
ollama_cloud:
api_key: "02a8f6e9f9744088885982ac645f1ce1.HB1MFh4AlTpyic9oJGxQjNuA"
base_url: "https://ollama.com/v1"
roles:
moderator:
name: "Moderator"
provider: ollama_cloud
model: "gpt-4o-mini"
temperature: 0.7
tools: "none"
system_prompt: |
You are the moderator of a discussion.
skeptic:
name: "Skeptic"
provider: ollama_cloud
model: "gpt-4o-mini"
temperature: 0.9
tools: "readonly"
system_prompt: |
You are a skeptic and critic.
defaults:
max_rounds: 8
language: "ru"
framework: "custom"
workdir: "."
"""
@pytest.fixture
def yaml_config() -> dict:
"""Parse SAMPLE_YAML the same way cli.py does."""
return yaml.safe_load(SAMPLE_YAML)
# ===========================================================================
# ProviderConfig
# ===========================================================================
class TestProviderConfig:
def test_basic(self):
p = ProviderConfig(base_url="https://api.example.com/v1", api_key="sk-123")
assert p.base_url == "https://api.example.com/v1"
assert p.api_key == "sk-123"
def test_extra_fields_forbidden(self):
with pytest.raises(Exception):
ProviderConfig(base_url="https://x", api_key="k", unknown_field="oops")
def test_missing_required(self):
with pytest.raises(Exception):
ProviderConfig(base_url="https://x") # missing api_key
# ===========================================================================
# RoleConfig
# ===========================================================================
class TestRoleConfig:
def test_basic(self):
r = RoleConfig(
name="Moderator",
provider="ollama_cloud",
model="gpt-4o-mini",
temperature=0.7,
tools="none",
system_prompt="You moderate.",
)
assert r.name == "Moderator"
assert r.temperature == 0.7
assert r.tools == "none"
def test_defaults(self):
r = RoleConfig(name="Bot", provider="p", model="m")
assert r.temperature == 0.7
assert r.tools == "none"
assert r.system_prompt == ""
def test_tools_string_values(self):
for val in ("all", "none", "readonly", "web", "full", "files"):
r = RoleConfig(name="X", provider="p", model="m", tools=val)
assert r.tools == val
def test_tools_list(self):
r = RoleConfig(name="X", provider="p", model="m", tools=["read_file", "web_search"])
assert r.tools == ["read_file", "web_search"]
def test_tools_coercion_list_items_to_str(self):
r = RoleConfig(name="X", provider="p", model="m", tools=["read_file"])
assert isinstance(r.tools[0], str)
def test_tools_invalid_type(self):
with pytest.raises(Exception):
RoleConfig(name="X", provider="p", model="m", tools=42)
def test_extra_fields_forbidden(self):
with pytest.raises(Exception):
RoleConfig(name="X", provider="p", model="m", rogue_field=True)
# ===========================================================================
# DefaultsConfig
# ===========================================================================
class TestDefaultsConfig:
def test_defaults(self):
d = DefaultsConfig()
assert d.max_rounds == 8
assert d.language == "ru"
assert d.framework == "custom"
assert d.workdir == "."
def test_custom(self):
d = DefaultsConfig(max_rounds=12, language="en", framework="autogen", workdir="/tmp")
assert d.max_rounds == 12
def test_extra_ignored(self):
# DefaultsConfig uses extra="ignore" — unknown keys are silently dropped
d = DefaultsConfig(max_rounds=5, unknown_key="oops")
assert d.max_rounds == 5
assert not hasattr(d, "unknown_key")
# ===========================================================================
# DiscussionConfig — the critical one
# ===========================================================================
class TestDiscussionConfig:
def test_from_yaml_config(self, yaml_config: dict):
"""DiscussionConfig(**yaml_config) must work without changes."""
cfg = DiscussionConfig(**yaml_config)
assert "routerai" in cfg.providers
assert "ollama_cloud" in cfg.providers
assert cfg.providers["routerai"].base_url == "https://routerai.ru/api/v1"
assert "moderator" in cfg.roles
assert "skeptic" in cfg.roles
assert cfg.roles["moderator"].name == "Moderator"
assert cfg.roles["skeptic"].tools == "readonly"
assert cfg.defaults.max_rounds == 8
def test_roundtrip_serialization(self, yaml_config: dict):
"""Model can be built from yaml, exported, and rebuilt."""
cfg = DiscussionConfig(**yaml_config)
exported = cfg.model_dump()
rebuilt = DiscussionConfig(**exported)
assert rebuilt.providers.keys() == cfg.providers.keys()
assert rebuilt.roles.keys() == cfg.roles.keys()
def test_empty_config(self):
cfg = DiscussionConfig()
assert cfg.providers == {}
assert cfg.roles == {}
assert cfg.defaults.max_rounds == 8
def test_extra_keys_ignored(self):
# DiscussionConfig uses extra="ignore"
cfg = DiscussionConfig(
providers={},
roles={},
defaults={},
unknown_top_level="should be dropped",
)
assert not hasattr(cfg, "unknown_top_level")
def test_nested_provider_is_typed(self, yaml_config: dict):
cfg = DiscussionConfig(**yaml_config)
prov = cfg.providers["routerai"]
assert isinstance(prov, ProviderConfig)
def test_nested_role_is_typed(self, yaml_config: dict):
cfg = DiscussionConfig(**yaml_config)
role = cfg.roles["moderator"]
assert isinstance(role, RoleConfig)
# ===========================================================================
# ScenarioConfig
# ===========================================================================
class TestScenarioConfig:
def test_defaults(self):
s = ScenarioConfig()
assert s.name == "Untitled"
assert s.participants == []
assert s.max_rounds == 8
assert s.problem == ""
def test_from_frontmatter(self):
data = {
"name": "SaaS Pricing",
"participants": ["moderator", "skeptic"],
"max_rounds": 6,
"problem": "How should we price our SaaS?",
}
s = ScenarioConfig(**data)
assert s.name == "SaaS Pricing"
assert s.participants == ["moderator", "skeptic"]
assert s.max_rounds == 6
def test_extra_ignored(self):
s = ScenarioConfig(name="X", surprise="drop me")
assert s.name == "X"
assert not hasattr(s, "surprise")
# ===========================================================================
# Message
# ===========================================================================
class TestMessage:
def test_basic(self):
m = Message(role_id="moderator", name="Moderator", content="Hello everyone!")
assert m.role_id == "moderator"
assert m.name == "Moderator"
assert m.content == "Hello everyone!"
def test_extra_forbidden(self):
with pytest.raises(Exception):
Message(role_id="mod", name="Mod", content="hi", extra=True)
def test_missing_required(self):
with pytest.raises(Exception):
Message(role_id="mod", name="Mod") # missing content
# ===========================================================================
# ToolCallRecord
# ===========================================================================
class TestToolCallRecord:
def test_basic(self):
t = ToolCallRecord(id="call_abc", name="read_file", arguments='{"path": "/tmp/x"}', result="file contents")
assert t.id == "call_abc"
assert t.name == "read_file"
assert t.result == "file contents"
def test_default_result(self):
t = ToolCallRecord(id="call_1", name="web_search", arguments='{"q": "test"}')
assert t.result == ""
def test_extra_forbidden(self):
with pytest.raises(Exception):
ToolCallRecord(id="1", name="n", arguments="{}", extra="bad")
# ===========================================================================
# Session
# ===========================================================================
class TestSession:
def test_defaults(self):
s = Session()
assert s.scenario.name == "Untitled"
assert s.messages == []
assert s.tool_calls == []
assert isinstance(s.created_at, datetime)
assert isinstance(s.updated_at, datetime)
def test_with_messages(self):
msg = Message(role_id="moderator", name="Moderator", content="Let's begin.")
s = Session(messages=[msg])
assert len(s.messages) == 1
assert s.messages[0].content == "Let's begin."
def test_timestamps_auto_set(self):
before = datetime.now(timezone.utc)
s = Session()
after = datetime.now(timezone.utc)
assert before <= s.created_at <= after
assert before <= s.updated_at <= after
def test_explicit_timestamps(self):
t = datetime(2025, 1, 1, 12, 0, 0, tzinfo=timezone.utc)
s = Session(created_at=t, updated_at=t)
assert s.created_at.year == 2025
def test_serialization_roundtrip(self):
msg = Message(role_id="mod", name="Mod", content="Hi")
tc = ToolCallRecord(id="c1", name="search", arguments='{}', result="ok")
s = Session(
scenario=ScenarioConfig(name="Test"),
messages=[msg],
tool_calls=[tc],
)
data = s.model_dump()
rebuilt = Session(**data)
assert rebuilt.messages[0].content == "Hi"
assert rebuilt.tool_calls[0].name == "search"
assert rebuilt.scenario.name == "Test"
def test_extra_ignored(self):
s = Session(unknown_field="drop me")
assert not hasattr(s, "unknown_field")
# ===========================================================================
# Edge cases with real config.yaml structure
# ===========================================================================
class TestConfigYamlCompatibility:
"""Verify that the models accept exactly what yaml.safe_load produces
from the actual config.yaml in the package.
"""
def test_full_config_yaml_structure(self):
"""Build DiscussionConfig from the full config.yaml structure
(minus env-var resolution) and verify every field."""
raw = yaml.safe_load(SAMPLE_YAML)
cfg = DiscussionConfig(**raw)
# Providers
assert len(cfg.providers) == 2
assert cfg.providers["routerai"].base_url == "https://routerai.ru/api/v1"
assert cfg.providers["routerai"].api_key == "sk-test-key"
assert cfg.providers["ollama_cloud"].base_url == "https://ollama.com/v1"
# Roles
assert len(cfg.roles) == 2
mod = cfg.roles["moderator"]
assert mod.name == "Moderator"
assert mod.provider == "ollama_cloud"
assert mod.model == "gpt-4o-mini"
assert mod.temperature == 0.7
assert mod.tools == "none"
assert "moderator" in mod.system_prompt.lower() or "moderator" in mod.system_prompt
sk = cfg.roles["skeptic"]
assert sk.name == "Skeptic"
assert sk.temperature == 0.9
assert sk.tools == "readonly"
# Defaults
assert cfg.defaults.max_rounds == 8
assert cfg.defaults.language == "ru"
def test_tools_as_list_in_role(self):
"""Roles can specify tools as a list of tool names."""
cfg = DiscussionConfig(
roles={
"analyst": RoleConfig(
name="Analyst",
provider="ollama_cloud",
model="gpt-4o-mini",
tools=["read_file", "web_search", "list_directory"],
),
},
)
assert cfg.roles["analyst"].tools == ["read_file", "web_search", "list_directory"]
def test_minimal_config(self):
"""Minimal valid config with just one provider and one role."""
cfg = DiscussionConfig(
providers={"local": ProviderConfig(base_url="http://localhost:11434/v1", api_key="ollama")},
roles={"bot": RoleConfig(name="Bot", provider="local", model="llama3")},
)
assert cfg.providers["local"].api_key == "ollama"
assert cfg.roles["bot"].model == "llama3"
assert cfg.defaults.max_rounds == 8 # default
Reference in New Issue View Git Blame Copy Permalink