diff --git a/test/integration/test_resolver.py b/test/integration/test_resolver.py index 07ddcb7a9..1b281edd4 100644 --- a/test/integration/test_resolver.py +++ b/test/integration/test_resolver.py @@ -10,7 +10,6 @@ from gi.repository import GLib from gajim.common import resolver from gajim_mocks import * -from xmpp_mocks import IdleQueueThread NONSENSE_NAME = 'sfsdfsdfsdf.sdfs.fsd' JABBERCZ_TXT_NAME = '_xmppconnect.jabber.cz' diff --git a/test/lib/xmpp_mocks.py b/test/lib/xmpp_mocks.py deleted file mode 100644 index 6617e27ae..000000000 --- a/test/lib/xmpp_mocks.py +++ /dev/null @@ -1,94 +0,0 @@ -''' -Module with dummy classes for unit testing of XMPP and related code. -''' - -import threading, time - -from mock import Mock - -from nbxmpp import idlequeue - -IDLEQUEUE_INTERVAL = 0.2 # polling interval. 200ms is used in Gajim as default -IDLEMOCK_TIMEOUT = 30 # how long we wait for an event - -class IdleQueueThread(threading.Thread): - ''' - Thread for regular processing of idlequeue. - ''' - def __init__(self): - self.iq = idlequeue.IdleQueue() - self.stop = threading.Event() # Event to stop the thread main loop. - self.stop.clear() - threading.Thread.__init__(self) - - def run(self): - while not self.stop.isSet(): - self.iq.process() - time.sleep(IDLEQUEUE_INTERVAL) - - def stop_thread(self): - self.stop.set() - - -class IdleMock: - ''' - Serves as template for testing objects that are normally controlled by GUI. - Allows to wait for asynchronous callbacks with wait() method. - ''' - def __init__(self): - self._event = threading.Event() - self._event.clear() - - def wait(self): - ''' - Block until some callback sets the event and clearing the event - subsequently. - Returns True if event was set, False on timeout - ''' - self._event.wait(IDLEMOCK_TIMEOUT) - if self._event.isSet(): - self._event.clear() - return True - else: - return False - - def set_event(self): - self._event.set() - -class MockConnection(IdleMock, Mock): - ''' - Class simulating Connection class from gajim/common/connection.py - - It is derived from Mock in order to avoid defining all methods - from real Connection that are called from NBClient or Dispatcher - ( _event_dispatcher for example) - ''' - - def __init__(self, *args): - self.connect_succeeded = True - IdleMock.__init__(self) - Mock.__init__(self, *args) - - def on_connect(self, success, *args): - ''' - Method called after connecting - after receiving - from server (NOT after TLS stream restart) or connect failure - ''' - self.connect_succeeded = success - self.set_event() - - - def on_auth(self, con, auth): - ''' - Method called after authentication, regardless of the result. - - :Parameters: - con : NonBlockingClient - reference to authenticated object - auth : string - type of authetication in case of success ('old_auth', 'sasl') or - None in case of auth failure - ''' - self.auth_connection = con - self.auth = auth - self.set_event()