Updated message stanza tests.

This commit is contained in:
Lance Stout 2010-07-20 01:55:44 -04:00
parent 9ca4bba2de
commit f505e229d6

View file

@ -1,18 +1,15 @@
import unittest from sleektest import *
from xml.etree import cElementTree as ET from sleekxmpp.stanza.message import Message
from sleekxmpp.stanza.htmlim import HTMLIM
class testmessagestanzas(unittest.TestCase): class TestMessageStanzas(SleekTest):
def setUp(self): def setUp(self):
import sleekxmpp.stanza.message as m registerStanzaPlugin(Message, HTMLIM)
from sleekxmpp.basexmpp import registerStanzaPlugin
from sleekxmpp.stanza.htmlim import HTMLIM
registerStanzaPlugin(m.Message, HTMLIM)
self.m = m
def testGroupchatReplyRegression(self): def testGroupchatReplyRegression(self):
"Regression groupchat reply should be to barejid" "Regression groupchat reply should be to barejid"
msg = self.m.Message() msg = self.Message()
msg['to'] = 'me@myserver.tld' msg['to'] = 'me@myserver.tld'
msg['from'] = 'room@someservice.someserver.tld/somenick' msg['from'] = 'room@someservice.someserver.tld/somenick'
msg['type'] = 'groupchat' msg['type'] = 'groupchat'
@ -22,23 +19,27 @@ class testmessagestanzas(unittest.TestCase):
def testAttribProperty(self): def testAttribProperty(self):
"Test attrib property returning self" "Test attrib property returning self"
msg = self.m.Message() msg = self.Message()
msg.attrib.attrib.attrib['to'] = 'usr@server.tld' msg.attrib.attrib.attrib['to'] = 'usr@server.tld'
self.failUnless(str(msg['to']) == 'usr@server.tld') self.failUnless(str(msg['to']) == 'usr@server.tld')
def testHTMLPlugin(self): def testHTMLPlugin(self):
"Test message/html/html stanza" "Test message/html/html stanza"
msgtxt = """<message to="fritzy@netflint.net/sleekxmpp" type="chat"><body>this is the plaintext message</body><html xmlns="http://jabber.org/protocol/xhtml-im"><body xmlns="http://www.w3.org/1999/xhtml"><p>This is the htmlim message</p></body></html></message>""" msg = self.Message()
msg = self.m.Message()
msg['to'] = "fritzy@netflint.net/sleekxmpp" msg['to'] = "fritzy@netflint.net/sleekxmpp"
msg['body'] = "this is the plaintext message" msg['body'] = "this is the plaintext message"
msg['type'] = 'chat' msg['type'] = 'chat'
p = ET.Element('{http://www.w3.org/1999/xhtml}p') p = ET.Element('{http://www.w3.org/1999/xhtml}p')
p.text = "This is the htmlim message" p.text = "This is the htmlim message"
msg['html']['html'] = p msg['html']['html'] = p
msg2 = self.m.Message() self.checkMessage(msg, """
values = msg.getValues() <message to="fritzy@netflint.net/sleekxmpp" type="chat">
msg2.setValues(values) <body>this is the plaintext message</body>
self.failUnless(msgtxt == str(msg) == str(msg2)) <html xmlns="http://jabber.org/protocol/xhtml-im">
<body xmlns="http://www.w3.org/1999/xhtml">
<p>This is the htmlim message</p>
</body>
</html>
</message>""")
suite = unittest.TestLoader().loadTestsFromTestCase(testmessagestanzas) suite = unittest.TestLoader().loadTestsFromTestCase(TestMessageStanzas)