Permalink
Browse files

Added stream tests for rosters.

  • Loading branch information...
1 parent d9aff3d commit d5288847234110cc9116aaf91a26c5a1fc395202 @legastero legastero committed Oct 24, 2010
Showing with 86 additions and 0 deletions.
  1. +86 −0 tests/test_stream_roster.py
View
86 tests/test_stream_roster.py
@@ -0,0 +1,86 @@
+from sleekxmpp.test import *
+import time
+import threading
+
+
+class TestStreamRoster(SleekTest):
+ """
+ Test handling roster updates.
+ """
+
+ def tearDown(self):
+ self.stream_close()
+
+ def testGetRoster(self):
+ """Test handling roster requests."""
+ self.stream_start(mode='client')
+ self.failUnless(self.xmpp.roster == {}, "Initial roster not empty.")
+
+ # Since get_roster blocks, we need to run it in a thread.
+ t = threading.Thread(name='get_roster', target=self.xmpp.get_roster)
+ t.start()
+
+ self.stream_send_iq("""
+ <iq type="get" id="1">
+ <query xmlns="jabber:iq:roster" />
+ </iq>
+ """)
+ self.stream_recv("""
+ <iq type="result" id="1">
+ <query xmlns="jabber:iq:roster">
+ <item jid="user@localhost"
+ name="User"
+ subscription="both">
+ <group>Friends</group>
+ <group>Examples</group>
+ </item>
+ </query>
+ </iq>
+ """)
+
+ # Wait for get_roster to return.
+ t.join()
+
+ roster = {'user@localhost': {'name': 'User',
+ 'subscription': 'both',
+ 'groups': ['Friends', 'Examples'],
+ 'presence': {},
+ 'in_roster': True}}
+ self.failUnless(self.xmpp.roster == roster,
+ "Unexpected roster values: %s" % self.xmpp.roster)
+
+ def testRosterSet(self):
+ """Test handling pushed roster updates."""
+ self.stream_start(mode='client')
+ self.failUnless(self.xmpp.roster == {}, "Initial roster not empty.")
+
+ self.stream_recv("""
+ <iq type="set" id="1">
+ <query xmlns="jabber:iq:roster">
+ <item jid="user@localhost"
+ name="User"
+ subscription="both">
+ <group>Friends</group>
+ <group>Examples</group>
+ </item>
+ </query>
+ </iq>
+ """)
+ self.stream_send_iq("""
+ <iq type="result" id="1">
+ <query xmlns="jabber:iq:roster" />
+ </iq>
+ """)
+
+ roster = {'user@localhost': {'name': 'User',
+ 'subscription': 'both',
+ 'groups': ['Friends', 'Examples'],
+ 'presence': {},
+ 'in_roster': True}}
+ self.failUnless(self.xmpp.roster == roster,
+ "Unexpected roster values: %s" % self.xmpp.roster)
+
+
+
+
+suite = unittest.TestLoader().loadTestsFromTestCase(TestStreamRoster)

0 comments on commit d528884

Please sign in to comment.