1 回答

TA貢獻1868條經驗 獲得超4個贊
如果要保存已打補丁的對象,則必須手動啟動和停止打補丁。如果您使用上下文管理器(例如with patch),補丁將在退出范圍時恢復。
您可以執行以下操作:
class TestSomething(unittest.TestCase):
def setUp(self):
self.patcher = patch('requests.get') # returns the patcher object
self.request = self.patcher.start() # returns the patched object
self.request.session = MockSession()
self.request.cookies = {}
def tearDown(self):
sef.patcher.stop()
def test_03_set_nonce(self):
self.web_session.set_nonce(self.request)
self.assertTrue(len(self.request.cookies) > 0, 'set_nonce failed.')
請注意,我沒有檢查實際測試——這取決于您的應用程序邏輯。
您也可以直接在測試中進行修補:
@patch('requests.get')
def test_03_set_nonce(self, request):
request.session = MockSession()
request.cookies = {}
self.web_session.set_nonce(request)
self.assertTrue(len(request.cookies) > 0, 'set_nonce failed.')
添加回答
舉報