#!/usr/bin/env python3
# SPDX-License-Identifier: GPL-3.0-or-later
"""Unit tests for the pure decision core of traffic_control.py (no network).

Covers: priority resolution (P-label / broken / draft / default P5), PASS 1
preemption (only P0 cancels strictly-lower; P1-P9 never bump; self / main /
equal-or-higher never cancelled), PASS 2 hold-back (yield to strictly-higher with
an active run; same-level running-first then oldest-first), and a few end-to-end
decision scenarios."""

from __future__ import annotations

import json
import unittest

import traffic_control as tc
from traffic_control import PullRequest


def pr(number, labels=(), *, draft=False, created_at="", status=tc.NONE, run_ids=()):
    """Terse PullRequest builder for tests."""
    return PullRequest(
        number=number,
        labels=tuple(labels),
        is_draft=draft,
        created_at=created_at,
        run_status=status,
        run_ids=tuple(run_ids),
    )


class EffectivePriorityTests(unittest.TestCase):
    def test_no_labels_defaults_to_p5(self):
        self.assertEqual(tc.effective_priority(pr(1)), 5)

    def test_non_priority_labels_ignored_default_p5(self):
        self.assertEqual(tc.effective_priority(pr(1, ["bug", "enhancement"])), 5)

    def test_single_p_label(self):
        self.assertEqual(tc.effective_priority(pr(1, ["P3"])), 3)
        self.assertEqual(tc.effective_priority(pr(1, ["P0"])), 0)

    def test_lowest_numbered_p_label_wins(self):
        self.assertEqual(tc.effective_priority(pr(1, ["P4", "P1", "P7"])), 1)

    def test_broken_is_bottom_p10_overriding_p0(self):
        self.assertEqual(tc.effective_priority(pr(1, ["broken", "P0"])), 10)

    def test_draft_is_bottom_p10_overriding_p0(self):
        self.assertEqual(tc.effective_priority(pr(1, ["P0"], draft=True)), 10)

    def test_draft_and_broken_still_p10(self):
        self.assertEqual(tc.effective_priority(pr(1, ["broken"], draft=True)), 10)

    def test_double_digit_pseudo_label_is_not_a_priority(self):
        # Only P0-P9 count (regex ^P[0-9]$); "P10" is not a valid priority label.
        self.assertEqual(tc.effective_priority(pr(1, ["P10"])), 5)

    def test_priority_label_text(self):
        self.assertEqual(tc.priority_label(0), "P0")
        self.assertEqual(tc.priority_label(5), "P5")
        self.assertIn("bottom", tc.priority_label(10))


class RunsToCancelTests(unittest.TestCase):
    def test_non_p0_self_cancels_nothing(self):
        me = pr(1, ["P1"])
        others = [pr(2, ["P5"], status=tc.RUNNING, run_ids=[200])]
        self.assertEqual(tc.runs_to_cancel(me, [me, *others]), [])

    def test_non_p0_self_never_cancels_even_broken(self):
        # Behaviour change vs the old bash (see module/PR notes): P1-P9 no longer
        # reclaim a broken target's runner — only P0 preempts.
        me = pr(1, ["P3"])
        broken = pr(2, ["broken"], status=tc.RUNNING, run_ids=[200])
        self.assertEqual(tc.runs_to_cancel(me, [me, broken]), [])

    def test_p0_cancels_strictly_lower_active_runs(self):
        me = pr(1, ["P0"])
        low = pr(2, ["P5"], status=tc.RUNNING, run_ids=[200])
        queued = pr(3, ["P9"], status=tc.QUEUED, run_ids=[300])
        self.assertEqual(
            sorted(tc.runs_to_cancel(me, [me, low, queued])), [200, 300])

    def test_p0_cancels_broken_and_draft_lower_runs(self):
        me = pr(1, ["P0"])
        broken = pr(2, ["broken"], status=tc.RUNNING, run_ids=[200])
        draft = pr(3, ["P2"], draft=True, status=tc.RUNNING, run_ids=[300])
        self.assertEqual(
            sorted(tc.runs_to_cancel(me, [me, broken, draft])), [200, 300])

    def test_p0_never_cancels_equal_priority_p0(self):
        me = pr(1, ["P0"])
        peer = pr(2, ["P0"], status=tc.RUNNING, run_ids=[200])
        self.assertEqual(tc.runs_to_cancel(me, [me, peer]), [])

    def test_p0_never_cancels_self(self):
        me = pr(1, ["P0"], status=tc.RUNNING, run_ids=[100])
        self.assertEqual(tc.runs_to_cancel(me, [me]), [])

    def test_p0_excludes_own_run_id_defensively(self):
        me = pr(1, ["P0"], status=tc.RUNNING, run_ids=[100])
        # A lower PR that somehow reports our own run id must not be cancelled.
        low = pr(2, ["P5"], status=tc.RUNNING, run_ids=[100, 200])
        self.assertEqual(
            tc.runs_to_cancel(me, [me, low], self_run_id=100), [200])

    def test_p0_skips_lower_with_no_active_run(self):
        me = pr(1, ["P0"])
        idle = pr(2, ["P5"], status=tc.NONE, run_ids=[])
        self.assertEqual(tc.runs_to_cancel(me, [me, idle]), [])


