From 70c770635199cfa473608162ec7d31e030a11c5f Mon Sep 17 00:00:00 2001 From: Folker Bernitt Date: Wed, 28 Oct 2015 12:05:59 +0100 Subject: Add markov-generate to load-mails - Allows to generat mails based on a sample mails - use it with: pixelated-maintenance markov-generate --seed 21 --limit 10 --- service/test/unit/maintenance/test_commands.py | 24 ++++++++++++++++++++++-- 1 file changed, 22 insertions(+), 2 deletions(-) (limited to 'service/test/unit/maintenance') 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) -- cgit v1.2.3