tools/devshell/worktree/tests/test_build_dir.py

# Copyright 2026 The Fuchsia Authors. All rights reserved.
# Use of this source code is governed by a BSD-style license that can be
# found in the LICENSE file.

import os
import sys
import tempfile
import unittest
from pathlib import Path

# Add tools/devshell/worktree to sys.path
worktree_dir = os.path.dirname(os.path.dirname(os.path.abspath(__file__)))
sys.path.insert(0, worktree_dir)

from build_dir import BuildDir, BuildStatus


class TestBuildDir(unittest.TestCase):
    def setUp(self) -> None:
        self.temp_dir = tempfile.TemporaryDirectory()
        self.wt_root = Path(self.temp_dir.name)
        self.build_path = self.wt_root / "out" / "default"
        self.build_path.mkdir(parents=True, exist_ok=True)
        self.bd = BuildDir(self.build_path)

    def tearDown(self) -> None:
        self.temp_dir.cleanup()

    def test_backup_and_restore(self) -> None:
        args_gn = self.build_path / "args.gn"
        args_gn_ref = self.build_path / "args.gn.ref"

        args_gn.write_text('key = "value"\n')

        self.bd.backup_args()
        self.assertTrue(args_gn_ref.exists())
        self.assertEqual(args_gn_ref.read_text(), 'key = "value"\n')

        args_gn.write_text('key = "new_value"\n')

        self.bd.restore_args()
        self.assertFalse(args_gn_ref.exists())
        self.assertEqual(args_gn.read_text(), 'key = "value"\n')

    def test_get_status(self) -> None:
        self.assertEqual(self.bd.get_build_status(), BuildStatus.NOT_CONFIGURED)
        args_gn = self.build_path / "args.gn"
        args_gn.write_text("")
        self.assertEqual(self.bd.get_build_status(), BuildStatus.CONFIGURED)


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