Files
napalm-vmware/tests/test_provision_placement.py
T
Christian Manivong 2b84ceae3a feat: VM provisioning, and reboot_host on ESXi
create_vm_from_cloud_init takes the same qcow2/raw cloud images netOrk
offers for Proxmox. ESXi can neither boot nor download them, so the
driver does both: download with checksum check and one retry, convert
with qemu-img to a streamOptimized VMDK (cached by URL), import through
a minimal OVF descriptor over NFC, pin requested MACs, grow the disk,
and attach a NoCloud seed ISO placed next to the VM's files. NoCloud
rather than guestinfo because it needs nothing in the guest; the
user-data installs open-vm-tools, which the driver declares as its
guest agent. A failure after the import removes the VM again.

Placement is a pure decision over inventory rows: a connected host
outside maintenance mode that sees the datastore and every port group,
with the resource pool and VM folder found by walking up to the
datacenter -- one path for a standalone host and for a vCenter.

Two faults vcsim surfaced and the tests now pin: a chunked upload body
next to a Content-Length is refused with 500, so the disk goes up as a
sized file object that also reports lease progress; and a device edit
replaces the device as sent, so disk and NIC edits start from the live
objects, backing included (vcsim panicked on a disk without one).

destroy_vm, get_vm_status, get_network_targets (port groups with their
fixed VLAN) and get_image_storages complete the contract.

reboot_host on ESXi uses RebootHost_Task and refuses outside
maintenance mode: force=True would cut power to running VMs.

Tested against vcsim in ESXi and vCenter mode, end to end.
2026-09-24 10:00:33 +02:00

115 lines
3.8 KiB
Python

"""Where a new VM goes: host, datastore, resource pool, folder, networks."""
from __future__ import annotations
import pytest
from napalm_vmware.provision.placement import Inventory as Inv
from napalm_vmware.provision.placement import choose_placement
def _host(moref, name, *, datastores, networks, parent="domain-c1", state="connected", maint=False):
return {
"_moref": moref,
"name": name,
"parent": parent,
"datastore": datastores,
"network": networks,
"runtime.connectionState": state,
"runtime.inMaintenanceMode": maint,
}
def _ds(moref, name, free, *, accessible=True, maintenance="normal"):
return {
"_moref": moref,
"name": name,
"summary": {"accessible": accessible, "freeSpace": free, "maintenanceMode": maintenance},
}
@pytest.fixture
def inv():
return Inv(
hosts=[
_host("host-1", "esx01", datastores=["ds-1", "ds-2"], networks=["net-1", "dvpg-1"]),
_host("host-2", "esx02", datastores=["ds-2", "ds-3"], networks=["net-2"]),
],
datastores=[_ds("ds-1", "local1", 50), _ds("ds-2", "san", 900), _ds("ds-3", "local2", 100)],
networks=[
{"_moref": "net-1", "name": "VM Network"},
{"_moref": "net-2", "name": "VM Network"},
{"_moref": "dvpg-1", "name": "DC0_DVPG0"},
],
compute_resources=[
{"_moref": "domain-c1", "parent": "group-h1", "resourcePool": "resgroup-1"}
],
folders=[
{"_moref": "group-h1", "parent": "datacenter-1"},
{"_moref": "group-v1", "parent": "datacenter-1"},
],
datacenters=[{"_moref": "datacenter-1", "name": "DC0", "vmFolder": "group-v1"}],
)
def test_most_free_space_wins_when_no_storage_is_named(inv):
p = choose_placement(inv, storage=None, networks=["VM Network"])
assert (p.host, p.host_name, p.datastore, p.datastore_name) == (
"host-1",
"esx01",
"ds-2",
"san",
)
assert (p.resource_pool, p.folder, p.datacenter_name) == ("resgroup-1", "group-v1", "DC0")
def test_the_network_moref_is_the_one_on_the_chosen_host(inv):
p = choose_placement(inv, storage="local2", networks=["VM Network"])
assert (p.host, p.networks) == ("host-2", ["net-2"])
def test_a_host_must_see_every_requested_network(inv):
p = choose_placement(inv, storage=None, networks=["VM Network", "DC0_DVPG0"])
assert p.host == "host-1"
assert p.networks == ["net-1", "dvpg-1"]
def test_named_datastore(inv):
assert choose_placement(inv, storage="local1", networks=["VM Network"]).datastore == "ds-1"
@pytest.mark.parametrize(
("change", "message"),
[
(lambda i: None, "no usable datastore named 'nope'"),
],
)
def test_unknown_datastore(inv, change, message):
change(inv)
with pytest.raises(ValueError, match=message):
choose_placement(inv, storage="nope", networks=["VM Network"])
def test_unknown_network(inv):
with pytest.raises(ValueError, match="no host"):
choose_placement(inv, storage=None, networks=["Missing"])
def test_hosts_in_maintenance_or_disconnected_are_skipped(inv):
inv.hosts[0]["runtime.inMaintenanceMode"] = True
inv.hosts[1]["runtime.connectionState"] = "notResponding"
with pytest.raises(ValueError, match="no host"):
choose_placement(inv, storage=None, networks=["VM Network"])
def test_inaccessible_datastores_are_skipped(inv):
inv.datastores[1]["summary"]["accessible"] = False
p = choose_placement(inv, storage=None, networks=["VM Network"])
assert p.datastore_name == "local2"
def test_datacenter_is_found_through_nested_folders(inv):
inv.compute_resources[0]["parent"] = "group-h2"
inv.folders.append({"_moref": "group-h2", "parent": "group-h1"})
assert choose_placement(inv, storage=None, networks=["VM Network"]).datacenter_name == "DC0"