概要
- 単体テストを書く際に、Privateメソッドの呼び出しに苦労したので方法を書き下しておく。
ケース１（通常呼び出し）
テスト対象ファイル
function.py
class Service:
def __private(self, name):
return name
テストファイル
test.py
from function import Service
def test_private():
name = "Smith"
s = Service()
r = s._Service__private(name)
assert r == "Smith"
[クラスのインスタンス]._[クラス名]__[Privateメソッド名]
ケース２（Mockする場合）
テスト対象ファイル
function.py
class Service:
def __private(self):
return 1
def no_private(self):
return self.__private()
テストファイル
test.py
from function import Service
from unittest.mock import patch
def test_mock_private(method):
s = Service()
with patch.object(s, '_Service__private', return_value=3) as method:
s.no_private()
method.assert_called_once_with()
もしくは
test.py
from function import Service
from unittest.mock import patch
@patch.object(Service, "_Service__private", return_value=3)
def test_mock_private(method):
s = Service()
s.no_private()
method.assert_called_once_with()
ケース３（複数のメソッドをMockする場合）
テスト対象ファイル
function.py
class Service:
def __private_1(self):
return 1
def __private_2(self):
return 2
def no_private(self):
self.__private_1()
self.__private_2()
test.py
from function import Service
from unittest.mock import patch
@patch.object(Service, "_Service__private_1", return_value=3)
@patch.object(Service, "_Service__private_2", return_value=3)
def test_mock_private(method1, method2):
s = Service()
s.no_private()
method.assert_called_once_with()
ケース４（テーブル定義をモックする場合）
example.py
def test_add_new_table():
td = Mock(TableDefinition)
td.table_name = "table_name"
patch.object(LocalDbService, "_Service__get_all_table_definitions", return_value=td)