Remove duplicated coded in test_client_nb.py
This commit is contained in:
Stephan Erb
parent
62a334b8fd
commit
a1f346e48a
|
|
@ -1,52 +1,37 @@
|
||||||
import unittest
|
|
||||||
from xmpp_mocks import *
|
|
||||||
|
|
||||||
import sys, os.path
|
|
||||||
|
|
||||||
gajim_root = os.path.join(os.path.abspath(os.path.dirname(__file__)), '..')
|
|
||||||
|
|
||||||
sys.path.append(gajim_root + '/src/common/xmpp')
|
|
||||||
|
|
||||||
import client_nb
|
|
||||||
|
|
||||||
'''
|
'''
|
||||||
Testing script for NonBlockingClient class (src/common/xmpp/client_nb.py)
|
Testing script for NonBlockingClient class (src/common/xmpp/client_nb.py)
|
||||||
It actually connects to a xmpp server so the connection values have to be
|
It actually connects to a xmpp server so the connection values have to be
|
||||||
changed before running.
|
changed before running.
|
||||||
'''
|
'''
|
||||||
|
|
||||||
|
import unittest
|
||||||
|
from xmpp_mocks import MockConnectionClass, IdleQueueThread
|
||||||
|
|
||||||
|
import sys, os.path
|
||||||
|
|
||||||
|
gajim_root = os.path.join(os.path.abspath(os.path.dirname(__file__)), '..')
|
||||||
|
sys.path.append(gajim_root + '/src/common/xmpp')
|
||||||
|
|
||||||
|
import client_nb
|
||||||
|
|
||||||
|
# (XMPP server hostname, c2s port). Script will connect to the machine.
|
||||||
xmpp_server_port = ('gajim.org', 5222)
|
xmpp_server_port = ('gajim.org', 5222)
|
||||||
'''
|
|
||||||
2-tuple - (XMPP server hostname, c2s port)
|
|
||||||
Script will connect to the machine.
|
|
||||||
'''
|
|
||||||
|
|
||||||
|
# [username, password, passphrase]. Script will authenticate to server above
|
||||||
credentials = ['loginn', 'passwo', 'testresour']
|
credentials = ['loginn', 'passwo', 'testresour']
|
||||||
'''
|
|
||||||
[username, password, passphrase]
|
|
||||||
Script will autheticate itself with this credentials on above mentioned server.
|
|
||||||
'''
|
|
||||||
|
|
||||||
class TestNonBlockingClient(unittest.TestCase):
|
class TestNonBlockingClient(unittest.TestCase):
|
||||||
'''
|
'''
|
||||||
Test Cases class for NonBlockingClient.
|
Test Cases class for NonBlockingClient.
|
||||||
'''
|
'''
|
||||||
|
|
||||||
def setUp(self):
|
def setUp(self):
|
||||||
'''
|
''' IdleQueue thread is run and dummy connection is created. '''
|
||||||
IdleQueue thread is run and dummy connection is created.
|
|
||||||
'''
|
|
||||||
|
|
||||||
self.idlequeue_thread = IdleQueueThread()
|
self.idlequeue_thread = IdleQueueThread()
|
||||||
self.connection = MockConnectionClass()
|
self.connection = MockConnectionClass()
|
||||||
|
|
||||||
self.idlequeue_thread.start()
|
self.idlequeue_thread.start()
|
||||||
|
|
||||||
def tearDown(self):
|
def tearDown(self):
|
||||||
'''
|
''' IdleQueue thread is stopped. '''
|
||||||
IdleQueue thread is stopped.
|
|
||||||
'''
|
|
||||||
self.idlequeue_thread.stop_thread()
|
self.idlequeue_thread.stop_thread()
|
||||||
self.idlequeue_thread.join()
|
self.idlequeue_thread.join()
|
||||||
|
|
||||||
|
|
@ -57,208 +42,21 @@ class TestNonBlockingClient(unittest.TestCase):
|
||||||
|
|
||||||
:param server_port: tuple of (hostname, port) for where the client should
|
:param server_port: tuple of (hostname, port) for where the client should
|
||||||
connect.
|
connect.
|
||||||
|
|
||||||
'''
|
'''
|
||||||
self.client = client_nb.NonBlockingClient(
|
self.client = client_nb.NonBlockingClient(
|
||||||
domain=server_port[0],
|
domain=server_port[0],
|
||||||
idlequeue=self.idlequeue_thread.iq,
|
idlequeue=self.idlequeue_thread.iq)
|
||||||
)
|
|
||||||
'''
|
'''
|
||||||
NonBlockingClient instance with parameters from global variables and with
|
NonBlockingClient instance with parameters from global variables and with
|
||||||
callbacks from dummy connection.
|
callbacks from dummy connection.
|
||||||
'''
|
'''
|
||||||
|
|
||||||
self.client.connect(
|
self.client.connect(
|
||||||
on_connect=lambda *args: self.connection.on_connect(True, *args),
|
|
||||||
on_connect_failure=lambda *args: self.connection.on_connect(False, *args),
|
|
||||||
)
|
|
||||||
|
|
||||||
print 'waiting for callback from client constructor'
|
|
||||||
self.connection.wait()
|
|
||||||
|
|
||||||
# if on_connect was called, client has to be connected and vice versa
|
|
||||||
if self.connection.connect_succeeded:
|
|
||||||
self.assert_(self.client.get_connect_type())
|
|
||||||
else:
|
|
||||||
self.assert_(not self.client.get_connect_type())
|
|
||||||
|
|
||||||
def client_auth(self, username, password, resource, sasl):
|
|
||||||
'''
|
|
||||||
Method authenticating connected client with supplied credentials. Returns
|
|
||||||
when authentication is over.
|
|
||||||
:param sasl: whether to use sasl (sasl=1) or old (sasl=0) authentication
|
|
||||||
|
|
||||||
:todo: to check and be more specific about when it returns (bind, session..)
|
|
||||||
'''
|
|
||||||
self.client.auth(username, password, resource, sasl,
|
|
||||||
on_auth=self.connection.on_auth)
|
|
||||||
|
|
||||||
print 'waiting for authentication...'
|
|
||||||
self.connection.wait()
|
|
||||||
|
|
||||||
def do_disconnect(self):
|
|
||||||
'''
|
|
||||||
Does disconnecting of connected client. Returns when TCP connection is closed.
|
|
||||||
'''
|
|
||||||
#self.client.start_disconnect(None, on_disconnect=self.connection.set_event)
|
|
||||||
self.client.RegisterDisconnectHandler(self.connection.set_event)
|
|
||||||
self.client.disconnect()
|
|
||||||
|
|
||||||
print 'waiting for disconnecting...'
|
|
||||||
self.connection.wait()
|
|
||||||
|
|
||||||
def test_proper_connect_sasl(self):
|
|
||||||
'''
|
|
||||||
The ideal testcase - client is connected, authenticated with SASL and
|
|
||||||
then disconnected.
|
|
||||||
'''
|
|
||||||
self.open_stream(xmpp_server_port)
|
|
||||||
|
|
||||||
# if client is not connected, lets raise the AssertionError
|
|
||||||
self.assert_(self.client.get_connect_type())
|
|
||||||
# (client.disconnect() is already called from NBClient._on_connected_failure
|
|
||||||
# so there's need to call it in this case
|
|
||||||
|
|
||||||
self.client_auth(credentials[0], credentials[1], credentials[2], sasl=1)
|
|
||||||
self.assert_(self.connection.con)
|
|
||||||
self.assert_(self.connection.auth=='sasl')
|
|
||||||
|
|
||||||
self.do_disconnect()
|
|
||||||
|
|
||||||
|
|
||||||
def test_proper_connect_oldauth(self):
|
|
||||||
'''
|
|
||||||
The ideal testcase - client is connected, authenticated with old auth and
|
|
||||||
then disconnected.
|
|
||||||
'''
|
|
||||||
self.open_stream(xmpp_server_port)
|
|
||||||
self.assert_(self.client.get_connect_type())
|
|
||||||
self.client_auth(credentials[0], credentials[1], credentials[2], sasl=0)
|
|
||||||
self.assert_(self.connection.con)
|
|
||||||
self.assert_(self.connection.auth=='old_auth')
|
|
||||||
self.do_disconnect()
|
|
||||||
|
|
||||||
def test_connect_to_nonexisting_host(self):
|
|
||||||
'''
|
|
||||||
Connect to nonexisting host. DNS request for A records should return nothing.
|
|
||||||
'''
|
|
||||||
self.open_stream(('fdsfsdf.fdsf.fss', 5222))
|
|
||||||
print 'nonexthost: %s' % self.client.get_connect_type()
|
|
||||||
self.assert_(not self.client.get_connect_type())
|
|
||||||
|
|
||||||
def test_connect_to_wrong_port(self):
|
|
||||||
'''
|
|
||||||
Connect to nonexisting host. DNS request for A records should return some IP
|
|
||||||
but there shouldn't be XMPP server running on specified port.
|
|
||||||
'''
|
|
||||||
self.open_stream((xmpp_server_port[0], 31337))
|
|
||||||
self.assert_(not self.client.get_connect_type())
|
|
||||||
|
|
||||||
def test_connect_with_wrong_creds(self):
|
|
||||||
'''
|
|
||||||
Connecting with invalid password.
|
|
||||||
'''
|
|
||||||
self.open_stream(xmpp_server_port)
|
|
||||||
self.assert_(self.client.get_connect_type())
|
|
||||||
self.client_auth(credentials[0], "wrong pass", credentials[2], sasl=1)
|
|
||||||
self.assert_(self.connection.auth is None)
|
|
||||||
self.do_disconnect()
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
if __name__ == '__main__':
|
|
||||||
|
|
||||||
suite = unittest.TestLoader().loadTestsFromTestCase(TestNonBlockingClient)
|
|
||||||
#suite = unittest.TestSuite()
|
|
||||||
#suite.addTest(TestNonBlockingClient('test_proper_connect_oldauth'))
|
|
||||||
#suite.addTest(TestNonBlockingClient('test_connect_to_nonexisting_host'))
|
|
||||||
|
|
||||||
unittest.TextTestRunner(verbosity=2).run(suite)
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
import unittest
|
|
||||||
from xmpp_mocks import *
|
|
||||||
|
|
||||||
import sys, os.path
|
|
||||||
|
|
||||||
gajim_root = os.path.join(os.path.abspath(os.path.dirname(__file__)), '..')
|
|
||||||
|
|
||||||
sys.path.append(gajim_root + '/src/common/xmpp')
|
|
||||||
|
|
||||||
import client_nb
|
|
||||||
|
|
||||||
'''
|
|
||||||
Testing script for NonBlockingClient class (src/common/xmpp/client_nb.py)
|
|
||||||
It actually connects to a xmpp server so the connection values have to be
|
|
||||||
changed before running.
|
|
||||||
'''
|
|
||||||
|
|
||||||
|
|
||||||
xmpp_server_port = ('xmpp.example.org',5222)
|
|
||||||
'''
|
|
||||||
2-tuple - (XMPP server hostname, c2s port)
|
|
||||||
Script will connect to the machine.
|
|
||||||
'''
|
|
||||||
|
|
||||||
credentials = ['loginn', 'passwo', 'testresour']
|
|
||||||
'''
|
|
||||||
[username, password, passphrase]
|
|
||||||
Script will autheticate itself with this credentials on above mentioned server.
|
|
||||||
'''
|
|
||||||
|
|
||||||
class TestNonBlockingClient(unittest.TestCase):
|
|
||||||
'''
|
|
||||||
Test Cases class for NonBlockingClient.
|
|
||||||
'''
|
|
||||||
|
|
||||||
def setUp(self):
|
|
||||||
'''
|
|
||||||
IdleQueue thread is run and dummy connection is created.
|
|
||||||
'''
|
|
||||||
|
|
||||||
self.idlequeue_thread = IdleQueueThread()
|
|
||||||
self.connection = MockConnectionClass()
|
|
||||||
|
|
||||||
self.idlequeue_thread.start()
|
|
||||||
|
|
||||||
def tearDown(self):
|
|
||||||
'''
|
|
||||||
IdleQueue thread is stopped.
|
|
||||||
'''
|
|
||||||
self.idlequeue_thread.stop_thread()
|
|
||||||
self.idlequeue_thread.join()
|
|
||||||
|
|
||||||
def open_stream(self, server_port):
|
|
||||||
'''
|
|
||||||
Method opening the XMPP connection. It returns when <stream:features>
|
|
||||||
is received from server.
|
|
||||||
|
|
||||||
:param server_port: tuple of (hostname, port) for where the client should
|
|
||||||
connect.
|
|
||||||
|
|
||||||
'''
|
|
||||||
self.client = client_nb.NonBlockingClient(
|
|
||||||
hostname=server_port[0],
|
hostname=server_port[0],
|
||||||
port=server_port[1],
|
port=server_port[1],
|
||||||
caller=self.connection,
|
|
||||||
idlequeue=self.idlequeue_thread.iq,
|
|
||||||
)
|
|
||||||
'''
|
|
||||||
NonBlockingClient instance with parameters from global variables and with
|
|
||||||
callbacks from dummy connection.
|
|
||||||
'''
|
|
||||||
|
|
||||||
self.client.connect(
|
|
||||||
on_connect=lambda *args: self.connection.on_connect(True, *args),
|
on_connect=lambda *args: self.connection.on_connect(True, *args),
|
||||||
on_connect_failure=lambda *args: self.connection.on_connect(False, *args),
|
on_connect_failure=lambda *args: self.connection.on_connect(
|
||||||
secure=False
|
False, *args))
|
||||||
)
|
|
||||||
|
|
||||||
print 'waiting for callback from client constructor'
|
print 'waiting for callback from client constructor'
|
||||||
self.connection.wait()
|
self.connection.wait()
|
||||||
|
|
@ -352,581 +150,7 @@ class TestNonBlockingClient(unittest.TestCase):
|
||||||
self.do_disconnect()
|
self.do_disconnect()
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
if __name__ == '__main__':
|
if __name__ == '__main__':
|
||||||
|
unittest.main()
|
||||||
|
|
||||||
suite = unittest.TestLoader().loadTestsFromTestCase(TestNonBlockingClient)
|
# vim: se ts=3:
|
||||||
#suite = unittest.TestSuite()
|
|
||||||
#suite.addTest(TestNonBlockingClient('test_proper_connect_oldauth'))
|
|
||||||
#suite.addTest(TestNonBlockingClient('test_connect_to_nonexisting_host'))
|
|
||||||
|
|
||||||
unittest.TextTestRunner(verbosity=2).run(suite)
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
import unittest
|
|
||||||
from xmpp_mocks import *
|
|
||||||
|
|
||||||
import sys, os.path
|
|
||||||
|
|
||||||
gajim_root = os.path.join(os.path.abspath(os.path.dirname(__file__)), '..')
|
|
||||||
|
|
||||||
sys.path.append(gajim_root + '/src/common/xmpp')
|
|
||||||
|
|
||||||
import client_nb
|
|
||||||
|
|
||||||
'''
|
|
||||||
Testing script for NonBlockingClient class (src/common/xmpp/client_nb.py)
|
|
||||||
It actually connects to a xmpp server so the connection values have to be
|
|
||||||
changed before running.
|
|
||||||
'''
|
|
||||||
|
|
||||||
|
|
||||||
xmpp_server_port = ('xmpp.example.org',5222)
|
|
||||||
'''
|
|
||||||
2-tuple - (XMPP server hostname, c2s port)
|
|
||||||
Script will connect to the machine.
|
|
||||||
'''
|
|
||||||
|
|
||||||
credentials = ['loginn', 'passwo', 'testresour']
|
|
||||||
'''
|
|
||||||
[username, password, passphrase]
|
|
||||||
Script will autheticate itself with this credentials on above mentioned server.
|
|
||||||
'''
|
|
||||||
|
|
||||||
class TestNonBlockingClient(unittest.TestCase):
|
|
||||||
'''
|
|
||||||
Test Cases class for NonBlockingClient.
|
|
||||||
'''
|
|
||||||
|
|
||||||
def setUp(self):
|
|
||||||
'''
|
|
||||||
IdleQueue thread is run and dummy connection is created.
|
|
||||||
'''
|
|
||||||
|
|
||||||
self.idlequeue_thread = IdleQueueThread()
|
|
||||||
self.connection = MockConnectionClass()
|
|
||||||
|
|
||||||
self.idlequeue_thread.start()
|
|
||||||
|
|
||||||
def tearDown(self):
|
|
||||||
'''
|
|
||||||
IdleQueue thread is stopped.
|
|
||||||
'''
|
|
||||||
self.idlequeue_thread.stop_thread()
|
|
||||||
self.idlequeue_thread.join()
|
|
||||||
|
|
||||||
def open_stream(self, server_port):
|
|
||||||
'''
|
|
||||||
Method opening the XMPP connection. It returns when <stream:features>
|
|
||||||
is received from server.
|
|
||||||
|
|
||||||
:param server_port: tuple of (hostname, port) for where the client should
|
|
||||||
connect.
|
|
||||||
|
|
||||||
'''
|
|
||||||
self.client = client_nb.NonBlockingClient(
|
|
||||||
hostname=server_port[0],
|
|
||||||
port=server_port[1],
|
|
||||||
caller=self.connection,
|
|
||||||
idlequeue=self.idlequeue_thread.iq,
|
|
||||||
)
|
|
||||||
'''
|
|
||||||
NonBlockingClient instance with parameters from global variables and with
|
|
||||||
callbacks from dummy connection.
|
|
||||||
'''
|
|
||||||
|
|
||||||
self.client.connect(
|
|
||||||
on_connect=lambda *args: self.connection.on_connect(True, *args),
|
|
||||||
on_connect_failure=lambda *args: self.connection.on_connect(False, *args),
|
|
||||||
secure=False
|
|
||||||
)
|
|
||||||
|
|
||||||
print 'waiting for callback from client constructor'
|
|
||||||
self.connection.wait()
|
|
||||||
|
|
||||||
# if on_connect was called, client has to be connected and vice versa
|
|
||||||
if self.connection.connect_succeeded:
|
|
||||||
self.assert_(self.client.get_connect_type())
|
|
||||||
else:
|
|
||||||
self.assert_(not self.client.get_connect_type())
|
|
||||||
|
|
||||||
def client_auth(self, username, password, resource, sasl):
|
|
||||||
'''
|
|
||||||
Method authenticating connected client with supplied credentials. Returns
|
|
||||||
when authentication is over.
|
|
||||||
:param sasl: whether to use sasl (sasl=1) or old (sasl=0) authentication
|
|
||||||
|
|
||||||
:todo: to check and be more specific about when it returns (bind, session..)
|
|
||||||
'''
|
|
||||||
self.client.auth(username, password, resource, sasl,
|
|
||||||
on_auth=self.connection.on_auth)
|
|
||||||
|
|
||||||
print 'waiting for authentication...'
|
|
||||||
self.connection.wait()
|
|
||||||
|
|
||||||
def do_disconnect(self):
|
|
||||||
'''
|
|
||||||
Does disconnecting of connected client. Returns when TCP connection is closed.
|
|
||||||
'''
|
|
||||||
#self.client.start_disconnect(None, on_disconnect=self.connection.set_event)
|
|
||||||
self.client.RegisterDisconnectHandler(self.connection.set_event)
|
|
||||||
self.client.disconnect()
|
|
||||||
|
|
||||||
print 'waiting for disconnecting...'
|
|
||||||
self.connection.wait()
|
|
||||||
|
|
||||||
def test_proper_connect_sasl(self):
|
|
||||||
'''
|
|
||||||
The ideal testcase - client is connected, authenticated with SASL and
|
|
||||||
then disconnected.
|
|
||||||
'''
|
|
||||||
self.open_stream(xmpp_server_port)
|
|
||||||
|
|
||||||
# if client is not connected, lets raise the AssertionError
|
|
||||||
self.assert_(self.client.get_connect_type())
|
|
||||||
# (client.disconnect() is already called from NBClient._on_connected_failure
|
|
||||||
# so there's need to call it in this case
|
|
||||||
|
|
||||||
self.client_auth(credentials[0], credentials[1], credentials[2], sasl=1)
|
|
||||||
self.assert_(self.connection.con)
|
|
||||||
self.assert_(self.connection.auth=='sasl')
|
|
||||||
|
|
||||||
self.do_disconnect()
|
|
||||||
|
|
||||||
|
|
||||||
def test_proper_connect_oldauth(self):
|
|
||||||
'''
|
|
||||||
The ideal testcase - client is connected, authenticated with old auth and
|
|
||||||
then disconnected.
|
|
||||||
'''
|
|
||||||
self.open_stream(xmpp_server_port)
|
|
||||||
self.assert_(self.client.get_connect_type())
|
|
||||||
self.client_auth(credentials[0], credentials[1], credentials[2], sasl=0)
|
|
||||||
self.assert_(self.connection.con)
|
|
||||||
self.assert_(self.connection.auth=='old_auth')
|
|
||||||
self.do_disconnect()
|
|
||||||
|
|
||||||
def test_connect_to_nonexisting_host(self):
|
|
||||||
'''
|
|
||||||
Connect to nonexisting host. DNS request for A records should return nothing.
|
|
||||||
'''
|
|
||||||
self.open_stream(('fdsfsdf.fdsf.fss', 5222))
|
|
||||||
print 'nonexthost: %s' % self.client.get_connect_type()
|
|
||||||
self.assert_(not self.client.get_connect_type())
|
|
||||||
|
|
||||||
def test_connect_to_wrong_port(self):
|
|
||||||
'''
|
|
||||||
Connect to nonexisting host. DNS request for A records should return some IP
|
|
||||||
but there shouldn't be XMPP server running on specified port.
|
|
||||||
'''
|
|
||||||
self.open_stream((xmpp_server_port[0], 31337))
|
|
||||||
self.assert_(not self.client.get_connect_type())
|
|
||||||
|
|
||||||
def test_connect_with_wrong_creds(self):
|
|
||||||
'''
|
|
||||||
Connecting with invalid password.
|
|
||||||
'''
|
|
||||||
self.open_stream(xmpp_server_port)
|
|
||||||
self.assert_(self.client.get_connect_type())
|
|
||||||
self.client_auth(credentials[0], "wrong pass", credentials[2], sasl=1)
|
|
||||||
self.assert_(self.connection.auth is None)
|
|
||||||
self.do_disconnect()
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
if __name__ == '__main__':
|
|
||||||
|
|
||||||
suite = unittest.TestLoader().loadTestsFromTestCase(TestNonBlockingClient)
|
|
||||||
#suite = unittest.TestSuite()
|
|
||||||
#suite.addTest(TestNonBlockingClient('test_proper_connect_oldauth'))
|
|
||||||
#suite.addTest(TestNonBlockingClient('test_connect_to_nonexisting_host'))
|
|
||||||
|
|
||||||
unittest.TextTestRunner(verbosity=2).run(suite)
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
import unittest
|
|
||||||
from xmpp_mocks import *
|
|
||||||
|
|
||||||
import sys, os.path
|
|
||||||
|
|
||||||
gajim_root = os.path.join(os.path.abspath(os.path.dirname(__file__)), '..')
|
|
||||||
|
|
||||||
sys.path.append(gajim_root + '/src/common/xmpp')
|
|
||||||
|
|
||||||
import client_nb
|
|
||||||
|
|
||||||
'''
|
|
||||||
Testing script for NonBlockingClient class (src/common/xmpp/client_nb.py)
|
|
||||||
It actually connects to a xmpp server so the connection values have to be
|
|
||||||
changed before running.
|
|
||||||
'''
|
|
||||||
|
|
||||||
|
|
||||||
xmpp_server_port = ('xmpp.example.org',5222)
|
|
||||||
'''
|
|
||||||
2-tuple - (XMPP server hostname, c2s port)
|
|
||||||
Script will connect to the machine.
|
|
||||||
'''
|
|
||||||
|
|
||||||
credentials = ['loginn', 'passwo', 'testresour']
|
|
||||||
'''
|
|
||||||
[username, password, passphrase]
|
|
||||||
Script will autheticate itself with this credentials on above mentioned server.
|
|
||||||
'''
|
|
||||||
|
|
||||||
class TestNonBlockingClient(unittest.TestCase):
|
|
||||||
'''
|
|
||||||
Test Cases class for NonBlockingClient.
|
|
||||||
'''
|
|
||||||
|
|
||||||
def setUp(self):
|
|
||||||
'''
|
|
||||||
IdleQueue thread is run and dummy connection is created.
|
|
||||||
'''
|
|
||||||
|
|
||||||
self.idlequeue_thread = IdleQueueThread()
|
|
||||||
self.connection = MockConnectionClass()
|
|
||||||
|
|
||||||
self.idlequeue_thread.start()
|
|
||||||
|
|
||||||
def tearDown(self):
|
|
||||||
'''
|
|
||||||
IdleQueue thread is stopped.
|
|
||||||
'''
|
|
||||||
self.idlequeue_thread.stop_thread()
|
|
||||||
self.idlequeue_thread.join()
|
|
||||||
|
|
||||||
def open_stream(self, server_port):
|
|
||||||
'''
|
|
||||||
Method opening the XMPP connection. It returns when <stream:features>
|
|
||||||
is received from server.
|
|
||||||
|
|
||||||
:param server_port: tuple of (hostname, port) for where the client should
|
|
||||||
connect.
|
|
||||||
|
|
||||||
'''
|
|
||||||
self.client = client_nb.NonBlockingClient(
|
|
||||||
hostname=server_port[0],
|
|
||||||
port=server_port[1],
|
|
||||||
caller=self.connection,
|
|
||||||
idlequeue=self.idlequeue_thread.iq,
|
|
||||||
)
|
|
||||||
'''
|
|
||||||
NonBlockingClient instance with parameters from global variables and with
|
|
||||||
callbacks from dummy connection.
|
|
||||||
'''
|
|
||||||
|
|
||||||
self.client.connect(
|
|
||||||
on_connect=lambda *args: self.connection.on_connect(True, *args),
|
|
||||||
on_connect_failure=lambda *args: self.connection.on_connect(False, *args),
|
|
||||||
secure=False
|
|
||||||
)
|
|
||||||
|
|
||||||
print 'waiting for callback from client constructor'
|
|
||||||
self.connection.wait()
|
|
||||||
|
|
||||||
# if on_connect was called, client has to be connected and vice versa
|
|
||||||
if self.connection.connect_succeeded:
|
|
||||||
self.assert_(self.client.get_connect_type())
|
|
||||||
else:
|
|
||||||
self.assert_(not self.client.get_connect_type())
|
|
||||||
|
|
||||||
def client_auth(self, username, password, resource, sasl):
|
|
||||||
'''
|
|
||||||
Method authenticating connected client with supplied credentials. Returns
|
|
||||||
when authentication is over.
|
|
||||||
:param sasl: whether to use sasl (sasl=1) or old (sasl=0) authentication
|
|
||||||
|
|
||||||
:todo: to check and be more specific about when it returns (bind, session..)
|
|
||||||
'''
|
|
||||||
self.client.auth(username, password, resource, sasl,
|
|
||||||
on_auth=self.connection.on_auth)
|
|
||||||
|
|
||||||
print 'waiting for authentication...'
|
|
||||||
self.connection.wait()
|
|
||||||
|
|
||||||
def do_disconnect(self):
|
|
||||||
'''
|
|
||||||
Does disconnecting of connected client. Returns when TCP connection is closed.
|
|
||||||
'''
|
|
||||||
#self.client.start_disconnect(None, on_disconnect=self.connection.set_event)
|
|
||||||
self.client.RegisterDisconnectHandler(self.connection.set_event)
|
|
||||||
self.client.disconnect()
|
|
||||||
|
|
||||||
print 'waiting for disconnecting...'
|
|
||||||
self.connection.wait()
|
|
||||||
|
|
||||||
def test_proper_connect_sasl(self):
|
|
||||||
'''
|
|
||||||
The ideal testcase - client is connected, authenticated with SASL and
|
|
||||||
then disconnected.
|
|
||||||
'''
|
|
||||||
self.open_stream(xmpp_server_port)
|
|
||||||
|
|
||||||
# if client is not connected, lets raise the AssertionError
|
|
||||||
self.assert_(self.client.get_connect_type())
|
|
||||||
# (client.disconnect() is already called from NBClient._on_connected_failure
|
|
||||||
# so there's need to call it in this case
|
|
||||||
|
|
||||||
self.client_auth(credentials[0], credentials[1], credentials[2], sasl=1)
|
|
||||||
self.assert_(self.connection.con)
|
|
||||||
self.assert_(self.connection.auth=='sasl')
|
|
||||||
|
|
||||||
self.do_disconnect()
|
|
||||||
|
|
||||||
|
|
||||||
def test_proper_connect_oldauth(self):
|
|
||||||
'''
|
|
||||||
The ideal testcase - client is connected, authenticated with old auth and
|
|
||||||
then disconnected.
|
|
||||||
'''
|
|
||||||
self.open_stream(xmpp_server_port)
|
|
||||||
self.assert_(self.client.get_connect_type())
|
|
||||||
self.client_auth(credentials[0], credentials[1], credentials[2], sasl=0)
|
|
||||||
self.assert_(self.connection.con)
|
|
||||||
self.assert_(self.connection.auth=='old_auth')
|
|
||||||
self.do_disconnect()
|
|
||||||
|
|
||||||
def test_connect_to_nonexisting_host(self):
|
|
||||||
'''
|
|
||||||
Connect to nonexisting host. DNS request for A records should return nothing.
|
|
||||||
'''
|
|
||||||
self.open_stream(('fdsfsdf.fdsf.fss', 5222))
|
|
||||||
print 'nonexthost: %s' % self.client.get_connect_type()
|
|
||||||
self.assert_(not self.client.get_connect_type())
|
|
||||||
|
|
||||||
def test_connect_to_wrong_port(self):
|
|
||||||
'''
|
|
||||||
Connect to nonexisting host. DNS request for A records should return some IP
|
|
||||||
but there shouldn't be XMPP server running on specified port.
|
|
||||||
'''
|
|
||||||
self.open_stream((xmpp_server_port[0], 31337))
|
|
||||||
self.assert_(not self.client.get_connect_type())
|
|
||||||
|
|
||||||
def test_connect_with_wrong_creds(self):
|
|
||||||
'''
|
|
||||||
Connecting with invalid password.
|
|
||||||
'''
|
|
||||||
self.open_stream(xmpp_server_port)
|
|
||||||
self.assert_(self.client.get_connect_type())
|
|
||||||
self.client_auth(credentials[0], "wrong pass", credentials[2], sasl=1)
|
|
||||||
self.assert_(self.connection.auth is None)
|
|
||||||
self.do_disconnect()
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
if __name__ == '__main__':
|
|
||||||
|
|
||||||
suite = unittest.TestLoader().loadTestsFromTestCase(TestNonBlockingClient)
|
|
||||||
#suite = unittest.TestSuite()
|
|
||||||
#suite.addTest(TestNonBlockingClient('test_proper_connect_oldauth'))
|
|
||||||
#suite.addTest(TestNonBlockingClient('test_connect_to_nonexisting_host'))
|
|
||||||
|
|
||||||
unittest.TextTestRunner(verbosity=2).run(suite)
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
import unittest
|
|
||||||
from xmpp_mocks import *
|
|
||||||
|
|
||||||
import sys, os.path
|
|
||||||
|
|
||||||
gajim_root = os.path.join(os.path.abspath(os.path.dirname(__file__)), '..')
|
|
||||||
|
|
||||||
sys.path.append(gajim_root + '/src/common/xmpp')
|
|
||||||
|
|
||||||
import client_nb
|
|
||||||
|
|
||||||
'''
|
|
||||||
Testing script for NonBlockingClient class (src/common/xmpp/client_nb.py)
|
|
||||||
It actually connects to a xmpp server so the connection values have to be
|
|
||||||
changed before running.
|
|
||||||
'''
|
|
||||||
|
|
||||||
|
|
||||||
xmpp_server_port = ('xmpp.example.org',5222)
|
|
||||||
'''
|
|
||||||
2-tuple - (XMPP server hostname, c2s port)
|
|
||||||
Script will connect to the machine.
|
|
||||||
'''
|
|
||||||
|
|
||||||
credentials = ['loginn', 'passwo', 'testresour']
|
|
||||||
'''
|
|
||||||
[username, password, passphrase]
|
|
||||||
Script will autheticate itself with this credentials on above mentioned server.
|
|
||||||
'''
|
|
||||||
|
|
||||||
class TestNonBlockingClient(unittest.TestCase):
|
|
||||||
'''
|
|
||||||
Test Cases class for NonBlockingClient.
|
|
||||||
'''
|
|
||||||
|
|
||||||
def setUp(self):
|
|
||||||
'''
|
|
||||||
IdleQueue thread is run and dummy connection is created.
|
|
||||||
'''
|
|
||||||
|
|
||||||
self.idlequeue_thread = IdleQueueThread()
|
|
||||||
self.connection = MockConnectionClass()
|
|
||||||
|
|
||||||
self.idlequeue_thread.start()
|
|
||||||
|
|
||||||
def tearDown(self):
|
|
||||||
'''
|
|
||||||
IdleQueue thread is stopped.
|
|
||||||
'''
|
|
||||||
self.idlequeue_thread.stop_thread()
|
|
||||||
self.idlequeue_thread.join()
|
|
||||||
|
|
||||||
def open_stream(self, server_port):
|
|
||||||
'''
|
|
||||||
Method opening the XMPP connection. It returns when <stream:features>
|
|
||||||
is received from server.
|
|
||||||
|
|
||||||
:param server_port: tuple of (hostname, port) for where the client should
|
|
||||||
connect.
|
|
||||||
|
|
||||||
'''
|
|
||||||
self.client = client_nb.NonBlockingClient(
|
|
||||||
hostname=server_port[0],
|
|
||||||
port=server_port[1],
|
|
||||||
caller=self.connection,
|
|
||||||
idlequeue=self.idlequeue_thread.iq,
|
|
||||||
)
|
|
||||||
'''
|
|
||||||
NonBlockingClient instance with parameters from global variables and with
|
|
||||||
callbacks from dummy connection.
|
|
||||||
'''
|
|
||||||
|
|
||||||
self.client.connect(
|
|
||||||
on_connect=lambda *args: self.connection.on_connect(True, *args),
|
|
||||||
on_connect_failure=lambda *args: self.connection.on_connect(False, *args),
|
|
||||||
secure=False
|
|
||||||
)
|
|
||||||
|
|
||||||
print 'waiting for callback from client constructor'
|
|
||||||
self.connection.wait()
|
|
||||||
|
|
||||||
# if on_connect was called, client has to be connected and vice versa
|
|
||||||
if self.connection.connect_succeeded:
|
|
||||||
self.assert_(self.client.get_connect_type())
|
|
||||||
else:
|
|
||||||
self.assert_(not self.client.get_connect_type())
|
|
||||||
|
|
||||||
def client_auth(self, username, password, resource, sasl):
|
|
||||||
'''
|
|
||||||
Method authenticating connected client with supplied credentials. Returns
|
|
||||||
when authentication is over.
|
|
||||||
:param sasl: whether to use sasl (sasl=1) or old (sasl=0) authentication
|
|
||||||
|
|
||||||
:todo: to check and be more specific about when it returns (bind, session..)
|
|
||||||
'''
|
|
||||||
self.client.auth(username, password, resource, sasl,
|
|
||||||
on_auth=self.connection.on_auth)
|
|
||||||
|
|
||||||
print 'waiting for authentication...'
|
|
||||||
self.connection.wait()
|
|
||||||
|
|
||||||
def do_disconnect(self):
|
|
||||||
'''
|
|
||||||
Does disconnecting of connected client. Returns when TCP connection is closed.
|
|
||||||
'''
|
|
||||||
#self.client.start_disconnect(None, on_disconnect=self.connection.set_event)
|
|
||||||
self.client.RegisterDisconnectHandler(self.connection.set_event)
|
|
||||||
self.client.disconnect()
|
|
||||||
|
|
||||||
print 'waiting for disconnecting...'
|
|
||||||
self.connection.wait()
|
|
||||||
|
|
||||||
def test_proper_connect_sasl(self):
|
|
||||||
'''
|
|
||||||
The ideal testcase - client is connected, authenticated with SASL and
|
|
||||||
then disconnected.
|
|
||||||
'''
|
|
||||||
self.open_stream(xmpp_server_port)
|
|
||||||
|
|
||||||
# if client is not connected, lets raise the AssertionError
|
|
||||||
self.assert_(self.client.get_connect_type())
|
|
||||||
# (client.disconnect() is already called from NBClient._on_connected_failure
|
|
||||||
# so there's need to call it in this case
|
|
||||||
|
|
||||||
self.client_auth(credentials[0], credentials[1], credentials[2], sasl=1)
|
|
||||||
self.assert_(self.connection.con)
|
|
||||||
self.assert_(self.connection.auth=='sasl')
|
|
||||||
|
|
||||||
self.do_disconnect()
|
|
||||||
|
|
||||||
|
|
||||||
def test_proper_connect_oldauth(self):
|
|
||||||
'''
|
|
||||||
The ideal testcase - client is connected, authenticated with old auth and
|
|
||||||
then disconnected.
|
|
||||||
'''
|
|
||||||
self.open_stream(xmpp_server_port)
|
|
||||||
self.assert_(self.client.get_connect_type())
|
|
||||||
self.client_auth(credentials[0], credentials[1], credentials[2], sasl=0)
|
|
||||||
self.assert_(self.connection.con)
|
|
||||||
self.assert_(self.connection.auth=='old_auth')
|
|
||||||
self.do_disconnect()
|
|
||||||
|
|
||||||
def test_connect_to_nonexisting_host(self):
|
|
||||||
'''
|
|
||||||
Connect to nonexisting host. DNS request for A records should return nothing.
|
|
||||||
'''
|
|
||||||
self.open_stream(('fdsfsdf.fdsf.fss', 5222))
|
|
||||||
print 'nonexthost: %s' % self.client.get_connect_type()
|
|
||||||
self.assert_(not self.client.get_connect_type())
|
|
||||||
|
|
||||||
def test_connect_to_wrong_port(self):
|
|
||||||
'''
|
|
||||||
Connect to nonexisting host. DNS request for A records should return some IP
|
|
||||||
but there shouldn't be XMPP server running on specified port.
|
|
||||||
'''
|
|
||||||
self.open_stream((xmpp_server_port[0], 31337))
|
|
||||||
self.assert_(not self.client.get_connect_type())
|
|
||||||
|
|
||||||
def test_connect_with_wrong_creds(self):
|
|
||||||
'''
|
|
||||||
Connecting with invalid password.
|
|
||||||
'''
|
|
||||||
self.open_stream(xmpp_server_port)
|
|
||||||
self.assert_(self.client.get_connect_type())
|
|
||||||
self.client_auth(credentials[0], "wrong pass", credentials[2], sasl=1)
|
|
||||||
self.assert_(self.connection.auth is None)
|
|
||||||
self.do_disconnect()
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
if __name__ == '__main__':
|
|
||||||
|
|
||||||
suite = unittest.TestLoader().loadTestsFromTestCase(TestNonBlockingClient)
|
|
||||||
#suite = unittest.TestSuite()
|
|
||||||
#suite.addTest(TestNonBlockingClient('test_proper_connect_oldauth'))
|
|
||||||
#suite.addTest(TestNonBlockingClient('test_connect_to_nonexisting_host'))
|
|
||||||
|
|
||||||
unittest.TextTestRunner(verbosity=2).run(suite)
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
|
||||||
Loading…
Reference in New Issue