我正在嘗試在我的一個模型中測試一個函數,并嘗試使用mock.patch. 無論我嘗試什么,它似乎都沒有攔截該方法。要測試的模型:應用程序/模型.pyfrom django.db import modelsfrom .utils.storageutils import get_filefrom .utils.datautils import derive_dataClass DataThing(models.Model): #defined here def set_data_from_file(self): data = derive_data(get_file('filepath')) setattr(self, 'derived_data', data) self.save()應用程序/實用程序/datautils.pyimport pandas as pddef derive_data(data_from_file): df = pd.DataFrame('...') #do stuff with dataframe return df應用程序/測試/unit_tests/tests_models.pyfrom django.test import TestCaseimport mockfrom app.models import DataThingclass DataThingModelTest(TestCase): @classmethod def setUpTestData(cls): cls.datathing = DataThing @mock.patch('app.models.derive_data') def test_set_data_from_file(self, mocked_derive_data): mocked_derive_data.return_value=('pretend_dataframe') self.datathing.set_data_from_file() self.assertEquals(self.datathing.derived_data, 'pretend_dataframe')我希望這會通過。但是,我收到一個錯誤,因為datathing.set_data_from_file()最終仍在調用utils.storageutils.get_file. 我試過修補,app.utils.datautils.derive_data但有同樣的問題。
1 回答
慕蓋茨4494581
TA貢獻1850條經驗 獲得超11個贊
我還需要修補get_file功能
@mock.patch('app.models.get_file')
@mock.patch('app.models.derive_data')
def test_set_data_from_file(self, mocked_derive_data, mocked_ger_file):
mocked_get_file.return_value=('placeholder')
mocked_derive_data.return_value=('pretend_dataframe')
self.datathing.set_data_from_file()
self.assertEquals(self.datathing.derived_data, 'pretend_dataframe')
添加回答
舉報
0/150
提交
取消
