diff options
Diffstat (limited to 'service/test/unit/maintenance/test_commands.py')
| -rw-r--r-- | service/test/unit/maintenance/test_commands.py | 24 |
1 files changed, 22 insertions, 2 deletions
diff --git a/service/test/unit/maintenance/test_commands.py b/service/test/unit/maintenance/test_commands.py index 52fe6ca2..a0fc58d6 100644 --- a/service/test/unit/maintenance/test_commands.py +++ b/service/test/unit/maintenance/test_commands.py @@ -80,8 +80,8 @@ class TestCommands(unittest.TestCase): def test_load_mails_adds_mails(self): # given mail_root = pkg_resources.resource_filename('test.unit.fixtures', 'mailset') - firstMailDeferred = defer.succeed(None) - secondMailDeferred = defer.succeed(None) + firstMailDeferred = defer.succeed(MagicMock()) + secondMailDeferred = defer.succeed(MagicMock()) self.mail_store.add_mail.side_effect = [firstMailDeferred, secondMailDeferred] self.mail_store.add_mailbox.return_value = defer.succeed(None) @@ -104,6 +104,26 @@ class TestCommands(unittest.TestCase): return d + def test_load_mails_supports_mbox(self): + # given + mbox_file = pkg_resources.resource_filename('test.unit.fixtures', 'mbox') + + d = load_mails(self.args, [mbox_file]) + + # then + def assert_mails_added(_): + self.assertTrue(self.mail_store.add_mail.called) + self.mail_store.add_mail.assert_any_call('INBOX', self._mail_content(mbox_file)) + + def error_callack(err): + print err + self.assertTrue(False) + + d.addCallback(assert_mails_added) + d.addErrback(error_callack) + + return d + def _mail_content(self, mail_file): with open(mail_file, 'r') as fp: m = email.message_from_file(fp) |