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 stanzaPlugin from sleekxmpp.stanza.htmlim import HTMLIM stanzaPlugin(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 = """this is the plaintext message

This is the htmlim message

""" msg = self.m.Message() msg['to'] = "" msg['body'] = "this is the plaintext message" msg['type'] = 'chat' p = ET.Element('{}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)