44 lines
1.6 KiB
Python
44 lines
1.6 KiB
Python
import unittest
|
|
from xml.etree import cElementTree as ET
|
|
|
|
class testmessagestanzas(unittest.TestCase):
|
|
|
|
def setUp(self):
|
|
import sleekxmpp.stanza.message as m
|
|
from sleekxmpp.basexmpp import registerStanzaPlugin
|
|
from sleekxmpp.stanza.htmlim import HTMLIM
|
|
registerStanzaPlugin(m.Message, HTMLIM)
|
|
self.m = m
|
|
|
|
def testGroupchatReplyRegression(self):
|
|
"Regression groupchat reply should be to barejid"
|
|
msg = self.m.Message()
|
|
msg['to'] = 'me@myserver.tld'
|
|
msg['from'] = 'room@someservice.someserver.tld/somenick'
|
|
msg['type'] = 'groupchat'
|
|
msg['body'] = "this is a message"
|
|
msg.reply()
|
|
self.failUnless(str(msg['to']) == 'room@someservice.someserver.tld')
|
|
|
|
def testAttribProperty(self):
|
|
"Test attrib property returning self"
|
|
msg = self.m.Message()
|
|
msg.attrib.attrib.attrib['to'] = 'usr@server.tld'
|
|
self.failUnless(str(msg['to']) == 'usr@server.tld')
|
|
|
|
def testHTMLPlugin(self):
|
|
"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.m.Message()
|
|
msg['to'] = "fritzy@netflint.net/sleekxmpp"
|
|
msg['body'] = "this is the plaintext message"
|
|
msg['type'] = 'chat'
|
|
p = ET.Element('{http://www.w3.org/1999/xhtml}p')
|
|
p.text = "This is the htmlim message"
|
|
msg['html']['html'] = p
|
|
msg2 = self.m.Message()
|
|
values = msg.getValues()
|
|
msg2.setValues(values)
|
|
self.failUnless(msgtxt == str(msg) == str(msg2))
|
|
|
|
suite = unittest.TestLoader().loadTestsFromTestCase(testmessagestanzas)
|