mirror of
https://github.com/correl/SleekXMPP.git
synced 2024-11-23 19:19:53 +00:00
Add more XEP-0047 tests.
This commit is contained in:
parent
d565e4be20
commit
caa967105c
2 changed files with 83 additions and 1 deletions
|
@ -46,7 +46,7 @@ class Data(ElementBase):
|
|||
self._set_attr('seq', str(value))
|
||||
|
||||
def get_data(self):
|
||||
b64_data = self.xml.text
|
||||
b64_data = self.xml.text.strip()
|
||||
if VALID_B64.match(b64_data).group() == b64_data:
|
||||
return from_b64(b64_data)
|
||||
else:
|
||||
|
|
|
@ -92,7 +92,89 @@ class TestInBandByteStreams(SleekTest):
|
|||
|
||||
self.assertEqual(events, set(['ibb_stream_start', 'callback']))
|
||||
|
||||
def testSendData(self):
|
||||
"""Test sending data over an in-band bytestream."""
|
||||
|
||||
streams = []
|
||||
data = []
|
||||
|
||||
def on_stream_start(stream):
|
||||
streams.append(stream)
|
||||
|
||||
def on_stream_data(d):
|
||||
data.append(d['data'])
|
||||
|
||||
self.xmpp.add_event_handler('ibb_stream_start', on_stream_start)
|
||||
self.xmpp.add_event_handler('ibb_stream_data', on_stream_data)
|
||||
|
||||
t = threading.Thread(name='open_stream',
|
||||
target=self.xmpp['xep_0047'].open_stream,
|
||||
args=('tester@localhost/receiver',),
|
||||
kwargs={'sid': 'testing'})
|
||||
t.start()
|
||||
|
||||
self.send("""
|
||||
<iq type="set" to="tester@localhost/receiver" id="1">
|
||||
<open xmlns="http://jabber.org/protocol/ibb"
|
||||
sid="testing"
|
||||
block-size="4096"
|
||||
stanza="iq" />
|
||||
</iq>
|
||||
""")
|
||||
|
||||
self.recv("""
|
||||
<iq type="result" id="1"
|
||||
to="tester@localhost"
|
||||
from="tester@localhost/receiver" />
|
||||
""")
|
||||
|
||||
t.join()
|
||||
|
||||
time.sleep(0.2)
|
||||
|
||||
stream = streams[0]
|
||||
|
||||
|
||||
# Test sending data out
|
||||
stream.send("Testing")
|
||||
|
||||
self.send("""
|
||||
<iq type="set" id="2"
|
||||
from="tester@localhost"
|
||||
to="tester@localhost/receiver">
|
||||
<data xmlns="http://jabber.org/protocol/ibb"
|
||||
seq="0"
|
||||
sid="testing">
|
||||
VGVzdGluZw==
|
||||
</data>
|
||||
</iq>
|
||||
""")
|
||||
|
||||
self.recv("""
|
||||
<iq type="result" id="2"
|
||||
to="tester@localhost"
|
||||
from="tester@localhost/receiver" />
|
||||
""")
|
||||
|
||||
# Test receiving data
|
||||
self.recv("""
|
||||
<iq type="set" id="A"
|
||||
to="tester@localhost"
|
||||
from="tester@localhost/receiver">
|
||||
<data xmlns="http://jabber.org/protocol/ibb"
|
||||
seq="0"
|
||||
sid="testing">
|
||||
aXQgd29ya3Mh
|
||||
</data>
|
||||
</iq>
|
||||
""")
|
||||
|
||||
self.send("""
|
||||
<iq type="result" id="A"
|
||||
to="tester@localhost/receiver" />
|
||||
""")
|
||||
|
||||
self.assertEqual(data, ['it works!'])
|
||||
|
||||
|
||||
suite = unittest.TestLoader().loadTestsFromTestCase(TestInBandByteStreams)
|
||||
|
|
Loading…
Reference in a new issue