mirror of
https://github.com/kevinveenbirkenbach/computer-playbook.git
synced 2025-07-18 14:34:24 +02:00
82 lines
3.0 KiB
Python
82 lines
3.0 KiB
Python
# tests/cli/test_fix/vars_main_files.py
|
|
import os
|
|
import shutil
|
|
import tempfile
|
|
import unittest
|
|
import yaml
|
|
|
|
# Adjust this import to match the real path in your project
|
|
from cli.fix.vars_main_files import run, ROLES_DIR
|
|
|
|
class TestEnsureVarsMain(unittest.TestCase):
|
|
def setUp(self):
|
|
# create a temporary directory to act as our roles dir
|
|
self.tmpdir = tempfile.mkdtemp()
|
|
self.roles_dir = os.path.join(self.tmpdir, "roles")
|
|
os.mkdir(self.roles_dir)
|
|
|
|
# Monkey-patch the module's ROLES_DIR to point here
|
|
self._orig_roles_dir = ROLES_DIR
|
|
setattr(__import__("cli.fix.vars_main_files", fromlist=["ROLES_DIR"]), "ROLES_DIR", self.roles_dir)
|
|
|
|
def tearDown(self):
|
|
# restore and cleanup
|
|
setattr(__import__("cli.fix.vars_main_files", fromlist=["ROLES_DIR"]), "ROLES_DIR", self._orig_roles_dir)
|
|
shutil.rmtree(self.tmpdir)
|
|
|
|
def _make_role(self, name, vars_content=None):
|
|
"""
|
|
Create a role under self.roles_dir/name
|
|
If vars_content is given, writes that to vars/main.yml
|
|
"""
|
|
role_path = os.path.join(self.roles_dir, name)
|
|
os.makedirs(os.path.join(role_path, "vars"))
|
|
if vars_content is not None:
|
|
with open(os.path.join(role_path, "vars", "main.yml"), "w") as f:
|
|
yaml.safe_dump(vars_content, f)
|
|
return role_path
|
|
|
|
def test_creates_missing_vars_main(self):
|
|
# Create a role with no vars/main.yml
|
|
role = self._make_role("desk-foobar")
|
|
# Ensure no file exists yet
|
|
self.assertFalse(os.path.exists(os.path.join(role, "vars", "main.yml")))
|
|
|
|
# Run with overwrite=False, preview=False
|
|
run(prefix="desk-", preview=False, overwrite=False)
|
|
|
|
# Now file must exist
|
|
vm = os.path.join(role, "vars", "main.yml")
|
|
self.assertTrue(os.path.exists(vm))
|
|
|
|
data = yaml.safe_load(open(vm))
|
|
# Expect application_id: 'foobar'
|
|
self.assertEqual(data.get("application_id"), "foobar")
|
|
|
|
def test_overwrite_updates_only_application_id(self):
|
|
# Create a role with an existing vars/main.yml
|
|
initial = {"application_id": "wrong", "foo": "bar"}
|
|
role = self._make_role("desk-baz", vars_content=initial.copy())
|
|
|
|
run(prefix="desk-", preview=False, overwrite=True)
|
|
|
|
path = os.path.join(role, "vars", "main.yml")
|
|
data = yaml.safe_load(open(path))
|
|
|
|
# application_id must be corrected...
|
|
self.assertEqual(data.get("application_id"), "baz")
|
|
# ...but other keys must survive
|
|
self.assertIn("foo", data)
|
|
self.assertEqual(data["foo"], "bar")
|
|
|
|
def test_preview_mode_does_not_write(self):
|
|
# Create a role directory but with no vars/main.yml
|
|
role = self._make_role("desk-preview")
|
|
vm = os.path.join(role, "vars", "main.yml")
|
|
# Run in preview => no file creation
|
|
run(prefix="desk-", preview=True, overwrite=False)
|
|
self.assertFalse(os.path.exists(vm))
|
|
|
|
if __name__ == "__main__":
|
|
unittest.main()
|