"""Tests for AI playbook validation and repair."""
from unittest.mock import patch
from souschef.converters import playbook as playbook_converter
def test_validate_and_fix_playbook_repairs_invalid_yaml() -> None:
"""Ensure invalid YAML triggers AI repair and returns valid YAML."""
invalid_playbook = """---
- name: Bad playbook
hosts: all
tasks:
- name: Bad task
when: ansible.builtin.stat:
"""
fixed_playbook = """---
- name: Repaired playbook
hosts: all
tasks:
- name: Check file
ansible.builtin.stat:
path: /tmp/file
register: file_stat
- name: Do work
ansible.builtin.command:
cmd: echo ok
when: file_stat.stat.exists
"""
with (
patch.object(
playbook_converter,
"_call_ai_api",
return_value=fixed_playbook,
) as mock_call,
patch.object(playbook_converter, "_run_ansible_lint", return_value=None),
):
result = playbook_converter._validate_and_fix_playbook(
invalid_playbook,
client=object(),
ai_provider="openai",
model="gpt-test",
temperature=0.2,
max_tokens=200,
)
assert result == fixed_playbook.strip()
assert mock_call.called
