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 = ['login', 'pass', 'testclient']
'''
[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.client = client_nb.NonBlockingClient(
				server=xmpp_server_port[0],
				port=xmpp_server_port[1],
				on_connect=lambda *args: self.connection.on_connect(True, *args),
				on_connect_failure=lambda *args: self.connection.on_connect(False, *args),
				caller=self.connection 
				)
		'''
		NonBlockingClient instance with parameters from global variables and with
		callbacks from dummy connection.
		'''

		self.client.set_idlequeue(self.idlequeue_thread.iq)	
		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.connect(server_port)

		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.isConnected())
		else:
			self.assert_(not self.client.isConnected())

	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)

		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.isConnected())
		# (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.isConnected())
		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))
		self.assert_(not self.client.isConnected())

	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.isConnected())

	def test_connect_with_wrong_creds(self):
		'''
		Connecting with invalid password.
		'''
		self.open_stream(xmpp_server_port)
		self.assert_(self.client.isConnected())
		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_sasl'))

	unittest.TextTestRunner(verbosity=2).run(suite)