52 lines
No EOL
1.9 KiB
Python
52 lines
No EOL
1.9 KiB
Python
import pytest
|
|
import os
|
|
import subprocess
|
|
import json
|
|
|
|
# Path discovery - We are in external/dynamic-infra-tooling/tests
|
|
# Correct project root is 3 levels up
|
|
PROJECT_ROOT = os.path.abspath(os.path.join(os.path.dirname(__file__), "..", "..", ".."))
|
|
ANSIBLE_BIN = "ansible-playbook"
|
|
PLAYBOOK_PATH = os.path.join(PROJECT_ROOT, "ansible", "playbooks", "test-infra-plugin.yml")
|
|
|
|
def test_ansible_lookup_plugin_integration():
|
|
"""Verifies the Ansible plugin can be loaded and executed by Ansible"""
|
|
if not os.path.exists(PLAYBOOK_PATH):
|
|
pytest.skip(f"Playbook not found at {PLAYBOOK_PATH}")
|
|
|
|
env = os.environ.copy()
|
|
env["LC_ALL"] = "en_US.UTF-8"
|
|
env["LANG"] = "en_US.UTF-8"
|
|
# Ensure Ansible finds the plugin
|
|
env["ANSIBLE_LOOKUP_PLUGINS"] = os.path.join(PROJECT_ROOT, "ansible", "plugins", "lookup")
|
|
|
|
# Run the test playbook
|
|
cmd = [ANSIBLE_BIN, PLAYBOOK_PATH]
|
|
result = subprocess.run(cmd, capture_output=True, text=True, env=env, cwd=os.path.join(PROJECT_ROOT, "ansible"))
|
|
|
|
assert result.returncode == 0
|
|
assert "The next available IP is" in result.stdout
|
|
assert "The certificate for loopaware.com is loopaware.com.pem" in result.stdout
|
|
|
|
def test_opentofu_external_wrapper():
|
|
"""Verifies the OpenTofu wrapper returns valid JSON for the external data source"""
|
|
wrapper_script = os.path.join(PROJECT_ROOT, "scripts", "tofu-infra-query.sh")
|
|
|
|
if not os.path.exists(wrapper_script):
|
|
pytest.skip(f"Wrapper script not found at {wrapper_script}")
|
|
|
|
# Simulate Tofu sending JSON query to stdin
|
|
query = json.dumps({"command": "ip next-free"})
|
|
|
|
# Use shell execution to ensure the wrapper can find its relative paths correctly if needed
|
|
result = subprocess.run(
|
|
[wrapper_script],
|
|
input=query,
|
|
capture_output=True,
|
|
text=True
|
|
)
|
|
|
|
assert result.returncode == 0
|
|
data = json.loads(result.stdout)
|
|
assert "result" in data
|
|
assert "10.32." in data["result"] |