class WaitBlockersTests(unittest.TestCase):
    def test_p0_never_waits(self):
        me = pr(1, ["P0"])
        higher = pr(2, ["P0"], status=tc.RUNNING)  # nothing outranks P0 anyway
        self.assertEqual(tc.wait_blockers(me, [me, higher]), [])

    def test_yields_to_strictly_higher_with_active_run(self):
        me = pr(2, ["P5"], status=tc.RUNNING)
        higher = pr(1, ["P2"], status=tc.RUNNING)
        blockers = tc.wait_blockers(me, [me, higher])
        self.assertEqual([b.number for b in blockers], [1])
        self.assertEqual(blockers[0].kind, "higher-priority")

    def test_does_not_yield_to_higher_without_active_run(self):
        me = pr(2, ["P5"], status=tc.RUNNING)
        higher_idle = pr(1, ["P2"], status=tc.NONE)
        self.assertEqual(tc.wait_blockers(me, [me, higher_idle]), [])

    def test_does_not_yield_to_lower_priority(self):
        me = pr(1, ["P2"], status=tc.RUNNING)
        lower = pr(2, ["P5"], status=tc.RUNNING)
        self.assertEqual(tc.wait_blockers(me, [me, lower]), [])

    def test_same_level_oldest_running_proceeds(self):
        me = pr(1, ["P5"], created_at="2026-07-01T00:00:00Z", status=tc.RUNNING)
        newer = pr(2, ["P5"], created_at="2026-07-02T00:00:00Z", status=tc.RUNNING)
        self.assertEqual(tc.wait_blockers(me, [me, newer]), [])

    def test_same_level_newer_running_yields_to_older(self):
        # Coordinator clarification: within a level, older createdAt goes first.
        older = pr(1, ["P5"], created_at="2026-07-01T00:00:00Z", status=tc.RUNNING)
        me = pr(2, ["P5"], created_at="2026-07-02T00:00:00Z", status=tc.RUNNING)
        blockers = tc.wait_blockers(me, [me, older])
        self.assertEqual([b.number for b in blockers], [1])
        self.assertEqual(blockers[0].kind, "same-level-ahead")

    def test_same_level_running_first_beats_older_waiting(self):
        # An in-flight peer keeps its place; a not-yet-running OLDER peer does not
        # jump ahead of us while we are the one already running.
        me = pr(2, ["P5"], created_at="2026-07-02T00:00:00Z", status=tc.RUNNING)
        older_waiting = pr(1, ["P5"], created_at="2026-07-01T00:00:00Z", status=tc.NONE)
        self.assertEqual(tc.wait_blockers(me, [me, older_waiting]), [])

    def test_same_level_waiting_orders_oldest_before_newer(self):
        # Neither running: strictly oldest-first among the waiting bucket.
        oldest = pr(1, ["P5"], created_at="2026-07-01T00:00:00Z", status=tc.NONE)
        middle = pr(2, ["P5"], created_at="2026-07-02T00:00:00Z", status=tc.NONE)
        me = pr(3, ["P5"], created_at="2026-07-03T00:00:00Z", status=tc.NONE)
        blockers = tc.wait_blockers(me, [oldest, middle, me])
        self.assertEqual([b.number for b in blockers], [1, 2])

    def test_same_level_queued_counts_as_waiting_ordered_by_age(self):
        # A queued peer is "waiting to start", not in-flight: ordered purely by age.
        me = pr(1, ["P5"], created_at="2026-07-01T00:00:00Z", status=tc.NONE)
        newer_queued = pr(2, ["P5"], created_at="2026-07-02T00:00:00Z", status=tc.QUEUED)
        self.assertEqual(tc.wait_blockers(me, [me, newer_queued]), [])

    def test_broken_self_yields_to_everyone_active(self):
        me = pr(1, ["broken"], status=tc.RUNNING)          # effective P10
        normal = pr(2, ["P5"], status=tc.RUNNING)
        blockers = tc.wait_blockers(me, [me, normal])
        self.assertEqual([b.number for b in blockers], [2])
        self.assertEqual(blockers[0].kind, "higher-priority")


