from __future__ import annotations

import json

import subprocess

import sys

import unittest

from pathlib import Path

from unittest import mock

sys.path.insert(0, str(Path(__file__).resolve().parents[1] / "scripts"))

import oak_field_map

class OakFieldMapTest(unittest.TestCase):

    def test_inspect_repo_is_read_only_and_summarizes_remote_open_branches(self) -> None:

        outputs = {

            ("oak", "status", "--json"): {

                "branch": "mrmrs-work",

                "head": "abcdef0123456789",

                "unmerged_commit_count": 0,

                "changes": [],

            },

            ("oak", "branch", "list", "--remote", "--status", "open", "--json"): [

                {"name": "main", "head": "1111111111119999"},

                {

                    "name": "mrmrs-open",

                    "head": "2222222222229999",

                    "description": "Clear PR description\n\nMore detail",

                },

            ],

            ("oak", "log", "-n", "8", "--json"): [

                {

                    "hash": "3333333333339999",

                    "branch": "main",

                    "description_or_subject": "Recent landing",

                }

            ],

        }

        calls: list[tuple[str, ...]] = []

        def fake_run(command, cwd, check, text, stdout, stderr):

            del cwd, check, text, stdout, stderr

            key = tuple(command)

            calls.append(key)

            return subprocess.CompletedProcess(command, 0, json.dumps(outputs[key]), "")

        with mock.patch("subprocess.run", side_effect=fake_run):

            field = oak_field_map.inspect_repo(Path("/tmp/worker/oak"))

        self.assertEqual(

            calls,

            [

                ("oak", "status", "--json"),

                ("oak", "branch", "list", "--remote", "--status", "open", "--json"),

                ("oak", "log", "-n", "8", "--json"),

            ],

        )

        self.assertEqual(field["remote_main"]["head"], "1111111111119999")

        self.assertEqual(field["remote_non_main_open_branches"][0]["name"], "mrmrs-open")

        rendered = oak_field_map.render_markdown([field])

        self.assertIn("Open non-main remote branches:", rendered)

        self.assertIn("`mrmrs-open` `222222222222` - Clear PR description", rendered)

        self.assertIn("`333333333333` `main` Recent landing", rendered)

    def test_main_returns_nonzero_when_a_probe_fails(self) -> None:

        def fake_run(command, cwd, check, text, stdout, stderr):

            del command, cwd, check, text, stdout, stderr

            return subprocess.CompletedProcess(["oak"], 7, "", "not an oak repo")

        with mock.patch("subprocess.run", side_effect=fake_run):

            with mock.patch("builtins.print"):

                self.assertEqual(oak_field_map.main(["--repo", "/tmp/nope"]), 1)

if __name__ == "__main__":

    unittest.main()