mirror of
https://github.com/kevinveenbirkenbach/computer-playbook.git
synced 2025-07-20 15:21:10 +02:00
67 lines
2.2 KiB
Python
67 lines
2.2 KiB
Python
import os
|
|
import sys
|
|
import json
|
|
import tempfile
|
|
import shutil
|
|
import unittest
|
|
from unittest.mock import patch
|
|
|
|
# Import the script as a module (assumes the script is named tree.py)
|
|
SCRIPT_PATH = os.path.abspath(
|
|
os.path.join(os.path.dirname(__file__), "../../../../cli/build/tree.py")
|
|
)
|
|
|
|
class TestTreeShadowFolder(unittest.TestCase):
|
|
def setUp(self):
|
|
# Create temp roles dir and a dummy role
|
|
self.roles_dir = tempfile.mkdtemp()
|
|
self.role_name = "dummyrole"
|
|
self.role_path = os.path.join(self.roles_dir, self.role_name)
|
|
os.makedirs(os.path.join(self.role_path, "meta"))
|
|
|
|
# Prepare shadow dir
|
|
self.shadow_dir = tempfile.mkdtemp()
|
|
|
|
# Patch sys.argv for the script
|
|
self.orig_argv = sys.argv[:]
|
|
sys.argv = [
|
|
SCRIPT_PATH,
|
|
"-d", self.roles_dir,
|
|
"-s", self.shadow_dir,
|
|
"-o", "json"
|
|
]
|
|
|
|
def tearDown(self):
|
|
sys.argv = self.orig_argv
|
|
shutil.rmtree(self.roles_dir)
|
|
shutil.rmtree(self.shadow_dir)
|
|
|
|
@patch("cli.build.tree.build_mappings")
|
|
@patch("cli.build.tree.output_graph")
|
|
def test_tree_json_written_to_shadow_folder(self, mock_output_graph, mock_build_mappings):
|
|
# Prepare dummy graph
|
|
dummy_graph = {"dummy": {"test": 42}}
|
|
mock_build_mappings.return_value = dummy_graph
|
|
|
|
# Run the script (as __main__)
|
|
import runpy
|
|
runpy.run_path(SCRIPT_PATH, run_name="__main__")
|
|
|
|
# Check file in shadow folder
|
|
expected_tree_path = os.path.join(
|
|
self.shadow_dir, self.role_name, "meta", "tree.json"
|
|
)
|
|
self.assertTrue(os.path.isfile(expected_tree_path), "tree.json not found in shadow folder")
|
|
|
|
# Check contents
|
|
with open(expected_tree_path) as f:
|
|
data = json.load(f)
|
|
self.assertEqual(data, dummy_graph, "tree.json content mismatch")
|
|
|
|
# Ensure nothing was written to original meta/
|
|
original_tree_path = os.path.join(self.role_path, "meta", "tree.json")
|
|
self.assertFalse(os.path.isfile(original_tree_path), "tree.json should NOT be in role's meta/")
|
|
|
|
if __name__ == "__main__":
|
|
unittest.main()
|