Files
FabledSteward/tests/test_playbook_variables.py
T
bvandeusen e5f6a11f94
CI / lint (push) Successful in 3s
CI / unit (push) Successful in 8s
CI / integration (push) Successful in 2m15s
CI / publish (push) Successful in 52s
feat(ansible): playbooks self-describe via "# description:" comment
Playbooks can ship a human description Steward reads and shows when one is
selected. Convention: a `# description: <text>` magic comment (Ansible rejects
unknown play keys, so a comment is the portable place — works for third-party
playbooks too); falls back to the first play's name:. sources
.discover_playbook_description().

Surfaced at the top of the shared _playbook_vars.html partial, which loads on
playbook selection in the host run form, schedules form, and browse run form.
All four bundled playbooks (provision/install/update/docker_prune) now carry a
description line. Unit tests added.

Scribe #900.

Co-Authored-By: Claude Opus 4.8 (1M context) <noreply@anthropic.com>
2026-06-17 11:25:25 -04:00

140 lines
4.1 KiB
Python

"""Unit tests for playbook variable discovery + run-time extra-vars file."""
import json
from steward.ansible.executor import build_extra_vars_file
from steward.ansible.sources import (
discover_playbook_description,
discover_playbook_variables,
)
def _names(variables):
return [v["name"] for v in variables]
def test_vars_block_defaults_surface_as_fields():
content = """
- hosts: all
vars:
steward_url: ""
agent_interval: 30
enabled: true
tasks: []
"""
variables = discover_playbook_variables(content)
assert _names(variables) == ["steward_url", "agent_interval", "enabled"]
interval = next(v for v in variables if v["name"] == "agent_interval")
assert interval["default"] == 30
assert interval["required"] is False
assert interval["secret"] is False
def test_secretish_names_flagged():
content = """
- hosts: all
vars:
db_password: ""
api_token: ""
vault_secret: ""
plain_value: "x"
"""
by = {v["name"]: v for v in discover_playbook_variables(content)}
assert by["db_password"]["secret"] is True
assert by["api_token"]["secret"] is True
assert by["vault_secret"]["secret"] is True
assert by["plain_value"]["secret"] is False
def test_vars_prompt_required_and_private():
content = """
- hosts: all
vars_prompt:
- name: release_tag
prompt: "Which release?"
- name: admin_pw
prompt: "Admin password"
private: true
default: ""
tasks: []
"""
by = {v["name"]: v for v in discover_playbook_variables(content)}
# No default → required; vars_prompt defaults to private=yes → secret.
assert by["release_tag"]["required"] is True
assert by["release_tag"]["secret"] is True
assert by["release_tag"]["prompt"] == "Which release?"
# Explicit private + has default → secret but not required.
assert by["admin_pw"]["secret"] is True
assert by["admin_pw"]["required"] is False
def test_vars_prompt_wins_on_name_collision():
content = """
- hosts: all
vars_prompt:
- name: dup
prompt: "from prompt"
private: false
vars:
dup: "from vars"
"""
variables = discover_playbook_variables(content)
assert _names(variables) == ["dup"]
assert variables[0]["prompt"] == "from prompt"
def test_non_scalar_vars_skipped():
content = """
- hosts: all
vars:
scalar: 1
a_list: [1, 2]
a_map: {k: v}
"""
assert _names(discover_playbook_variables(content)) == ["scalar"]
def test_malformed_yaml_returns_empty():
assert discover_playbook_variables("this: : : not valid") == []
assert discover_playbook_variables("just a string") == []
def test_extra_vars_file_roundtrip_and_space_safe():
merged = {"agent_interval": "30", "msg": "hello world", "q": 'a "quote"'}
args, files = build_extra_vars_file(merged, "/td")
assert args == ["-e", "@/td/extravars.json"]
path, content = files[0]
assert path == "/td/extravars.json"
# JSON keeps spaces/quotes intact — no shlex hazard like -e key=value.
assert json.loads(content) == merged
def test_extra_vars_file_empty_is_noop():
assert build_extra_vars_file({}, "/td") == ([], [])
assert build_extra_vars_file(None, "/td") == ([], [])
# ── Playbook description discovery ────────────────────────────────────────────
def test_description_from_magic_comment():
content = """---
# description: Reclaim disk by pruning Docker.
- hosts: all
name: Docker prune
tasks: []
"""
assert discover_playbook_description(content) == "Reclaim disk by pruning Docker."
def test_description_case_insensitive_key():
assert discover_playbook_description("# DESCRIPTION: trimmed \n- hosts: all") == "trimmed"
def test_description_falls_back_to_play_name():
content = "- hosts: all\n name: Configure web servers\n tasks: []\n"
assert discover_playbook_description(content) == "Configure web servers"
def test_description_empty_when_neither_present():
assert discover_playbook_description("- hosts: all\n tasks: []\n") == ""
assert discover_playbook_description("not: : valid") == ""