| 123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163 |
- """Focused regression tests for module base commands."""
- from __future__ import annotations
- from types import SimpleNamespace
- from cli.core.module.base_commands import GenerationConfig, apply_output_name, generate_template, list_templates
- def _noop(*_args, **_kwargs) -> None:
- return None
- def _raise_destination_prompt(slug: str):
- raise AssertionError(f"prompt_generation_destination called for {slug}")
- def _raise_output_check(*_args, **_kwargs):
- raise AssertionError("check_output_directory should not run")
- class _DisplayCapture:
- def __init__(self) -> None:
- self.lines: list[str] = []
- self.templates = SimpleNamespace(
- render_template_header=_noop,
- render_file_tree=_noop,
- )
- self.variables = SimpleNamespace(render_variables_table=_noop)
- def text(self, value: str, style: str | None = None) -> None:
- del style
- self.lines.append(value)
- def success(self, value: str, *args, **kwargs) -> None:
- del args, kwargs
- self.lines.append(value)
- def warning(self, value: str, *args, **kwargs) -> None:
- del args, kwargs
- self.lines.append(value)
- def error(self, value: str, *args, **kwargs) -> None:
- del args, kwargs
- self.lines.append(value)
- def data_table(self, *args, **kwargs) -> None:
- del args, kwargs
- raise AssertionError("data_table should not be used for raw output")
- def info(self, *args, **kwargs) -> None:
- del args, kwargs
- raise AssertionError("info should not be used when templates exist")
- def test_list_templates_raw_outputs_tab_separated_rows() -> None:
- """Raw listing should emit one tab-separated row per template."""
- template = SimpleNamespace(
- id="whoami",
- metadata=SimpleNamespace(
- name="Whoami",
- tags=["docker", "test"],
- version=SimpleNamespace(name="1.0.0"),
- library="default",
- library_type="git",
- ),
- )
- display = _DisplayCapture()
- module_instance = SimpleNamespace(
- name="compose",
- display=display,
- _load_all_templates=lambda: [template],
- )
- returned_templates = list_templates(module_instance, raw=True)
- assert returned_templates == [template]
- assert display.lines == ["whoami\tWhoami\tdocker,test\t1.0.0\tdefault"]
- def test_apply_output_name_renames_top_level_paths_only() -> None:
- """Named generation should rename top-level outputs while preserving nested names."""
- rendered_files = {
- "files/test.txt": "nested",
- "main.tf": "main",
- "dns.tf": "dns",
- }
- assert apply_output_name(rendered_files, "servertest1") == {
- "servertest1_files/test.txt": "nested",
- "servertest1.tf": "main",
- "servertest1_dns.tf": "dns",
- }
- def test_generate_template_applies_output_name_before_writing(monkeypatch, tmp_path) -> None:
- """Generate should write renamed paths when --name is provided."""
- display = _DisplayCapture()
- template = SimpleNamespace(id="terraform", slug="terraform")
- module_instance = SimpleNamespace(name="terraform", display=display)
- written: dict[str, object] = {}
- monkeypatch.setattr("cli.core.module.base_commands._prepare_template", lambda *_args, **_kwargs: template)
- monkeypatch.setattr(
- "cli.core.module.base_commands._render_template",
- lambda *_args, **_kwargs: ({"files/test.txt": "nested", "main.tf": "main", "dns.tf": "dns"}, {}),
- )
- monkeypatch.setattr("cli.core.module.base_commands.check_output_directory", lambda *_args, **_kwargs: [])
- def capture_write(output_dir, rendered_files):
- written["output_dir"] = output_dir
- written["rendered_files"] = rendered_files
- monkeypatch.setattr("cli.core.module.base_commands.write_rendered_files", capture_write)
- generate_template(
- module_instance,
- GenerationConfig(
- id="terraform",
- output=str(tmp_path),
- interactive=False,
- name="servertest1",
- ),
- )
- assert written["output_dir"] == tmp_path
- assert written["rendered_files"] == {
- "servertest1_files/test.txt": "nested",
- "servertest1.tf": "main",
- "servertest1_dns.tf": "dns",
- }
- def test_generate_template_dry_run_skips_destination_prompt_and_overwrite_check(
- monkeypatch,
- ) -> None:
- """Dry runs without explicit destinations should not ask where to write or confirm overwrites."""
- display = _DisplayCapture()
- template = SimpleNamespace(id="whoami", slug="whoami")
- module_instance = SimpleNamespace(name="compose", display=display)
- monkeypatch.setattr("cli.core.module.base_commands._prepare_template", lambda *_args, **_kwargs: template)
- monkeypatch.setattr(
- "cli.core.module.base_commands._render_template",
- lambda *_args, **_kwargs: ({"compose.yaml": "services:\n"}, {}),
- )
- monkeypatch.setattr("cli.core.module.base_commands.prompt_generation_destination", _raise_destination_prompt)
- monkeypatch.setattr(
- "cli.core.module.base_commands.check_output_directory",
- _raise_output_check,
- )
- generate_template(
- module_instance,
- GenerationConfig(
- id="whoami",
- interactive=True,
- dry_run=True,
- ),
- )
- assert any("boilerplate rendered successfully" in line for line in display.lines)
- assert any("preview only" in line for line in display.lines)
|