Add missing jmc.jabber.tests.message package
darcs-hash:20071112180923-86b55-49820f615136881c7a666a1480f173b50c52a942.gz
This commit is contained in:
David Rousselie
253
src/jmc/jabber/tests/message.py
Normal file
253
src/jmc/jabber/tests/message.py
Normal file
@@ -0,0 +1,253 @@
|
||||
##
|
||||
## message.py
|
||||
## Login : <dax@happycoders.org>
|
||||
## Started on Tue Nov 6 19:00:22 2007 David Rousselie
|
||||
## $Id$
|
||||
##
|
||||
## Copyright (C) 2007 David Rousselie
|
||||
## This program is free software; you can redistribute it and/or modify
|
||||
## it under the terms of the GNU General Public License as published by
|
||||
## the Free Software Foundation; either version 2 of the License, or
|
||||
## (at your option) any later version.
|
||||
##
|
||||
## This program is distributed in the hope that it will be useful,
|
||||
## but WITHOUT ANY WARRANTY; without even the implied warranty of
|
||||
## MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
||||
## GNU General Public License for more details.
|
||||
##
|
||||
## You should have received a copy of the GNU General Public License
|
||||
## along with this program; if not, write to the Free Software
|
||||
## Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
|
||||
##
|
||||
|
||||
import unittest
|
||||
import re
|
||||
|
||||
from pyxmpp.message import Message
|
||||
|
||||
from jcl.tests import JCLTestCase
|
||||
from jcl.model.account import Account, User
|
||||
|
||||
from jmc.model.account import SMTPAccount
|
||||
from jmc.jabber.message import SendMailMessageHandler, \
|
||||
RootSendMailMessageHandler
|
||||
from jmc.lang import Lang
|
||||
|
||||
class MockSMTPAccount(object):
|
||||
def __init__(self):
|
||||
self.email_sent = 0
|
||||
self.default_from = "user1@test.com"
|
||||
self.email = None
|
||||
|
||||
def send_email(self, email):
|
||||
self.email = email
|
||||
self.email_sent += 1
|
||||
|
||||
def create_email(self, from_addr, to_addr, subject, body,
|
||||
other_headers=None):
|
||||
return (from_addr, to_addr, subject, body, other_headers)
|
||||
|
||||
class SendMailMessageHandler_TestCase(unittest.TestCase):
|
||||
def setUp(self):
|
||||
self.handler = SendMailMessageHandler(None)
|
||||
|
||||
def test_get_email_to_headers_from_message(self):
|
||||
to_regexp = re.compile("^\s*(?i)to\s*:\s*(?P<to_email>.*)")
|
||||
(message, to_header) = self.handler.get_email_headers_from_message(\
|
||||
"To: dest@test.com\ntest body\n", [to_regexp], ["to_email"])
|
||||
self.assertEquals(message, "test body\n")
|
||||
self.assertEquals(to_header, ["dest@test.com"])
|
||||
|
||||
def test_get_email_headers_from_message(self):
|
||||
to_regexp = re.compile("^\s*(?i)to\s*:\s*(?P<to_email>.*)")
|
||||
cc_regexp = re.compile("^\s*(?i)cc\s*:\s*(?P<cc_email>.*)")
|
||||
bcc_regexp = re.compile("^\s*(?i)bcc\s*:\s*(?P<bcc_email>.*)")
|
||||
subject_regexp = re.compile("^\s*(?i)subject\s*:\s*(?P<subject_email>.*)")
|
||||
(message, headers) = self.handler.get_email_headers_from_message(\
|
||||
"To: dest@test.com\nCc: cc@test.com\n"
|
||||
+ "Bcc: bcc@test.com\n"
|
||||
+ "Subject: test subject\ntest body\n",
|
||||
[to_regexp, cc_regexp, bcc_regexp, subject_regexp],
|
||||
["to_email", "cc_email", "bcc_email", "subject_email"])
|
||||
self.assertEquals(message, "test body\n")
|
||||
self.assertEquals(headers, ["dest@test.com",
|
||||
"cc@test.com",
|
||||
"bcc@test.com",
|
||||
"test subject"])
|
||||
|
||||
def test_get_email_headers_from_message_unordered(self):
|
||||
to_regexp = re.compile("^\s*(?i)to\s*:\s*(?P<to_email>.*)")
|
||||
cc_regexp = re.compile("^\s*(?i)cc\s*:\s*(?P<cc_email>.*)")
|
||||
bcc_regexp = re.compile("^\s*(?i)bcc\s*:\s*(?P<bcc_email>.*)")
|
||||
subject_regexp = re.compile("^\s*(?i)subject\s*:\s*(?P<subject_email>.*)")
|
||||
(message, headers) = self.handler.get_email_headers_from_message(\
|
||||
"To: dest@test.com\nCc: cc@test.com\n"
|
||||
+ "Bcc: bcc@test.com\n"
|
||||
+ "Subject: test subject\ntest body\n",
|
||||
[cc_regexp, to_regexp, subject_regexp, bcc_regexp],
|
||||
["cc_email", "to_email", "subject_email", "bcc_email"])
|
||||
self.assertEquals(message, "test body\n")
|
||||
self.assertEquals(headers, ["cc@test.com",
|
||||
"dest@test.com",
|
||||
"test subject",
|
||||
"bcc@test.com"])
|
||||
|
||||
def test_handle(self):
|
||||
mock_account = MockSMTPAccount()
|
||||
message = Message(from_jid="user1@test.com",
|
||||
to_jid="real_dest%test.com@jmc.test.com",
|
||||
subject="real subject",
|
||||
body="To: dest@test.com\nCc: cc@test.com\n" \
|
||||
+ "Bcc: bcc@test.com\n" \
|
||||
+ "Subject: test subject\ntest body\n")
|
||||
result = self.handler.handle(\
|
||||
message, Lang.en, [mock_account])
|
||||
self.assertEquals(len(result), 1)
|
||||
self.assertEquals(result[0].stanza_type, "message")
|
||||
self.assertEquals(result[0].get_from(), "real_dest%test.com@jmc.test.com")
|
||||
self.assertEquals(result[0].get_to(), "user1@test.com")
|
||||
self.assertEquals(result[0].get_subject(),
|
||||
Lang.en.send_mail_ok_subject)
|
||||
self.assertEquals(result[0].get_body(),
|
||||
Lang.en.send_mail_ok_body % \
|
||||
("real_dest@test.com, dest@test.com"))
|
||||
self.assertEquals(mock_account.email_sent, 1)
|
||||
self.assertEquals(mock_account.email[0], "user1@test.com")
|
||||
self.assertEquals(mock_account.email[1], "real_dest@test.com, dest@test.com")
|
||||
self.assertEquals(mock_account.email[2], "real subject")
|
||||
self.assertEquals(mock_account.email[3], "test body\n")
|
||||
self.assertEquals(mock_account.email[4], {u"Bcc": "bcc@test.com",
|
||||
u"Cc": "cc@test.com"})
|
||||
|
||||
class RootSendMailMessageHandler_TestCase(JCLTestCase):
|
||||
def setUp(self):
|
||||
JCLTestCase.setUp(self, tables=[Account, SMTPAccount, User])
|
||||
self.handler = RootSendMailMessageHandler(None)
|
||||
|
||||
def test_filter(self):
|
||||
user1 = User(jid="user1@test.com")
|
||||
account11 = SMTPAccount(user=user1,
|
||||
name="account11",
|
||||
jid="account11@jmc.test.com")
|
||||
account11.default_account = True
|
||||
account12 = SMTPAccount(user=user1,
|
||||
name="account12",
|
||||
jid="account12@jmc.test.com")
|
||||
message = Message(from_jid="user1@test.com",
|
||||
to_jid="account11@jmc.test.com",
|
||||
body="message")
|
||||
accounts = self.handler.filter(message, None)
|
||||
self.assertEquals(accounts.count(), 1)
|
||||
|
||||
def test_filter_no_default_account(self):
|
||||
user1 = User(jid="user1@test.com")
|
||||
account11 = SMTPAccount(user=user1,
|
||||
name="account11",
|
||||
jid="account11@jmc.test.com")
|
||||
account12 = SMTPAccount(user=user1,
|
||||
name="account12",
|
||||
jid="account12@jmc.test.com")
|
||||
message = Message(from_jid="user1@test.com",
|
||||
to_jid="account11@jmc.test.com",
|
||||
body="message")
|
||||
accounts = self.handler.filter(message, None)
|
||||
self.assertEquals(accounts.count(), 2)
|
||||
self.assertEquals(accounts[0].name, "account11")
|
||||
|
||||
def test_filter_wrong_dest(self):
|
||||
user1 = User(jid="user1@test.com")
|
||||
account11 = SMTPAccount(user=user1,
|
||||
name="account11",
|
||||
jid="account11@jmc.test.com")
|
||||
account12 = SMTPAccount(user=user1,
|
||||
name="account12",
|
||||
jid="account12@jmc.test.com")
|
||||
message = Message(from_jid="user1@test.com",
|
||||
to_jid="user2%test.com@jmc.test.com",
|
||||
body="message")
|
||||
accounts = self.handler.filter(message, None)
|
||||
self.assertEquals(accounts.count(), 2)
|
||||
|
||||
def test_filter_wrong_user(self):
|
||||
user1 = User(jid="user1@test.com")
|
||||
account11 = SMTPAccount(user=user1,
|
||||
name="account11",
|
||||
jid="account11@jmc.test.com")
|
||||
account12 = SMTPAccount(user=user1,
|
||||
name="account12",
|
||||
jid="account12@jmc.test.com")
|
||||
message = Message(from_jid="user2@test.com",
|
||||
to_jid="account11@jmc.test.com",
|
||||
body="message")
|
||||
accounts = self.handler.filter(message, None)
|
||||
self.assertEquals(accounts.count(), 0)
|
||||
|
||||
def test_handle(self):
|
||||
mock_account = MockSMTPAccount()
|
||||
message = Message(from_jid="user1@test.com",
|
||||
to_jid="jmc.test.com",
|
||||
subject="real subject",
|
||||
body="To: dest@test.com\nCc: cc@test.com\n" \
|
||||
+ "Bcc: bcc@test.com\n" \
|
||||
+ "Subject: test subject\ntest body\n")
|
||||
result = self.handler.handle(\
|
||||
message, Lang.en, [mock_account])
|
||||
self.assertEquals(len(result), 1)
|
||||
self.assertEquals(result[0].get_type(), None)
|
||||
self.assertEquals(result[0].get_from(), "jmc.test.com")
|
||||
self.assertEquals(result[0].get_to(), "user1@test.com")
|
||||
self.assertEquals(result[0].get_subject(),
|
||||
Lang.en.send_mail_ok_subject)
|
||||
self.assertEquals(result[0].get_body(),
|
||||
Lang.en.send_mail_ok_body % ("dest@test.com"))
|
||||
self.assertEquals(mock_account.email_sent, 1)
|
||||
self.assertEquals(mock_account.email[0], "user1@test.com")
|
||||
self.assertEquals(mock_account.email[1], "dest@test.com")
|
||||
self.assertEquals(mock_account.email[2], "real subject")
|
||||
self.assertEquals(mock_account.email[3], "test body\n")
|
||||
self.assertEquals(mock_account.email[4], {u"Bcc": "bcc@test.com",
|
||||
u"Cc": "cc@test.com"})
|
||||
|
||||
def test_handle_email_not_found_in_header(self):
|
||||
message = Message(from_jid="user1@test.com",
|
||||
to_jid="jmc.test.com",
|
||||
subject="message subject",
|
||||
body="message body")
|
||||
accounts = [MockSMTPAccount()]
|
||||
result = self.handler.handle(message, Lang.en, accounts)
|
||||
self.assertEquals(len(result), 1)
|
||||
self.assertEquals(result[0].get_type(), "error")
|
||||
self.assertEquals(result[0].get_from(), "jmc.test.com")
|
||||
self.assertEquals(result[0].get_to(), "user1@test.com")
|
||||
self.assertEquals(result[0].get_subject(),
|
||||
Lang.en.send_mail_error_no_to_header_subject)
|
||||
self.assertEquals(result[0].get_body(),
|
||||
Lang.en.send_mail_error_no_to_header_body)
|
||||
|
||||
def test_handle_no_jabber_subject(self):
|
||||
mock_account = MockSMTPAccount()
|
||||
message = Message(from_jid="user1@test.com",
|
||||
to_jid="jmc.test.com",
|
||||
subject="",
|
||||
body="To: dest@test.com\nCc: cc@test.com\n" \
|
||||
+ "Bcc: bcc@test.com\n" \
|
||||
+ "Subject: test subject\ntest body\n")
|
||||
message_to_send = self.handler.handle(\
|
||||
message, Lang.en, [mock_account])
|
||||
self.assertNotEquals(message_to_send, None)
|
||||
self.assertEquals(mock_account.email_sent, 1)
|
||||
self.assertEquals(mock_account.email[0], "user1@test.com")
|
||||
self.assertEquals(mock_account.email[1], "dest@test.com")
|
||||
self.assertEquals(mock_account.email[2], "test subject")
|
||||
self.assertEquals(mock_account.email[3], "test body\n")
|
||||
self.assertEquals(mock_account.email[4], {u"Bcc": "bcc@test.com",
|
||||
u"Cc": "cc@test.com"})
|
||||
|
||||
def suite():
|
||||
suite = unittest.TestSuite()
|
||||
suite.addTest(unittest.makeSuite(SendMailMessageHandler_TestCase, 'test'))
|
||||
suite.addTest(unittest.makeSuite(RootSendMailMessageHandler_TestCase, 'test'))
|
||||
return suite
|
||||
|
||||
if __name__ == '__main__':
|
||||
unittest.main(defaultTest='suite')
|
||||
Reference in New Issue
Block a user