"""Tests for command classification and error-text pattern matching."""

from __future__ import annotations

import sys

from pathlib import Path

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

import unittest

from oakbench.classify import UNKNOWN_COMMAND_PATTERN, classify_command, unwrap_shell_command

class UnwrapTest(unittest.TestCase):

    def test_unwraps_login_shell(self) -> None:

        self.assertEqual(unwrap_shell_command("/bin/zsh -lc 'git status'"), "git status")

        self.assertEqual(unwrap_shell_command("/bin/zsh -c 'git status'"), "git status")

        self.assertEqual(unwrap_shell_command("bash -lc 'oak diff'"), "oak diff")

    def test_passthrough(self) -> None:

        self.assertEqual(unwrap_shell_command("git status"), "git status")

class ClassifyTest(unittest.TestCase):

    def test_vcs_detection(self) -> None:

        self.assertIn("vcs", classify_command("git status --short"))

        self.assertIn("vcs", classify_command("/bin/zsh -lc 'oak commit --no-verify'"))

        self.assertNotIn("vcs", classify_command("ls -la"))

    def test_test_detection(self) -> None:

        self.assertIn("test", classify_command("python3 -m unittest discover -s tests"))

        self.assertIn("test", classify_command("python3 validate_configs.py"))

    def test_search_and_read(self) -> None:

        self.assertIn("search", classify_command("rg -n pattern ."))

        self.assertIn("search", classify_command("grep -rnE pattern src"))

        self.assertIn("search", classify_command("/usr/bin/grep -rnE pattern src"))

        self.assertIn("read", classify_command("sed -n '1,120p' app/pricing.py"))

        self.assertIn("read", classify_command("/bin/cat app/pricing.py"))

    def test_help_detection(self) -> None:

        # Help/discovery calls signal the agent does not know the tool zero-shot.

        self.assertIn("help", classify_command("oak --help"))

        self.assertIn("help", classify_command("git help commit"))

        self.assertIn("help", classify_command("man git-status"))

        self.assertNotIn("help", classify_command("ls -lh"))

class UnknownCommandPatternTest(unittest.TestCase):

    def test_matches_real_error_shapes(self) -> None:

        samples = [

            "git: 'stats' is not a git command. See 'git --help'.",

            "error: unrecognized subcommand 'stage-all-now'",

            "error: unknown option '--porcelain'",

            "zsh: no such command: oka",

        ]

        for sample in samples:

            self.assertIsNotNone(UNKNOWN_COMMAND_PATTERN.search(sample), sample)

    def test_does_not_match_ordinary_failures(self) -> None:

        for sample in ("nothing to commit, working tree clean", "merge conflict in app/pricing.py"):

            self.assertIsNone(UNKNOWN_COMMAND_PATTERN.search(sample), sample)

if __name__ == "__main__":

    unittest.main()