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)