Initial Commit

tests/conftest.py

@@ -0,0 +1,27 @@
+import pytest
+from fastapi.testclient import TestClient
+from sqlmodel import Session, SQLModel, create_engine
+from sqlmodel.pool import StaticPool
+
+from uptime_pie_api.main import app, get_session
+
+
+@pytest.fixture(name="session")
+def session_fixture():
+    engine = create_engine(
+        "sqlite://", connect_args={"check_same_thread": False}, poolclass=StaticPool
+    )
+    SQLModel.metadata.create_all(engine)
+    with Session(engine) as session:
+        yield session
+
+
+@pytest.fixture(name="client")
+def client_fixture(session: Session):
+    def get_session_override():
+        return session
+
+    app.dependency_overrides[get_session] = get_session_override
+    client = TestClient(app)
+    yield client
+    app.dependency_overrides.clear()

tests/test_main.py

@@ -0,0 +1,94 @@
+import json
+import re
+
+from fastapi.testclient import TestClient
+from sqlmodel import Session
+
+from uptime_pie_api.main import Agent
+
+
+uuid_re = re.compile(
+    r"^[0-9A-F]{8}-[0-9A-F]{4}-[4][0-9A-F]{3}-[89AB][0-9A-F]{3}-[0-9A-F]{12}$",
+    flags=re.IGNORECASE,
+)
+
+
+def test_ping(client: TestClient):
+    response = client.get("/ping")
+    assert response.status_code == 200
+    assert response.text == "OK"
+
+
+def test_create_agents(client: TestClient):
+    payload = {"name": "test_agent"}
+    response = client.post("/agents", content=json.dumps(payload))
+    json_response = response.json()
+
+    assert response.status_code == 200
+    assert json_response["name"] == "test_agent"
+    assert uuid_re.match(json_response["id"])
+    assert json_response["api_key"] is not None
+
+
+def test_duplicate_create_agent(client: TestClient):
+    payload = {"name": "test_agent", "api_key": "test_api_key"}
+    response = client.post("/agents", content=json.dumps(payload))
+    assert response.status_code == 200
+
+    response = client.post("/agents", content=json.dumps(payload))
+    json_response = response.json()
+    assert response.status_code == 400
+    assert json_response["detail"] == "Agent already exists"
+
+
+def test_list_agents(client: TestClient, session: Session):
+    session.add(Agent(name="test_agent", api_key="test_api_key"))
+    session.add(Agent(name="test_agent2", api_key="test_api_key2"))
+    session.commit()
+
+    response = client.get("/agents")
+    json_response = response.json()
+
+    assert response.status_code == 200
+    assert len(json_response) == 2
+    assert json_response[0]["name"] == "test_agent"
+    assert json_response[1]["name"] == "test_agent2"
+    assert uuid_re.match(json_response[0]["id"])
+    assert uuid_re.match(json_response[1]["id"])
+
+
+def test_get_agent(client: TestClient, session: Session):
+    test_agent = Agent(name="test_agent", api_key="test_api_key")
+    session.add(test_agent)
+    session.commit()
+    session.refresh(test_agent)
+
+    response = client.get(f"/agents/{test_agent.id}")
+    json_response = response.json()
+
+    assert response.status_code == 200
+    assert json_response["name"] == "test_agent"
+    assert json_response["id"] == str(test_agent.id)
+    assert json_response["api_key"] == "test_api_key"
+
+
+def test_get_agent_not_found(client: TestClient):
+    response = client.get("/agents/00000000-aaaa-1111-bbbb-222222222222")
+    json_response = response.json()
+
+    assert response.status_code == 404
+    assert json_response["detail"] == "Agent not found"
+
+
+def test_delete_agent(client: TestClient, session: Session):
+    test_agent = Agent(name="test_agent", api_key="test_api_key")
+    session.add(test_agent)
+    session.commit()
+    session.refresh(test_agent)
+
+    response = client.delete(f"/agents/{test_agent.id}")
+    json_response = response.json()
+
+    assert response.status_code == 200
+    assert json_response["message"] == "Agent deleted"
+    assert session.get(Agent, test_agent.id) is None