class EndToEndDecisionTests(unittest.TestCase):
    def test_p0_emergency_cancels_lower_and_proceeds(self):
        me = pr(10, ["P0"], status=tc.RUNNING, run_ids=[1000])
        prs = [
            me,
            pr(11, ["P2"], status=tc.RUNNING, run_ids=[1100]),
            pr(12, ["P5"], status=tc.QUEUED, run_ids=[1200]),
            pr(13, ["P0"], status=tc.RUNNING, run_ids=[1300]),  # equal — spared
        ]
        dec = tc.decide(me, prs, self_run_id=1000)
        self.assertEqual(sorted(dec.cancel_run_ids), [1100, 1200])
        self.assertTrue(dec.proceed)

    def test_p5_waits_behind_running_higher(self):
        me = pr(20, ["P5"], status=tc.RUNNING, run_ids=[2000])
        higher = pr(21, ["P2"], status=tc.RUNNING, run_ids=[2100])
        dec = tc.decide(me, [me, higher])
        self.assertEqual(dec.cancel_run_ids, ())     # not P0 — cancels nothing
        self.assertFalse(dec.proceed)
        self.assertEqual([b.number for b in dec.blockers], [21])

    def test_lone_p5_proceeds(self):
        me = pr(30, ["P5"], status=tc.RUNNING, run_ids=[3000])
        dec = tc.decide(me, [me])
        self.assertEqual(dec.cancel_run_ids, ())
        self.assertTrue(dec.proceed)


class SnapshotParsingTests(unittest.TestCase):
    def test_from_json_label_objects_and_fields(self):
        obj = {
            "number": 7,
            "labels": [{"name": "P3"}, {"name": "bug"}],
            "isDraft": True,
            "createdAt": "2026-07-01T00:00:00Z",
            "runStatus": "running",
            "runIds": [42, 43],
        }
        p = PullRequest.from_json(obj)
        self.assertEqual(p.number, 7)
        self.assertEqual(p.labels, ("P3", "bug"))
        self.assertTrue(p.is_draft)
        self.assertEqual(p.run_status, tc.RUNNING)
        self.assertEqual(p.run_ids, (42, 43))
        self.assertEqual(tc.effective_priority(p), 10)   # draft => bottom

    def test_from_json_plain_string_labels_and_unknown_status(self):
        p = PullRequest.from_json(
            {"number": 8, "labels": ["P1"], "runStatus": "bogus"})
        self.assertEqual(p.labels, ("P1",))
        self.assertEqual(p.run_status, tc.NONE)          # unknown -> none

    def test_load_snapshot_roundtrip(self):
        text = json.dumps({
            "self": 2,
            "self_run_id": 222,
            "prs": [
                {"number": 1, "labels": ["P2"], "runStatus": "running",
                 "runIds": [111]},
                {"number": 2, "labels": ["P5"], "runStatus": "running",
                 "runIds": [222]},
            ],
        })
        this_pr, all_prs, self_run_id = tc._load_snapshot(text)
        self.assertEqual(this_pr.number, 2)
        self.assertEqual(len(all_prs), 2)
        self.assertEqual(self_run_id, 222)


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