예제 #1
0
    def test__run_virtualbox_method(self, host_mock):
        host_object_mock = mock.MagicMock(spec_set=['find_vm'])
        func_mock = mock.MagicMock(spec_set=[])
        vm_object_mock = mock.MagicMock(spec_set=['foo'], foo=func_mock)
        host_mock.return_value = host_object_mock
        host_object_mock.find_vm.return_value = vm_object_mock
        func_mock.return_value = 'return-value'

        return_value = virtualbox._run_virtualbox_method(self.node,
                'some-ironic-method', 'foo', 'args', kwarg='kwarg')

        host_mock.assert_called_once_with(vmname='baremetal1',
                                          host='10.0.2.2',
                                          username='******',
                                          password='******',
                                          port=12345)
        host_object_mock.find_vm.assert_called_once_with('baremetal1')
        func_mock.assert_called_once_with('args', kwarg='kwarg')
        self.assertEqual('return-value', return_value)
예제 #2
0
    def test__run_virtualbox_method(self, host_mock):
        host_object_mock = mock.MagicMock()
        func_mock = mock.MagicMock()
        vm_object_mock = mock.MagicMock(foo=func_mock)
        host_mock.return_value = host_object_mock
        host_object_mock.find_vm.return_value = vm_object_mock
        func_mock.return_value = 'return-value'

        return_value = virtualbox._run_virtualbox_method(self.node,
                                                         'some-ironic-method',
                                                         'foo',
                                                         'args',
                                                         kwarg='kwarg')

        host_mock.assert_called_once_with(vmname='baremetal1',
                                          host='10.0.2.2',
                                          username='******',
                                          password='******',
                                          port=12345)
        host_object_mock.find_vm.assert_called_once_with('baremetal1')
        func_mock.assert_called_once_with('args', kwarg='kwarg')
        self.assertEqual('return-value', return_value)