Mock pythonを使ってメソッド呼び出しをアサートする

Question

私はPythonでモックライブラリを使ってユニットテストをしようとしています。私はbへのモック呼び出しが行われたときbのための呼び出しが行われたことを主張するにはどうすればよい

def a(): 
    print 'a' 

def b(): 
    print 'b' 
    if some condition 
     a() 

：私は、次のコードを持っていますか？私は、次のコードを試してみましたが、それは失敗しました：

mymock=Mock() 
mymock.b() 
assertTrue(a.__call__ in mymock.mock_calls) 

何らかの理由で、私はmymock.b()はメソッドb()とは何の関係もないと思います。これには何ができるでしょうか？

Answer 1

あなたがaにパッチを適用した場合、あなたはそれはそうのように呼ばれていたことを確認することができます

with mock.patch('__main__.a') as fake_a: 
    b() 
    fake_a.assert_called_with() 

あなたの方法は、別のモジュール内にある場合：

import mymodule 

with mock.patch('mymodule.a') as fake_a: 
    mymodule.b() 
    fake_a.assert_called_with() 

Answer 2

Somehow, I think that the mymock.b() has nothing to do with the method b() What can be done for this?

あなたが正しいです。オブジェクトをモックすると、モックが背後で何をしているのか気にしないことを意味しています。 aがbから呼び出されたことを確認したい場合は、patchaをbに設定します。

>>> from mock import patch 
>>> with patch('__main__.a') as patch_a: 
...  b() 
...  patch_a.assert_called_with() 

だから、この話の教訓は、あなたが実際に呼び出さずに測定したい対象物ではなく、その実装あなたが気にオブジェクトの上モックまたはパッチです。この場合、あなたはbに関心があり、どのようにそれがaを使用しているのか調べたいと思う。 aが何をしているのか気にしないので、呼び出されているだけで、パッチを当てることができます。

さらに、assert_called_withの代わりにmock_callsというパッチにアクセスして行われたすべてのコールを分析できます。この場合、patch_a.mock_callsとなります。