def workload(num_vms, additional_places, server, port): client = DirectClient("Host", Host.Client, server, port) client.connect() for _ in xrange(num_vms): vm = VmWrapper(client) for _ in xrange(additional_places): vm.place() vm.create() client.close()
def clear(self): """Remove all the VMs and disks""" for host in self.hosts: client = DirectClient("Host", Host.Client, host, 8835) client.connect() request = GetResourcesRequest() response = rpc_call(client.get_resources, request) vm_wrapper = VmWrapper(client) for resource in response.resources: disk_ids = [disk.id for disk in resource.disks] delete_request = Host.DeleteVmRequest(resource.vm.id, disk_ids) vm_wrapper.delete(request=delete_request) vm_wrapper.delete_disks(disk_ids, validate=True) client.close()