def test_install_machines(self, target_platforms, isos, dryrun):
'''
Collect list of missing VMs
'''
missing_platforms = []
print(target_platforms)
index = 0
for target_platform in target_platforms:
presents, _ = self.__vm_presents(target_platform["os_name"],
target_platform["architecture"])
if not presents:
missing_platforms.append((index, target_platform))
index += 1
res, adapter_name, ip_address = utils.get_connected_network_adapter()
assert res, "I did not find any connected network adapters on the host machine"
print(
f"I am going to use network adapter {adapter_name}, {ip_address}")
if len(missing_platforms):
print(f"Missing: {missing_platforms}")
for index, target_platform in missing_platforms:
os_name, architecture = target_platform[
"os_name"], target_platform["architecture"]
autounattend_vfd = self.__get_autounattend_vfd(os_name)
if not os.path.isfile(autounattend_vfd):
autounattend_command = self.__get_autounattend_vfd_command(
os_name)
assert False, f"Floppy image {autounattend_vfd} is not found. Try {autounattend_command}"
iso = isos[index]
assert len(
isos
) > index, f"No ISO is specified for the missing {target_platform}"
uuid, settings_file = self.__install_machine(os_name, architecture)
target_platform["uuid"] = uuid
if not dryrun:
self.__setup_machine(target_platform, settings_file, uuid,
dryrun)
vbox = virtualbox_shell.VirtualBox()
if not dryrun:
print(f"Add boot ISO {iso} to {uuid}")
vbox.add_boot_disk(uuid, iso)
if not dryrun:
print(f"Add floppy ISO {autounattend_vfd} to {uuid}")
vbox.add_floppy_disk(uuid, autounattend_vfd)
vbox = virtualbox_shell.VirtualBox()
# I configure the VM bridge to use a connected (has an assigned IP) Host
# network interface
# The patch is for laptops which switch between WiFi/Wired interfacces often
for target_platform in target_platforms:
os_name, architecture = target_platform[
"os_name"], target_platform["architecture"]
presents, machine = self.__vm_presents(os_name, architecture)
print(f"Setup network addater {adapter_name} to {machine.uuid}")
if not dryrun:
vbox.set_network_adapter(machine.uuid, adapter_name)
def __vm_running(self, os_name, architecture):
vm_name = self.__get_vm_name(os_name, architecture)
vbox = virtualbox_shell.VirtualBox()
for vm in vbox.running_machines():
if vm_name == vm.name:
return True, vm
return False, None
def test_wait_for_ip(self, target_platforms):
'''
Wait for the VM to boot. Can be a fresh VM which running the setup
Wait for the setup to complete
'''
vbox = virtualbox_shell.VirtualBox()
uuids_macs = {}
for target_platform in target_platforms:
os_name, architecture = target_platform[
"os_name"], target_platform["architecture"]
presents, machine = self.__vm_presents(os_name, architecture)
mac = vbox.get_mac_address(machine.uuid)
target_platform["mac"] = mac
uuids_macs[machine.uuid] = (mac, machine.name)
print(f"Waiting for ping from {uuids_macs}")
time_end = datetime.datetime.now() + datetime.timedelta(minutes=20)
while len(uuids_macs) and (datetime.datetime.now() < time_end):
uuid = next(iter(uuids_macs))
mac, machine_name = uuids_macs[uuid]
res, hostname, ipaddress = utils.find_ip_by_mac(mac)
if res:
res = utils.run_shell_command(f"ping -c 1 {ipaddress}", None,
None, True)
if res:
print(
f"Got a response from {ipaddress} {hostname} for {mac} {machine_name} {uuid}"
)
target_platform["ipaddress"] = ipaddress
del uuids_macs[uuid]
time.sleep(5.0)
assert len(
uuids_macs
) == 0, "Failed to get a response to ping for " + str(uuids_macs)
def __find_machine_type(self, os_name, architecture):
# Dictionary which translates my OS target to vbox.guest_os_types.description
machine_types = {"win8": "Windows 8", "win10": "Windows 10"}
machine_type = machine_types[os_name]
vbox = virtualbox_shell.VirtualBox()
os_types = vbox.guest_os_types()
for os_type in os_types:
if machine_type in os_type.description and architecture in os_type.description:
print(f"Hit OS type {os_type}")
return os_type.id
return None
def __install_machine(self, os_name, architecture):
machine_path = self.__get_vm_path(os_name, architecture)
machine_name = self.__get_vm_name(os_name, architecture)
os_type_id = self.__find_machine_type(os_name, architecture)
vbox = virtualbox_shell.VirtualBox()
res, uuid, settings_file = vbox.create_machine(machine_path,
machine_name,
os_type_id)
assert res, f"Failed to create {machine_name}"
print(f"Created machine {machine_name} uuid={uuid} in {settings_file}")
vbox.register_machine(uuid)
return uuid, settings_file
def test_start_machines(self, target_platforms, headless_vms, dryrun):
'''
Start required VMs
'''
vbox = virtualbox_shell.VirtualBox()
for target_platform in target_platforms:
os_name, architecture = target_platform[
"os_name"], target_platform["architecture"]
presents, machine = self.__vm_presents(os_name, architecture)
assert presents, "At this point the VM shall exist"
print(f"Start machine {machine.uuid}, headless is {headless_vms}")
if not dryrun:
vbox.start_machine(machine.uuid, headless_vms)
def __setup_machine(self, target_platform, settings_file, uuid, dryrun):
vbox = virtualbox_shell.VirtualBox()
memory = 2 * 1024
os_name, architecture = target_platform["os_name"], target_platform[
"architecture"]
presents, machine = self.__vm_presents(os_name, architecture)
assert presents, f"Failed to find machine {os_name} {architecture}"
print(f"Set memory size {memory}")
if not dryrun:
vbox.set_machine(machine.uuid, memory)
print(f"Add hard disk to {uuid}")
if not dryrun:
vbox.add_hard_disk(settings_file, uuid, 32 * 1024)
def test_remove_machines(self, remove_vms, target_platforms, dryrun):
if not remove_vms:
return
print("Removing all VMs")
vbox = virtualbox_shell.VirtualBox()
for target_platform in target_platforms:
presents, vm = self.__vm_presents(target_platform["os_name"],
target_platform["architecture"])
if presents:
print(f"Remove {vm.name}")
if not dryrun:
vbox.remove_machine(vm.uuid)
# a short delay in case a human being watches the GUI - all VNs are disappearing here
time.sleep(0.5)
def test_stop_machines(self, target_platforms, dryrun):
'''
Stop all running VMs before I start to modify them
'''
vbox = virtualbox_shell.VirtualBox()
for target_platform in target_platforms:
running, vm = self.__vm_running(target_platform["os_name"],
target_platform["architecture"])
if running:
print(f"Stop {vm.name}")
if not dryrun:
#vbox.stop_machine(vm.uuid)
# I want to shutdown the VM, because I am going to setup the network adapter
# and need the VM in a mutable state.
vbox.power_down_machine(vm.uuid)
# a short delay in case a human being watches the GUI - all VNs are disappearing here
time.sleep(0.5)
def test_clean_arp(self, target_platforms):
'''
Remove IP address of the machines. I am going to ping the created machines
using cached by arp IP address
'''
vbox = virtualbox_shell.VirtualBox()
time_end = datetime.datetime.now() + datetime.timedelta(minutes=8)
uuids_macs = {}
for target_platform in target_platforms:
os_name, architecture = target_platform[
"os_name"], target_platform["architecture"]
presents, machine = self.__vm_presents(os_name, architecture)
if not presents:
continue
mac = vbox.get_mac_address(machine.uuid)
res, hostname, ipaddress = utils.find_ip_by_mac(mac)
if not res:
continue
utils.run_shell_command(f"ping -c 1 {ipaddress}")
def test_environment(self):
assert virtualbox_shell.VirtualBox().is_ready(
), "No VBoxManage in the path? Try apt-get install virtualbox"
def __get_vm_path(self, os_name, architecture):
vbox = virtualbox_shell.VirtualBox()
machine_folder = vbox.default_machine_folder()
return machine_folder