Я написал тест, как показано ниже:
class TestLoader(TestCase):
@pytest.fixture(autouse=True)
@patch('loaders.myloader.DSFactory')
def _initialize_(self, mock_ds_factory):
self.loader = MyLoader()
mock_ds = Mock()
mock_ds_factory.get_ds_for_env.return_value = mock_ds
self.loader.ds = mock_ds
def test_load(self):
self.loader.ds.read_file.return_value = json.dumps(self.get_data())
self.loader.load("test_s3_key") #####IN THIS LINE I AM GETTING ERROR AS MENTIONED BELOW##
@staticmethod
def get_data():
return {"key1":"value1","key2":"value2"}
Связанный источник находится здесь: loaders-> myloader.py. myloader.py выглядит следующим образом:
from common.ds_factory import DSFactory
class MyLoader:
def __init__(self):
self.ds = DSFactory.get_ds_for_env()
def load(self, file_key):
print(f"ds : {self.ds}")
print(f"file read is : {self.ds.read_file(S3_BUCKET, file_key)}"}
data_dict = json.loads(self.ds.read_file(S3_BUCKET, file_key))
Но во время тестирования я получаю сообщение об ошибке следующим образом:
ds is :<MagicMock name='DSFactory.get_ds_for_env()' id='140634163567528'>
file read is :<MagicMock name='DSFactory.get_ds_for_env().read_file()' id='140635257259568'>
E TypeError: the JSON object must be str, bytes or bytearray, not 'MagicMock'
Я не понимаю, почему даже после поддельного возвращаемого значения файла read_file с
self.loader.ds.read_file.return_value = json.dumps(self.get_data())
Я получаю MagickMock
объект. Я застрял, не понимаю, как решить эту проблему.