def test_connect_vnic_to_network(self): # arrange connect_command = VirtualSwitchConnectCommand(self.pv_service, self.dv_connector, self.dv_port_name_gen, self.vlan_spec_factory, self.vlan_id_range_parser) mapping = VmNetworkMapping() mapping.vnic_name = 'name' mapping.vlan_id = 'vlan_id' mapping.vlan_spec = 'trunc' mapping.dv_port_name = 'port_name' mapping.network = Mock() # act connect_results = connect_command.connect_to_networks(si=self.si, logger=Mock(), vm_uuid=self.vm_uuid, vm_network_mappings=[mapping], default_network_name='default_network', reserved_networks=[]) # assert self.assertTrue(self.vlan_id_range_parser.parse_vlan_id.called_with(self.vlan_id)) self.assertTrue( self.dv_port_name_gen.generate_port_group_name.called_with(self.vlan_id, self.vlan_spec_factory)) self.assertTrue(self.vlan_spec_factory.get_vlan_spec.called_with(self.spec_type)) self.assertTrue(self.dv_connector.connect_by_mapping.called_with(self.si, self.vm, [mapping])) self.assertEqual(connect_results[0].mac_address, 'AA-BB')
def test_connect_vnic_to_network(self): # arrange connect_command = VirtualSwitchConnectCommand( self.pv_service, self.dv_connector, self.dv_port_name_gen, self.vlan_spec_factory, self.vlan_id_range_parser) mapping = VmNetworkMapping() mapping.vnic_name = 'name' mapping.vlan_id = 'vlan_id' mapping.vlan_spec = 'trunc' mapping.dv_port_name = 'port_name' mapping.network = Mock() # act connect_results = connect_command.connect_to_networks( si=self.si, logger=Mock(), vm_uuid=self.vm_uuid, vm_network_mappings=[mapping], default_network_name='default_network', reserved_networks=[], dv_switch_name='') # assert self.assertTrue( self.vlan_id_range_parser.parse_vlan_id.called_with(self.vlan_id)) self.assertTrue( self.dv_port_name_gen.generate_port_group_name.called_with( self.vlan_id, self.vlan_spec_factory)) self.assertTrue( self.vlan_spec_factory.get_vlan_spec.called_with(self.spec_type)) self.assertTrue( self.dv_connector.connect_by_mapping.called_with( self.si, self.vm, [mapping])) self.assertEqual(connect_results[0].mac_address, 'AA-BB')
def _create_map(self, vlan_id, mode, vnic_name): vnic_to_network = VmNetworkMapping() vnic_to_network.vnic_name = self._validate_vnic_name(vnic_name) vnic_to_network.dv_switch_path = self.dv_switch_path vnic_to_network.dv_switch_name = self.dv_switch_name vnic_to_network.vlan_id = vlan_id vnic_to_network.vlan_spec = mode return vnic_to_network