63 lines
2.2 KiB
Python
63 lines
2.2 KiB
Python
#!/usr/bin/env python3
|
|
"""Tests for Phase 0: Fleet Context Grounding"""
|
|
|
|
import pytest
|
|
from datetime import datetime, timezone
|
|
from pathlib import Path
|
|
import sys
|
|
sys.path.insert(0, str(Path(__file__).parent.parent))
|
|
|
|
from fleet_context import FleetContext, GiteaFleetClient, build_fleet_context
|
|
|
|
|
|
class TestFleetContext:
|
|
"""Test suite for fleet context dataclass."""
|
|
|
|
def test_to_markdown_format(self):
|
|
ctx = FleetContext(
|
|
generated_at=datetime.now(timezone.utc).isoformat(),
|
|
repos=[{"name": "the-nexus", "open_issues_count": 3, "open_prs_count": 1}],
|
|
open_issues=[{"repo": "the-nexus", "number": 830, "title": "Deep Dive", "state": "open"}],
|
|
recent_commits=[{"repo": "timmy-config", "message": "docs: update", "author": "ezra", "when": "2026-04-05T12:00:00Z"}],
|
|
open_prs=[{"repo": "hermes-agent", "number": 42, "title": "feat: tools", "state": "open"}],
|
|
)
|
|
md = ctx.to_markdown()
|
|
assert "Fleet Context Snapshot" in md
|
|
assert "the-nexus" in md
|
|
assert "#830" in md
|
|
assert "docs: update" in md
|
|
|
|
def test_to_prompt_text(self):
|
|
ctx = FleetContext(
|
|
generated_at="2026-04-05T17:00:00Z",
|
|
repos=[],
|
|
open_issues=[],
|
|
recent_commits=[],
|
|
open_prs=[],
|
|
)
|
|
assert ctx.to_prompt_text() == ctx.to_markdown()
|
|
|
|
|
|
class TestGiteaFleetClient:
|
|
"""Test suite for Gitea API client (mocked)."""
|
|
|
|
def test_client_headers_with_token(self):
|
|
client = GiteaFleetClient("http://example.com", token="testtoken")
|
|
assert client.headers["Authorization"] == "token testtoken"
|
|
|
|
def test_client_headers_without_token(self):
|
|
client = GiteaFleetClient("http://example.com")
|
|
assert "Authorization" not in client.headers
|
|
|
|
|
|
class TestBuildFleetContext:
|
|
"""Test configuration-driven builder."""
|
|
|
|
def test_disabled_returns_none(self):
|
|
config = {"fleet_context": {"enabled": False}}
|
|
assert build_fleet_context(config) is None
|
|
|
|
def test_no_repos_returns_none(self):
|
|
config = {"fleet_context": {"enabled": True, "repos": []}}
|
|
assert build_fleet_context(config) is None
|