Source code for tests.vm_gen_plugin

from linux.ubuntu2204 import Ubuntu2204Server

from firewheel.control.experiment_graph import Vertex, AbstractPlugin


[docs] class VmGen(AbstractPlugin): """ This plugin will create as many :py:class:`Ubuntu2204Server <linux.ubuntu2204.Ubuntu2204Server>` VMs as the size specifies. The new VMs will be give the name server-X where X is an integer. """
[docs] def run(self, size="1"): """ This plugin will create as many :py:class:`Ubuntu2204Server <linux.ubuntu2204.Ubuntu2204Server>` VMs as the size specifies. Args: size (str): The number of VMs to create. """ size = int(size) for i in range(size): vm = Vertex(self.g, f"server-{i}") vm.decorate(Ubuntu2204Server)