def test_build_empty(self): item = RosterItem(JID("*****@*****.**")) self.assertEqual(item.jid, JID("*****@*****.**")) self.assertIsNone(item.name) self.assertIsNone(item.subscription) self.assertIsNone(item.ask) self.assertFalse(item.approved) self.assertEqual(item.groups, set()) xml = item.as_xml() self.assertEqual(xml.tag, "{jabber:iq:roster}item") self.assertEqual(len(xml), 0) self.assertEqual(xml.get("jid"), u"*****@*****.**") self.assertEqual(xml.get("name"), None) self.assertEqual(xml.get("subscription"), None) self.assertEqual(xml.get("ask"), None) self.assertEqual(xml.get("approved"), None) # check if serializable self.assertTrue(ElementTree.tostring(xml))
def test_build_full(self): item = RosterItem(JID("*****@*****.**"), "NAME", ["G1", "G2"], "from", "subscribe", "true") self.assertEqual(item.jid, JID("*****@*****.**")) self.assertEqual(item.name, "NAME") self.assertEqual(item.subscription, "from") self.assertEqual(item.ask, "subscribe") self.assertTrue(item.approved) self.assertEqual(item.groups, set(["G1", "G2"])) xml = item.as_xml() self.assertEqual(xml.tag, "{jabber:iq:roster}item") self.assertEqual(len(xml), 2) self.assertEqual(xml.get("jid"), u"*****@*****.**") self.assertEqual(xml.get("name"), "NAME") self.assertEqual(xml.get("subscription"), "from") self.assertEqual(xml.get("ask"), "subscribe") self.assertEqual(xml.get("approved"), "true") self.assertEqual(xml[0].tag, "{jabber:iq:roster}group") self.assertEqual(xml[1].tag, "{jabber:iq:roster}group") self.assertEqual(set([xml[0].text, xml[1].text]), set(["G1", "G2"])) # check if serializable self.assertTrue(ElementTree.tostring(xml))