def test_stanza_get_custom_payload(self):
stanza6 = Stanza(ElementTree.XML(STANZA6))
payload = stanza6.get_payload(TestPayload)
self.assertIsInstance(payload, TestPayload)
self.assertIsNone(payload.data) # pylint: disable=E1103
self.assertTrue(xml_elements_equal(ElementTree.XML(STANZA6)[0],
payload.as_xml()))
def test_stanza_get_custom_payload(self):
stanza6 = Stanza(ElementTree.XML(STANZA6))
payload = stanza6.get_payload(TestPayload)
self.assertIsInstance(payload, TestPayload)
self.assertIsNone(payload.data) # pylint: disable=E1103
self.assertTrue(
xml_elements_equal(ElementTree.XML(STANZA6)[0], payload.as_xml()))
def test_stanza_set_custom_payload(self):
stanza7 = Stanza("iq", from_jid = JID('[email protected]/d'),
to_jid = JID('[email protected]/h'), stanza_id = '666',
stanza_type='get')
payload = TestPayload(data = u"Test")
stanza7.set_payload(payload)
payload1 = stanza7.get_payload(TestPayload)
self.assertTrue(payload1 is payload)
self.assertTrue(xml_elements_equal(ElementTree.XML(STANZA7),
stanza7.as_xml(), True))
def test_stanza_set_custom_payload(self):
stanza7 = Stanza("iq",
from_jid=JID('[email protected]/d'),
to_jid=JID('[email protected]/h'),
stanza_id='666',
stanza_type='get')
payload = TestPayload(data=u"Test")
stanza7.set_payload(payload)
payload1 = stanza7.get_payload(TestPayload)
self.assertTrue(payload1 is payload)
self.assertTrue(
xml_elements_equal(ElementTree.XML(STANZA7), stanza7.as_xml(),
True))
