This repository has been archived by the owner on Jan 30, 2018. It is now read-only.
forked from fritzy/SleekXMPP
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
- Loading branch information
Showing
1 changed file
with
86 additions
and
0 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -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) |