def test_missingIMAPMessages(self):
"""
Make sure L{IMAP4DownloadProtocol.cbGotMessage} can deal with missing messages.
"""
class DummyResult(object):
def __init__(self):
self._values = []
def values(self):
return self._values
noResult = DummyResult()
missingKey = DummyResult()
missingKey.values().append({})
imap4 = IMAP4DownloadProtocol()
imap4.messageUIDs = []
imap4.fetchNextMessage = lambda: None
result = yield imap4.cbGotMessage(noResult, [])
self.assertTrue(result is None)
result = yield imap4.cbGotMessage(missingKey, [])
self.assertTrue(result is None)
def test_Twisted_16_6_bug(self):
"""
Make sure L{IMAP4DownloadProtocol.cbGotSearch} works.
There is a bug in Twisted 16.6 where the IMAP4Client._fetch method does not
accept a MessageSet argument - though it previously did.
We also need to double check that IMAP4Client._store works with a MessageSet.
"""
imap4 = IMAP4DownloadProtocol()
imap4.sendCommand = lambda cmd: succeed(cmd)
imap4._cbFetch = lambda result, requestedParts, structured: {
123: {
"UID": "456"
}
}
imap4.cbGotMessage = lambda results, messageList: succeed(True)
result = yield imap4.cbGotSearch((123, ))
self.assertTrue(result is None)
imap4 = IMAP4DownloadProtocol()
imap4.sendCommand = lambda cmd: succeed(cmd)
imap4._cbFetch = lambda result, requestedParts, structured: {
"123": {
"UID": "456"
}
}
imap4.cbMessageUnseen = lambda results, messageList: succeed(True)
ms = MessageSet()
ms.add(123)
result = yield imap4.cbFlagUnseen(ms)
self.assertTrue(result is None)
imap4 = IMAP4DownloadProtocol()
imap4.sendCommand = lambda cmd: succeed(cmd)
imap4._cbFetch = lambda result, requestedParts, structured: {
"123": {
"UID": "456"
}
}
imap4.cbMessageDeleted = lambda results, messageList: succeed(True)
ms = MessageSet()
ms.add(123)
result = yield imap4.cbFlagDeleted(ms)
self.assertTrue(result is None)
def test_deletion(self):
"""
Verify the IMAP protocol will delete messages only when the right
conditions are met. Either:
A) We've been told to delete all mail
B) We've not been told to delete all mail, but it was a message
we processed
"""
def stubFetchNextMessage():
pass
def stubCbFlagDeleted(result):
self.flagDeletedResult = result
return succeed(None)
proto = IMAP4DownloadProtocol()
self.patch(proto, "fetchNextMessage", stubFetchNextMessage)
self.patch(proto, "cbFlagDeleted", stubCbFlagDeleted)
results = {
"ignored": (
{
"RFC822": "a message"
}
)
}
# Delete all mail = False; action taken = submitted; result = deletion
proto.factory = StubFactory(MailReceiver.INJECTION_SUBMITTED, False)
self.flagDeletedResult = None
yield proto.cbGotMessage(results, "xyzzy")
self.assertEquals(self.flagDeletedResult, "xyzzy")
# Delete all mail = False; action taken = not submitted; result = no deletion
proto.factory = StubFactory(MailReceiver.NO_TOKEN, False)
self.flagDeletedResult = None
yield proto.cbGotMessage(results, "xyzzy")
self.assertEquals(self.flagDeletedResult, None)
# Delete all mail = True; action taken = submitted; result = deletion
proto.factory = StubFactory(MailReceiver.INJECTION_SUBMITTED, True)
self.flagDeletedResult = None
yield proto.cbGotMessage(results, "xyzzy")
self.assertEquals(self.flagDeletedResult, "xyzzy")
# Delete all mail = True; action taken = not submitted; result = deletion
proto.factory = StubFactory(MailReceiver.NO_TOKEN, True)
self.flagDeletedResult = None
yield proto.cbGotMessage(results, "xyzzy")
self.assertEquals(self.flagDeletedResult, "xyzzy")
def test_IMAP4Logger_err(self):
"""
Make sure L{IMAP4Logger.err} works when twisted.mail.imap4 calls log.err().
"""
class DummyTransport(object):
def __init__(self):
self.called = False
def loseConnection(self):
self.called = True
imap4 = IMAP4DownloadProtocol()
imap4.transport = DummyTransport()
imap4.state = "FOOBAR"
imap4.dispatchCommand("A001", "OK")
self.assertTrue(imap4.transport.called)