def test_disconnect_event(): """ Ensure on_disconnect() is triggered on terminate(). """ identity = unique_identity() class ConnectPlugin(object): def __init__(self): self.got_connect = False self.got_disconnect = False def bind(self, client): signals.on_connect.connect(self.connected, sender=client) signals.on_disconnect.connect(self.disconnected, sender=client) def connected(self, client): self.got_connect = True def disconnected(self, client): self.got_disconnect = True connect_plugin = ConnectPlugin() client = CoreClient(identity, 'localhost', plugins=[connect_plugin]) result = client.connect() assert(result.get() is True) client.terminate() assert(connect_plugin.got_connect) assert(connect_plugin.got_disconnect)
def test_connect_success(): """ Ensure a basic connection can be successfully created. """ identity = unique_identity() client = CoreClient(identity, 'localhost') result = client.connect() assert(result.get() is True)
def test_handshake_success(): """ Ensure the handshake works when configured properly. """ identity = unique_identity() rec_plugin = RecPlugin(terminate_on=('001',)) client = CoreClient(identity, 'localhost', plugins=[ HandshakePlugin, rec_plugin ]) result = client.connect() assert(result.get() is True) client._io_workers.join(timeout=5) assert(rec_plugin.did_receive('001'))