def test_request_call_writes_request_message(self):
conn = mock.Mock()
doc = {'amessage': 'foru'}
requester = request.RequestRPC(doc, conn, "XYZ")
requester.send()
(param_list, keywords) = conn.send.call_args
send_doc = param_list[0]
self._validate_request_message(send_doc, doc)
self.assertEqual(conn.send.call_count, 1)
def test_request_no_ack_timeout(self):
conn = mock.Mock()
doc = {'amessage': 'foru'}
requester = request.RequestRPC(doc, conn, "XYZ", timeout=1)
requester.send()
dcm_events.poll(timeblock=1.5)
self.assertGreater(conn.send.call_count, 1)
(param_list, keywords) = conn.send.call_args
self._validate_request_message(param_list[0], doc)
requester.cleanup()
def test_standard_with_callback_path(self):
self.called = False
def reply_called(*args, **kwargs):
self.called = True
conn = mock.Mock()
doc = {'amessage': 'foru'}
requester = request.RequestRPC(doc,
conn,
"XYZ",
reply_callback=reply_called)
requester.send()
self.assertEqual(conn.send.call_count, 1)
(param_list, keywords) = conn.send.call_args
send_doc = param_list[0]
reply_doc = {
'type': types.MessageTypes.ACK,
'message_id': send_doc['message_id'],
'request_id': send_doc['request_id']
}
requester.incoming_message(reply_doc)
self.assertEqual('REQUESTED', requester._sm._current_state)
reply_doc = {
'type': types.MessageTypes.REPLY,
'message_id': send_doc['message_id'],
'request_id': send_doc['request_id']
}
requester.incoming_message(reply_doc)
while requester._sm._current_state !=\
states.RequesterStates.ACK_SENT:
dcm_events.poll()
self.assertEqual(states.RequesterStates.ACK_SENT,
requester._sm._current_state)
(param_list, keywords) = conn.send.call_args
send_doc = param_list[0]
self.assertEqual(reply_doc['message_id'], send_doc['message_id'])
self.assertTrue('request_id' in send_doc)
self.assertTrue('type' in send_doc)
self.assertEqual(send_doc['type'], types.MessageTypes.ACK)
requester.ack_sent_timeout()
def test_reply_after_ack(self):
conn = mock.Mock()
doc = {'amessage': 'foru'}
requester = request.RequestRPC(doc, conn, "XYZ")
requester.send()
self.assertEqual(conn.send.call_count, 1)
(param_list, keywords) = conn.send.call_args
send_doc = param_list[0]
reply_doc = {
'type': types.MessageTypes.ACK,
'message_id': send_doc['message_id'],
'request_id': send_doc['request_id']
}
requester.incoming_message(reply_doc)
self.assertEqual('REQUESTED', requester._sm._current_state)
reply_doc = {
'type': types.MessageTypes.REPLY,
'message_id': send_doc['message_id'],
'request_id': send_doc['request_id']
}
requester.incoming_message(reply_doc)
self.assertEqual(states.RequesterStates.USER_CALLBACK,
requester._sm._current_state)
requester.incoming_message(reply_doc)
reply = requester.get_reply()
requester.got_reply()
self.assertEqual(reply, reply_doc)
requester.incoming_message(reply_doc)
(param_list, keywords) = conn.send.call_args
send_doc = param_list[0]
self.assertEqual(reply_doc['message_id'], send_doc['message_id'])
self.assertTrue('request_id' in send_doc)
self.assertTrue('type' in send_doc)
self.assertEqual(send_doc['type'], types.MessageTypes.ACK)
requester.ack_sent_timeout()
requester.cleanup()
def test_requesting_reply(self):
conn = mock.Mock()
doc = {'amessage': 'foru'}
requester = request.RequestRPC(doc, conn, "XYZ")
requester.send()
self.assertEqual(conn.send.call_count, 1)
(param_list, keywords) = conn.send.call_args
send_doc = param_list[0]
reply_doc = {
'type': types.MessageTypes.REPLY,
'message_id': send_doc['message_id']
}
requester.incoming_message(reply_doc)
self.assertEqual(states.RequesterStates.USER_CALLBACK,
requester._sm._current_state)
def _rpc_wait_reply(self, doc):
def reply_callback():
pass
reqRPC = request.RequestRPC(doc,
self.req_conn,
self.agent_id,
reply_callback=reply_callback)
self.reply_conn.set_request_side(reqRPC)
reqRPC.send()
# wait for message completion:
while not self._event.isSet():
dcm_events.poll()
self._event.clear()
reqRPC.cleanup()
self.shutting_down = False
return reqRPC