34 lines
1.3 KiB
Python
34 lines
1.3 KiB
Python
# tests/test_tpl.py
|
|
import os, tempfile, unittest
|
|
from engine.lib.tpl import Tpl
|
|
from engine.lib.log import Log
|
|
|
|
class TestTpl(unittest.TestCase):
|
|
def setUp(self):
|
|
self.cfg = tempfile.mkdtemp()
|
|
self.tpl = Tpl(self.cfg, Log())
|
|
|
|
def _w(self, rel, body):
|
|
p = os.path.join(self.cfg, rel)
|
|
os.makedirs(os.path.dirname(p), exist_ok=True)
|
|
open(p, "w").write(body)
|
|
|
|
def test_render_substitutes_known_vars(self):
|
|
self._w("cron/crontab", "root=${APEX_TIER1} node=$APEX_NODE\n")
|
|
out = self.tpl.render("cron/crontab", {"APEX_TIER1": "/mnt/t1", "APEX_NODE": "daedalus"})
|
|
self.assertEqual(out, "root=/mnt/t1 node=daedalus\n")
|
|
|
|
def test_render_leaves_shell_command_substitution(self):
|
|
self._w("c", "x $(cat /f) ${APEX_NODE}\n")
|
|
out = self.tpl.render("c", {"APEX_NODE": "d"})
|
|
self.assertEqual(out, "x $(cat /f) d\n") # $(...) survives; ${APEX_NODE} replaced
|
|
|
|
def test_render_dir_returns_relpath_map(self):
|
|
self._w("fail2ban/jail.local", "a=${X}\n")
|
|
self._w("fail2ban/sub/extra.conf", "b=${Y}\n")
|
|
out = self.tpl.render_dir("fail2ban", {"X": "1", "Y": "2"})
|
|
self.assertEqual(out, {"jail.local": "a=1\n", "sub/extra.conf": "b=2\n"})
|
|
|
|
if __name__ == "__main__":
|
|
unittest.main()
|