78 lines
3.3 KiB
Python
78 lines
3.3 KiB
Python
'''
|
|
Unit test for tranports classes.
|
|
'''
|
|
|
|
import unittest
|
|
|
|
import lib
|
|
lib.setup_env()
|
|
|
|
from common.xmpp import transports_nb
|
|
|
|
|
|
class TestModuleLevelFunctions(unittest.TestCase):
|
|
'''
|
|
Test class for functions defined at module level
|
|
'''
|
|
def test_urisplit(self):
|
|
def check_uri(uri, proto, host, port, path):
|
|
_proto, _host, _port, _path = transports_nb.urisplit(uri)
|
|
self.assertEqual(proto, _proto)
|
|
self.assertEqual(host, _host)
|
|
self.assertEqual(path, _path)
|
|
self.assertEqual(port, _port)
|
|
|
|
check_uri('http://httpcm.jabber.org:5280/webclient', proto='http',
|
|
host='httpcm.jabber.org', port=5280, path='/webclient')
|
|
|
|
check_uri('http://httpcm.jabber.org/webclient', proto='http',
|
|
host='httpcm.jabber.org', port=80, path='/webclient')
|
|
|
|
check_uri('https://httpcm.jabber.org/webclient', proto='https',
|
|
host='httpcm.jabber.org', port=443, path='/webclient')
|
|
|
|
def test_get_proxy_data_from_dict(self):
|
|
def check_dict(proxy_dict, host, port, user, passwd):
|
|
_host, _port, _user, _passwd = transports_nb.get_proxy_data_from_dict(
|
|
proxy_dict)
|
|
self.assertEqual(_host, host)
|
|
self.assertEqual(_port, port)
|
|
self.assertEqual(_user, user)
|
|
self.assertEqual(_passwd, passwd)
|
|
|
|
bosh_dict = {'bosh_content': u'text/xml; charset=utf-8',
|
|
'bosh_hold': 2,
|
|
'bosh_http_pipelining': False,
|
|
'bosh_uri': u'http://gajim.org:5280/http-bind',
|
|
'bosh_useproxy': False,
|
|
'bosh_wait': 30,
|
|
'bosh_wait_for_restart_response': False,
|
|
'host': u'172.16.99.11',
|
|
'pass': u'pass',
|
|
'port': 3128,
|
|
'type': u'bosh',
|
|
'useauth': True,
|
|
'user': u'user'}
|
|
check_dict(bosh_dict, host=u'gajim.org', port=5280, user=u'user',
|
|
passwd=u'pass')
|
|
|
|
proxy_dict = {'bosh_content': u'text/xml; charset=utf-8',
|
|
'bosh_hold': 2,
|
|
'bosh_http_pipelining': False,
|
|
'bosh_port': 5280,
|
|
'bosh_uri': u'',
|
|
'bosh_useproxy': True,
|
|
'bosh_wait': 30,
|
|
'bosh_wait_for_restart_response': False,
|
|
'host': u'172.16.99.11',
|
|
'pass': u'pass',
|
|
'port': 3128,
|
|
'type': 'socks5',
|
|
'useauth': True,
|
|
'user': u'user'}
|
|
check_dict(proxy_dict, host=u'172.16.99.11', port=3128, user=u'user',
|
|
passwd=u'pass')
|
|
|
|
|
|
if __name__ == '__main__':
|
|
unittest.main()
|