mirror of
https://github.com/kevinveenbirkenbach/computer-playbook.git
synced 2026-11-08 11:46:35 +00:00
120 lines
4.6 KiB
Python
120 lines
4.6 KiB
Python
import importlib.util
|
|
import unittest
|
|
from pathlib import Path
|
|
from unittest.mock import patch
|
|
|
|
from ansible.errors import AnsibleError
|
|
|
|
|
|
def _repo_root() -> Path:
|
|
return Path(__file__).resolve().parents[4]
|
|
|
|
|
|
def _load_module(rel_path: str, name: str):
|
|
path = _repo_root() / rel_path
|
|
spec = importlib.util.spec_from_file_location(name, str(path))
|
|
mod = importlib.util.module_from_spec(spec)
|
|
assert spec and spec.loader
|
|
spec.loader.exec_module(mod)
|
|
return mod
|
|
|
|
|
|
class TestLookupDocker(unittest.TestCase):
|
|
@classmethod
|
|
def setUpClass(cls):
|
|
cls.mod = _load_module("plugins/lookup/container.py", "lookup_plugins_docker")
|
|
cls.LookupModule = cls.mod.LookupModule
|
|
|
|
def setUp(self):
|
|
self.lookup = self.LookupModule()
|
|
self.vars_ok = {"DIR_COMPOSITIONS": "/opt/compose/"}
|
|
|
|
self.fake_paths = {
|
|
"directories": {
|
|
"instance": "/opt/compose/app/",
|
|
"env": "/opt/compose/app/.env/",
|
|
"services": "/opt/compose/app/services/",
|
|
"volumes": "/opt/compose/app/volumes/",
|
|
"config": "/opt/compose/app/config/",
|
|
},
|
|
"files": {
|
|
"env": "/opt/compose/app/.env/app.env",
|
|
"compose": "/opt/compose/app/compose.yml",
|
|
"compose_override": "/opt/compose/app/compose.override.yml",
|
|
"dockerfile": "/opt/compose/app/Dockerfile",
|
|
},
|
|
}
|
|
|
|
def test_returns_full_dict_when_only_application_id(self):
|
|
with patch.object(
|
|
self.mod, "get_docker_paths", return_value=self.fake_paths
|
|
) as p:
|
|
out = self.lookup.run(["web-app-mailu"], variables=self.vars_ok)
|
|
self.assertEqual(out, [self.fake_paths])
|
|
p.assert_called_once_with("web-app-mailu", "/opt/compose/")
|
|
|
|
def test_returns_subdict_for_top_level_key(self):
|
|
with patch.object(self.mod, "get_docker_paths", return_value=self.fake_paths):
|
|
out = self.lookup.run(
|
|
["web-app-mailu", "directories"], variables=self.vars_ok
|
|
)
|
|
self.assertEqual(out, [self.fake_paths["directories"]])
|
|
|
|
out = self.lookup.run(["web-app-mailu", "files"], variables=self.vars_ok)
|
|
self.assertEqual(out, [self.fake_paths["files"]])
|
|
|
|
def test_returns_deep_value_for_dotted_path(self):
|
|
with patch.object(self.mod, "get_docker_paths", return_value=self.fake_paths):
|
|
out = self.lookup.run(
|
|
["web-app-mailu", "directories.instance"], variables=self.vars_ok
|
|
)
|
|
self.assertEqual(out, ["/opt/compose/app/"])
|
|
|
|
out = self.lookup.run(
|
|
["web-app-mailu", "files.env"], variables=self.vars_ok
|
|
)
|
|
self.assertEqual(out, ["/opt/compose/app/.env/app.env"])
|
|
|
|
def test_invalid_path_raises(self):
|
|
with patch.object(self.mod, "get_docker_paths", return_value=self.fake_paths):
|
|
with self.assertRaises(AnsibleError):
|
|
self.lookup.run(
|
|
["web-app-mailu", "doesnotexist"], variables=self.vars_ok
|
|
)
|
|
|
|
with self.assertRaises(AnsibleError):
|
|
self.lookup.run(
|
|
["web-app-mailu", "directories.doesnotexist"],
|
|
variables=self.vars_ok,
|
|
)
|
|
|
|
with self.assertRaises(AnsibleError):
|
|
self.lookup.run(
|
|
["web-app-mailu", "directories.instance.foo"],
|
|
variables=self.vars_ok,
|
|
)
|
|
|
|
def test_missing_instances_base_var_raises(self):
|
|
with patch.object(self.mod, "get_docker_paths", return_value=self.fake_paths):
|
|
with self.assertRaises(AnsibleError) as ctx:
|
|
self.lookup.run(["web-app-mailu"], variables={})
|
|
self.assertIn("DIR_COMPOSITIONS not set", str(ctx.exception))
|
|
|
|
def test_empty_application_id_raises(self):
|
|
with patch.object(self.mod, "get_docker_paths", return_value=self.fake_paths):
|
|
with self.assertRaises(AnsibleError) as ctx:
|
|
self.lookup.run([" "], variables=self.vars_ok)
|
|
self.assertIn("application_id is empty", str(ctx.exception))
|
|
|
|
def test_wrong_arity_raises(self):
|
|
with patch.object(self.mod, "get_docker_paths", return_value=self.fake_paths):
|
|
with self.assertRaises(AnsibleError):
|
|
self.lookup.run([], variables=self.vars_ok)
|
|
|
|
with self.assertRaises(AnsibleError):
|
|
self.lookup.run(["a", "b", "c"], variables=self.vars_ok)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
unittest.main()
|