nrf_rpc: Add test for ot_message allocations.

Commit adds unit tests for ot message manipulation in nrf rpc Openthread
part.

Signed-off-by: Przemyslaw Bida <[email protected]>

subsys/net/openthread/rpc/Kconfig

@@ -59,6 +59,18 @@ config OPENTHREAD_RPC_CLIENT_NUM_SENT_COAP_REQUESTS
 
 endmenu # "OpenThread over RPC client configuration"
 
+menu "OpenThread over RPC client configuration"
+	depends on OPENTHREAD_RPC_SERVER
+
+	config OPENTHREAD_RPC_MESSAGE_POOL
+		int "Maximum number of messages that can be allocated on server"
+		default 8
+		help
+		  Defines maximum number of messages that can be allocated on server at
+		  the same time.
+
+endmenu
+
 config OPENTHREAD_RPC_INITIALIZE_NRF_RPC
 	bool "Automatically initialize nRF RPC library"
 	default n

subsys/net/openthread/rpc/client/ot_rpc_message.c

@@ -45,7 +45,7 @@ otMessage *otUdpNewMessage(otInstance *aInstance, const otMessageSettings *aSett
 
 	NRF_RPC_CBOR_ALLOC(&ot_group, ctx, sizeof(otMessageSettings) + 3);
 
-	nrf_rpc_encode_buffer(&ctx, (const void *)aSettings, sizeof(otMessageSettings));
+	ot_rpc_encode_message_settings(&ctx, aSettings);
 
 	nrf_rpc_cbor_cmd_rsp_no_err(&ot_group, OT_RPC_CMD_UDP_NEW_MESSAGE, &ctx);
 

subsys/net/openthread/rpc/server/ot_rpc_message.c

@@ -16,12 +16,16 @@
 
 #include <zephyr/net/openthread.h>
 
-#define OT_MESSAGES_POOL 8
+#define OT_MESSAGES_POOL CONFIG_OPENTHREAD_RPC_MESSAGE_POOL
 
 static otMessage *ot_message_registry[OT_MESSAGES_POOL];
 
 ot_msg_key ot_reg_msg_alloc(otMessage *msg)
 {
+	if (msg == NULL) {
+		return 0;
+	}
+
 	for (ot_msg_key i = 0; i < OT_MESSAGES_POOL; i++) {
 		if (ot_message_registry[i] == NULL) {
 			ot_message_registry[i] = msg;
@@ -118,9 +122,9 @@ static void ot_rpc_msg_udp_new(const struct nrf_rpc_group *group, struct nrf_rpc
 	ot_msg_key key;
 	struct nrf_rpc_cbor_ctx rsp_ctx;
 	otMessageSettings settings;
-	otMessageSettings *pSettings;
+	otMessageSettings *pSettings = &settings;
 
-	pSettings = nrf_rpc_decode_buffer(ctx, &settings, sizeof(otMessageSettings));
+	pSettings = ot_rpc_decode_message_settings(ctx, &settings);
 
 	if (!nrf_rpc_decoding_done_and_check(group, ctx)) {
 		ot_rpc_report_cmd_decoding_error(OT_RPC_CMD_UDP_NEW_MESSAGE);

subsys/net/openthread/rpc/server/ot_rpc_udp.c

@@ -104,6 +104,8 @@ static void ot_rpc_udp_open(const struct nrf_rpc_group *group, struct nrf_rpc_cb
 		return;
 	}
 
+	printf("dupa\n");
+
 	socket = nrf_udp_alloc_socket(key);
 
 	if (socket == NULL) {

tests/subsys/net/openthread/rpc/server/src/coap_suite.c

@@ -15,6 +15,8 @@
 
 #include <openthread/coap.h>
 
+#include "common_fakes.h"
+
 /* Message address used when testing serialization of a function that takes otMessage* */
 #define MSG_ADDR UINT32_MAX
 
@@ -28,12 +30,6 @@
 
 /* Fake functions */
 
-FAKE_VALUE_FUNC(otMessage *, otUdpNewMessage, otInstance *, const otMessageSettings *);
-FAKE_VALUE_FUNC(uint16_t, otMessageGetLength, const otMessage *);
-FAKE_VALUE_FUNC(uint16_t, otMessageGetOffset, const otMessage *);
-FAKE_VALUE_FUNC(uint16_t, otMessageRead, const otMessage *, uint16_t, void *, uint16_t);
-FAKE_VOID_FUNC(otMessageFree, otMessage *);
-FAKE_VALUE_FUNC(otError, otMessageAppend, otMessage *, const void *, uint16_t);
 FAKE_VALUE_FUNC(otMessage *, otCoapNewMessage, otInstance *, const otMessageSettings *);
 FAKE_VOID_FUNC(otCoapMessageInit, otMessage *, otCoapType, otCoapCode);
 FAKE_VALUE_FUNC(otError, otCoapMessageInitResponse, otMessage *, const otMessage *, otCoapType,
@@ -56,7 +52,6 @@ FAKE_VALUE_FUNC(otError, otCoapSendResponseWithParameters, otInstance *, otMessa
 		const otMessageInfo *, const otCoapTxParameters *);
 
 #define FOREACH_FAKE(f)                                                                            \
-	f(otUdpNewMessage);                                                                        \
 	f(otMessageGetLength);                                                                     \
 	f(otMessageGetOffset);                                                                     \
 	f(otMessageRead);                                                                          \

tests/subsys/net/openthread/rpc/server/src/common_fakes.c

@@ -0,0 +1,14 @@
+/*
+ * Copyright (c) 2024 Nordic Semiconductor ASA
+ *
+ * SPDX-License-Identifier: LicenseRef-Nordic-5-Clause
+ */
+
+#include "common_fakes.h"
+
+DEFINE_FAKE_VALUE_FUNC(uint16_t, otMessageGetLength, const otMessage *);
+DEFINE_FAKE_VALUE_FUNC(uint16_t, otMessageGetOffset, const otMessage *);
+DEFINE_FAKE_VALUE_FUNC(uint16_t, otMessageRead, const otMessage *, uint16_t, void *, uint16_t);
+DEFINE_FAKE_VOID_FUNC(otMessageFree, otMessage *);
+DEFINE_FAKE_VALUE_FUNC(otError, otMessageAppend, otMessage *, const void *, uint16_t);
+DEFINE_FAKE_VALUE_FUNC(otMessage *, otUdpNewMessage, otInstance *, const otMessageSettings *);

tests/subsys/net/openthread/rpc/server/src/common_fakes.h

@@ -0,0 +1,19 @@
+/*
+ * Copyright (c) 2024 Nordic Semiconductor ASA
+ *
+ * SPDX-License-Identifier: LicenseRef-Nordic-5-Clause
+ */
+
+#ifndef __COMMON_FAKES_H__
+#define __COMMON_FAKES_H__
+#include <zephyr/fff.h>
+#include <openthread/message.h>
+
+DECLARE_FAKE_VALUE_FUNC(otMessage *, otUdpNewMessage, otInstance *, const otMessageSettings *);
+DECLARE_FAKE_VALUE_FUNC(uint16_t, otMessageGetLength, const otMessage *);
+DECLARE_FAKE_VALUE_FUNC(uint16_t, otMessageGetOffset, const otMessage *);
+DECLARE_FAKE_VALUE_FUNC(uint16_t, otMessageRead, const otMessage *, uint16_t, void *, uint16_t);
+DECLARE_FAKE_VOID_FUNC(otMessageFree, otMessage *);
+DECLARE_FAKE_VALUE_FUNC(otError, otMessageAppend, otMessage *, const void *, uint16_t);
+
+#endif

tests/subsys/net/openthread/rpc/server/src/message_suite.c

@@ -0,0 +1,140 @@
+/*
+ * Copyright (c) 2024 Nordic Semiconductor ASA
+ *
+ * SPDX-License-Identifier: LicenseRef-Nordic-5-Clause
+ */
+
+#include <mock_nrf_rpc_transport.h>
+#include <ot_rpc_ids.h>
+#include <ot_rpc_message.h>
+#include <test_rpc_env.h>
+
+#include <zephyr/fff.h>
+#include <zephyr/kernel.h>
+#include <zephyr/ztest.h>
+
+#include <openthread/message.h>
+#include "common_fakes.h"
+#include <assert.h>
+#include <stddef.h>
+
+static void nrf_rpc_err_handler(const struct nrf_rpc_err_report *report)
+{
+	zassert_ok(report->code);
+}
+
+#define FOREACH_FAKE(f)                                                                            \
+	f(otMessageGetLength);                                                                     \
+	f(otMessageGetOffset);                                                                     \
+	f(otMessageRead);                                                                          \
+	f(otMessageFree);                                                                          \
+	f(otMessageAppend);                                                                        \
+	f(otUdpNewMessage);
+
+static void tc_setup(void *f)
+{
+	mock_nrf_rpc_tr_expect_add(RPC_INIT_REQ, RPC_INIT_RSP);
+	zassert_ok(nrf_rpc_init(nrf_rpc_err_handler));
+	mock_nrf_rpc_tr_expect_reset();
+
+	FOREACH_FAKE(RESET_FAKE);
+	FFF_RESET_HISTORY();
+}
+
+ZTEST(ot_rpc_message, test_otUdpNewMessage_failing)
+{
+	otUdpNewMessage_fake.return_val = NULL;
+
+	mock_nrf_rpc_tr_expect_add(RPC_RSP(0), NO_RSP);
+	mock_nrf_rpc_tr_receive(RPC_CMD(OT_RPC_CMD_UDP_NEW_MESSAGE, CBOR_NULL));
+
+	mock_nrf_rpc_tr_expect_add(RPC_RSP(0), NO_RSP);
+	mock_nrf_rpc_tr_receive(RPC_CMD(OT_RPC_CMD_UDP_NEW_MESSAGE, CBOR_TRUE, CBOR_UINT8(40)));
+	mock_nrf_rpc_tr_expect_done();
+
+	zassert_equal(otUdpNewMessage_fake.call_count, 2);
+}
+
+ZTEST(ot_rpc_message, test_otUdpNewMessage_free_working)
+{
+	char buf[16];
+	otUdpNewMessage_fake.return_val = (otMessage *)buf;
+
+	mock_nrf_rpc_tr_expect_add(RPC_RSP(1), NO_RSP);
+	mock_nrf_rpc_tr_receive(RPC_CMD(OT_RPC_CMD_UDP_NEW_MESSAGE, CBOR_NULL));
+
+	mock_nrf_rpc_tr_expect_add(RPC_RSP(2), NO_RSP);
+	mock_nrf_rpc_tr_receive(RPC_CMD(OT_RPC_CMD_UDP_NEW_MESSAGE, CBOR_TRUE, CBOR_UINT8(40)));
+
+	zassert_equal(otUdpNewMessage_fake.call_count, 2);
+
+	mock_nrf_rpc_tr_expect_add(RPC_RSP(), NO_RSP);
+	mock_nrf_rpc_tr_receive(RPC_CMD(OT_RPC_CMD_MESSAGE_FREE, 1));
+
+	mock_nrf_rpc_tr_expect_add(RPC_RSP(), NO_RSP);
+	mock_nrf_rpc_tr_receive(RPC_CMD(OT_RPC_CMD_MESSAGE_FREE, 2));
+
+	mock_nrf_rpc_tr_expect_done();
+
+	zassert_equal(otMessageFree_fake.call_count, 2);
+}
+
+ZTEST(ot_rpc_message, test_ot_reg_msg_alloc_free)
+{
+
+	for (size_t i = 1; i < CONFIG_OPENTHREAD_RPC_MESSAGE_POOL + 1; i++) {
+		zassert_equal(ot_reg_msg_alloc((otMessage *)i), i);
+	}
+
+	zassert_equal(ot_reg_msg_alloc((otMessage *)UINT32_MAX), 0);
+
+	for (size_t i = 1; i < CONFIG_OPENTHREAD_RPC_MESSAGE_POOL + 1; i++) {
+		ot_msg_free(i);
+	}
+}
+
+ZTEST(ot_rpc_message, test_ot_reg_msg_get)
+{
+
+	zassert_equal(ot_reg_msg_alloc((otMessage *)1), 1);
+
+	zassert_equal(ot_msg_get(1), (otMessage *)1);
+
+	zassert_equal(ot_msg_get(CONFIG_OPENTHREAD_RPC_MESSAGE_POOL), NULL);
+
+	ot_msg_free(1);
+}
+
+ZTEST(ot_rpc_message, test_get_length_offset)
+{
+	char buf[16];
+	otUdpNewMessage_fake.return_val = (otMessage *)buf;
+	otMessageGetLength_fake.return_val = 1;
+	otMessageGetOffset_fake.return_val = 1;
+
+	// OT_RPC_CMD_MESSAGE_GET_OFFSET
+
+	mock_nrf_rpc_tr_expect_add(RPC_RSP(1), NO_RSP);
+	mock_nrf_rpc_tr_receive(RPC_CMD(OT_RPC_CMD_UDP_NEW_MESSAGE, CBOR_NULL));
+
+	zassert_equal(otUdpNewMessage_fake.call_count, 1);
+
+	mock_nrf_rpc_tr_expect_add(RPC_RSP(1), NO_RSP);
+	mock_nrf_rpc_tr_receive(RPC_CMD(OT_RPC_CMD_MESSAGE_GET_LENGTH, 1));
+
+	mock_nrf_rpc_tr_expect_add(RPC_RSP(1), NO_RSP);
+	mock_nrf_rpc_tr_receive(RPC_CMD(OT_RPC_CMD_MESSAGE_GET_OFFSET, 1));
+
+	mock_nrf_rpc_tr_expect_add(RPC_RSP(0), NO_RSP);
+	mock_nrf_rpc_tr_receive(RPC_CMD(OT_RPC_CMD_MESSAGE_GET_LENGTH, 0));
+
+	mock_nrf_rpc_tr_expect_add(RPC_RSP(), NO_RSP);
+	mock_nrf_rpc_tr_receive(RPC_CMD(OT_RPC_CMD_MESSAGE_FREE, 1));
+
+	mock_nrf_rpc_tr_expect_done();
+
+	zassert_equal(otMessageGetLength_fake.call_count, 1);
+	zassert_equal(otMessageFree_fake.call_count, 1);
+}
+
+ZTEST_SUITE(ot_rpc_message, NULL, NULL, tc_setup, NULL, NULL);