16 lines
926 B
Python
16 lines
926 B
Python
from central.tests.test_control_plane import setup_db
|
|
from central.scripts.generate_control_panel_fixture import generate_fixture
|
|
from central.app.models import AgentEvent, AgentSnapshot, Installation, Job
|
|
|
|
|
|
def test_fixture_creates_one_hundred_varied_installations_and_events():
|
|
session_factory = setup_db()
|
|
with session_factory() as session:
|
|
result = generate_fixture(session, agent_count=100, seed=11)
|
|
assert result == {"agents": 100, "events": 500, "blockers": 50, "jobs": 100}
|
|
assert session.query(Installation).count() == 100
|
|
assert session.query(AgentEvent).count() == 500
|
|
assert session.query(Job).count() == 100
|
|
assert {row.agent_health for row in session.query(AgentSnapshot).all()} == {"ONLINE", "DEGRADED", "OFFLINE", "FAILED", "UNKNOWN"}
|
|
assert {row.version_drift for row in session.query(AgentSnapshot).all()} == {"CURRENT", "OUTDATED", "UNKNOWN"}
|