類型錯誤：例外必須是老式類或BaseException衍生，不MagicMock

Question

我要測試的異常處理在我的代碼：

def test_get_mails__exception_in_search(self): 
    with mock.patch('imaplib.IMAP4', autospec=True) as imap_mock: 
     imap_mock.return_value.create.return_value = ('OK', ['']) 
     imap_mock.return_value.search.side_effect=imap_mock.error 
     self.get_mails() 

但模擬哭喊：

TypeError: exceptions must be old-style classes or derived from BaseException, 
not MagicMock 

如何測試我的代碼：我想imaplib.search提高imaplib.error

Answer 1

我找到了一個解決方案：

with mock.patch('imaplib.IMAP4', autospec=True) as imap_mock: 
     imap_mock.error=Exception # <---- This is needed, don't ask me why 
     imap_mock.return_value.create.return_value = ('OK', ['']) 
     imap_mock.return_value.search.side_effect=imap_mock.error 
     self.get_mails()