gnh1201/wasm-micro-runtime
1
0
Fork 0
mirror of https://github.com/bytecodealliance/wasm-micro-runtime.git synced 2026-04-18 18:18:44 +00:00
Code Issues Actions 18 Packages Projects Releases Wiki Activity

fix(tests): resolve memory leak in bh_queue_test_suite and improve message handling

Browse Source
This commit is contained in:
liang.he@intel.com 2026-03-30 14:33:10 +08:00
parent 9ec6237241
commit 0cb713d3af
1 changed files with 54 additions and 13 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

67
tests/unit/shared-utils/bh_queue_test.cc
View File

@ -8,6 +8,9 @@
#include "bh_platform.h" #include "bh_platform.h"
// FIXME: Resolve memory leak in bh_queue_test_suite.
// It includes release created queue and messages.
class bh_queue_test_suite : public testing::Test class bh_queue_test_suite : public testing::Test
{ {
protected: protected:
@ -87,6 +90,14 @@ enum {
// If RES_CMP == 1, the function bh_queue_enter_loop_run run error. // If RES_CMP == 1, the function bh_queue_enter_loop_run run error.
int RES_CMP = 0; int RES_CMP = 0;
/* Don't touch .ro data in msg body */
static void
local_ro_msg_body_cleaner(void *body)
{
(void)body;
return;
}
TEST_F(bh_queue_test_suite, bh_queue_create) TEST_F(bh_queue_test_suite, bh_queue_create)
{ {
EXPECT_NE(nullptr, bh_queue_create()); EXPECT_NE(nullptr, bh_queue_create());
@ -111,33 +122,44 @@ TEST_F(bh_queue_test_suite, bh_message_payload)
{ {
bh_message_t msg_ptr; bh_message_t msg_ptr;
msg_ptr = bh_new_msg(RESTFUL_REQUEST, (void *)"test_msg_body", msg_ptr =
sizeof("test_msg_body"), nullptr); bh_new_msg(RESTFUL_REQUEST, (void *)"test_msg_body",
sizeof("test_msg_body"), (void *)local_ro_msg_body_cleaner);
EXPECT_EQ("test_msg_body", bh_message_payload(msg_ptr)); EXPECT_EQ("test_msg_body", bh_message_payload(msg_ptr));
bh_free_msg(msg_ptr);
} }
TEST_F(bh_queue_test_suite, bh_message_payload_len) TEST_F(bh_queue_test_suite, bh_message_payload_len)
{ {
bh_message_t msg_ptr; bh_message_t msg_ptr;
msg_ptr = bh_new_msg(RESTFUL_REQUEST, (void *)"test_msg_body", msg_ptr =
sizeof("test_msg_body"), nullptr); bh_new_msg(RESTFUL_REQUEST, (void *)"test_msg_body",
sizeof("test_msg_body"), (void *)local_ro_msg_body_cleaner);
EXPECT_EQ(sizeof("test_msg_body"), bh_message_payload_len(msg_ptr)); EXPECT_EQ(sizeof("test_msg_body"), bh_message_payload_len(msg_ptr));
bh_free_msg(msg_ptr);
} }
TEST_F(bh_queue_test_suite, bh_message_type) TEST_F(bh_queue_test_suite, bh_message_type)
{ {
bh_message_t msg_ptr; bh_message_t msg_ptr;
msg_ptr = bh_new_msg(RESTFUL_REQUEST, (void *)"test_msg_body", msg_ptr =
sizeof("test_msg_body"), nullptr); bh_new_msg(RESTFUL_REQUEST, (void *)"test_msg_body",
sizeof("test_msg_body"), (void *)local_ro_msg_body_cleaner);
EXPECT_EQ(RESTFUL_REQUEST, bh_message_type(msg_ptr)); EXPECT_EQ(RESTFUL_REQUEST, bh_message_type(msg_ptr));
bh_free_msg(msg_ptr);
} }
TEST_F(bh_queue_test_suite, bh_new_msg) TEST_F(bh_queue_test_suite, bh_new_msg)
{ {
EXPECT_NE(nullptr, bh_new_msg(RESTFUL_REQUEST, (void *)"test_msg_body", bh_message_t msg_ptr =
sizeof("test_msg_body"), nullptr)); bh_new_msg(RESTFUL_REQUEST, (void *)"test_msg_body",
sizeof("test_msg_body"), (void *)local_ro_msg_body_cleaner);
EXPECT_NE(nullptr, msg_ptr);
bh_free_msg(msg_ptr);
} }
void void
@ -215,11 +237,14 @@ TEST_F(bh_queue_test_suite, bh_queue_get_message_count)
bh_message_t msg_ptr; bh_message_t msg_ptr;
bh_queue *queue_ptr = bh_queue_create(); bh_queue *queue_ptr = bh_queue_create();
// Normally.
msg_ptr = bh_new_msg(RESTFUL_REQUEST, (void *)"test_msg_body",
sizeof("test_msg_body"), nullptr);
for (i = 1; i <= 20; i++) { for (i = 1; i <= 20; i++) {
bh_post_msg2(queue_ptr, msg_ptr); msg_ptr = bh_new_msg(RESTFUL_REQUEST, (void *)"test_msg_body",
sizeof("test_msg_body"),
(void *)local_ro_msg_body_cleaner);
EXPECT_NE(nullptr, msg_ptr);
bool post_result = bh_post_msg2(queue_ptr, msg_ptr);
EXPECT_EQ(true, post_result);
} }
i = i - 1; i = i - 1;
// The count of msg is less than queue_ptr->max. // The count of msg is less than queue_ptr->max.
@ -227,7 +252,21 @@ TEST_F(bh_queue_test_suite, bh_queue_get_message_count)
// The count of msg is more than queue_ptr->max. // The count of msg is more than queue_ptr->max.
for (j = 1; j <= 60; j++) { for (j = 1; j <= 60; j++) {
bh_post_msg2(queue_ptr, msg_ptr); msg_ptr = bh_new_msg(RESTFUL_REQUEST, (void *)"test_msg_body",
sizeof("test_msg_body"),
(void *)local_ro_msg_body_cleaner);
EXPECT_NE(nullptr, msg_ptr);
bool post_result = bh_post_msg2(queue_ptr, msg_ptr);
// The first 30 messages should be posted successfully, and the rest
// should be dropped.
if (j <= 30) {
EXPECT_EQ(true, post_result);
}
else {
EXPECT_EQ(false, post_result);
}
} }
j = j - 1; j = j - 1;
EXPECT_EQ(queue_ptr->max, bh_queue_get_message_count(queue_ptr)); EXPECT_EQ(queue_ptr->max, bh_queue_get_message_count(queue_ptr));
@ -235,6 +274,8 @@ TEST_F(bh_queue_test_suite, bh_queue_get_message_count)
// Illegal parameters. // Illegal parameters.
EXPECT_EQ(0, bh_queue_get_message_count(nullptr)); EXPECT_EQ(0, bh_queue_get_message_count(nullptr));
bh_queue_destroy(queue_ptr);
} }
void void