def send_burst(mrc, fmt, mtype=1, num=13, counter=0): """ Internal function to support test_rcv_all. Send a burst of messages optionally giving the type, payload and number to send. """ sbuf_send = rmr.rmr_alloc_msg(MRC_SEND, SIZE) # seed message buffer for i in range(num): payload = bytes(fmt % counter, "UTF-8") counter += 1 rmr.set_payload_and_length(payload, sbuf_send) sbuf_send.contents.mtype = mtype max_retries = 5 while max_retries > 0: sbuf_send = rmr.rmr_send_msg(mrc, sbuf_send) ms = rmr.message_summary(sbuf_send) if ms["message state"] != 10: # 10 is retry break max_retries -= 1 time.sleep(0.75) assert ms["message state"] == 0 assert max_retries > 0
def test_send_rcv_subid_good(): """ test send and receive where subid is used for routing """ pay = b"\x01\x00\x80" test_mtype = 46656 test_subid = 777 # send a message sbuf_send = rmr.rmr_alloc_msg(MRC_SEND, 3, pay, mtype=test_mtype, sub_id=test_subid) pre_send_summary = rmr.message_summary(sbuf_send) sbuf_send = rmr.rmr_send_msg(MRC_SEND, sbuf_send) send_summary = rmr.message_summary(sbuf_send) # receive it in other context time.sleep(0.5) sbuf_rcv = rmr.rmr_alloc_msg(MRC_RCV, 3) sbuf_rcv = rmr.rmr_torcv_msg(MRC_RCV, sbuf_rcv, 2000) rcv_summary = rmr.message_summary(sbuf_rcv) # asserts assert send_summary["message state"] == rcv_summary["message state"] == 0 assert send_summary["message status"] == rcv_summary[ "message status"] == "RMR_OK" assert pre_send_summary["payload"] == rcv_summary["payload"] == pay assert pre_send_summary["message type"] == rcv_summary[ "message type"] == test_mtype assert pre_send_summary["subscription id"] == rcv_summary[ "subscription id"] == test_subid
def test_send_rcv_subid_bad_mtype(): """ test send and receive where subid is used for routing but nobody recieves this mtype """ sbuf_send = rmr.rmr_alloc_msg(MRC_SEND, 3, b"\x01\x00\x80", mtype=46657, sub_id=777) sbuf_send = rmr.rmr_send_msg(MRC_SEND, sbuf_send) assert rmr.message_summary(sbuf_send)["message state"] == 2 assert rmr.message_summary( sbuf_send)["message status"] == "RMR_ERR_NOENDPT"
def _send_msg(self, pay, mtype, subid): """ sends a msg """ for _ in range(0, RETRY_TIMES): sbuf = rmr.rmr_alloc_msg(self.mrc, len(pay), payload=pay, gen_transaction_id=True, mtype=mtype, sub_id=subid) sbuf.contents.sub_id = subid pre_send_summary = rmr.message_summary(sbuf) sbuf = rmr.rmr_send_msg(self.mrc, sbuf) # send if self._assert_good_send(sbuf, pre_send_summary): rmr.rmr_free_msg(sbuf) # free break
def test_send_rcv(): """ test send and receive """ pay = b"\x01\x00\x80" # send a message sbuf_send = rmr.rmr_alloc_msg(MRC_SEND, SIZE) _assert_new_sbuf(sbuf_send) rmr.set_payload_and_length(pay, sbuf_send) sbuf_send.contents.mtype = 0 sbuf_send = rmr.rmr_send_msg(MRC_SEND, sbuf_send) send_summary = rmr.message_summary(sbuf_send) assert send_summary[ "message state"] == 0 # if send fails don't attempt receive assert send_summary["message status"] == "RMR_OK" time.sleep(0.5) # receive it in other context sbuf_rcv = rmr.rmr_alloc_msg(MRC_RCV, SIZE) sbuf_rcv = rmr.rmr_torcv_msg(MRC_RCV, sbuf_rcv, 2000) rcv_summary = rmr.message_summary(sbuf_rcv) assert rcv_summary["message state"] == 0 assert rcv_summary["message status"] == "RMR_OK" assert rcv_summary["message type"] == 0 assert rcv_summary["payload"] == pay # send an ACK back ack_pay = b"message received" sbuf_rcv = rmr.rmr_rts_msg(MRC_RCV, sbuf_rcv, payload=ack_pay, mtype=6666) rcv_ack_summary = rmr.message_summary(sbuf_rcv) # have the sender receive it sbuf_send = rmr.rmr_torcv_msg(MRC_SEND, sbuf_send, 2000) send_ack_summary = rmr.message_summary(sbuf_send) assert send_ack_summary["message state"] == rcv_ack_summary[ "message state"] == 0 assert send_ack_summary["message status"] == rcv_ack_summary[ "message status"] == "RMR_OK" assert send_ack_summary["payload"] == ack_pay assert send_ack_summary["message type"] == 6666
def test_send_mock(monkeypatch): """ tests the send mock """ monkeypatch.setattr("rmr.rmr.rmr_send_msg", rmr_mocks.send_mock_generator(12)) rmr_mocks.patch_rmr(monkeypatch) sbuf = rmr.rmr_alloc_msg(MRC, SIZE) rmr.set_payload_and_length("testttt".encode("utf8"), sbuf) expected = { "meid": None, "message source": "localtest:80", "message state": 0, "message type": 0, "message status": "RMR_OK", "payload": b"testttt", "payload length": 7, "payload max size": 4096, "subscription id": 0, } _partial_dict_comparison(expected, rmr.message_summary(sbuf)) # set the mtype sbuf.contents.mtype = 666 # send it (the fake send sets the state, and touches nothing else) sbuf = rmr.rmr_send_msg(MRC, sbuf) expected = { "meid": None, "message source": "localtest:80", "message state": 12, "message type": 666, "message status": "RMR_ERR_TIMEOUT", "payload": None, "payload length": 7, "payload max size": 4096, "subscription id": 0, } _partial_dict_comparison(expected, rmr.message_summary(sbuf))
mrc = rmr.rmr_init(PORT.encode("utf-8"), rmr.RMR_MAX_RCV_BYTES, rmr.RMRFL_MTCALL) test_type = 1006001 while rmr.rmr_ready(mrc) == 0: time.sleep(1) print("not yet ready") print("listening ON {}".format(PORT)) # loop while True: # do query pay = {"policy_type_id": test_type} sbuf_send = rmr.rmr_alloc_msg(mrc, 4096, payload=json.dumps(pay).encode("utf-8"), gen_transaction_id=True, mtype=20012) sbuf_send = rmr.rmr_send_msg(mrc, sbuf_send) post_send_summary = rmr.message_summary(sbuf_send) if not (post_send_summary["message state"] == 0 and post_send_summary["message status"] == "RMR_OK"): print("was unable to send query to a1!") time.sleep(1) else: # query worked, wait 2 seconds, then receive everything we have time.sleep(1) print("reading messages") # this is a hacked up version of rmr_rcvall_msgs in the rmr package # we need the actual messages, not the summaries, to use rts sbuf_rcv = rmr.rmr_alloc_msg(mrc, 4096) # allocate buffer to have something for a return status while True: sbuf_rcv = rmr.rmr_torcv_msg(mrc, sbuf_rcv, 0) # set the timeout to 0 so this doesn't block!!
sbuf = rmr.rmr_alloc_msg(mrc, 256) # capture ctrl-c signal.signal(signal.SIGINT, signal_handler) while True: # generate a random value between 1 and 256 bytes, then gen some random bytes with several nulls thrown in for val in [ ''.join([ random.choice(string.ascii_letters + string.digits) for n in range(random.randint(1, 256)) ]).encode("utf8"), b"\x00" + os.urandom(4) + b"\x00" + os.urandom(4) + b"\x00" ]: rmr.set_payload_and_length(val, sbuf) rmr.generate_and_set_transaction_id(sbuf) sbuf.contents.state = 0 sbuf.contents.mtype = 0 print("Pre send summary: {}".format(rmr.message_summary(sbuf))) sbuf = rmr.rmr_send_msg(mrc, sbuf) print("Post send summary: {}".format(rmr.message_summary(sbuf))) print("Waiting for return, will timeout after 2000ms") sbuf = rmr.rmr_torcv_msg(mrc, sbuf, 2000) summary = rmr.message_summary(sbuf) if summary['message state'] == 12: print("Nothing received yet") else: print("Ack Message received!: {}".format(summary)) time.sleep(1)