"""Unit tests for weekly_activity.aggregate — stubbed sources, no network."""

from __future__ import annotations

import os
import unittest
from datetime import datetime
from unittest import mock

from weekly_activity.aggregate import (
    SourceSpec,
    build_specs,
    collect_specs,
    format_combined,
)
from weekly_activity.config import (
    ActivityConfig,
    BtsSettings,
    GithubSettings,
    GitlabSettings,
    LaunchpadSettings,
)
from weekly_activity.model import ActivityReport, Section


class StubSource:
    def __init__(self, name: str = "stub", *, fail: bool = False) -> None:
        self.name = name
        self._fail = fail

    def collect(self, username: str, since: datetime, until: datetime) -> ActivityReport:
        if self._fail:
            raise RuntimeError("kaput")
        return ActivityReport(
            source=self.name,
            username=username,
            since=since,
            until=until,
            sections=[Section("Things", ["did a thing"])],
            warnings=["some query failed"],
        )


def window() -> tuple[datetime, datetime]:
    return datetime(2026, 8, 19), datetime(2026, 8, 26)


class BuildSpecsTest(unittest.TestCase):
    def test_maps_every_section_in_order(self) -> None:
        config = ActivityConfig(
            launchpad=LaunchpadSettings(username="lp"),
            github=GithubSettings(username="gh"),
            gitlab=GitlabSettings(username="gl"),
            bts=BtsSettings(email="bts@mail"),
        )
        specs = build_specs(config)
        self.assertEqual([s.name for s in specs], ["launchpad", "github", "gitlab", "bts"])
        self.assertEqual([s.username for s in specs], ["lp", "gh", "gl", "bts@mail"])

    def test_disabled_sections_are_skipped(self) -> None:
        config = ActivityConfig(github=GithubSettings(username="only-one"))
        self.assertEqual([s.name for s in build_specs(config)], ["github"])

    def test_launchpad_spec_expands_tilde_credentials_file(self) -> None:
        settings = LaunchpadSettings(
            username="u", mode="credentials", credentials_file="~/creds.json"
        )
        spec = build_specs(ActivityConfig(launchpad=settings))[0]
        with mock.patch("weekly_activity.aggregate.LaunchpadSource") as fake_source:
            spec.make()
        kwargs = fake_source.call_args.kwargs
        self.assertEqual(kwargs["credentials_file"], os.path.expanduser("~") + "/creds.json")
        self.assertFalse(kwargs["anonymous"])


class CollectAndFormatTest(unittest.TestCase):
    def test_failure_isolated_per_source(self) -> None:
        since, until = window()
        specs = [
            SourceSpec(name="alpha", username="a", make=lambda: StubSource("alpha")),
            SourceSpec(name="beta", username="b", make=lambda: StubSource("beta", fail=True)),
        ]
        collected = collect_specs(specs, since, until)
        self.assertIsNotNone(collected[0].report)
        self.assertIn("RuntimeError: kaput", collected[1].error or "")

        text = format_combined(collected, since, until)
        self.assertIn("Weekly activity report — 2 source(s)", text)
        self.assertIn("Activity report — alpha / ~a", text)
        self.assertIn("[skipped: beta — RuntimeError: kaput]", text)

    def test_warning_surfaced_inside_block(self) -> None:
        since, until = window()
        specs = [SourceSpec(name="gamma", username="g", make=lambda: StubSource("gamma"))]
        text = format_combined(collect_specs(specs, since, until), since, until)
        self.assertIn("Warnings (partial data — some queries failed):", text)
        self.assertIn("some query failed", text)


if __name__ == "__main__":
    unittest.main()