def test_queue():
r"""Test creation/removal of queue."""
mq = IPCComm.get_queue()
key = str(mq.key)
assert (CommBase.is_registered('IPCComm', key))
CommBase.unregister_comm('IPCComm', key, dont_close=True)
nt.assert_raises(KeyError, IPCComm.remove_queue, mq)
CommBase.register_comm('IPCComm', key, mq)
IPCComm.remove_queue(mq)
assert (not CommBase.is_registered('IPCComm', key))
def close_sockets(self):
r"""Close the sockets."""
self.debug('Closing sockets')
if self.cli_socket:
self.cli_socket.close()
self.cli_socket = None
if self.srv_socket:
self.srv_socket.close()
self.srv_socket = None
CommBase.unregister_comm('ZMQComm',
'ROUTER_server_' + self.cli_address)
CommBase.unregister_comm('ZMQComm',
'DEALER_server_' + self.srv_address)
def remove_queue(mq):
r"""Remove a sysv_ipc.MessageQueue and unregister it.
Args:
mq (:class:`sysv_ipc.MessageQueue`) Message queue.
Raises:
KeyError: If the provided queue is not registered.
"""
key = str(mq.key)
if not CommBase.is_registered('IPCComm', key):
raise KeyError("Queue not registered.")
CommBase.unregister_comm('IPCComm', key)
def test_registry():
r"""Test registry of comm."""
comm_class = 'CommBase'
key = 'key1'
value = None
assert(not CommBase.is_registered(comm_class, key))
assert(not CommBase.unregister_comm(comm_class, key))
nt.assert_equal(CommBase.get_comm_registry(None), {})
nt.assert_equal(CommBase.get_comm_registry(comm_class), {})
CommBase.register_comm(comm_class, key, value)
assert(key in CommBase.get_comm_registry(comm_class))
assert(CommBase.is_registered(comm_class, key))
assert(not CommBase.unregister_comm(comm_class, key, dont_close=True))
CommBase.register_comm(comm_class, key, value)
assert(not CommBase.unregister_comm(comm_class, key))
def terminate(self, *args, **kwargs):
CommBase.unregister_comm('IPCComm', self.srv_address)
super(IPCServer, self).terminate(*args, **kwargs)