Birim testi: Aynı yöntemle birden fazla çağrı nasıl onaylanır?

Question

Başka bir yöntemi farklı argümanlarla iki kez çağıran bir yöntem var.

class A(object): 
    def helper(self, arg_one, arg_two): 
     """Return something which depends on arguments.""" 

    def caller(self): 
     value_1 = self.helper(foo, bar) # First call. 
     value_2 = self.helper(foo_bar, bar_foo) # Second call! 

assert_called_with kullanılması sadece ilk çağrıyı değil, ikinci bir iddia bana yardımcı oluyor. assert_called_once_with bile yardımcı olmuyor. Burada neyi özlüyorum? Bu tür çağrıları test etmenin herhangi bir yolu var mı?

Answer 1

Bir yöntemde yapılan tüm çağrıları içeren mock_calls'u kullanabilirsiniz. Bu liste ilk aramayı, ikinci aramayı ve sonraki tüm çağrıları içerir. Böylece ikinci görüşme hakkında bir şey belirtmek için mock_calls[1] ile iddialar yazabilirsiniz. m = mock.Mock() ve kod m.method(123) sonra yazıyor Örneğin

---

:

m.method çağrı listesi tam olarak bir çağrı, argüman 123 ile yani bir çağrı olduğunu iddia

assert m.method.mock_calls == [mock.call(123)] 

.

Answer 2

Simon Visser'in cevabı eklemek için, bunun yerine Birim test bağlam si daha iyi bir uygulamadır söyleyebilirim assert sözdizimi ait unittest.TestCaseself.assertEqual() yöntemi kullanabilirsiniz nce, bir şeyler yanlış gittiğinde görüntülenecek olan yorumları da ekleyebilirsiniz. Örneğin bu şekilde

:

self.assertEqual(
    [ 
     mock.call(1, 'ValueA', True)), 
     mock.call(2, 'ValueB', False)), 
     mock.call(3, 'ValueC', False)) 
    ], 
    mock_cur.execute.mock_calls, 
    "The method was not called with the correct arguments." 
)