diff options
| author | Lance Stout <lancestout@gmail.com> | 2010-10-26 23:47:17 -0400 |
|---|---|---|
| committer | Lance Stout <lancestout@gmail.com> | 2010-10-26 23:47:17 -0400 |
| commit | 65aa6573df3a8f298e7df96473014c19216971ef (patch) | |
| tree | d9a40d480071bd5e1983f3de41aa1860be1baa98 /tests/test_stream_roster.py | |
| parent | b888610525a441dce622bf6132a4b3cab76a26b2 (diff) | |
| download | slixmpp-65aa6573df3a8f298e7df96473014c19216971ef.tar.gz slixmpp-65aa6573df3a8f298e7df96473014c19216971ef.tar.bz2 slixmpp-65aa6573df3a8f298e7df96473014c19216971ef.tar.xz slixmpp-65aa6573df3a8f298e7df96473014c19216971ef.zip | |
First pass at integrating the new roster manager.
Diffstat (limited to 'tests/test_stream_roster.py')
| -rw-r--r-- | tests/test_stream_roster.py | 40 |
1 files changed, 17 insertions, 23 deletions
diff --git a/tests/test_stream_roster.py b/tests/test_stream_roster.py index 6eda7e3e..67da6e6c 100644 --- a/tests/test_stream_roster.py +++ b/tests/test_stream_roster.py @@ -13,8 +13,7 @@ class TestStreamRoster(SleekTest): def testGetRoster(self): """Test handling roster requests.""" - self.stream_start(mode='client') - self.failUnless(self.xmpp.roster == {}, "Initial roster not empty.") + self.stream_start(mode='client', jid='tester@localhost') # Since get_roster blocks, we need to run it in a thread. t = threading.Thread(name='get_roster', target=self.xmpp.get_roster) @@ -26,11 +25,12 @@ class TestStreamRoster(SleekTest): </iq> """) self.stream_recv(""" - <iq type="result" id="1"> + <iq to='tester@localhost' type="result" id="1"> <query xmlns="jabber:iq:roster"> <item jid="user@localhost" name="User" - subscription="both"> + subscription="from" + ask="subscribe"> <group>Friends</group> <group>Examples</group> </item> @@ -41,21 +41,20 @@ class TestStreamRoster(SleekTest): # 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) + print self.xmpp.rosters['tester@localhost']['user@localhost']._state + self.check_roster('tester@localhost', 'user@localhost', + name='User', + subscription='from', + afrom=True, + pending_out=True, + groups=['Friends', 'Examples']) def testRosterSet(self): """Test handling pushed roster updates.""" - self.stream_start(mode='client') - self.failUnless(self.xmpp.roster == {}, "Initial roster not empty.") + self.stream_start(mode='client', jid='tester@localhost') self.stream_recv(""" - <iq type="set" id="1"> + <iq to='tester@localhost' type="set" id="1"> <query xmlns="jabber:iq:roster"> <item jid="user@localhost" name="User" @@ -72,15 +71,10 @@ class TestStreamRoster(SleekTest): </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) - - + self.check_roster('tester@localhost', 'user@localhost', + name='User', + subscription='both', + groups=['Friends', 'Examples']) suite = unittest.TestLoader().loadTestsFromTestCase(TestStreamRoster) |