mirror of
https://github.com/kevinveenbirkenbach/computer-playbook.git
synced 2026-11-09 12:17:13 +00:00
kevinveenbirkenbachandClaude Sonnet 4.6
3cfdc4bdf7
Adds inventory consistency guard that fails fast when SERVICES_DISABLED conflicts with an already-deployed inventory, and ensures the env var is forwarded through the dedicated runner, development exec command, and local test scripts. Co-Authored-By: Claude Sonnet 4.6 <noreply@anthropic.com>
61 lines
1.8 KiB
Python
61 lines
1.8 KiB
Python
from __future__ import annotations
|
|
|
|
import argparse
|
|
import os
|
|
import unittest
|
|
import unittest.mock
|
|
|
|
from cli.deploy.development import exec as exec_cmd
|
|
|
|
|
|
class TestDevelopmentExec(unittest.TestCase):
|
|
def test_handler_forwards_services_disabled(self):
|
|
compose = unittest.mock.Mock()
|
|
compose.exec.return_value.returncode = 0
|
|
|
|
args = argparse.Namespace(distro="debian", cmd=["--", "sh", "-lc", "true"])
|
|
|
|
with unittest.mock.patch(
|
|
"cli.deploy.development.exec.make_compose",
|
|
return_value=compose,
|
|
):
|
|
with unittest.mock.patch.dict(
|
|
os.environ, {"SERVICES_DISABLED": "email"}, clear=False
|
|
):
|
|
rc = exec_cmd.handler(args)
|
|
|
|
self.assertEqual(rc, 0)
|
|
compose.exec.assert_called_once_with(
|
|
["sh", "-lc", "true"],
|
|
check=False,
|
|
extra_env={"SERVICES_DISABLED": "email"},
|
|
)
|
|
|
|
def test_handler_omits_extra_env_when_services_disabled_unset(self):
|
|
compose = unittest.mock.Mock()
|
|
compose.exec.return_value.returncode = 0
|
|
|
|
args = argparse.Namespace(distro="debian", cmd=["echo", "hi"])
|
|
|
|
env = {k: v for k, v in os.environ.items() if k != "SERVICES_DISABLED"}
|
|
with unittest.mock.patch(
|
|
"cli.deploy.development.exec.make_compose",
|
|
return_value=compose,
|
|
):
|
|
with unittest.mock.patch.dict(os.environ, env, clear=True):
|
|
rc = exec_cmd.handler(args)
|
|
|
|
self.assertEqual(rc, 0)
|
|
compose.exec.assert_called_once_with(
|
|
["echo", "hi"], check=False, extra_env=None
|
|
)
|
|
|
|
def test_handler_requires_command(self):
|
|
args = argparse.Namespace(distro="debian", cmd=[])
|
|
with self.assertRaises(SystemExit):
|
|
exec_cmd.handler(args)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
unittest.main()
|