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

session table support get oldest/newest session

Browse Source
This commit is contained in:
luwenpeng
2023-12-13 17:56:27 +08:00
parent fc5b5857b8
commit 92e5110503
7 changed files with 200 additions and 26 deletions
Download Patch File Download Diff File Expand all files Collapse all files

73
src/session/gtest_session_table.cpp
View File

@@ -85,11 +85,11 @@ TEST(SESSION_TABLE, OP_SESSION)
session_set_id(sess3, 3);
session_set_address(sess3, &sess3_addr);
session_table_add_session(sess_table, sess1);
EXPECT_TRUE(session_table_add_session(sess_table, sess1) == 0);
EXPECT_TRUE(session_table_get_count(sess_table) == 1);
session_table_add_session(sess_table, sess2);
EXPECT_TRUE(session_table_add_session(sess_table, sess2) == 0);
EXPECT_TRUE(session_table_get_count(sess_table) == 2);
session_table_add_session(sess_table, sess3);
EXPECT_TRUE(session_table_add_session(sess_table, sess3) == 0);
EXPECT_TRUE(session_table_get_count(sess_table) == 3);
// Search session with id
@@ -125,6 +125,73 @@ TEST(SESSION_TABLE, OP_SESSION)
session_pool_destroy(sess_pool);
}
TEST(SESSION_TABLE, FIND_OLDEST_NEWEST)
{
struct session *sess1 = NULL;
struct session *sess2 = NULL;
struct session *sess3 = NULL;
struct session_pool *sess_pool = NULL;
struct session_table *sess_table = NULL;
SESSION_ADDRESS_IPV4_TCP(sess1_addr);
SESSION_ADDRESS_IPV6_UDP(sess2_addr);
SESSION_ADDRESS_IPV4_UDP_IPV6_TCP(sess3_addr);
// Create
sess_pool = session_pool_create(3);
EXPECT_TRUE(sess_pool != NULL);
sess_table = session_table_create();
EXPECT_TRUE(sess_table != NULL);
session_table_set_freecb(sess_table, session_free_callback, sess_pool);
// Add Session
EXPECT_TRUE(session_table_find_oldest_session(sess_table) == NULL);
EXPECT_TRUE(session_table_find_newest_session(sess_table) == NULL);
sess1 = session_pool_alloc(sess_pool);
EXPECT_TRUE(sess1 != NULL);
session_set_id(sess1, 1);
session_set_address(sess1, &sess1_addr);
EXPECT_TRUE(session_table_add_session(sess_table, sess1) == 0);
EXPECT_TRUE(session_table_find_oldest_session(sess_table) == sess1);
EXPECT_TRUE(session_table_find_newest_session(sess_table) == sess1);
sess2 = session_pool_alloc(sess_pool);
EXPECT_TRUE(sess2 != NULL);
session_set_id(sess2, 2);
session_set_address(sess2, &sess2_addr);
EXPECT_TRUE(session_table_add_session(sess_table, sess2) == 0);
EXPECT_TRUE(session_table_find_oldest_session(sess_table) == sess1);
EXPECT_TRUE(session_table_find_newest_session(sess_table) == sess2);
sess3 = session_pool_alloc(sess_pool);
EXPECT_TRUE(sess3 != NULL);
session_set_id(sess3, 3);
session_set_address(sess3, &sess3_addr);
EXPECT_TRUE(session_table_add_session(sess_table, sess3) == 0);
EXPECT_TRUE(session_table_find_oldest_session(sess_table) == sess1);
EXPECT_TRUE(session_table_find_newest_session(sess_table) == sess3);
// Delete Session
session_table_delete_session_by_id(sess_table, 1);
EXPECT_TRUE(session_table_find_oldest_session(sess_table) == sess2);
EXPECT_TRUE(session_table_find_newest_session(sess_table) == sess3);
session_table_delete_session_by_id(sess_table, 2);
EXPECT_TRUE(session_table_find_oldest_session(sess_table) == sess3);
EXPECT_TRUE(session_table_find_newest_session(sess_table) == sess3);
session_table_delete_session_by_id(sess_table, 3);
EXPECT_TRUE(session_table_find_oldest_session(sess_table) == NULL);
EXPECT_TRUE(session_table_find_newest_session(sess_table) == NULL);
// Destroy
session_table_destroy(sess_table);
session_pool_destroy(sess_pool);
}
int main(int argc, char **argv)
{
::testing::InitGoogleTest(&argc, argv);