import sleekxmpp from sleektest import * from sleekxmpp.stanza.presence import Presence class TestPresenceStanzas(SleekTest): def testPresenceShowRegression(self): """Regression check presence['type'] = 'dnd' show value working""" p = self.Presence() p['type'] = 'dnd' self.checkPresence(p, "dnd") def testPresenceType(self): """Test manipulating presence['type']""" p = self.Presence() p['type'] = 'available' self.checkPresence(p, "") self.failUnless(p['type'] == 'available', "Incorrect presence['type'] for type 'available'") for showtype in ['away', 'chat', 'dnd', 'xa']: p['type'] = showtype self.checkPresence(p, """ %s """ % showtype) self.failUnless(p['type'] == showtype, "Incorrect presence['type'] for type '%s'" % showtype) p['type'] = None self.checkPresence(p, "") def testPresenceUnsolicitedOffline(self): """ Unsolicted offline presence does not spawn changed_status or update the roster. """ p = self.Presence() p['type'] = 'unavailable' p['from'] = 'bill@chadmore.com/gmail15af' c = sleekxmpp.ClientXMPP('crap@wherever', 'password') happened = [] def handlechangedpresence(event): happened.append(True) c.add_event_handler("changed_status", handlechangedpresence) c._handlePresence(p) self.failUnless(happened == [], "changed_status event triggered for extra unavailable presence") self.failUnless(c.roster == {}, "Roster updated for superfulous unavailable presence") def testNickPlugin(self): """Test presence/nick/nick stanza.""" p = self.Presence() p['nick']['nick'] = 'A nickname!' self.checkPresence(p, """ A nickname! """) suite = unittest.TestLoader().loadTestsFromTestCase(TestPresenceStanzas)