from __future__ import annotations

import os

import sys

import tempfile

import unittest

from pathlib import Path

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

from oakbench.environment import ENV_ISOLATION_VERSION, base_env  # noqa: E402

class EnvironmentPolicyTest(unittest.TestCase):

    def test_base_env_isolates_host_home_git_config_and_locale(self) -> None:

        env = base_env()

        bench_home = Path(env["HOME"])

        self.assertNotEqual(bench_home, Path.home())

        self.assertTrue(bench_home.is_dir())

        self.assertEqual(env["XDG_CONFIG_HOME"], str(bench_home / ".config"))

        self.assertEqual(env["XDG_CACHE_HOME"], str(bench_home / ".cache"))

        self.assertEqual(env["ZDOTDIR"], str(bench_home))

        self.assertEqual(env["BASH_ENV"], os.devnull)

        self.assertEqual(env["ENV"], os.devnull)

        self.assertEqual(env["GIT_CONFIG_GLOBAL"], os.devnull)

        self.assertEqual(env["GIT_CONFIG_NOSYSTEM"], "1")

        self.assertEqual(env["LC_ALL"], "C")

        self.assertEqual(env["LANG"], "C")

        self.assertEqual(env["TZ"], "UTC")

        self.assertEqual(env["OAK_BENCH_ENV_ISOLATION_VERSION"], ENV_ISOLATION_VERSION)

    def test_base_env_accepts_explicit_stable_home(self) -> None:

        with tempfile.TemporaryDirectory() as tmp:

            home = Path(tmp) / "bench-home"

            first = base_env(home=home)

            second = base_env(home=home)

        self.assertEqual(first["HOME"], second["HOME"])

        self.assertEqual(first["XDG_CONFIG_HOME"], str(home / ".config"))

    def test_base_env_preserves_existing_path_and_explicit_tokens(self) -> None:

        old_token = os.environ.get("OAK_API_KEY")

        try:

            os.environ["OAK_API_KEY"] = "test-token"

            env = base_env()

        finally:

            if old_token is None:

                os.environ.pop("OAK_API_KEY", None)

            else:

                os.environ["OAK_API_KEY"] = old_token

        self.assertEqual(env.get("PATH"), os.environ.get("PATH"))

        self.assertEqual(env["OAK_API_KEY"], "test-token")

if __name__ == "__main__":

    unittest.main()