# Copyright 2025 The Chromium Authors
# Use of this source code is governed by a BSD-style license that can be
# found in the LICENSE file.

import pathlib
import unittest
import textwrap

from compiler import _maybe_cache
from compiler import Compiler
from modularize import SOURCE_ROOT
from platforms import Cpu
from platforms import Os


class TestableCompiler(Compiler):
    n = 1

    @_maybe_cache
    def cached_n(self):
        return self.n


class CompilerTest(unittest.TestCase):
    def setUp(self):
        super().setUp()
        self.compiler1 = TestableCompiler(
            gn_out=pathlib.Path('/tmp/compiler1'),
            source_root=SOURCE_ROOT,
            error_dir=None,
            use_cache=True,
            os=Os.Linux,
            cpu=Cpu.x64,
        )
        self.compiler1_uncached = TestableCompiler(
            gn_out=pathlib.Path('/tmp/compiler1'),
            source_root=SOURCE_ROOT,
            error_dir=None,
            use_cache=False,
            os=Os.Linux,
            cpu=Cpu.x64,
        )
        self.compiler2 = TestableCompiler(
            gn_out=pathlib.Path('/tmp/compiler2'),
            source_root=SOURCE_ROOT,
            error_dir=None,
            use_cache=True,
            os=Os.Linux,
            cpu=Cpu.x64,
        )

    def test_maybe_cache(self):
        # Uncached compilers should write to the cache, but not read from it.
        self.compiler1.n = 2
        self.assertEqual(self.compiler1_uncached.cached_n(), 1)
        self.assertEqual(self.compiler1.cached_n(), 1)
        self.compiler1_uncached.n = 3
        self.assertEqual(self.compiler1_uncached.cached_n(), 3)
        self.assertEqual(self.compiler1.cached_n(), 3)

        # This one should be unrelated since it has a different gn_out dir.
        self.compiler2.n = 4
        self.assertEqual(self.compiler2.cached_n(), 4)

    def test_parse_depfile(self):
        self.assertListEqual(
            self.compiler1._parse_depfile(
                textwrap.dedent("""\
          /dev/null: /tmp/main.cc \\
            /path/to/foo.txt \\
            ../up.h \\
            path/to/relative \\
            /path/to/absolute.hpp \\
            path\\ with\\ spaces path2.h \\
          """)
            ),
            [
                pathlib.Path('/tmp/up.h'),
                pathlib.Path('/tmp/compiler1/path/to/relative'),
                pathlib.Path('/path/to/absolute.hpp'),
                pathlib.Path('/tmp/compiler1/path with spaces'),
                pathlib.Path('/tmp/compiler1/path2.h'),
            ],
        )


if __name__ == '__main__':
    unittest.main()