Restructured CLI logic

tests/unit/cli/fix/test_vars_main_files.py

@@ -0,0 +1,81 @@
+# 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.ensure.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.ensure.vars_main_files", fromlist=["ROLES_DIR"]), "ROLES_DIR", self.roles_dir)
+
+    def tearDown(self):
+        # restore and cleanup
+        setattr(__import__("cli.ensure.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()