gfwleak/stellar-stellar
Archived
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Move test cases to test dir

Browse Source
This commit is contained in:
luwenpeng
2023-12-22 14:54:25 +08:00
parent 2aeb699359
commit 6ed321caea
19 changed files with 1553 additions and 1535 deletions
Download Patch File Download Diff File Expand all files Collapse all files

118
src/session/test/gtest_session.cpp Normal file
View File

@@ -0,0 +1,118 @@
#include <gtest/gtest.h>
#include "session_private.h"
#define SESSION_KEY_IPV4_TCP(name) \
struct tuple6 name; \
memset(&name, 0, sizeof(struct tuple6)); \
(name).ip_type = IP_TYPE_V4; \
(name).src_addr.v4.s_addr = inet_addr("192.168.1.2"); \
(name).dst_addr.v4.s_addr = inet_addr("192.168.1.3"); \
(name).src_port = htons(1234); \
(name).dst_port = htons(5678); \
(name).ip_proto = IPPROTO_TCP; \
(name).security_zone = 0;
#define SESSION_KEY_IPV6_UDP(name) \
struct tuple6 name; \
memset(&name, 0, sizeof(struct tuple6)); \
(name).ip_type = IP_TYPE_V6; \
inet_pton(AF_INET6, "2001:db8:0:0:0:ff00:42:8329", &(name).src_addr.v6); \
inet_pton(AF_INET6, "2001:db8:0:0:0:ff00:42:832a", &(name).dst_addr.v6); \
(name).src_port = htons(1234); \
(name).dst_port = htons(5678); \
(name).ip_proto = IPPROTO_UDP; \
(name).security_zone = 0;
#define SESSION_KEY_IPV6_TCP(name) \
struct tuple6 name; \
memset(&name, 0, sizeof(struct tuple6)); \
(name).ip_type = IP_TYPE_V6; \
inet_pton(AF_INET6, "2001:db8:0:0:0:ff00:42:8329", &(name).src_addr.v6); \
inet_pton(AF_INET6, "2001:db8:0:0:0:ff00:42:832a", &(name).dst_addr.v6); \
(name).src_port = htons(1234); \
(name).dst_port = htons(5678); \
(name).ip_proto = IPPROTO_TCP; \
(name).security_zone = 0;
void free_ex_data(struct session *sess, uint8_t idx, void *ex_ptr, void *arg)
{
if (ex_ptr)
{
printf("free_ex_data: %s\n", (char *)ex_ptr);
free(ex_ptr);
ex_ptr = NULL;
}
}
TEST(SESSION, EX_NEW_INDEX)
{
uint8_t idx1 = session_get_ex_new_index("key1", NULL, NULL);
uint8_t idx2 = session_get_ex_new_index("key2", NULL, NULL);
uint8_t idx3 = session_get_ex_new_index("key1", NULL, NULL);
EXPECT_TRUE(idx1 != idx2);
EXPECT_TRUE(idx1 == idx3);
}
TEST(SESSION, EX_SET_GET)
{
struct session sess;
memset(&sess, 0, sizeof(sess));
uint8_t idx = session_get_ex_new_index("ex_set_get", NULL, NULL);
session_set_ex_data(&sess, idx, (void *)0x1234);
EXPECT_TRUE(session_get0_ex_data(&sess, idx) == (void *)0x1234);
}
TEST(SESSION, EX_FREE_BY_RESET_NULL)
{
struct session sess;
memset(&sess, 0, sizeof(sess));
uint8_t idx = session_get_ex_new_index("ex_free_by_reset_null", free_ex_data, NULL);
char *ptr = strdup("hello");
session_set_ex_data(&sess, idx, ptr);
session_set_ex_data(&sess, idx, NULL);
}
TEST(SESSION, EX_FREE_BY_CALL_API)
{
struct session sess;
memset(&sess, 0, sizeof(sess));
uint8_t idx = session_get_ex_new_index("ex_free_by_call_api", free_ex_data, NULL);
char *ptr = strdup("hello");
session_set_ex_data(&sess, idx, ptr);
session_free_ex_data(&sess, idx);
}
TEST(SESSION, EV_QUEUE)
{
uint32_t event = 0;
struct session sess;
session_init(&sess);
EXPECT_TRUE(session_pop_event(&sess, &event) == false);
EXPECT_TRUE(session_push_event(&sess, 0x1234) == true);
EXPECT_TRUE(session_pop_event(&sess, &event) == true);
EXPECT_TRUE(event == 0x1234);
EXPECT_TRUE(session_pop_event(&sess, &event) == false);
for (int j = 0; j < 10; j++)
{
for (uint32_t i = 0; i < SESSION_EVENT_QUEUE_SIZE - 1; i++)
{
EXPECT_TRUE(session_push_event(&sess, i) == true);
}
EXPECT_TRUE(session_push_event(&sess, 0) == false);
for (uint32_t i = 0; i < SESSION_EVENT_QUEUE_SIZE - 1; i++)
{
EXPECT_TRUE(session_pop_event(&sess, &event) == true);
EXPECT_TRUE(event == i);
}
EXPECT_TRUE(session_pop_event(&sess, &event) == false);
}
}
int main(int argc, char **argv)
{
::testing::InitGoogleTest(&argc, argv);
return RUN_ALL_TESTS();
}