def test_generate_loopbacks_returned_loopback_dict(mock_random):
expected_loopback_dict = {
'blackhole': '127.6.6.6',
'sw01': '127.255.255.1',
'sw02': '127.255.255.2'
}
assert generate_loopbacks(mock_guest_data) == expected_loopback_dict
def test_render_from_template(mock_random, mock_uuid, mock_time):
time_now = time.strftime(TIMESTAMP_FORMAT)
loopbacks = generate_loopbacks(mock_guest_data)
vagrantfile = render_from_template(
template_name='guest.j2',
template_directory=TEMPLATES_DIR,
custom_filters=custom_filters,
guests=mock_guest_data,
loopbacks=loopbacks,
interface_mappings=interface_mappings,
domain_uuid=get_uuid(),
creation_time=time_now,
blackhole_interfaces={},
)
assert vagrantfile == mock_vagrantfile
def test_render_vagrantfile_with_additional_storage_interfaces(mock_storage_size, mock_random,
mock_uuid, mock_time):
time_now = time.strftime(TIMESTAMP_FORMAT)
guest_data = copy.deepcopy(mock_guest_data)
loopbacks = generate_loopbacks(guest_data)
guest_data['sw01']['provider_config']['additional_storage_volumes'] = mock_additional_storage_volumes
update_guest_additional_storage(guest_data)
vagrantfile = render_from_template(
template_name='guest.j2',
template_directory=TEMPLATES_DIR,
custom_filters=custom_filters,
guests=guest_data,
loopbacks=loopbacks,
interface_mappings=generate_guest_interface_mappings(),
domain_uuid=get_uuid(),
creation_time=time_now,
blackhole_interfaces={},
)
assert mock_vagrantfile_with_additional_storage_volumes == vagrantfile
def test_generate_loopbacks_guest_dict_empty_raises_exception():
with pytest.raises(ValueError):
generate_loopbacks(guest_dict={})
def test_generate_loopbacks_guest_dict_is_none_raises_exception():
with pytest.raises(AttributeError):
generate_loopbacks(guest_dict=None)