def test_add_single_nic_on_vm(self, client_mock, mock_vm_set, mock_vm_get): #pylint: disable=line-too-long new_nic_name = 'new_nic' new_nic_id = '/subscriptions/123/resourceGroups/rg1/providers/Microsoft.Network/networkInterfaces/' + new_nic_name new_nic = NetworkInterfaceReference(new_nic_id, False) #stub to get the vm which has no nic vm = FakedVM() vm.network_profile = None mock_vm_get.return_value = vm #stub to return nic on 'get' network_client_fake = mock.MagicMock() network_client_fake.network_interfaces.get.return_value = new_nic client_mock.return_value = network_client_fake #execute vm_add_nics('rg1', 'vm1', [new_nic_id], []) #assert self.assertTrue(mock_vm_get.called) mock_vm_set.assert_called_once_with(vm) network_client_fake.network_interfaces.get.assert_called_once_with('rg1', new_nic_name) self.assertEqual(len(vm.network_profile.network_interfaces), 1) self.assertEqual(vm.network_profile.network_interfaces[0].id, new_nic_id) self.assertTrue(vm.network_profile.network_interfaces[0].primary)
def test_add_nics_on_vm(self, client_mock, mock_vm_set, mock_vm_get): #pylint: disable=line-too-long new_nic_name = 'new_nic' new_nic_id = '/subscriptions/123/resourceGroups/rg1/providers/Microsoft.Network/networkInterfaces/' + new_nic_name new_nic = NetworkInterfaceReference(new_nic_id, False) existing_nic_id = '/subscriptions/123/resourceGroups/rg1/providers/Microsoft.Network/networkInterfaces/existing_nic' existing_nic = NetworkInterfaceReference(existing_nic_id, True) existing_nic_name2 = 'existing_nic2' existing_nic_id2 = '/subscriptions/123/resourceGroups/rg1/providers/Microsoft.Network/networkInterfaces/' + existing_nic_name2 existing_nic2 = NetworkInterfaceReference(existing_nic_id2, False) #stub to get the vm vm = FakedVM([existing_nic, existing_nic2]) mock_vm_get.return_value = vm #stub to return nic on 'get' network_client_fake = mock.MagicMock() network_client_fake.network_interfaces.get.return_value = new_nic client_mock.return_value = network_client_fake #execute vm_add_nics('rg1', 'vm1', [new_nic_id], [], existing_nic_name2) #assert self.assertTrue(mock_vm_get.called) mock_vm_set.assert_called_once_with(vm) network_client_fake.network_interfaces.get.assert_called_once_with('rg1', new_nic_name) self.assertEqual(len(vm.network_profile.network_interfaces), 3) self.assertEqual(vm.network_profile.network_interfaces[0].id, existing_nic_id) self.assertEqual(vm.network_profile.network_interfaces[1].id, existing_nic_id2) self.assertEqual(vm.network_profile.network_interfaces[2].id, new_nic_id) self.assertFalse(vm.network_profile.network_interfaces[0].primary) self.assertTrue(vm.network_profile.network_interfaces[1].primary) self.assertFalse(vm.network_profile.network_interfaces[2].primary)
def test_add_single_nic_on_vm(self, client_mock, mock_vm_set, mock_vm_get): #pylint: disable=line-too-long new_nic_name = 'new_nic' new_nic_id = '/subscriptions/123/resourceGroups/rg1/providers/Microsoft.Network/networkInterfaces/' + new_nic_name new_nic = NetworkInterfaceReference(new_nic_id, False) #stub to get the vm which has no nic vm = FakedVM() vm.network_profile = None mock_vm_get.return_value = vm #stub to return nic on 'get' network_client_fake = mock.MagicMock() network_client_fake.network_interfaces.get.return_value = new_nic client_mock.return_value = network_client_fake #execute vm_add_nics('rg1', 'vm1', [new_nic_id], []) #assert self.assertTrue(mock_vm_get.called) mock_vm_set.assert_called_once_with(vm) network_client_fake.network_interfaces.get.assert_called_once_with( 'rg1', new_nic_name) self.assertEqual(len(vm.network_profile.network_interfaces), 1) self.assertEqual(vm.network_profile.network_interfaces[0].id, new_nic_id) self.assertTrue(vm.network_profile.network_interfaces[0].primary)
def test_add_nics_on_vm(self, client_mock, mock_vm_set, mock_vm_get): #pylint: disable=line-too-long new_nic_name = 'new_nic' new_nic_id = '/subscriptions/123/resourceGroups/rg1/providers/Microsoft.Network/networkInterfaces/' + new_nic_name new_nic = NetworkInterfaceReference(new_nic_id, False) existing_nic_id = '/subscriptions/123/resourceGroups/rg1/providers/Microsoft.Network/networkInterfaces/existing_nic' existing_nic = NetworkInterfaceReference(existing_nic_id, True) existing_nic_name2 = 'existing_nic2' existing_nic_id2 = '/subscriptions/123/resourceGroups/rg1/providers/Microsoft.Network/networkInterfaces/' + existing_nic_name2 existing_nic2 = NetworkInterfaceReference(existing_nic_id2, False) #stub to get the vm vm = FakedVM([existing_nic, existing_nic2]) mock_vm_get.return_value = vm #stub to return nic on 'get' network_client_fake = mock.MagicMock() network_client_fake.network_interfaces.get.return_value = new_nic client_mock.return_value = network_client_fake #execute vm_add_nics('rg1', 'vm1', [new_nic_id], [], existing_nic_name2) #assert self.assertTrue(mock_vm_get.called) mock_vm_set.assert_called_once_with(vm) network_client_fake.network_interfaces.get.assert_called_once_with( 'rg1', new_nic_name) self.assertEqual(len(vm.network_profile.network_interfaces), 3) self.assertEqual(vm.network_profile.network_interfaces[0].id, existing_nic_id) self.assertEqual(vm.network_profile.network_interfaces[1].id, existing_nic_id2) self.assertEqual(vm.network_profile.network_interfaces[2].id, new_nic_id) self.assertFalse(vm.network_profile.network_interfaces[0].primary) self.assertTrue(vm.network_profile.network_interfaces[1].primary) self.assertFalse(vm.network_profile.network_interfaces[2].primary)