50 lines
1.3 KiB
Python
50 lines
1.3 KiB
Python
# Copyright (c) 2014, Menno Smits
|
|
# Released subject to the New BSD License
|
|
# Please see http://en.wikipedia.org/wiki/BSD_licenses
|
|
|
|
from __future__ import unicode_literals
|
|
|
|
from .imapclient import IMAPClient
|
|
|
|
try:
|
|
from unittest.mock import Mock
|
|
except ImportError:
|
|
try:
|
|
from mock import Mock
|
|
except ImportError:
|
|
raise ImportError(
|
|
"mock library could not be loaded. Please install Python 3.3 or newer "
|
|
"or install the `mock` third-party package through PyPi."
|
|
)
|
|
|
|
|
|
class TestableIMAPClient(IMAPClient):
|
|
"""Wrapper of :py:class:`imapclient.IMAPClient` that mocks all
|
|
interaction with real IMAP server.
|
|
|
|
This class should only be used in tests, where you can safely
|
|
interact with imapclient without running commands on a real
|
|
IMAP account.
|
|
"""
|
|
|
|
def __init__(self):
|
|
super(TestableIMAPClient, self).__init__("somehost")
|
|
|
|
def _create_IMAP4(self):
|
|
return MockIMAP4()
|
|
|
|
|
|
class MockIMAP4(Mock):
|
|
def __init__(self, *args, **kwargs):
|
|
super(Mock, self).__init__(*args, **kwargs)
|
|
self.use_uid = True
|
|
self.sent = b"" # Accumulates what was given to send()
|
|
self.tagged_commands = {}
|
|
self._starttls_done = False
|
|
|
|
def send(self, data):
|
|
self.sent += data
|
|
|
|
def _new_tag(self):
|
|
return "tag"
|