def worker_with_fake_imap(): from tinymail.imap_worker import ImapWorker, ConnectionErrorWrapper worker = ImapWorker() imap_conn = Mock(spec=imaplib.IMAP4) worker.conn = ConnectionErrorWrapper(imap_conn) return worker, imap_conn