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)