# -*- coding: utf-8 -*-

from ..syntax import macros, test, test_raises, the  # noqa: F401
from ..test.fixtures import session, testset

import os

from ..environ import override

def runtests():
    with testset("environ.override"):
        # Basic override and restore
        os.environ["_UNPYTHONIC_TEST_VAR"] = "original"
        with override(_UNPYTHONIC_TEST_VAR="overridden"):
            test[the[os.environ["_UNPYTHONIC_TEST_VAR"]] == "overridden"]
        test[the[os.environ["_UNPYTHONIC_TEST_VAR"]] == "original"]
        del os.environ["_UNPYTHONIC_TEST_VAR"]

        # Adding a variable that didn't exist before
        key = "_UNPYTHONIC_TEST_NEW"
        if key in os.environ:
            del os.environ[key]
        with override(**{key: "added"}):
            test[the[os.environ[key]] == "added"]
        test[key not in os.environ]

        # Multiple overrides at once
        os.environ["_UNPYTHONIC_TEST_A"] = "a_orig"
        os.environ["_UNPYTHONIC_TEST_B"] = "b_orig"
        with override(_UNPYTHONIC_TEST_A="a_new", _UNPYTHONIC_TEST_B="b_new"):
            test[the[os.environ["_UNPYTHONIC_TEST_A"]] == "a_new"]
            test[the[os.environ["_UNPYTHONIC_TEST_B"]] == "b_new"]
        test[the[os.environ["_UNPYTHONIC_TEST_A"]] == "a_orig"]
        test[the[os.environ["_UNPYTHONIC_TEST_B"]] == "b_orig"]
        del os.environ["_UNPYTHONIC_TEST_A"]
        del os.environ["_UNPYTHONIC_TEST_B"]

        # Nested overrides (same-thread; RLock allows this)
        os.environ["_UNPYTHONIC_TEST_VAR"] = "level0"
        with override(_UNPYTHONIC_TEST_VAR="level1"):
            test[the[os.environ["_UNPYTHONIC_TEST_VAR"]] == "level1"]
            with override(_UNPYTHONIC_TEST_VAR="level2"):
                test[the[os.environ["_UNPYTHONIC_TEST_VAR"]] == "level2"]
            test[the[os.environ["_UNPYTHONIC_TEST_VAR"]] == "level1"]
        test[the[os.environ["_UNPYTHONIC_TEST_VAR"]] == "level0"]
        del os.environ["_UNPYTHONIC_TEST_VAR"]

        # Restore on exception
        os.environ["_UNPYTHONIC_TEST_VAR"] = "before"
        try:
            with override(_UNPYTHONIC_TEST_VAR="during"):
                test[the[os.environ["_UNPYTHONIC_TEST_VAR"]] == "during"]
                raise RuntimeError("boom")
        except RuntimeError:
            pass
        test[the[os.environ["_UNPYTHONIC_TEST_VAR"]] == "before"]
        del os.environ["_UNPYTHONIC_TEST_VAR"]

if __name__ == '__main__':  # pragma: no cover
    with session(__file__):
        runtests()