def test_get_network_ref_flat_dhcp(self):
self.mox.StubOutWithMock(vif, 'ensure_vlan_bridge')
vif.ensure_vlan_bridge(self.session,
self.vif,
cluster=self.cluster,
create_vlan=False)
self.mox.ReplayAll()
vif.get_network_ref(self.session, self.cluster, self.vif, False)
def test_ensure_vlan_bridge_without_vlan(self):
self.mox.StubOutWithMock(network_util, "get_network_with_the_name")
self.mox.StubOutWithMock(network_util, "get_vswitch_for_vlan_interface")
self.mox.StubOutWithMock(network_util, "check_if_vlan_interface_exists")
self.mox.StubOutWithMock(network_util, "create_port_group")
network_util.get_network_with_the_name(self.session, "fa0", self.cluster).AndReturn(None)
network_util.get_vswitch_for_vlan_interface(self.session, "vmnet0", self.cluster).AndReturn("vmnet0")
network_util.check_if_vlan_interface_exists(self.session, "vmnet0", self.cluster).AndReturn(True)
network_util.create_port_group(self.session, "fa0", "vmnet0", 0, self.cluster)
self.mox.ReplayAll()
vif.ensure_vlan_bridge(self.session, self.vif, create_vlan=False)
def test_ensure_vlan_bridge_with_network(self):
self.mox.StubOutWithMock(network_util, 'get_network_with_the_name')
self.mox.StubOutWithMock(network_util,
'get_vswitch_for_vlan_interface')
self.mox.StubOutWithMock(network_util,
'check_if_vlan_interface_exists')
self.mox.StubOutWithMock(network_util, 'create_port_group')
vm_network = {'name': 'VM Network', 'type': 'Network'}
network_util.get_network_with_the_name(self.session, 'fa0',
self.cluster).AndReturn(vm_network)
self.mox.ReplayAll()
vif.ensure_vlan_bridge(self.session, self.vif, create_vlan=False)
def test_ensure_vlan_bridge_with_network(self):
self.mox.StubOutWithMock(network_util, 'get_network_with_the_name')
self.mox.StubOutWithMock(network_util,
'get_vswitch_for_vlan_interface')
self.mox.StubOutWithMock(network_util,
'check_if_vlan_interface_exists')
self.mox.StubOutWithMock(network_util, 'create_port_group')
vm_network = {'name': 'VM Network', 'type': 'Network'}
network_util.get_network_with_the_name(
self.session, 'fa0', self.cluster).AndReturn(vm_network)
self.mox.ReplayAll()
vif.ensure_vlan_bridge(self.session, self.vif, create_vlan=False)
def test_ensure_vlan_bridge_with_network(self, mock_create_port_group,
mock_check_if_vlan_exists,
mock_get_vswitch_for_vlan,
mock_get_network_with_name):
vm_network = {'name': 'VM Network', 'type': 'Network'}
mock_get_network_with_name.return_value = vm_network
vif.ensure_vlan_bridge(self.session, self.vif, create_vlan=False)
mock_get_network_with_name.assert_called_once_with(
self.session, 'fa0', self.cluster)
mock_check_if_vlan_exists.assert_not_called()
mock_get_vswitch_for_vlan.assert_not_called()
mock_create_port_group.assert_not_called()
def test_ensure_vlan_bridge_with_network(self,
mock_create_port_group,
mock_check_if_vlan_exists,
mock_get_vswitch_for_vlan,
mock_get_network_with_name
):
vm_network = {'name': 'VM Network', 'type': 'Network'}
mock_get_network_with_name.return_value = vm_network
vif.ensure_vlan_bridge(self.session, self.vif, create_vlan=False)
mock_get_network_with_name.assert_called_once_with(self.session,
'fa0',
self.cluster)
mock_check_if_vlan_exists.assert_not_called()
mock_get_vswitch_for_vlan.assert_not_called()
mock_create_port_group.assert_not_called()
def test_ensure_vlan_bridge_without_vlan(self,
mock_create_port_group,
mock_check_if_vlan_exists,
mock_get_vswitch_for_vlan,
mock_get_network_with_name):
vif.ensure_vlan_bridge(self.session, self.vif, create_vlan=False)
expected_calls = [mock.call(self.session, 'fa0', self.cluster),
mock.call(self.session, 'fa0', None)]
mock_get_network_with_name.assert_has_calls(expected_calls)
self.assertEqual(2, mock_get_network_with_name.call_count)
mock_get_vswitch_for_vlan.assert_called_once_with(
self.session, 'vmnet0', self.cluster)
mock_check_if_vlan_exists.assert_called_once_with(
self.session, 'vmnet0', self.cluster)
mock_create_port_group.assert_called_once_with(
self.session, 'fa0', 'vmnet0', 0, self.cluster)
def test_ensure_vlan_bridge_without_vlan(self,
mock_create_port_group,
mock_check_if_vlan_exists,
mock_get_vswitch_for_vlan,
mock_get_network_with_name):
vif.ensure_vlan_bridge(self.session, self.vif, create_vlan=False)
expected_calls = [mock.call(self.session, 'fa0', self.cluster),
mock.call(self.session, 'fa0', None)]
mock_get_network_with_name.assert_has_calls(expected_calls)
self.assertEqual(2, mock_get_network_with_name.call_count)
mock_get_vswitch_for_vlan.assert_called_once_with(
self.session, 'vmnet0', self.cluster)
mock_check_if_vlan_exists.assert_called_once_with(
self.session, 'vmnet0', self.cluster)
mock_create_port_group.assert_called_once_with(
self.session, 'fa0', 'vmnet0', 0, self.cluster)
def test_ensure_vlan_bridge_without_vlan(self):
self.mox.StubOutWithMock(network_util, 'get_network_with_the_name')
self.mox.StubOutWithMock(network_util,
'get_vswitch_for_vlan_interface')
self.mox.StubOutWithMock(network_util,
'check_if_vlan_interface_exists')
self.mox.StubOutWithMock(network_util, 'create_port_group')
network_util.get_network_with_the_name(self.session, 'fa0',
self.cluster).AndReturn(None)
network_util.get_vswitch_for_vlan_interface(
self.session, 'vmnet0', self.cluster).AndReturn('vmnet0')
network_util.check_if_vlan_interface_exists(
self.session, 'vmnet0', self.cluster).AndReturn(True)
network_util.create_port_group(self.session, 'fa0', 'vmnet0', 0,
self.cluster)
self.mox.ReplayAll()
vif.ensure_vlan_bridge(self.session, self.vif, create_vlan=False)
def test_ensure_vlan_bridge_without_vlan(self):
self.mox.StubOutWithMock(network_util, 'get_network_with_the_name')
self.mox.StubOutWithMock(network_util,
'get_vswitch_for_vlan_interface')
self.mox.StubOutWithMock(network_util,
'check_if_vlan_interface_exists')
self.mox.StubOutWithMock(network_util, 'create_port_group')
network_util.get_network_with_the_name(self.session, 'fa0',
self.cluster).AndReturn(None)
network_util.get_vswitch_for_vlan_interface(self.session, 'vmnet0',
self.cluster).AndReturn('vmnet0')
network_util.check_if_vlan_interface_exists(self.session, 'vmnet0',
self.cluster).AndReturn(True)
network_util.create_port_group(self.session, 'fa0', 'vmnet0', 0,
self.cluster)
self.mox.ReplayAll()
vif.ensure_vlan_bridge(self.session, self.vif, create_vlan=False)
def _get_vif_infos():
vif_infos = []
for (network, mapping) in network_info:
mac_address = mapping["mac"]
network_name = network["bridge"]
if mapping.get("should_create_vlan"):
network_ref = vmwarevif.ensure_vlan_bridge(self._session, network)
else:
network_ref = _check_if_network_bridge_exists(network_name)
vif_infos.append({"network_name": network_name, "mac_address": mac_address, "network_ref": network_ref})
return vif_infos
def test_get_network_ref_bridge(self):
self.mox.StubOutWithMock(vif, "ensure_vlan_bridge")
vif.ensure_vlan_bridge(self.session, self.vif, cluster=self.cluster, create_vlan=True)
self.mox.ReplayAll()
network = network_model.Network(
id=0, bridge="fa0", label="fake", vlan=3, bridge_interface="eth0", injected=True, should_create_vlan=True
)
self.vif = network_model.NetworkInfo(
[
network_model.VIF(
id=None,
address="DE:AD:BE:EF:00:00",
network=network,
type=None,
devname=None,
ovs_interfaceid=None,
rxtx_cap=3,
)
]
)[0]
vif.get_network_ref(self.session, self.cluster, self.vif, False)
def test_get_network_ref_bridge(self):
self.mox.StubOutWithMock(vif, 'ensure_vlan_bridge')
vif.ensure_vlan_bridge(self.session, self.vif, cluster=self.cluster,
create_vlan=True)
self.mox.ReplayAll()
network = network_model.Network(id=0,
bridge='fa0',
label='fake',
vlan=3,
bridge_interface='eth0',
injected=True,
should_create_vlan=True)
self.vif = network_model.NetworkInfo([
network_model.VIF(id=None,
address='DE:AD:BE:EF:00:00',
network=network,
type=None,
devname=None,
ovs_interfaceid=None,
rxtx_cap=3)
])[0]
vif.get_network_ref(self.session, self.cluster, self.vif, False)
def _get_vif_infos():
vif_infos = []
for (network, mapping) in network_info:
mac_address = mapping['mac']
network_name = network['bridge']
if mapping.get('should_create_vlan'):
network_ref = vmwarevif.ensure_vlan_bridge(
self._session, network)
else:
network_ref = _check_if_network_bridge_exists(network_name)
vif_infos.append({'network_name': network_name,
'mac_address': mac_address,
'network_ref': network_ref,
})
return vif_infos
def _get_vif_infos():
vif_infos = []
for (network, mapping) in network_info:
mac_address = mapping['mac']
network_name = network['bridge']
if mapping.get('should_create_vlan'):
network_ref = vmwarevif.ensure_vlan_bridge(
self._session, network)
else:
network_ref = _check_if_network_bridge_exists(network_name)
vif_infos.append({
'network_name': network_name,
'mac_address': mac_address,
'network_ref': network_ref,
})
return vif_infos
def test_ensure_vlan_bridge_with_existing_dvs(self):
network_ref = {
'dvpg': 'dvportgroup-2062',
'type': 'DistributedVirtualPortgroup'
}
self.mox.StubOutWithMock(network_util, 'get_network_with_the_name')
self.mox.StubOutWithMock(network_util,
'get_vswitch_for_vlan_interface')
self.mox.StubOutWithMock(network_util,
'check_if_vlan_interface_exists')
self.mox.StubOutWithMock(network_util, 'create_port_group')
network_util.get_network_with_the_name(
self.session, 'fa0', self.cluster).AndReturn(network_ref)
self.mox.ReplayAll()
ref = vif.ensure_vlan_bridge(self.session, self.vif, create_vlan=False)
self.assertThat(ref, matchers.DictMatches(network_ref))
def test_ensure_vlan_bridge_with_existing_dvs(self):
network_ref = {'dvpg': 'dvportgroup-2062',
'type': 'DistributedVirtualPortgroup'}
self.mox.StubOutWithMock(network_util, 'get_network_with_the_name')
self.mox.StubOutWithMock(network_util,
'get_vswitch_for_vlan_interface')
self.mox.StubOutWithMock(network_util,
'check_if_vlan_interface_exists')
self.mox.StubOutWithMock(network_util, 'create_port_group')
network_util.get_network_with_the_name(self.session, 'fa0',
self.cluster).AndReturn(network_ref)
self.mox.ReplayAll()
ref = vif.ensure_vlan_bridge(self.session,
self.vif,
create_vlan=False)
self.assertThat(ref, matchers.DictMatches(network_ref))
def test_ensure_vlan_bridge_with_existing_dvs(self, mock_create_port_group,
mock_check_if_vlan_exists,
mock_get_vswitch_for_vlan,
mock_get_network_with_name):
network_ref = {
'dvpg': 'dvportgroup-2062',
'type': 'DistributedVirtualPortgroup'
}
mock_get_network_with_name.return_value = network_ref
ref = vif.ensure_vlan_bridge(self.session, self.vif, create_vlan=False)
self.assertThat(ref, matchers.DictMatches(network_ref))
mock_get_network_with_name.assert_called_once_with(
self.session, 'fa0', self.cluster)
mock_check_if_vlan_exists.assert_not_called()
mock_get_vswitch_for_vlan.assert_not_called()
mock_create_port_group.assert_not_called()
def test_ensure_vlan_bridge_with_existing_dvs(self,
mock_create_port_group,
mock_check_if_vlan_exists,
mock_get_vswitch_for_vlan,
mock_get_network_with_name
):
network_ref = {'dvpg': 'dvportgroup-2062',
'type': 'DistributedVirtualPortgroup'}
mock_get_network_with_name.return_value = network_ref
ref = vif.ensure_vlan_bridge(self.session,
self.vif,
create_vlan=False)
self.assertThat(ref, matchers.DictMatches(network_ref))
mock_get_network_with_name.assert_called_once_with(self.session,
'fa0',
self.cluster)
mock_check_if_vlan_exists.assert_not_called()
mock_get_vswitch_for_vlan.assert_not_called()
mock_create_port_group.assert_not_called()
def test_get_network_ref_flat_dhcp(self):
self.mox.StubOutWithMock(vif, 'ensure_vlan_bridge')
vif.ensure_vlan_bridge(self.session, self.vif, cluster=self.cluster,
create_vlan=False)
self.mox.ReplayAll()
vif.get_network_ref(self.session, self.cluster, self.vif, False)
