Python 模拟使用不同参数调用两次的函数
假设我有以下函数:Python 模拟使用不同参数调用两次的函数,python,unit-testing,mocking,python-unittest,magicmock,Python,Unit Testing,Mocking,Python Unittest,Magicmock,假设我有以下函数: def function_to_test(context): list_of_invocations = [ func('arg%s' % number) for number in [1,2]] email_one = list_of_invocations[0].render(context) email_two = list_of_invocations[1].render(context) instance = class_to_moc
def function_to_test(context):
list_of_invocations = [ func('arg%s' % number) for number in [1,2]]
email_one = list_of_invocations[0].render(context)
email_two = list_of_invocations[1].render(context)
instance = class_to_mock(email_one, arg_1, arg_2)
instance.method_for_class(email_two, 'text/html')
instance.send()
@mock.patch('app.foo.class_to_mock')
@mock.patch('app.foo.func')
def test_render_and_send_email(self, func_mock, mock_to_class):
render_method = mock.Mock()
render_method.return_value = mock.Mock()
class_method = mock.Mock()
class_method.return_value = mock.Mock()
func_mock.return_value = mock.MagicMock(render=render_method)
mock_to_class.return_value = mock.Magic(method_for_class=class_method)
function_to_test('context goes heres')
self.assertEqual(func_mock.call_count, 2)
self.assertEqual(render_method.call_count, 2)
self.assertEqual(mock_to_class.call_count, 1)
self.assertEqual(method_class.call_count,1)
我试图用mock测试这个函数,但是我在决定如何测试这个函数时遇到了困难。想到的两个选项是使用副作用
或assert\u has\u调用
。下面是我试图做的事情的概要
- 我想确保调用了两次
(完成)func
- 我想为每个
调用(未完成)创建一个模拟func
- 我想确保render被调用两次(完成)
- 我想确保使用
email\u one调用
这就是我需要一个模拟的class\u to\u mock
和email\u-one
(未完成)email\u-two
`mock_to_class.assert_called_once_with(*MOCK EMAIL ONE*,'one', 'two')`
- 对于“类”的“方法”,我想要相同的,看起来像这样的(未完成):
class\u方法。用(*MOCK EMAIL TWO*,'text/html')断言\u调用了一次\u
希望这是有道理的。基本上,我需要从
func
进行两次不同的调用,以确保class\u到\u mock
和method\u中有参数用于\u class
您可以在@patch.object
中使用新的
属性:
email_one=Mock()
email_two=Mock()
func_call_count = 0
def my_func_mock():
global func_call_count
func_call_count += 1
if func_call_count == 1:
do_invocation_1()
if func_call_count == 2:
do_invocation_2()
def do_invocation_1():
return [email_one, email_two]
def do_invocation_2():
return [email_one, email_two]
@mock.patch('app.foo.class_to_mock')
@mock.patch('app.foo.func', new=my_func_mock)
def test_render_and_send_email(self, func_mock, mock_to_class):
render_method = mock.Mock()
render_method.return_value = mock.Mock()
class_method = mock.Mock()
class_method.return_value = mock.Mock()
func_mock.return_value = mock.MagicMock(render=render_method)
mock_to_class.return_value = mock.Magic(method_for_class=class_method)
function_to_test('context goes heres')
self.assertEqual(func_call_count, 2) # Use the func_call_count instead
self.assertEqual(render_method.call_count, 2)
mock_to_class.assert_called_once_With(email_one, email_two) # Now you can assert it is called once with email_one & two
self.assertEqual(method_class.call_count,1)