Add test suite for confgen

app/tests/test_template.py

@@ -0,0 +1,137 @@
+"""Tests for template engine."""
+
+from pathlib import Path
+import tempfile
+
+from src.confgen.template import TemplateEngine
+
+
+class TestTemplateEngine:
+    """Tests for TemplateEngine."""
+
+    def setup_method(self):
+        """Set up test fixtures."""
+        self.engine = TemplateEngine()
+
+    def test_render_simple_template(self):
+        """Test rendering a simple template with variables."""
+        template = "app_name: {{APP_NAME}}\nport: {{PORT}}"
+        variables = {"APP_NAME": "myapp", "PORT": "8080"}
+
+        result = self.engine.render(template, variables)
+
+        assert "app_name: myapp" in result
+        assert "port: 8080" in result
+
+    def test_render_with_conditionals(self):
+        """Test rendering template with conditionals."""
+        template = """
+{% if DEBUG %}
+debug_mode: true
+{% endif %}
+app_name: {{APP_NAME}}
+"""
+        variables = {"DEBUG": True, "APP_NAME": "myapp"}
+
+        result = self.engine.render(template, variables)
+
+        assert "debug_mode: true" in result
+        assert "app_name: myapp" in result
+
+    def test_render_with_loops(self):
+        """Test rendering template with loops."""
+        template = """
+services:
+{% for service in SERVICES %}
+  - name: {{service.name}}
+    port: {{service.port}}
+{% endfor %}
+"""
+        variables = {
+            "SERVICES": [
+                {"name": "web", "port": "80"},
+                {"name": "api", "port": "8080"},
+            ]
+        }
+
+        result = self.engine.render(template, variables)
+
+        assert "- name: web" in result
+        assert "- name: api" in result
+        assert "port: 80" in result
+        assert "port: 8080" in result
+
+    def test_extract_variables(self):
+        """Test extracting variables from template."""
+        template = """
+app_name: {{APP_NAME}}
+database:
+  host: {{DB_HOST}}
+  port: {{DB_PORT}}
+"""
+        variables = self.engine.extract_variables(template)
+
+        assert "APP_NAME" in variables
+        assert "DB_HOST" in variables
+        assert "DB_PORT" in variables
+
+    def test_extract_variables_ignores_env_vars(self):
+        """Test that env variables are not extracted."""
+        template = """
+password: {{env.SECRET_PASSWORD}}
+api_key: {{vault.SECRET_API_KEY}}
+"""
+        variables = self.engine.extract_variables(template)
+
+        assert len(variables) == 0
+
+    def test_render_file(self):
+        """Test rendering a template file."""
+        with tempfile.NamedTemporaryFile(
+            mode="w", suffix=".j2", delete=False
+        ) as f:
+            f.write("app_name: {{APP_NAME}}\nport: {{PORT}}")
+            temp_path = f.name
+
+        try:
+            variables = {"APP_NAME": "testapp", "PORT": "3000"}
+            result = self.engine.render_file(temp_path, variables)
+
+            assert "app_name: testapp" in result
+            assert "port: 3000" in result
+        finally:
+            Path(temp_path).unlink()
+
+    def test_has_conditionals(self):
+        """Test detecting conditionals in template."""
+        template_with_if = "{% if DEBUG %}debug{% endif %}"
+        template_without_if = "app: {{APP_NAME}}"
+
+        assert self.engine.has_conditionals(template_with_if) is True
+        assert self.engine.has_conditionals(template_without_if) is False
+
+    def test_has_loops(self):
+        """Test detecting loops in template."""
+        template_with_for = "{% for item in ITEMS %}{{item}}{% endfor %}"
+        template_without_for = "app: {{APP_NAME}}"
+
+        assert self.engine.has_loops(template_with_for) is True
+        assert self.engine.has_loops(template_without_for) is False
+
+    def test_render_with_default_filters(self):
+        """Test rendering with Jinja2 default filters."""
+        template = "host: {{HOST | default('localhost')}}"
+        variables = {}
+
+        result = self.engine.render(template, variables)
+
+        assert "host: localhost" in result
+
+    def test_render_with_upper_filter(self):
+        """Test rendering with upper filter."""
+        template = "name: {{NAME | upper}}"
+        variables = {"NAME": "myapp"}
+
+        result = self.engine.render(template, variables)
+
+        assert "name: MYAPP" in result