gajim-plural/test/unit/test_jingle.py

154 lines
5.7 KiB
Python
Raw Normal View History

2011-07-18 00:28:38 +02:00
'''
Tests for dispatcher_nb.py
'''
import unittest
import lib
lib.setup_env()
from mock import Mock
2017-06-13 23:58:06 +02:00
from gajim.common.protocol.bytestream import ConnectionIBBytestream, ConnectionSocks5Bytestream
2018-08-18 19:11:26 +02:00
from nbxmpp import dispatcher_nb
from nbxmpp import protocol
2017-06-13 23:58:06 +02:00
from gajim.common.jingle import ConnectionJingle
2018-08-18 19:11:26 +02:00
from gajim.common import app
2017-06-13 23:58:06 +02:00
from gajim.common.socks5 import SocksQueue
2011-07-18 00:28:38 +02:00
2013-01-01 19:44:25 +01:00
session_init = '''
2011-07-18 00:28:38 +02:00
<iq xmlns="jabber:client" to="jingleft@thiessen.im/Gajim" type="set" id="43">
<jingle xmlns="urn:xmpp:jingle:1" action="session-initiate" initiator="jtest@thiessen.im/Gajim" sid="38">
<content name="fileWL1Y2JIPTM5RAD68" creator="initiator">
<security xmlns="urn:xmpp:jingle:security:xtls:0">
<method name="x509" />
</security>
<description xmlns="urn:xmpp:jingle:apps:file-transfer:1">
<offer>
<file xmlns="http://jabber.org/protocol/si/profile/file-transfer" name="to" size="2273">
<desc />
</file>
</offer>
</description>
<transport xmlns="urn:xmpp:jingle:transports:s5b:1" sid="39">
<candidate jid="jtest@thiessen.im/Gajim" cid="40" priority="8257536" host="192.168.2.100" type="direct" port="28011" />
<candidate jid="proxy.thiessen.im" cid="41" priority="655360" host="192.168.2.100" type="proxy" port="5000" />
<candidate jid="proxy.jabbim.cz" cid="42" priority="655360" host="192.168.2.100" type="proxy" port="7777" />
</transport>
</content>
</jingle>
2013-01-01 19:44:25 +01:00
</iq>
2011-07-18 00:28:38 +02:00
'''
2013-01-01 19:44:25 +01:00
2011-07-18 00:28:38 +02:00
transport_info = '''
<iq from='jtest@thiessen.im/Gajim'
id='hjdi8'
to='jingleft@thiessen.im/Gajim'
type='set'>
<jingle xmlns='urn:xmpp:jingle:1'
action='transport-info'
initiator='jtest@thiessen.im/Gajim'
sid='38'>
<content creator='initiator' name='fileWL1Y2JIPTM5RAD68'>
<transport xmlns='urn:xmpp:jingle:transports:s5b:1'
sid='vj3hs98y'>
<candidate-used cid='hr65dqyd'/>
</transport>
</content>
</jingle>
</iq>
'''
class Connection(Mock, ConnectionJingle, ConnectionSocks5Bytestream,
ConnectionIBBytestream):
2013-01-01 19:44:25 +01:00
2011-07-18 00:28:38 +02:00
def __init__(self):
Mock.__init__(self)
ConnectionJingle.__init__(self)
ConnectionSocks5Bytestream.__init__(self)
ConnectionIBBytestream.__init__(self)
self.connected = 2 # This tells gajim we are connected
2013-01-01 19:44:25 +01:00
2011-07-18 00:28:38 +02:00
def send(self, stanza=None, when=None):
# Called when gajim wants to send something
2013-01-01 19:44:25 +01:00
print(str(stanza))
2018-08-18 19:11:26 +02:00
2011-07-18 00:28:38 +02:00
class TestJingle(unittest.TestCase):
def setUp(self):
self.dispatcher = dispatcher_nb.XMPPDispatcher()
2018-08-18 19:11:26 +02:00
app.nec = Mock()
app.socks5queue = SocksQueue(Mock())
2011-07-18 00:28:38 +02:00
# Setup mock client
self.client = Connection()
self.client.__str__ = lambda: 'Mock' # FIXME: why do I need this one?
self.client._caller = Connection()
self.client.defaultNamespace = protocol.NS_CLIENT
self.client.Connection = Connection() # mock transport
self.con = self.client.Connection
self.con.server_resource = None
self.con.connection = Connection()
2013-01-01 19:44:25 +01:00
'''
2018-06-22 00:47:29 +02:00
Fake file_props when we receive a file. Gajim creates a file_props
out of a FileRequestReceive event and from then on it changes in
2013-01-01 19:44:25 +01:00
a lot of places. It is easier to just copy it in here.
If the session_initiate stanza changes, this also must change.
2011-07-18 00:28:38 +02:00
'''
2018-08-18 19:11:26 +02:00
self.receive_file = {'stream-methods':
2013-01-01 19:44:25 +01:00
'http://jabber.org/protocol/bytestreams',
'sender': 'jtest@thiessen.im/Gajim',
2018-08-18 19:11:26 +02:00
'file-name': 'test_received_file',
2013-01-01 19:44:25 +01:00
'request-id': '43', 'sid': '39',
'session-sid': '38', 'session-type': 'jingle',
'transfered_size': [], 'receiver':
'jingleft@thiessen.im/Gajim', 'desc': '',
'size': '2273', 'type': 'r',
'streamhosts': [{'initiator':
'jtest@thiessen.im/Gajim',
'target': 'jingleft@thiessen.im/Gajim',
'cid': '41', 'state': 0, 'host': '192.168.2.100',
'type': 'direct', 'port': '28011'},
{'initiator': 'jtest@thiessen.im/Gajim',
'target': 'jingleft@thiessen.im/Gajim',
'cid': '42', 'state': 0, 'host': '192.168.2.100',
'type': 'proxy', 'port': '5000'}],
'name': 'to'}
2011-07-18 00:28:38 +02:00
def tearDown(self):
# Unplug if needed
if hasattr(self.dispatcher, '_owner'):
self.dispatcher.PlugOut()
def _simulate_connect(self):
self.dispatcher.PlugIn(self.client) # client is owner
2013-01-01 19:44:25 +01:00
# Simulate that we have established a connection
2011-07-18 00:28:38 +02:00
self.dispatcher.StreamInit()
self.dispatcher.ProcessNonBlocking("<stream:stream xmlns:stream='http://etherx.jabber.org/streams' xmlns='jabber:client'>")
2013-01-01 19:44:25 +01:00
2011-07-18 00:28:38 +02:00
def _simulate_jingle_session(self):
2013-01-01 19:44:25 +01:00
2018-08-18 19:11:26 +02:00
self.dispatcher.RegisterHandler('iq', self.con._JingleCB, 'set',
protocol.NS_JINGLE)
2011-07-18 00:28:38 +02:00
self.dispatcher.ProcessNonBlocking(session_init)
2013-01-02 13:54:02 +01:00
session = list(self.con._sessions.values())[0] # The only session we have
jft = list(session.contents.values())[0] # jingleFT object
2018-08-18 19:11:26 +02:00
jft.file_props = self.receive_file # We plug file_props manually
# The user accepts to receive the file
2011-07-18 00:28:38 +02:00
# we have to manually simulate this behavior
session.approve_session()
2018-08-18 19:11:26 +02:00
self.con.send_file_approval(self.receive_file)
2013-01-01 19:44:25 +01:00
2011-07-18 00:28:38 +02:00
self.dispatcher.ProcessNonBlocking(transport_info)
2013-01-01 19:44:25 +01:00
2011-07-18 00:28:38 +02:00
def test_jingle_session(self):
self._simulate_connect()
self._simulate_jingle_session()
2013-01-01 19:44:25 +01:00
2011-07-18 00:28:38 +02:00
if __name__ == '__main__':
unittest.main()