from __future__ import annotations

import sys

import tempfile

import unittest

from pathlib import Path

from unittest.mock import patch

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

from oakbench import configio

class TomlFallbackTest(unittest.TestCase):

    def test_fallback_parser_preserves_hash_inside_quoted_strings(self) -> None:

        with tempfile.TemporaryDirectory() as tmp:

            path = Path(tmp) / "subjects.toml"

            path.write_text(

                """

[subjects.oak]

label = "Oak # release"

enabled = true # ordinary comment

args = ["one#two", "three"] # ordinary comment

"""

            )

            with patch.object(configio, "tomllib", None):

                data = configio.load_toml(path)

        self.assertEqual(data["subjects"]["oak"]["label"], "Oak # release")

        self.assertEqual(data["subjects"]["oak"]["args"], ["one#two", "three"])

        self.assertTrue(data["subjects"]["oak"]["enabled"])

    def test_fallback_parser_treats_backslash_as_literal_in_single_quoted_strings(self) -> None:

        with tempfile.TemporaryDirectory() as tmp:

            path = Path(tmp) / "subjects.toml"

            path.write_text(

                """

[subjects.oak]

label = 'Oak \\# release' # ordinary comment

"""

            )

            with patch.object(configio, "tomllib", None):

                data = configio.load_toml(path)

        self.assertEqual(data["subjects"]["oak"]["label"], "Oak \\# release")

if __name__ == "__main__":

    unittest.main()