def setUp(self):
super(TestLibvirtTemplate, self).setUp()
# speed up retry
self.sleep_mock = self.patch('time.sleep')
# mock open
self.open_mock = mock.mock_open(read_data='image_data')
self.patch('devops.driver.libvirt.libvirt_driver.open',
self.open_mock, create=True)
self.os_mock = self.patch('devops.helpers.helpers.os')
self.os_mock.urandom = os.urandom
# noinspection PyPep8Naming
Size = collections.namedtuple('Size', ['st_size'])
self.file_sizes = {
'/tmp/admin.iso': Size(st_size=500),
}
self.os_mock.stat.side_effect = self.file_sizes.get
# Create Environment
self.full_conf = yaml.load(ENV_TMPLT)
self.env = Environment.create_environment(self.full_conf)
self.d = self.env.get_group(name='rack-01').driver
def _create_env_from_config(self, config):
env_name = config['template']['devops_settings']['env_name']
for env in Environment.list_all():
if env.name == env_name:
print("Please, set another environment name")
raise SystemExit()
self.env = Environment.create_environment(config)
self.env.define()
# Start all l2 network devices
for group in self.env.get_groups():
for net in group.get_l2_network_devices():
net.start()
def _create_env_from_config(self, config):
env_name = config['template']['devops_settings']['env_name']
for env in Environment.list_all():
if env.name == env_name:
print("Please, set another environment name")
raise SystemExit()
self.env = Environment.create_environment(config)
self.env.define()
# Start all l2 network devices
for group in self.env.get_groups():
for net in group.get_l2_network_devices():
net.start()
def d_env(self):
if self._virt_env is None:
if not self._config:
try:
return Environment.get(name=settings.ENV_NAME)
except Exception:
self._virt_env = Environment.describe_environment(boot_from=settings.ADMIN_BOOT_DEVICE)
self._virt_env.define()
else:
try:
return Environment.get(name=self._config["template"]["devops_settings"]["env_name"])
except Exception:
self._virt_env = Environment.create_environment(full_config=self._config)
self._virt_env.define()
return self._virt_env
def d_env(self):
if self._virt_env is None:
if not self._config:
try:
return Environment.get(name=settings.ENV_NAME)
except Exception:
self._virt_env = Environment.describe_environment(
boot_from=settings.ADMIN_BOOT_DEVICE)
self._virt_env.define()
else:
try:
return Environment.get(name=self._config['template']
['devops_settings']['env_name'])
except Exception:
self._virt_env = Environment.create_environment(
full_config=self._config)
self._virt_env.define()
return self._virt_env
def setUp(self):
super(TestIPMITemplate, self).setUp()
# Create Environment
self.full_conf = yaml.load(ENV_TMPLT)
self.env = Environment.create_environment(self.full_conf)
self.ipmiclient_mock = self.patch(
'devops.driver.baremetal.ipmi_client.IpmiClient')
self.wait_mock = self.patch('devops.helpers.helpers.wait')
self.ipmiclient1 = mock.Mock(spec=IpmiClient)
self.ipmiclient2 = mock.Mock(spec=IpmiClient)
def get_client(*args):
"""Tricky way to return necessary node """
if args and args[6] == 'slave-01':
return self.ipmiclient1
elif args and args[6] == 'slave-02':
return self.ipmiclient2
self.ipmiclient_mock.side_effect = get_client
def setUp(self):
super(TestIPMITemplate, self).setUp()
# Create Environment
self.full_conf = yaml.load(ENV_TMPLT)
self.env = Environment.create_environment(self.full_conf)
self.ipmiclient_mock = self.patch(
'devops.driver.baremetal.ipmi_client.IpmiClient')
self.wait_mock = self.patch(
'devops.helpers.helpers.wait')
self.ipmiclient1 = mock.Mock(spec=IpmiClient)
self.ipmiclient2 = mock.Mock(spec=IpmiClient)
def get_client(*args):
"""Tricky way to return necessary node """
if args and args[6] == 'slave-01':
return self.ipmiclient1
elif args and args[6] == 'slave-02':
return self.ipmiclient2
self.ipmiclient_mock.side_effect = get_client
def define_from_config(conf):
env = Environment.create_environment(conf)
env.define()
env.start()
print get_ips(env)
def define_from_config(conf):
env = Environment.create_environment(conf)
env.define()
env.start()
