http://nao-y.hatenablog.com/entry/2018/08/01/231336 のコードではまった原因
Pythonでは import したモジュールはキャッシュされ、もう一度 import してもインポート処理は省略され、キャッシュが使われる
import_module 関数を使っても import 文を使っても、 fortune.py は1回しか実行されないreload を使っているこのgistのテストコードでは、patchデコレータの引数を変えた版を用意した(test2.py)。
また、テストメソッドにはmock後のオブジェクトが渡されるため、引数に m を受け取るようにした。
| import random | |
| fortunes = { | |
| 1: '凶', | |
| 2: '吉', | |
| 3: '大吉' | |
| } | |
| def main(): | |
| number = random.choice([1, 2, 3]) | |
| fortune = fortunes[number] | |
| return fortune | |
| if __name__ == '__main__': | |
| print(main()) |
| import unittest | |
| from unittest.mock import patch | |
| class FortuneTestCase(unittest.TestCase): | |
| @patch('random.choice', lambda x: 1) | |
| def test_bad(self): | |
| import fortune | |
| expect = ('凶') | |
| self.assertEqual(fortune.main(), expect) | |
| @patch('random.choice', lambda x: 3) | |
| def test_good(self): | |
| import fortune | |
| expect = ('大吉') | |
| self.assertEqual(fortune.main(), expect) |
| import unittest | |
| from unittest.mock import patch | |
| class FortuneTestCase(unittest.TestCase): | |
| @patch('random.choice', return_value=1) | |
| def test_bad(self, m): | |
| import fortune | |
| expect = ('凶') | |
| self.assertEqual(fortune.main(), expect) | |
| @patch('random.choice', return_value=3) | |
| def test_good(self, m): | |
| import fortune | |
| expect = ('大吉') | |
| self.assertEqual(fortune.main(), expect) |