279 lines
9.5 KiB
Python
279 lines
9.5 KiB
Python
import sys
|
|
import time
|
|
import threading
|
|
|
|
from sleekxmpp.test import *
|
|
|
|
|
|
class TestStreamPubsub(SleekTest):
|
|
|
|
"""
|
|
Test using the XEP-0030 plugin.
|
|
"""
|
|
|
|
def setUp(self):
|
|
self.stream_start()
|
|
|
|
def tearDown(self):
|
|
self.stream_close()
|
|
|
|
def testCreateInstantNode(self):
|
|
"""Test creating an instant node"""
|
|
t = threading.Thread(name='create_node',
|
|
target=self.xmpp['xep_0060'].create_node,
|
|
args=('pubsub.example.com', None))
|
|
t.start()
|
|
|
|
self.send("""
|
|
<iq type="set" id="1" to="pubsub.example.com">
|
|
<pubsub xmlns="http://jabber.org/protocol/pubsub">
|
|
<create />
|
|
</pubsub>
|
|
</iq>
|
|
""")
|
|
|
|
self.recv("""
|
|
<iq type="result" id="1"
|
|
to="tester@localhost" from="pubsub.example.com">
|
|
<pubsub xmlns="http://jabber.org/protocol/pubsub">
|
|
<create node="25e3d37dabbab9541f7523321421edc5bfeb2dae" />
|
|
</pubsub>
|
|
</iq>
|
|
""")
|
|
|
|
t.join()
|
|
|
|
def testCreateNodeNoConfig(self):
|
|
"""Test creating a node without a config"""
|
|
t = threading.Thread(name='create_node',
|
|
target=self.xmpp['xep_0060'].create_node,
|
|
args=('pubsub.example.com', 'princely_musings'))
|
|
t.start()
|
|
|
|
self.send("""
|
|
<iq type="set" id="1" to="pubsub.example.com">
|
|
<pubsub xmlns="http://jabber.org/protocol/pubsub">
|
|
<create node="princely_musings" />
|
|
</pubsub>
|
|
</iq>
|
|
""")
|
|
|
|
self.recv("""
|
|
<iq type="result" id="1"
|
|
to="tester@localhost" from="pubsub.example.com" />
|
|
""")
|
|
|
|
t.join()
|
|
|
|
def testCreateNodeConfig(self):
|
|
"""Test creating a node with a config"""
|
|
form = self.xmpp['xep_0004'].stanza.Form()
|
|
form['type'] = 'submit'
|
|
form.add_field(var='pubsub#access_model', value='whitelist')
|
|
|
|
t = threading.Thread(name='create_node',
|
|
target=self.xmpp['xep_0060'].create_node,
|
|
args=('pubsub.example.com', 'princely_musings'),
|
|
kwargs={'config': form})
|
|
t.start()
|
|
|
|
self.send("""
|
|
<iq type="set" id="1" to="pubsub.example.com">
|
|
<pubsub xmlns="http://jabber.org/protocol/pubsub">
|
|
<create node="princely_musings" />
|
|
<configure>
|
|
<x xmlns="jabber:x:data" type="submit">
|
|
<field var="pubsub#access_model">
|
|
<value>whitelist</value>
|
|
</field>
|
|
<field var="FORM_TYPE">
|
|
<value>http://jabber.org/protocol/pubsub#node_config</value>
|
|
</field>
|
|
</x>
|
|
</configure>
|
|
</pubsub>
|
|
</iq>
|
|
""")
|
|
|
|
self.recv("""
|
|
<iq type="result" id="1"
|
|
to="tester@localhost" from="pubsub.example.com" />
|
|
""")
|
|
|
|
t.join()
|
|
|
|
def testDeleteNode(self):
|
|
"""Test deleting a node"""
|
|
t = threading.Thread(name='delete_node',
|
|
target=self.xmpp['xep_0060'].delete_node,
|
|
args=('pubsub.example.com', 'some_node'))
|
|
t.start()
|
|
|
|
self.send("""
|
|
<iq type="set" to="pubsub.example.com" id="1">
|
|
<pubsub xmlns="http://jabber.org/protocol/pubsub#owner">
|
|
<delete node="some_node" />
|
|
</pubsub>
|
|
</iq>
|
|
""")
|
|
|
|
self.recv("""
|
|
<iq type="result" id="1"
|
|
to="tester@localhost" from="pubsub.example.com" />
|
|
""")
|
|
|
|
t.join()
|
|
|
|
def testSubscribe(self):
|
|
"""Test subscribing to a node"""
|
|
|
|
def run_test(jid, bare, ifrom, send, recv):
|
|
t = threading.Thread(name='subscribe',
|
|
target=self.xmpp['xep_0060'].subscribe,
|
|
args=('pubsub.example.com', 'some_node'),
|
|
kwargs={'subscribee': jid,
|
|
'bare': bare,
|
|
'ifrom': ifrom})
|
|
t.start()
|
|
self.send(send)
|
|
self.recv(recv)
|
|
t.join()
|
|
|
|
# Case 1: No subscribee, default 'from' JID, bare JID
|
|
run_test(None, True, None,
|
|
"""
|
|
<iq type="set" id="1" to="pubsub.example.com">
|
|
<pubsub xmlns="http://jabber.org/protocol/pubsub">
|
|
<subscribe node="some_node" jid="tester@localhost" />
|
|
</pubsub>
|
|
</iq>
|
|
""",
|
|
"""
|
|
<iq type="result" id="1"
|
|
to="tester@localhost" from="pubsub.example.com" />
|
|
""")
|
|
|
|
# Case 2: No subscribee, given 'from' JID, bare JID
|
|
run_test(None, True, 'foo@comp.example.com/bar',
|
|
"""
|
|
<iq type="set" id="2"
|
|
to="pubsub.example.com" from="foo@comp.example.com/bar">
|
|
<pubsub xmlns="http://jabber.org/protocol/pubsub">
|
|
<subscribe node="some_node" jid="foo@comp.example.com" />
|
|
</pubsub>
|
|
</iq>
|
|
""",
|
|
"""
|
|
<iq type="result" id="2"
|
|
to="foo@comp.example.com/bar" from="pubsub.example.com" />
|
|
""")
|
|
|
|
# Case 3: No subscribee, given 'from' JID, full JID
|
|
run_test(None, False, 'foo@comp.example.com/bar',
|
|
"""
|
|
<iq type="set" id="3"
|
|
to="pubsub.example.com" from="foo@comp.example.com/bar">
|
|
<pubsub xmlns="http://jabber.org/protocol/pubsub">
|
|
<subscribe node="some_node" jid="foo@comp.example.com/bar" />
|
|
</pubsub>
|
|
</iq>
|
|
""",
|
|
"""
|
|
<iq type="result" id="3"
|
|
to="foo@comp.example.com/bar" from="pubsub.example.com" />
|
|
""")
|
|
|
|
# Case 4: Subscribee
|
|
run_test('user@example.com/foo', True, 'foo@comp.example.com/bar',
|
|
"""
|
|
<iq type="set" id="4"
|
|
to="pubsub.example.com" from="foo@comp.example.com/bar">
|
|
<pubsub xmlns="http://jabber.org/protocol/pubsub">
|
|
<subscribe node="some_node" jid="user@example.com/foo" />
|
|
</pubsub>
|
|
</iq>
|
|
""",
|
|
"""
|
|
<iq type="result" id="4"
|
|
to="foo@comp.example.com/bar" from="pubsub.example.com" />
|
|
""")
|
|
|
|
def testSubscribeWithOptions(self):
|
|
pass
|
|
|
|
def testUnubscribe(self):
|
|
"""Test unsubscribing from a node"""
|
|
|
|
def run_test(jid, bare, ifrom, send, recv):
|
|
t = threading.Thread(name='unsubscribe',
|
|
target=self.xmpp['xep_0060'].unsubscribe,
|
|
args=('pubsub.example.com', 'some_node'),
|
|
kwargs={'subscribee': jid,
|
|
'bare': bare,
|
|
'ifrom': ifrom})
|
|
t.start()
|
|
self.send(send)
|
|
self.recv(recv)
|
|
t.join()
|
|
|
|
# Case 1: No subscribee, default 'from' JID, bare JID
|
|
run_test(None, True, None,
|
|
"""
|
|
<iq type="set" id="1" to="pubsub.example.com">
|
|
<pubsub xmlns="http://jabber.org/protocol/pubsub">
|
|
<unsubscribe node="some_node" jid="tester@localhost" />
|
|
</pubsub>
|
|
</iq>
|
|
""",
|
|
"""
|
|
<iq type="result" id="1"
|
|
to="tester@localhost" from="pubsub.example.com" />
|
|
""")
|
|
|
|
# Case 2: No subscribee, given 'from' JID, bare JID
|
|
run_test(None, True, 'foo@comp.example.com/bar',
|
|
"""
|
|
<iq type="set" id="2"
|
|
to="pubsub.example.com" from="foo@comp.example.com/bar">
|
|
<pubsub xmlns="http://jabber.org/protocol/pubsub">
|
|
<unsubscribe node="some_node" jid="foo@comp.example.com" />
|
|
</pubsub>
|
|
</iq>
|
|
""",
|
|
"""
|
|
<iq type="result" id="2"
|
|
to="foo@comp.example.com/bar" from="pubsub.example.com" />
|
|
""")
|
|
|
|
# Case 3: No subscribee, given 'from' JID, full JID
|
|
run_test(None, False, 'foo@comp.example.com/bar',
|
|
"""
|
|
<iq type="set" id="3"
|
|
to="pubsub.example.com" from="foo@comp.example.com/bar">
|
|
<pubsub xmlns="http://jabber.org/protocol/pubsub">
|
|
<unsubscribe node="some_node" jid="foo@comp.example.com/bar" />
|
|
</pubsub>
|
|
</iq>
|
|
""",
|
|
"""
|
|
<iq type="result" id="3"
|
|
to="foo@comp.example.com/bar" from="pubsub.example.com" />
|
|
""")
|
|
|
|
# Case 4: Subscribee
|
|
run_test('user@example.com/foo', True, 'foo@comp.example.com/bar',
|
|
"""
|
|
<iq type="set" id="4"
|
|
to="pubsub.example.com" from="foo@comp.example.com/bar">
|
|
<pubsub xmlns="http://jabber.org/protocol/pubsub">
|
|
<unsubscribe node="some_node" jid="user@example.com/foo" />
|
|
</pubsub>
|
|
</iq>
|
|
""",
|
|
"""
|
|
<iq type="result" id="4"
|
|
to="foo@comp.example.com/bar" from="pubsub.example.com" />
|
|
""")
|
|
|
|
|
|
suite = unittest.TestLoader().loadTestsFromTestCase(TestStreamPubsub)
|