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): 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()