zev/mailbot
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

missing test + flake8

Browse Source
This commit is contained in:
Mathieu Agopian
2013-03-28 15:47:48 +01:00
parent dafc93198b
commit c254c9ee60
2 changed files with 15 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
mailbot/callback.py
View File

@@ -54,7 +54,7 @@ class Callback(object):
value = message[item] value = message[item]
# decode header (might be encoded as latin-1, utf-8... # decode header (might be encoded as latin-1, utf-8...
value = ' '.join(chunk.decode(encoding or 'ASCII') value = ' '.join(chunk.decode(encoding or 'ASCII')
for chunk, encoding in decode_header(value)) for chunk, encoding in decode_header(value))
for regexp in regexps: # store all captures for easy access for regexp in regexps: # store all captures for easy access
self.matches[item] += findall(regexp, value) self.matches[item] += findall(regexp, value)

14
mailbot/tests/test_mailbot.py
View File

@@ -123,6 +123,7 @@ class MailBotTest(MailBotClientTest):
self.bot.timeout = None # don't reset messages, no timeout! self.bot.timeout = None # don't reset messages, no timeout!
self.bot.reset_timeout_messages() self.bot.reset_timeout_messages()
self.assertFalse(self.bot.client.search.mock_calls) self.assertFalse(self.bot.client.search.mock_calls)
self.assertFalse(self.bot.client.fetch.mock_calls)
self.assertFalse(self.bot.client.remove_flags.mock_calls) self.assertFalse(self.bot.client.remove_flags.mock_calls)
def test_reset_timeout_messages_timeout(self): def test_reset_timeout_messages_timeout(self):
@@ -141,3 +142,16 @@ class MailBotTest(MailBotClientTest):
[sentinel.id1, sentinel.id2], ['INTERNALDATE']) [sentinel.id1, sentinel.id2], ['INTERNALDATE'])
self.bot.client.remove_flags.assert_called_once_with( self.bot.client.remove_flags.assert_called_once_with(
[sentinel.id1], ['\\Flagged', '\\Seen']) [sentinel.id1], ['\\Flagged', '\\Seen'])
def test_reset_timeout_messages_timeout_no_old_messages(self):
self.bot.timeout = 180 # 3 minutes ago
self.bot.client.search.return_value = [sentinel.id1, sentinel.id2]
self.bot.client.fetch.return_value = {
sentinel.id1: {'INTERNALDATE': datetime.utcnow(), 'SEQ': 1},
sentinel.id2: {'INTERNALDATE': datetime.utcnow(), 'SEQ': 2}}
self.bot.reset_timeout_messages()
self.bot.client.search.assert_called_once_with(['Flagged', 'Seen'])
self.assertFalse(self.bot.client.remove_flags.mock_calls)