#include <gtest/gtest.h>

#include "session_private.h"

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();
}