类型错误：例外必须是老式类或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()