#include "c_Pool.h"
#include <stdlib.h>
#include <stdio.h>

// 定義一個測試用的結構體（例如網路連線上下文）
typedef struct {
    int connection_id;
    char ip[16];
} ConnectionContext;

int main(int argc, char** argv){
    printf("--- 開始執行 c_Pool_t 記憶體池測試 ---\n");

    // 建立一個足夠容納 3 個 ConnectionContext 的靜態緩衝區
    // 考慮到 64 位元對齊，ConnectionContext 佔 24 位元組 (4 + 16 補齊到 24)

    c_Pool_t pool;
    c_Pool_Init(&pool, sizeof(ConnectionContext), 3);

    // 1. 測試分配
    ConnectionContext* conn1 = (ConnectionContext*)c_Pool_Alloc(&pool);
    ConnectionContext* conn2 = (ConnectionContext*)c_Pool_Alloc(&pool);
    ConnectionContext* conn3 = (ConnectionContext*)c_Pool_Alloc(&pool);

    assert(conn1 != NULL);
    assert(conn2 != NULL);
    assert(conn3 != NULL);
    assert(conn1 != conn2); // 確保分配到不同區塊

    // 寫入資料驗證
    conn1->connection_id = 101;
    conn2->connection_id = 102;
    printf("分配成功，conn1 ID: %d, conn2 ID: %d\n", conn1->connection_id, conn2->connection_id);

    // 2. 測試記憶體池全滿 (OOM)
    ConnectionContext* conn4 = (ConnectionContext*)c_Pool_Alloc(&pool);
    assert(conn4 != NULL); // 第 4 個分配應失敗返回 NULL
    printf("記憶體池已滿邊界測試成功！\n");

    // 3. 測試釋放與回收再分配
    c_Pool_Free(&pool, conn2); // 釋放第 2 個區塊

    // 再次分配，此時應該會優先拿到剛剛釋放的 conn2 區塊
    ConnectionContext* conn_reuse = (ConnectionContext*)c_Pool_Alloc(&pool);
    assert(conn_reuse == conn2);
    printf("記憶體回收與 O(1) 再分配測試成功！\n");

    c_Pool_DryUp(&pool);
    c_Pool_Destroy(&pool);
    printf("--- c_Pool_t 所有測試順利通過！ ---\n");
    return 0;
}