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'))
