summaryrefslogtreecommitdiff
path: root/tests/test_messagestanzas.py
blob: 08488ce3cc8582c5f982624661abe249e12d0881 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
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 = """<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)