python - 入力に基づく MagicMock の戻り値

Question

テストをフレックスモックからモックにリファクタリングしようとしています。flexmock からの次の構文が与えられます。

flexmock(subprocess).should_receive('check_output').with_args('ls /').and_return(output)

Mock を使用してこれをどのように書き換えることができますか? 特に、モックを使用して特定の入力に戻り値を固定するにはどうすればよいですか?

Answer 1

属性を使用してそれを行うことができます：side_effect patch

>>> from unittest.mock import *
>>> root = '''bin   cdrom  etc   initrd.img  lib32  libx32      media  opt   root  sbin  sys  usr  vmlinuz
... boot  dev    home  lib         lib64  lost+found  mnt    proc  run   srv   tmp  var'''
>>> answer = {'ls /': root}
>>> import subprocess

>>> with patch('subprocess.check_output', side_effect=lambda arg, *args, **kwargs: answer[arg]) as mock_check_output :
...     assert root == subprocess.check_output('ls /')
...     mock_check_output.assert_called_with('ls /')
...     subprocess.check_output('something else')
... 
Traceback (most recent call last):
  File "<stdin>", line 4, in <module>
  File "/usr/lib/python3.4/unittest/mock.py", line 896, in __call__
    return _mock_self._mock_call(*args, **kwargs)
  File "/usr/lib/python3.4/unittest/mock.py", line 962, in _mock_call
    ret_val = effect(*args, **kwargs)
  File "<stdin>", line 1, in <lambda>
KeyError: 'something else'
>>> 

の構文に比べて少し難しく感じると思いますがflexmock、必要なのはモックというよりもスタブです。テストを分離するように設計し、セットアップ段階でモック/スタブを構成できる場合は、この構文が適切であり、すべてのmockアサートとpatchオプションが非常に強力であることがわかるでしょう。