106 lines
4.3 KiB
Python
106 lines
4.3 KiB
Python
import importlib.util
|
|
import pathlib
|
|
import types
|
|
import pytest
|
|
|
|
_PATH = pathlib.Path(__file__).resolve().parent.parent / "scripts" / "integration-vm.py"
|
|
_spec = importlib.util.spec_from_file_location("integration_vm", _PATH)
|
|
ivm = importlib.util.module_from_spec(_spec)
|
|
_spec.loader.exec_module(ivm)
|
|
|
|
|
|
def test_valid_tiers():
|
|
assert ivm.VALID_TIERS == ("internal", "le-staging", "le-prod-wildcard")
|
|
|
|
|
|
def test_vm_name_prefix_and_suffix():
|
|
assert ivm.vm_name("askari", "ab12cd34") == "boma-it-askari-ab12cd34"
|
|
|
|
def test_vm_name_generates_suffix():
|
|
n = ivm.vm_name("askari")
|
|
assert n.startswith("boma-it-askari-") and len(n.split("-")[-1]) == 8
|
|
|
|
def test_free_mib_parses_memavailable():
|
|
sample = "MemTotal: 16331156 kB\nMemAvailable: 8388608 kB\n"
|
|
assert ivm.free_mib(sample) == 8192
|
|
|
|
def test_parse_lease_ip_extracts_ipv4():
|
|
out = (" Name MAC address Protocol Address\n"
|
|
"-------------------------------------------------------------------\n"
|
|
" vnet0 52:54:00:aa:bb:cc ipv4 192.168.150.42/24\n")
|
|
assert ivm.parse_lease_ip(out) == "192.168.150.42"
|
|
|
|
def test_parse_lease_ip_none_when_absent():
|
|
assert ivm.parse_lease_ip("no leases\n") is None
|
|
|
|
def test_parse_lease_ip_format_is_source_agnostic():
|
|
# virsh domifaddr --source arp output format is identical to --source lease;
|
|
# this test only proves the regex is format-agnostic (both sources produce the
|
|
# same table). The behavioral arp-fallback in wait_for_ip is covered by
|
|
# test_wait_for_ip_falls_back_to_arp below.
|
|
out = (" Name MAC address Protocol Address\n"
|
|
"-------------------------------------------------------------------\n"
|
|
" vnet0 52:54:00:de:ad:be ipv4 192.168.150.73/24\n")
|
|
assert ivm.parse_lease_ip(out) == "192.168.150.73"
|
|
|
|
|
|
def test_wait_for_ip_falls_back_to_arp(monkeypatch):
|
|
# wait_for_ip polls virsh domifaddr with --source lease first, then --source arp.
|
|
# Simulate lease returning empty (no DHCP lease yet) and arp returning a real address.
|
|
arp_out = (" Name MAC address Protocol Address\n"
|
|
"-------------------------------------------------------------------\n"
|
|
" vnet0 52:54:00:aa:bb:cc ipv4 192.168.150.142/24\n")
|
|
|
|
def fake_sh(cmd, **kwargs):
|
|
if "arp" in cmd:
|
|
return types.SimpleNamespace(stdout=arp_out)
|
|
return types.SimpleNamespace(stdout="")
|
|
|
|
monkeypatch.setattr(ivm, "sh", fake_sh)
|
|
monkeypatch.setattr(ivm.time, "sleep", lambda _: None)
|
|
assert ivm.wait_for_ip("dummy") == "192.168.150.142"
|
|
|
|
|
|
def test_meta_data_has_instance_and_hostname():
|
|
md = ivm.render_meta_data("iid-askari-x", "boma-it-askari-x")
|
|
assert "instance-id: iid-askari-x" in md
|
|
assert "local-hostname: boma-it-askari-x" in md
|
|
|
|
def test_user_data_injects_key_and_ansible_user():
|
|
ud = ivm.render_user_data("ssh-ed25519 AAAA... claude@ubongo", "ansible")
|
|
assert ud.startswith("#cloud-config")
|
|
assert "name: ansible" in ud
|
|
assert "ssh-ed25519 AAAA... claude@ubongo" in ud
|
|
assert "NOPASSWD:ALL" in ud
|
|
|
|
|
|
def test_cert_file_valid_tier():
|
|
p = ivm.cert_file("le-staging")
|
|
assert p.name == "le-staging.yml" and p.parent.name == "certs"
|
|
|
|
def test_cert_file_rejects_bad_tier():
|
|
with pytest.raises(ValueError):
|
|
ivm.cert_file("bogus")
|
|
|
|
def test_render_run_hosts_single_host_in_groups():
|
|
out = ivm.render_run_hosts("boma-it-askari-x", "192.168.150.42",
|
|
"ansible", ["offsite_hosts"])
|
|
assert "offsite_hosts:" in out
|
|
assert "boma-it-askari-x:" in out
|
|
assert "ansible_host: 192.168.150.42" in out
|
|
assert "ansible_user: ansible" in out
|
|
assert "askari:" not in out.replace("boma-it-askari-x:", "")
|
|
|
|
def test_free_mib_returns_zero_when_absent():
|
|
assert ivm.free_mib("MemTotal: 16384 kB\n") == 0
|
|
|
|
def test_render_run_hosts_multiple_groups():
|
|
out = ivm.render_run_hosts("boma-it-x-1", "192.168.150.5", "ansible",
|
|
["offsite_hosts", "docker_hosts"])
|
|
assert "offsite_hosts:" in out
|
|
assert "docker_hosts:" in out
|
|
|
|
def test_render_run_hosts_dedups_groups():
|
|
out = ivm.render_run_hosts("boma-it-x-1", "192.168.150.5", "ansible",
|
|
["docker_hosts", "docker_hosts"])
|
|
assert out.count("docker_hosts:") == 1
|