wehub-resource-sync
80 lines
2.1 KiB
Python
80 lines
2.1 KiB
Python
"""Tests for policy entity dataclass."""
|
|
|
|
from __future__ import annotations
|
|
|
|
from omnigent.entities.policy import Policy
|
|
|
|
|
|
def test_policy_minimal() -> None:
|
|
pol = Policy(
|
|
id="pol_abc123",
|
|
name="block_push",
|
|
session_id="conv_1",
|
|
scope="session",
|
|
created_at=1700000000,
|
|
type="python",
|
|
handler="omnigent.policies.builtins.safety.block_push",
|
|
)
|
|
assert pol.id == "pol_abc123"
|
|
assert pol.name == "block_push"
|
|
assert pol.session_id == "conv_1"
|
|
assert pol.scope == "session"
|
|
assert pol.type == "python"
|
|
assert pol.handler == "omnigent.policies.builtins.safety.block_push"
|
|
assert pol.factory_params is None
|
|
assert pol.enabled is True
|
|
assert pol.updated_at is None
|
|
assert pol.created_by is None
|
|
|
|
|
|
def test_policy_full() -> None:
|
|
pol = Policy(
|
|
id="pol_xyz",
|
|
name="cost_budget",
|
|
session_id=None,
|
|
scope="default",
|
|
created_at=1700000000,
|
|
type="python",
|
|
handler="omnigent.policies.builtins.cost.cost_budget",
|
|
factory_params={"limit": 10.0, "currency": "USD"},
|
|
enabled=False,
|
|
updated_at=1700001000,
|
|
created_by="admin@example.com",
|
|
)
|
|
assert pol.session_id is None # server-wide default
|
|
assert pol.scope == "default"
|
|
assert pol.factory_params == {"limit": 10.0, "currency": "USD"}
|
|
assert pol.enabled is False
|
|
assert pol.updated_at == 1700001000
|
|
assert pol.created_by == "admin@example.com"
|
|
|
|
|
|
def test_policy_url_type() -> None:
|
|
pol = Policy(
|
|
id="pol_url1",
|
|
name="external_check",
|
|
session_id="conv_1",
|
|
scope="session",
|
|
created_at=1700000000,
|
|
type="url",
|
|
handler="https://hooks.example.com/policy",
|
|
)
|
|
assert pol.type == "url"
|
|
assert pol.handler.startswith("https://")
|
|
|
|
|
|
def test_policy_is_mutable() -> None:
|
|
pol = Policy(
|
|
id="pol_1",
|
|
name="p",
|
|
session_id=None,
|
|
scope="default",
|
|
created_at=1,
|
|
type="python",
|
|
handler="mod.func",
|
|
)
|
|
pol.enabled = False
|
|
pol.updated_at = 2
|
|
assert pol.enabled is False
|
|
assert pol.updated_at == 2
|