#include "test_utils.h"

#if 1
TEST(TIMEOUT, TCP_TIMEOUT_INIT1)
{
    struct packet pkt;
    struct session *sess = NULL;
    struct session_manager *mgr = NULL;

    timestamp_update();

    mgr = session_manager_new(&config);
    EXPECT_TRUE(mgr != NULL);

    // C2S SYN Packet
    printf("=> packet parse: TCP C2S SYN packet\n");
    packet_parse(&pkt, (const char *)tcp_pkt1_c2s_syn, sizeof(tcp_pkt1_c2s_syn));
    printf("<= packet parse\n\n");
    sess = session_manager_update_session(mgr, &pkt);
    EXPECT_TRUE(sess);

    __session_manager_check_counter(mgr, 1, 0, 0, 0, 0, 0, 0, 0, 0, 0);

    __session_manager_check_tcp_timeout_init(mgr, &config);
    __session_manager_check_tcp_timeout_time_wait(mgr, &config);

    __session_manager_check_counter(mgr, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0);

    session_manager_free(mgr);
}
#endif

#if 1
TEST(TIMEOUT, TCP_TIMEOUT_INIT2)
{
    struct packet pkt;
    struct session *sess = NULL;
    struct session_manager *mgr = NULL;

    timestamp_update();

    mgr = session_manager_new(&config);
    EXPECT_TRUE(mgr != NULL);

    // S2C SYNACK Packet
    printf("=> packet parse: TCP S2C SYNACK packet\n");
    packet_parse(&pkt, (const char *)tcp_pkt2_s2c_syn_ack, sizeof(tcp_pkt2_s2c_syn_ack));
    printf("<= packet parse\n\n");
    sess = session_manager_update_session(mgr, &pkt);
    EXPECT_TRUE(sess);

    __session_manager_check_counter(mgr, 1, 0, 0, 0, 0, 0, 0, 0, 0, 0);

    __session_manager_check_tcp_timeout_init(mgr, &config);
    __session_manager_check_tcp_timeout_time_wait(mgr, &config);

    __session_manager_check_counter(mgr, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0);

    session_manager_free(mgr);
}
#endif

int main(int argc, char **argv)
{
    ::testing::InitGoogleTest(&argc, argv);
    return RUN_ALL_TESTS();
}