_rmr_alloc_msg.restype = POINTER(rmr_mbuf_t)
-def rmr_alloc_msg(vctx, size):
+def rmr_alloc_msg(vctx, size, payload=None, gen_transaction_id=False, mtype=None, meid=None):
"""
Refer to the rmr C documentation for rmr_alloc_msg
extern rmr_mbuf_t* rmr_alloc_msg(void* vctx, int size)
+
+ if payload is not None, attempts to set the payload
+ if gen_transaction_id is True, it generates and sets a transaction id
+ if mtype is not None, sets the sbuf's message type
+ if meid is not None, sets the sbuf's meid
+
"""
sbuf = _rmr_alloc_msg(vctx, size)
+ # make sure it's good
try:
sbuf.contents
+ if payload:
+ set_payload_and_length(payload, sbuf)
+
+ if gen_transaction_id:
+ generate_and_set_transaction_id(sbuf)
+
+ if mtype:
+ sbuf.contents.mtype = mtype
+
+ if meid:
+ rmr_set_meid(sbuf, meid)
+
+ return sbuf
+
except ValueError:
raise BadBufferAllocation
- return sbuf
_rmr_free_msg = rmr_c_lib.rmr_free_msg
_rmr_bytes2meid.restype = c_int
-def rmr_bytes2meid(ptr_mbuf, src, length):
+def rmr_set_meid(ptr_mbuf, byte_str):
"""
Refer to the rmr C documentation for rmr_bytes2meid
extern int rmr_bytes2meid(rmr_mbuf_t* mbuf, unsigned char const* src, int len);
"""
- return _rmr_bytes2meid(ptr_mbuf, src, length)
-
+ return _rmr_bytes2meid(ptr_mbuf, byte_str, len(byte_str))
-# this is an alias to rmr_bytes2meid using familiar set/get terminoloigy
-rmr_set_meid = rmr_bytes2meid
# CAUTION: Some of the C functions expect a mutable buffer to copy the bytes into;
# if there is a get_* function below, use it to set up and return the
sz = _get_constants().get("RMR_MAX_MEID", 64) # size for buffer to fill
buf = create_string_buffer(sz)
_rmr_get_meid(ptr_mbuf, buf)
- return buf.value.decode() # decode turns into a string
+ return buf.value
_rmr_get_src = rmr_c_lib.rmr_get_src
assert summary["transaction id"] == b""
assert summary["message state"] == 0
assert summary["message status"] == "RMR_OK"
- assert summary["meid"] == ""
+ assert summary["meid"] == b""
assert summary["errno"] == 0
"""
sbuf = rmr.rmr_alloc_msg(MRC_SEND, SIZE)
- rmr.rmr_set_meid(sbuf, b"\x01\x02", 2)
- assert rmr.rmr_get_meid(sbuf) == rmr.message_summary(sbuf)["meid"] == "\x01\x02"
+ rmr.rmr_set_meid(sbuf, b"\x01\x02")
+ assert rmr.rmr_get_meid(sbuf) == rmr.message_summary(sbuf)["meid"] == b"\x01\x02"
assert len(rmr.rmr_get_meid(sbuf)) == 2
- rmr.rmr_set_meid(sbuf, b"\x00" * 32, 32)
- assert rmr.rmr_get_meid(sbuf) == rmr.message_summary(sbuf)["meid"] == "" # NULL bytes get truncated
+ rmr.rmr_set_meid(sbuf, b"\x00" * 32)
+ assert rmr.rmr_get_meid(sbuf) == rmr.message_summary(sbuf)["meid"] == b"" # NULL bytes get truncated
- rmr.rmr_set_meid(sbuf, b"6" * 32, 32)
- assert rmr.rmr_get_meid(sbuf) == rmr.message_summary(sbuf)["meid"] == "6" * 32 # string in string out
+ rmr.rmr_set_meid(sbuf, b"6" * 32)
+ assert rmr.rmr_get_meid(sbuf) == rmr.message_summary(sbuf)["meid"] == b"6" * 32 # string in string out
- rmr.rmr_set_meid(sbuf, b"\x01\x02", 2)
+ rmr.rmr_set_meid(sbuf, b"\x01\x02")
assert (
- rmr.rmr_get_meid(sbuf) == rmr.message_summary(sbuf)["meid"] == "\x01\x02" + "6" * 30
+ rmr.rmr_get_meid(sbuf) == rmr.message_summary(sbuf)["meid"] == b"\x01\x02" + b"6" * 30
) # bytes in string out, 6s left over
assert len(rmr.rmr_get_meid(sbuf)) == 32
assert len(summary["transaction id"]) == 32
# test meid
- assert rmr.rmr_get_meid(sbuf) == summary["meid"] == ""
- rmr.rmr_set_meid(sbuf, b"666\x01\x00\x01", 6)
+ assert rmr.rmr_get_meid(sbuf) == summary["meid"] == b""
+ rmr.rmr_set_meid(sbuf, b"666\x01\x00\x01")
summary = rmr.message_summary(sbuf)
- assert rmr.rmr_get_meid(sbuf) == summary["meid"] == "666\x01"
+ assert rmr.rmr_get_meid(sbuf) == summary["meid"] == b"666\x01"
assert (len(summary["meid"])) == 4
"""test that we get a proper exception when the buffer has a null pointer"""
with pytest.raises(exceptions.BadBufferAllocation):
rmr.rmr_alloc_msg(None, 4096)
+
+
+def test_alloc_fancy():
+ """test allocation with setting payload, trans, mtype"""
+ pay = b"yoo\x01\x00\x80"
+ sbuf = rmr.rmr_alloc_msg(MRC_SEND, SIZE, payload=pay, gen_transaction_id=True, mtype=14, meid=b"asdf")
+ summary = rmr.message_summary(sbuf)
+ assert summary["payload"] == pay
+ assert summary["payload length"] == 6
+ assert summary["transaction id"] != b"" # hard to test what it will be, but make sure not empty
+ assert summary["message state"] == 0
+ assert summary["message type"] == 14
+ assert summary["meid"] == b"asdf"