def _init_sockets(self, control_socket, event_socket):
logging.info("Initializing message handlers")
zmqctx = zmq.Context.instance()
self._socket_notifiers = []
self._control_socket = control_socket
self._control_socket_queue = messaging.MessageQueue(
control_socket, "control socket queue",
messaging.REPLY_SUCCESS_PREFIX,
{
HELLO_COMMAND: self._handle_hello_reply,
GET_COMMAND: self._handle_get_reply,
CALL_COMMAND: self._handle_call_reply,
PLAY_COMMAND: self._handle_play_reply,
})
self._connect_socket_to_notifier(
control_socket, self._control_socket_queue)
self._event_socket = event_socket
self._event_socket_queue = messaging.MessageQueue(
event_socket, "event socket queue", [],
{
DEAL_COMMAND: self._handle_deal_event,
CALL_COMMAND: self._handle_call_event,
BIDDING_COMMAND: self._handle_bidding_event,
PLAY_COMMAND: self._handle_play_event,
DUMMY_COMMAND: self._handle_dummy_event,
TRICK_COMMAND: self._handle_trick_event,
DEALEND_COMMAND: self._handle_dealend_event,
})
self._connect_socket_to_notifier(event_socket, self._event_socket_queue)
sendCommand(control_socket, HELLO_COMMAND)
def testSendCommand(self):
sendCommand(self._front_socket, COMMAND, arg=1)
self.assertEqual(
self._back_socket.recv_multipart(flags=zmq.NOBLOCK),
[b'', COMMAND, b'arg', b'1'])
def _send_play_command(self, card):
sendCommand(self._control_socket, PLAY_COMMAND, card=card._asdict())
def _send_call_command(self, call):
sendCommand(self._control_socket, CALL_COMMAND, call=call)
def _request(self, *args):
sendCommand(self._control_socket, GET_COMMAND, keys=args)
def testSendCommand(self):
sendCommand(self._front_socket, COMMAND, arg=1)
self.assertEqual(
self._back_socket.recv_multipart(flags=zmq.NOBLOCK),
[b'', COMMAND, COMMAND, b'arg', b'1'])
