stellar-stellar/infra/plugin_manager/test/plugin_manager_gtest_main.cpp

#pragma GCC diagnostic ignored "-Wunused-parameter"

#include <gtest/gtest.h>
#include "stellar/utils.h"

#include "plugin_manager_gtest_mock.h"

#define STELLAR_INTRINSIC_TOPIC_NUM 0
#define TOPIC_NAME_MAX 512

void whitebox_test_plugin_manager_intrisic_metadata(struct stellar *st, struct plugin_manager_schema *plug_mgr)
{
	SCOPED_TRACE("whitebox test intrisic metadata"); 

	EXPECT_TRUE(plug_mgr!=NULL);

	EXPECT_EQ(plug_mgr->st, st);

	//load spec null
	EXPECT_TRUE(plug_mgr->plugin_load_specs_array==NULL);

	//session exdata schema null
	EXPECT_TRUE(plug_mgr->stellar_exdata_schema_array==NULL);

	//stellar mq schema null
	EXPECT_TRUE(plug_mgr->stellar_mq_schema_array==NULL);

	//registered plugin array null
	EXPECT_TRUE(plug_mgr->registered_polling_plugin_array==NULL);
	EXPECT_TRUE(plug_mgr->registered_packet_plugin_array==NULL);

	EXPECT_TRUE(plug_mgr->per_thread_data!=NULL);
	int thread_num=stellar_get_worker_thread_num(st);
	for(int i=0; i<thread_num; i++)
	{
		EXPECT_TRUE(plug_mgr->per_thread_data[i].per_thread_pkt_exdata_array.exdata_array==NULL);
		EXPECT_TRUE(plug_mgr->per_thread_data[i].dealth_letter_queue==NULL);
		for(int j=0; j<STELLAR_MQ_PRIORITY_MAX; j++)
			EXPECT_TRUE(plug_mgr->per_thread_data[i].priority_mq[j]==NULL);
	}
}

/***********************************
 * TEST PLUGIN MANAGER INIT & EXIT *
 ***********************************/

//TODO: test plugin_specs_load

TEST(plugin_manager_init, init_with_null_toml) {

	struct stellar st={0};
	struct plugin_manager_schema *plug_mgr = plugin_manager_init(&st, NULL);
	whitebox_test_plugin_manager_intrisic_metadata(&st, plug_mgr);
	plugin_manager_exit(plug_mgr);
}

/******************************************
 * TEST PLUGIN MANAGER PACKET PLUGIN INIT *
 ******************************************/

static void test_mock_packet_exdata_free(int idx, void *ex_ptr, void *arg){}

static void test_mock_overwrite_packet_exdata_free(int idx, void *ex_ptr, void *arg){}

TEST(plugin_manager_init, packet_exdata_new_index_overwrite) {
	struct stellar st={0};
	struct plugin_manager_schema *plug_mgr = plugin_manager_init(&st, NULL);
	whitebox_test_plugin_manager_intrisic_metadata(&st, plug_mgr);

	const char *exdata_name="PACKET_EXDATA";
	int exdata_idx=stellar_exdata_new_index(&st,exdata_name, test_mock_packet_exdata_free, &st);
	EXPECT_GE(exdata_idx, 0);
	int overwrite_idx=stellar_exdata_new_index(&st,exdata_name, test_mock_overwrite_packet_exdata_free, plug_mgr);
	EXPECT_GE(overwrite_idx, 0);
	EXPECT_EQ(overwrite_idx, exdata_idx);

    {
        SCOPED_TRACE("White-box test, check stellar internal schema");
        struct stellar_exdata_schema *exdata_schema = (struct stellar_exdata_schema *)utarray_eltptr(
            plug_mgr->stellar_exdata_schema_array, (unsigned int)exdata_idx);
        EXPECT_EQ(exdata_schema->free_func, (void *)test_mock_overwrite_packet_exdata_free);
        EXPECT_EQ(exdata_schema->free_arg, plug_mgr);
        EXPECT_EQ(exdata_schema->idx, exdata_idx);
        EXPECT_STREQ(exdata_schema->name, exdata_name);

        int exdata_num = utarray_len(plug_mgr->stellar_exdata_schema_array);
        EXPECT_EQ(exdata_num, 1);
    }

    plugin_manager_exit(plug_mgr);
}

void test_mock_packet_msg_free(void *msg, void *msg_free_arg){}
void test_mock_overwrite_packet_msg_free(void *msg, void *msg_free_arg){}

TEST(plugin_manager_init, packet_mq_topic_create_and_update) {
	struct stellar st={0};
	struct plugin_manager_schema *plug_mgr = plugin_manager_init(&st, NULL);
	whitebox_test_plugin_manager_intrisic_metadata(&st, plug_mgr);

	const char *topic_name="PACKET_TOPIC";

    EXPECT_EQ(stellar_mq_get_topic_id(&st, topic_name), -1); // illegal topic_name

    int topic_id = stellar_mq_create_topic(&st, topic_name, test_mock_packet_msg_free, &st);
    EXPECT_GE(topic_id, 0);
    struct stellar_mq_topic_schema *topic_schema = NULL;
    {
        SCOPED_TRACE("White-box test, check stellar internal schema");
        topic_schema =
            (struct stellar_mq_topic_schema *)utarray_eltptr(plug_mgr->stellar_mq_schema_array, (unsigned int)topic_id);
        EXPECT_EQ(topic_schema->free_cb, (void *)test_mock_packet_msg_free);
        EXPECT_EQ(topic_schema->free_cb_arg, &st);
        EXPECT_EQ(topic_schema->topic_id, topic_id);
        EXPECT_STREQ(topic_schema->topic_name, topic_name);
    }

    EXPECT_EQ(stellar_mq_get_topic_id(&st, topic_name), topic_id);
    EXPECT_EQ(stellar_mq_create_topic(&st, topic_name, test_mock_overwrite_packet_msg_free, plug_mgr),
              -1); // duplicate create, return error
    {
        SCOPED_TRACE("White-box test, check stellar internal schema");
        topic_schema =
            (struct stellar_mq_topic_schema *)utarray_eltptr(plug_mgr->stellar_mq_schema_array, (unsigned int)topic_id);
        EXPECT_EQ(topic_schema->free_cb, (void *)test_mock_packet_msg_free);
        EXPECT_EQ(topic_schema->free_cb_arg, &st);
        EXPECT_EQ(topic_schema->topic_id, topic_id);
        EXPECT_STREQ(topic_schema->topic_name, topic_name);
    }

    EXPECT_EQ(stellar_mq_update_topic(&st, topic_id, test_mock_overwrite_packet_msg_free, plug_mgr), 0);

    {
        SCOPED_TRACE("White-box test, check stellar internal schema");
        topic_schema =
            (struct stellar_mq_topic_schema *)utarray_eltptr(plug_mgr->stellar_mq_schema_array, (unsigned int)topic_id);
        EXPECT_EQ(topic_schema->free_cb, (void *)test_mock_overwrite_packet_msg_free);
        EXPECT_EQ(topic_schema->free_cb_arg, plug_mgr);
        EXPECT_EQ(topic_schema->topic_id, topic_id);
        EXPECT_STREQ(topic_schema->topic_name, topic_name);
        EXPECT_EQ(utarray_len(plug_mgr->stellar_mq_schema_array), 1+STELLAR_INTRINSIC_TOPIC_NUM);
    }

    EXPECT_EQ(stellar_mq_destroy_topic(&st, 10), -1); // illgeal topic_id

    EXPECT_EQ(stellar_mq_destroy_topic(&st, topic_id), 1);
    EXPECT_EQ(stellar_mq_destroy_topic(&st, topic_id), 0); // duplicate destroy, return 0;

    {
        SCOPED_TRACE("White-box test, check stellar internal schema");
        EXPECT_EQ(utarray_len(plug_mgr->stellar_mq_schema_array), 1+STELLAR_INTRINSIC_TOPIC_NUM); // destory won't delete the topic schema
    }
    plugin_manager_exit(plug_mgr);
}

void test_mock_on_packet_msg(int topic_id, const void *msg, void *plugin_env){}

void test_mock_overwrite_on_packet_msg(int topic_id, const void *msg, void *plugin_env){}

TEST(plugin_manager_init, packet_mq_subscribe) {

	struct stellar st={0};
	struct plugin_manager_schema *plug_mgr = plugin_manager_init(&st, NULL);
	whitebox_test_plugin_manager_intrisic_metadata(&st, plug_mgr);


	const char *topic_name="PACKET_TOPIC";

	int topic_id=stellar_mq_create_topic(&st, topic_name, test_mock_packet_msg_free, &st);
	EXPECT_GE(topic_id, 0);

	EXPECT_EQ(stellar_mq_subscribe(&st, topic_id, test_mock_on_packet_msg, 10),-1);//illgeal plugin_id
	EXPECT_EQ(stellar_mq_subscribe(&st, 10, test_mock_on_packet_msg, 10),-1);//illgeal topic_id & plugin_id

	int plugin_id=stellar_plugin_register(&st, 6, NULL, NULL,&st);
	EXPECT_GE(plugin_id, 0);

	EXPECT_EQ(stellar_mq_subscribe(&st, topic_id, test_mock_on_packet_msg, plugin_id),0);
	EXPECT_EQ(stellar_mq_subscribe(&st, topic_id, test_mock_overwrite_on_packet_msg, plugin_id),0);//duplicate subscribe, return 0, won't overwrite
	struct stellar_mq_topic_schema *topic_schema;
    {
        SCOPED_TRACE("White-box test, check stellar internal schema");
        topic_schema = (struct stellar_mq_topic_schema *)utarray_eltptr(plug_mgr->stellar_mq_schema_array, (unsigned int)topic_id);
        EXPECT_EQ(topic_schema->free_cb, (void *)test_mock_packet_msg_free);
        EXPECT_EQ(topic_schema->free_cb_arg, &st);
        EXPECT_EQ(topic_schema->topic_id, topic_id);
        EXPECT_STREQ(topic_schema->topic_name, topic_name);
    }

    EXPECT_EQ(topic_schema->subscriber_cnt, 1);
	EXPECT_EQ(topic_schema->subscribers->plugin_msg_cb, (void *)test_mock_overwrite_on_packet_msg);

	plugin_manager_exit(plug_mgr);
}


/*******************************************
 * TEST PLUGIN MANAGER PACKET PLUGIN RUNTIME*
 *******************************************/

#define PACKET_PROTO_PLUGIN_NUM 128
#define PACKET_EXDATA_NUM 2
#define PACKET_TOPIC_NUM 2
#define PACKET_MQ_SUB_NUM 2
struct packet_plugin_env
{
	struct plugin_manager_schema *plug_mgr;
	int basic_on_packet_called;
	int proto_filter_plugin_id[PACKET_PROTO_PLUGIN_NUM];
	int proto_filter_plugin_called[PACKET_PROTO_PLUGIN_NUM];
	int exdata_set_on_packet_called;
	int exdata_get_on_packet_called;
	unsigned int packet_exdata_idx[PACKET_EXDATA_NUM];
	int exdata_free_called[PACKET_EXDATA_NUM];
	unsigned int packet_topic_id[PACKET_TOPIC_NUM];
	unsigned int packet_mq_sub_plugin_id[PACKET_MQ_SUB_NUM];
	int msg_pub_cnt;
	int msg_sub_cnt;
	int msg_free_cnt;
};

static void test_basic_on_packet(struct packet *pkt, unsigned char ip_protocol, void *plugin_env)
{
	struct packet_plugin_env *env = (struct packet_plugin_env *)plugin_env;
	EXPECT_TRUE(env!=NULL);
	EXPECT_EQ(pkt->ip_proto, ip_protocol);
	EXPECT_EQ(pkt->st, env->plug_mgr->st);
	EXPECT_EQ(packet_exdata_set(pkt, 2, pkt), -1);// illegal set
	EXPECT_EQ(packet_exdata_get(pkt, 2), nullptr);// illegal get
	env->basic_on_packet_called+=1;
	return;
}

TEST(plugin_manager, packet_plugin_illegal_exdata) {

	struct stellar st={0};
	struct plugin_manager_schema *plug_mgr = plugin_manager_init(&st, NULL);
	whitebox_test_plugin_manager_intrisic_metadata(&st, plug_mgr);

	unsigned char ip_proto=6;
	struct packet_plugin_env env;
	memset(&env, 0, sizeof(struct packet_plugin_env));
	env.plug_mgr=plug_mgr;
	int plugin_id=stellar_plugin_register(&st, ip_proto, test_basic_on_packet, NULL,&env);
	EXPECT_GE(plugin_id, 0);

    {
        SCOPED_TRACE("White-box test, check stellar internal schema");
        int packet_plugin_num = utarray_len(plug_mgr->registered_packet_plugin_array);
        EXPECT_EQ(packet_plugin_num, 1);
    }

    struct packet pkt={&st, IPv4, ip_proto};
	plugin_manager_on_packet_input(plug_mgr, &pkt);
	plugin_manager_on_packet_output(plug_mgr, &pkt);
	
	plugin_manager_exit(plug_mgr);

	EXPECT_EQ(env.basic_on_packet_called, 1);
}

static void test_proto_filter_on_packet(struct packet *pkt, unsigned char ip_protocol, void *plugin_env)
{
	struct packet_plugin_env *env = (struct packet_plugin_env *)plugin_env;
	EXPECT_TRUE(env!=NULL);
	EXPECT_EQ(pkt->ip_proto, ip_protocol);
	EXPECT_EQ(pkt->st, env->plug_mgr->st);
	EXPECT_EQ(packet_exdata_set(pkt, 2, pkt), -1);// illegal set
	EXPECT_EQ(packet_exdata_get(pkt, 2), nullptr);// illegal get
	env->proto_filter_plugin_called[ip_protocol]+=1;
	return;
}

TEST(plugin_manager, packet_plugins_with_proto_filter) {

	struct stellar st={0};
	struct plugin_manager_schema *plug_mgr = plugin_manager_init(&st, NULL);
	whitebox_test_plugin_manager_intrisic_metadata(&st, plug_mgr);

	struct packet_plugin_env env;
	memset(&env, 0, sizeof(struct packet_plugin_env));
	env.plug_mgr=plug_mgr;

	int proto_filter_plugin_num=(int)(sizeof(env.proto_filter_plugin_id) / sizeof(env.proto_filter_plugin_id[0]));
    for (int i = 0; i < proto_filter_plugin_num; i++)
    {
        env.proto_filter_plugin_id[i] = stellar_plugin_register(&st, i, test_proto_filter_on_packet, NULL,&env);
        EXPECT_GE(env.proto_filter_plugin_id[i], 0);


    }

	{
        SCOPED_TRACE("White-box test, check stellar internal schema");
        EXPECT_EQ(utarray_len(plug_mgr->registered_packet_plugin_array), proto_filter_plugin_num);
    }

    struct packet pkt={&st, IPv4, 0};

	int N_packet=10;
    for (int j = 0; j < N_packet; j++)
    {
        for (int i = 0; i < proto_filter_plugin_num; i++)
        {
            pkt.ip_proto = i;
            plugin_manager_on_packet_input(plug_mgr, &pkt);
            plugin_manager_on_packet_output(plug_mgr, &pkt);
        }
    }
    plugin_manager_exit(plug_mgr);

	for (int i = 0; i < proto_filter_plugin_num; i++)
	{
		EXPECT_EQ(env.proto_filter_plugin_called[i], N_packet);
	}
}


struct test_exdata
{
	struct packet *pkt;
	struct session *sess;
	long long value;
};

static void test_exdata_set_on_packet(struct packet *pkt, unsigned char ip_protocol, void *plugin_env)
{
	struct packet_plugin_env *env = (struct packet_plugin_env *)plugin_env;
	EXPECT_TRUE(env!=NULL);
	EXPECT_EQ(pkt->ip_proto, ip_protocol);
	EXPECT_EQ(pkt->st, env->plug_mgr->st);
	env->exdata_set_on_packet_called+=1;

	int exdata_idx_len=(int)(sizeof(env->packet_exdata_idx) / sizeof(env->packet_exdata_idx[0]));

	for(int i=0; i<exdata_idx_len; i++)
	{
		struct test_exdata *exdata_val=CALLOC(struct test_exdata , 1);
		exdata_val->value=i;
		exdata_val->pkt=pkt;
		EXPECT_EQ(packet_exdata_set(pkt, env->packet_exdata_idx[i], exdata_val), 0);
	}
	return;
}

static void test_exdata_get_on_packet(struct packet *pkt, unsigned char ip_protocol, void *plugin_env)
{
	struct packet_plugin_env *env = (struct packet_plugin_env *)plugin_env;
	EXPECT_TRUE(env!=NULL);
	EXPECT_EQ(pkt->ip_proto, ip_protocol);
	EXPECT_EQ(pkt->st, env->plug_mgr->st);

	int exdata_idx_len=(int)(sizeof(env->packet_exdata_idx) / sizeof(env->packet_exdata_idx[0]));
	
	for(int i=0; i<exdata_idx_len; i++)
	{
		struct test_exdata *exdata_val=(struct test_exdata *)packet_exdata_get(pkt, env->packet_exdata_idx[i]);
		EXPECT_EQ(exdata_val->value, i);
	}
	env->exdata_get_on_packet_called+=1;
	return;
}

static void test_packet_exdata_free(int idx, void *ex_ptr, void *arg)
{
	struct packet_plugin_env *env = (struct packet_plugin_env *)arg;
	struct test_exdata *exdata_val=(struct test_exdata *)ex_ptr;
	EXPECT_EQ(env->packet_exdata_idx[idx], idx);
	EXPECT_EQ(exdata_val->value, idx);

	EXPECT_EQ(packet_exdata_get(exdata_val->pkt, idx), nullptr);// illegal get in exdata_free callback
	EXPECT_EQ(packet_exdata_set(exdata_val->pkt, idx, exdata_val->pkt), -1);// illegal set in exdata_free callback
	
	FREE(ex_ptr);
	env->exdata_free_called[idx]+=1;

	return;
}


TEST(plugin_manager, packet_plugins_share_exdata) {

	struct stellar st={0};
	struct plugin_manager_schema *plug_mgr = plugin_manager_init(&st, NULL);
	whitebox_test_plugin_manager_intrisic_metadata(&st, plug_mgr);

	unsigned char ip_proto=6;
	struct packet_plugin_env env;
	memset(&env, 0, sizeof(struct packet_plugin_env));
	env.plug_mgr=plug_mgr;

	char exdata_name[PACKET_EXDATA_NUM][TOPIC_NAME_MAX];
	int exdata_idx_len=(int)(sizeof(env.packet_exdata_idx) / sizeof(env.packet_exdata_idx[0]));
	for(int i=0; i<exdata_idx_len; i++)
	{
		sprintf(exdata_name[i], "PACKET_EXDATA_%d", i);
		env.packet_exdata_idx[i]=stellar_exdata_new_index(&st, exdata_name[i], test_packet_exdata_free, &env);
        {
            SCOPED_TRACE("White-box test, check stellar internal schema");
            struct stellar_exdata_schema *exdata_schema = (struct stellar_exdata_schema *)utarray_eltptr(
                plug_mgr->stellar_exdata_schema_array, env.packet_exdata_idx[i]);

            EXPECT_EQ(exdata_schema->free_func, (void *)test_packet_exdata_free);
            EXPECT_EQ(exdata_schema->free_arg, &env);
            EXPECT_EQ(exdata_schema->idx, env.packet_exdata_idx[i]);
            EXPECT_STREQ(exdata_schema->name, exdata_name[i]);
        }
    }

    {
        SCOPED_TRACE("White-box test, check stellar internal schema");
        EXPECT_EQ(utarray_len(plug_mgr->stellar_exdata_schema_array), exdata_idx_len);
    }

    int exdata_set_plugin_id=stellar_plugin_register(&st, ip_proto, test_exdata_set_on_packet, NULL,&env);
	EXPECT_GE(exdata_set_plugin_id, 0);

	int exdata_get_plugin_id=stellar_plugin_register(&st, ip_proto, test_exdata_get_on_packet, NULL,&env);
	EXPECT_GE(exdata_get_plugin_id, 0);

    {
        SCOPED_TRACE("White-box test, check stellar internal schema");
        EXPECT_EQ(utarray_len(plug_mgr->registered_packet_plugin_array), 2); // Fix plugin number
    }

    struct packet pkt={&st, IPv4, ip_proto};

	int N_packet=10;

	for(int i=0; i < N_packet; i++)
	{
		plugin_manager_on_packet_input(plug_mgr, &pkt);
		plugin_manager_on_packet_output(plug_mgr, &pkt);
	}
	plugin_manager_exit(plug_mgr);

	EXPECT_EQ(env.exdata_set_on_packet_called, N_packet);
	EXPECT_EQ(env.exdata_get_on_packet_called, N_packet);

	for(int i=0; i < exdata_idx_len; i++)
	{
		EXPECT_EQ(env.exdata_free_called[i], N_packet);
	}
}

static void test_packet_msg_free_cb_func(void *msg, void *msg_free_arg)
{
	struct packet_plugin_env *env = (struct packet_plugin_env *)msg_free_arg;
	env->msg_free_cnt+=1;
	return;
}

static void test_mq_on_packet_msg(int topic_id, const void *msg, void *plugin_env)
{
	struct packet_plugin_env *env = (struct packet_plugin_env *)plugin_env;
	EXPECT_TRUE(env!=NULL);
	env->msg_sub_cnt+=1;
	return;
}

static void test_mq_pub_on_packet(struct packet *pkt, unsigned char ip_protocol, void *plugin_env)
{
	struct packet_plugin_env *env = (struct packet_plugin_env *)plugin_env;
	EXPECT_TRUE(env!=NULL);
	EXPECT_EQ(pkt->ip_proto, ip_protocol);
	EXPECT_EQ(pkt->st, env->plug_mgr->st);
	int topic_id_num=(int)(sizeof(env->packet_topic_id) / sizeof(env->packet_topic_id[0]));
	for(int i=0; i<topic_id_num; i++)
	{
		EXPECT_EQ(stellar_mq_publish_message(env->plug_mgr->st, env->packet_topic_id[i], pkt), 0);
		env->msg_pub_cnt+=1;
	}
	return;
}

TEST(plugin_manager, packet_plugins_mq_pub_sub) {

	struct stellar st={0};
	struct plugin_manager_schema *plug_mgr = plugin_manager_init(&st, NULL);
	whitebox_test_plugin_manager_intrisic_metadata(&st, plug_mgr);

	unsigned char ip_proto=6;
	struct packet_plugin_env env;
	memset(&env, 0, sizeof(struct packet_plugin_env));
	env.plug_mgr=plug_mgr;
	char topic_name[PACKET_TOPIC_NUM][TOPIC_NAME_MAX];

	int topic_id_num=(int)(sizeof(env.packet_topic_id) / sizeof(env.packet_topic_id[0]));

	for(int i=0; i<topic_id_num; i++)
	{
		sprintf(topic_name[i], "PACKET_TOPIC_%d", i);
		env.packet_topic_id[i]=stellar_mq_create_topic(&st, topic_name[i], test_packet_msg_free_cb_func, &env);
		EXPECT_GE(env.packet_topic_id[i], 0);
        {
            SCOPED_TRACE("White-box test, check stellar internal schema");
            struct stellar_mq_topic_schema *topic = (struct stellar_mq_topic_schema *)utarray_eltptr(
                plug_mgr->stellar_mq_schema_array, env.packet_topic_id[i]);
            EXPECT_EQ(topic->free_cb, test_packet_msg_free_cb_func);
            EXPECT_EQ(topic->free_cb_arg, &env);
            EXPECT_EQ(topic->topic_id, env.packet_topic_id[i]);
            EXPECT_STREQ(topic->topic_name, topic_name[i]);
        }
    }

	{
		SCOPED_TRACE("White-box test, check stellar internal schema");
		EXPECT_EQ(utarray_len(plug_mgr->stellar_mq_schema_array), topic_id_num+STELLAR_INTRINSIC_TOPIC_NUM);
	}

	int pub_plugin_id=stellar_plugin_register(&st, ip_proto, test_mq_pub_on_packet, NULL,&env);
	EXPECT_GE(pub_plugin_id, 0);

	int topic_sub_num=(int)(sizeof(env.packet_mq_sub_plugin_id) / sizeof(env.packet_mq_sub_plugin_id[0]));

    for (int i = 0; i < topic_sub_num; i++)
    {
        env.packet_mq_sub_plugin_id[i] = stellar_plugin_register(&st, ip_proto, NULL, NULL,&env);// empty on_packet is ok
        EXPECT_GE(env.packet_mq_sub_plugin_id[i], 0);
		for(int j = 0; j < topic_id_num; j++)
		{
			EXPECT_EQ(stellar_mq_subscribe(&st, env.packet_topic_id[j], test_mq_on_packet_msg, env.packet_mq_sub_plugin_id[i]), 0);
		}
    }

	{
		SCOPED_TRACE("White-box test, check stellar internal schema");
		EXPECT_EQ(utarray_len(plug_mgr->registered_packet_plugin_array), topic_sub_num+1);
	}

	struct packet pkt={&st, IPv4, ip_proto};

	int N_packet=10;
    for (int i = 0; i < N_packet; i++)
    {
        plugin_manager_on_packet_input(plug_mgr, &pkt);
        plugin_manager_on_packet_output(plug_mgr, &pkt);
    }

    plugin_manager_exit(plug_mgr);
	EXPECT_EQ(N_packet*topic_id_num, env.msg_pub_cnt);
	EXPECT_EQ(env.msg_free_cnt, env.msg_pub_cnt);
	EXPECT_EQ(env.msg_sub_cnt, env.msg_pub_cnt*topic_sub_num);
}

static void overlimit_packet_msg_free_cb_func(void *msg, void *msg_free_arg)
{
	struct packet_plugin_env *env = (struct packet_plugin_env *)msg_free_arg;
	env->msg_free_cnt+=1;
	FREE(msg);
	return;
}

static void overlimit_sub_on_packet_msg(int topic_id, const void *msg, void *plugin_env)
{
	struct packet_plugin_env *env = (struct packet_plugin_env *)plugin_env;
	EXPECT_TRUE(env!=NULL);
	env->msg_sub_cnt+=1;
	return;
}

static void overlimit_pub_on_packet(struct packet *pkt, unsigned char ip_protocol, void *plugin_env)
{
	struct packet_plugin_env *env = (struct packet_plugin_env *)plugin_env;
	EXPECT_TRUE(env!=NULL);
	EXPECT_EQ(pkt->ip_proto, ip_protocol);
	int topic_id_num=(int)(sizeof(env->packet_topic_id) / sizeof(env->packet_topic_id[0]));
	int cnt=0;
	int *msg;
	for(int i=0; i<topic_id_num; i++)
	{
		for(int j=0; j < MAX_MSG_PER_DISPATCH; j++)
		{
			msg=CALLOC(int, 1);
			*msg=cnt;
			int pub_ret=stellar_mq_publish_message(env->plug_mgr->st, env->packet_topic_id[i], msg);
			if(cnt < MAX_MSG_PER_DISPATCH)
			{
				ASSERT_EQ(pub_ret, 0);
				env->msg_pub_cnt+=1;
			}
			else
			{
				ASSERT_EQ(pub_ret, -1);
			}
			if(pub_ret!=0)FREE(msg);
			cnt+=1;
		}
	}
	return;
}

TEST(plugin_manager, packet_plugins_pub_overlimit) {

	struct stellar st={0};
	struct plugin_manager_schema *plug_mgr = plugin_manager_init(&st, NULL);
	whitebox_test_plugin_manager_intrisic_metadata(&st, plug_mgr);

	unsigned char ip_proto=6;
	struct packet_plugin_env env;
	memset(&env, 0, sizeof(struct packet_plugin_env));
	env.plug_mgr=plug_mgr;
	char topic_name[PACKET_TOPIC_NUM][TOPIC_NAME_MAX];

	int topic_id_num=(int)(sizeof(env.packet_topic_id) / sizeof(env.packet_topic_id[0]));

	for(int i=0; i<topic_id_num; i++)
	{
		sprintf(topic_name[i], "PACKET_TOPIC_%d", i);
		env.packet_topic_id[i]=stellar_mq_create_topic(&st, topic_name[i], overlimit_packet_msg_free_cb_func, &env);
		EXPECT_GE(env.packet_topic_id[i], 0);
        {
            SCOPED_TRACE("White-box test, check stellar internal schema");
            struct stellar_mq_topic_schema *topic = (struct stellar_mq_topic_schema *)utarray_eltptr(
                plug_mgr->stellar_mq_schema_array, env.packet_topic_id[i]);
            EXPECT_EQ(topic->free_cb, overlimit_packet_msg_free_cb_func);
            EXPECT_EQ(topic->free_cb_arg, &env);
            EXPECT_EQ(topic->topic_id, env.packet_topic_id[i]);
            EXPECT_STREQ(topic->topic_name, topic_name[i]);
        }
    }

	{
		SCOPED_TRACE("White-box test, check stellar internal schema");
		EXPECT_EQ(utarray_len(plug_mgr->stellar_mq_schema_array), topic_id_num+STELLAR_INTRINSIC_TOPIC_NUM);
	}

	int pub_plugin_id=stellar_plugin_register(&st, ip_proto, overlimit_pub_on_packet, NULL,&env);
	EXPECT_GE(pub_plugin_id, 0);

	int topic_sub_num=(int)(sizeof(env.packet_mq_sub_plugin_id) / sizeof(env.packet_mq_sub_plugin_id[0]));

    for (int i = 0; i < topic_sub_num; i++)
    {
        env.packet_mq_sub_plugin_id[i] = stellar_plugin_register(&st, ip_proto, NULL, NULL, &env);// empty on_packet is ok
        EXPECT_GE(env.packet_mq_sub_plugin_id[i], 0);
		for(int j = 0; j < topic_id_num; j++)
		{
			EXPECT_EQ(stellar_mq_subscribe(&st, env.packet_topic_id[j], overlimit_sub_on_packet_msg, env.packet_mq_sub_plugin_id[i]), 0);
		}
    }

	{
		SCOPED_TRACE("White-box test, check stellar internal schema");
		EXPECT_EQ(utarray_len(plug_mgr->registered_packet_plugin_array), topic_sub_num+1);
	}

	struct packet pkt={&st, IPv4, ip_proto};

	int N_packet=10;
    for (int i = 0; i < N_packet; i++)
    {
        plugin_manager_on_packet_input(plug_mgr, &pkt);
        plugin_manager_on_packet_output(plug_mgr, &pkt);
    }

    plugin_manager_exit(plug_mgr);
	EXPECT_EQ(N_packet*MAX_MSG_PER_DISPATCH, env.msg_pub_cnt);
	EXPECT_EQ(env.msg_free_cnt, env.msg_pub_cnt);
	EXPECT_EQ(env.msg_sub_cnt, env.msg_pub_cnt*topic_sub_num);
}



static void test_exdata_free_pub_msg_exdata_free(int idx, void *ex_ptr, void *arg)
{
	struct packet_plugin_env *env = (struct packet_plugin_env *)arg;
	EXPECT_EQ(env->packet_exdata_idx[idx], idx);
	env->exdata_free_called[idx]+=1;
	EXPECT_EQ(stellar_mq_publish_message(env->plug_mgr->st, env->packet_topic_id[0], (struct packet *)ex_ptr), -1);// publish message in packet exdata_free is illegal
	env->msg_pub_cnt+=1;
	return;
}

static void test_exdata_free_pub_msg_free( void *msg, void *msg_free_arg)
{
	struct packet_plugin_env *env = (struct packet_plugin_env *)msg_free_arg;
	env->msg_free_cnt+=1;
	EXPECT_EQ(stellar_mq_publish_message(env->plug_mgr->st, env->packet_topic_id[0], msg), -1 );// publish message in packet msg_free is illegal
	return;
}

static void test_exdata_free_pub_msg_on_packet(struct packet *pkt, unsigned char ip_protocol, void *plugin_env)
{
	struct packet_plugin_env *env = (struct packet_plugin_env *)plugin_env;
	EXPECT_TRUE(env!=NULL);
	EXPECT_EQ(pkt->ip_proto, ip_protocol);
	EXPECT_EQ(pkt->st, env->plug_mgr->st);
	EXPECT_EQ(packet_exdata_set(pkt, env->packet_exdata_idx[0], pkt), 0);
	env->basic_on_packet_called+=1;
	return;
}

static void test_exdata_free_pub_msg_on_packet_msg(int topic_id, const void *msg, void *plugin_env)
{
	struct packet_plugin_env *env = (struct packet_plugin_env *)plugin_env;
	EXPECT_EQ(topic_id, env->packet_topic_id[0]);
	env->msg_sub_cnt+=1;
}

TEST(plugin_manager, packet_plugin_exdata_free_pub_msg) {

	struct stellar st={0};
	struct plugin_manager_schema *plug_mgr = plugin_manager_init(&st, NULL);
	whitebox_test_plugin_manager_intrisic_metadata(&st, plug_mgr);

	unsigned char ip_proto=6;
	struct packet_plugin_env env;
	memset(&env, 0, sizeof(struct packet_plugin_env));
	env.plug_mgr=plug_mgr;
	int plugin_id=stellar_plugin_register(&st, ip_proto, test_exdata_free_pub_msg_on_packet, NULL,&env);
	EXPECT_GE(plugin_id, 0);

	env.packet_exdata_idx[0]=stellar_exdata_new_index(&st, "PACKET_EXDATA", test_exdata_free_pub_msg_exdata_free, &env);
	env.packet_topic_id[0]=stellar_mq_create_topic(&st, "PACKET_TOPIC", test_exdata_free_pub_msg_free, &env);

	EXPECT_EQ(stellar_mq_subscribe(&st, env.packet_topic_id[0], test_exdata_free_pub_msg_on_packet_msg, plugin_id),0);
	

    struct packet pkt={&st, IPv4, ip_proto};
	plugin_manager_on_packet_input(plug_mgr, &pkt);
	plugin_manager_on_packet_output(plug_mgr, &pkt);
	
	plugin_manager_exit(plug_mgr);

	EXPECT_EQ(env.basic_on_packet_called, 1);
	EXPECT_EQ(env.msg_pub_cnt, 1);
	EXPECT_EQ(env.msg_sub_cnt, 0);
	EXPECT_EQ(env.msg_free_cnt, 0);
	EXPECT_EQ(env.exdata_free_called[0], 1);
}

/**********************************************
 * TEST PLUGIN MANAGER ON SESSION PLUGIN INIT *
 **********************************************/

static void test_mock_session_exdata_free(int idx, void *ex_ptr, void *arg){}
static void test_mock_overwrite_session_exdata_free(int idx, void *ex_ptr, void *arg){}

TEST(plugin_manager_init, session_exdata_new_index_overwrite) {
	struct stellar st={0};
	struct plugin_manager_schema *plug_mgr = plugin_manager_init(&st, NULL);
	whitebox_test_plugin_manager_intrisic_metadata(&st, plug_mgr);

	const char *exdata_name="SESSION_EXDATA";
	int exdata_idx=stellar_exdata_new_index(&st,exdata_name, test_mock_session_exdata_free, &st);
	EXPECT_GE(exdata_idx, 0);
	int overwrite_idx=stellar_exdata_new_index(&st,exdata_name, test_mock_overwrite_session_exdata_free, plug_mgr);
	EXPECT_GE(overwrite_idx, 0);
	EXPECT_EQ(overwrite_idx, exdata_idx);

    {
        SCOPED_TRACE("White-box test, check stellar internal schema");
        struct stellar_exdata_schema *exdata_schema = (struct stellar_exdata_schema *)utarray_eltptr(
            plug_mgr->stellar_exdata_schema_array, (unsigned int)exdata_idx);
        EXPECT_EQ(exdata_schema->free_func, (void *)test_mock_overwrite_session_exdata_free);
        EXPECT_EQ(exdata_schema->free_arg, plug_mgr);
        EXPECT_EQ(exdata_schema->idx, exdata_idx);
        EXPECT_STREQ(exdata_schema->name, exdata_name);

        int exdata_num = utarray_len(plug_mgr->stellar_exdata_schema_array);
        EXPECT_EQ(exdata_num, 1);
    }
    plugin_manager_exit(plug_mgr);
}

void test_mock_session_msg_free(void *msg, void *msg_free_arg){}
void test_mock_overwrite_session_msg_free(void *msg, void *msg_free_arg){}

TEST(plugin_manager_init, session_mq_topic_create_and_update) {
	struct stellar st={0};
	struct plugin_manager_schema *plug_mgr = plugin_manager_init(&st, NULL);
	whitebox_test_plugin_manager_intrisic_metadata(&st, plug_mgr);

	const char *topic_name="SESSION_TOPIC";

	EXPECT_EQ(stellar_mq_get_topic_id(&st, topic_name), -1);// illegal topic_name

	int topic_id=stellar_mq_create_topic(&st, topic_name, test_mock_session_msg_free, &st);
	EXPECT_GE(topic_id, 0);
	struct stellar_mq_topic_schema *topic_schema;
    {
        SCOPED_TRACE("White-box test, check stellar internal schema");
        topic_schema =
            (struct stellar_mq_topic_schema *)utarray_eltptr(plug_mgr->stellar_mq_schema_array, (unsigned int)topic_id);
        EXPECT_EQ(topic_schema->free_cb, (void *)test_mock_session_msg_free);
        EXPECT_EQ(topic_schema->free_cb_arg, &st);
        EXPECT_EQ(topic_schema->topic_id, topic_id);
        EXPECT_STREQ(topic_schema->topic_name, topic_name);
    }

    EXPECT_EQ(stellar_mq_get_topic_id(&st, topic_name), topic_id);
	EXPECT_EQ(stellar_mq_create_topic(&st, topic_name, test_mock_overwrite_session_msg_free, plug_mgr), -1); // duplicate create, return error

    {
        SCOPED_TRACE("White-box test, check stellar internal schema");
        topic_schema =
            (struct stellar_mq_topic_schema *)utarray_eltptr(plug_mgr->stellar_mq_schema_array, (unsigned int)topic_id);
        EXPECT_EQ(topic_schema->free_cb, (void *)test_mock_session_msg_free);
        EXPECT_EQ(topic_schema->free_cb_arg, &st);
        EXPECT_EQ(topic_schema->topic_id, topic_id);
        EXPECT_STREQ(topic_schema->topic_name, topic_name);
    }

    EXPECT_EQ(stellar_mq_update_topic(&st, topic_id, test_mock_overwrite_session_msg_free, plug_mgr), 0);

    {
        SCOPED_TRACE("White-box test, check stellar internal schema");
        topic_schema =
            (struct stellar_mq_topic_schema *)utarray_eltptr(plug_mgr->stellar_mq_schema_array, (unsigned int)topic_id);
        EXPECT_EQ(topic_schema->free_cb, (void *)test_mock_overwrite_session_msg_free);
        EXPECT_EQ(topic_schema->free_cb_arg, plug_mgr);
        EXPECT_EQ(topic_schema->topic_id, topic_id);
        EXPECT_STREQ(topic_schema->topic_name, topic_name);

        EXPECT_EQ(utarray_len(plug_mgr->stellar_mq_schema_array), STELLAR_INTRINSIC_TOPIC_NUM+1); // 5 intrinsic topic + 1 created topic
    }

	EXPECT_EQ(stellar_mq_destroy_topic(&st, 10), -1);// illgeal session topic_id

	EXPECT_EQ(stellar_mq_destroy_topic(&st, topic_id), 1);
	EXPECT_EQ(stellar_mq_destroy_topic(&st, topic_id), 0);//duplicate destroy, return 0;

	{
		SCOPED_TRACE("White-box test, check stellar internal schema");
		EXPECT_EQ(utarray_len(plug_mgr->stellar_mq_schema_array), STELLAR_INTRINSIC_TOPIC_NUM+1);//destory won't delete the topic schema
	}
	EXPECT_EQ(plug_mgr->stellar_mq_topic_num, STELLAR_INTRINSIC_TOPIC_NUM);//intrinsic topic number

	plugin_manager_exit(plug_mgr);
}

void test_mock_on_session_msg(int topic_id, const void *msg,void *plugin_env){}
void test_mock_overwrite_on_session_msg(int topic_id, const void *msg, void *plugin_env){}


TEST(plugin_manager_init, session_mq_subscribe_overwrite) {

	struct stellar st={0};
	struct plugin_manager_schema *plug_mgr = plugin_manager_init(&st, NULL);
	whitebox_test_plugin_manager_intrisic_metadata(&st, plug_mgr);

	const char *topic_name="SESSION_TOPIC";

	int topic_id=stellar_mq_create_topic(&st, topic_name, test_mock_session_msg_free, &st);
	EXPECT_GE(topic_id, 0);

	EXPECT_EQ(stellar_mq_subscribe(&st, topic_id, test_mock_on_session_msg, 10),-1);//illgeal plugin_id
	EXPECT_EQ(stellar_mq_subscribe(&st, 10, test_mock_on_session_msg, 10),-1);//illgeal topic_id & plugin_id

	int plugin_id=stellar_plugin_register(&st, 0, NULL, NULL, NULL);
	EXPECT_GE(plugin_id, 0);

	EXPECT_EQ(stellar_mq_subscribe(&st, topic_id, test_mock_on_session_msg, plugin_id),0);
	EXPECT_EQ(stellar_mq_subscribe(&st, topic_id, test_mock_overwrite_on_session_msg, plugin_id),0);//duplicate subscribe, return 0, won't overwrite 

	struct stellar_mq_topic_schema *topic_schema;
	{
		SCOPED_TRACE("White-box test, check stellar internal schema");
		topic_schema = (struct stellar_mq_topic_schema *)utarray_eltptr(plug_mgr->stellar_mq_schema_array,(unsigned int)topic_id);
		EXPECT_EQ(topic_schema->free_cb, (void *)test_mock_session_msg_free);
		EXPECT_EQ(topic_schema->free_cb_arg, &st);
		EXPECT_EQ(topic_schema->topic_id, topic_id);
		EXPECT_STREQ(topic_schema->topic_name, topic_name);
	}

	EXPECT_EQ(topic_schema->subscriber_cnt, 1);
	EXPECT_EQ(topic_schema->subscribers->plugin_msg_cb, (void *)test_mock_overwrite_on_session_msg);

	plugin_manager_exit(plug_mgr);
}
/**********************************************
 * TEST PLUGIN MANAGER ON SESSION PLUGIN RUNTIME *
 **********************************************/

#define TOPIC_TCP_INPUT "TCP_INPUT" //topic message: session
#define TOPIC_UDP_INPUT "UDP_INPUT" //topic message: session

#define TOPIC_TCP_OUTPUT "TCP_OUTPUT" //topic message: session
#define TOPIC_UDP_OUTPUT "UDP_OUTPUT" //topic message: session

struct session_plugin_env
{
	struct plugin_manager_schema *plug_mgr;
	int N_session;
	int N_per_session_pkt_cnt;
	int intrinsc_tcp_topic_id;
	int intrinsc_egress_topic_id;
	int basic_exdata_idx;
	int basic_exdata_free_called;
	int basic_on_session_ingress_called;
	int basic_on_session_egress_called;
	int basic_ctx_new_called;
	int basic_ctx_free_called;
	int test_mq_pub_plugin_id;
	int test_mq_sub_plugin_id;
	int test_mq_pub_called;
	int test_mq_sub_called;
	int test_mq_free_called;
	int test_mq_topic_id;
	int plugin_id_1;
	int plugin_id_2;
	int plugin_id_1_called;
	int plugin_id_2_called;
	int exdata_ctx_1_id;
	int exdata_ctx_2_id;
};

TEST(plugin_manager, no_plugin_register_runtime) {

	struct stellar st={0};

// init stage
	struct plugin_manager_schema *plug_mgr = plugin_manager_init(&st, NULL);
	whitebox_test_plugin_manager_intrisic_metadata(&st, plug_mgr);

// init plugin

// prepare packet and session

	struct session_plugin_env env;
	memset(&env, 0, sizeof(struct session_plugin_env));
	env.plug_mgr=plug_mgr;
	env.N_per_session_pkt_cnt=10;
	env.N_session=10;

	struct packet pkt={&st, TCP, 6};

	struct session sess[env.N_session];
	memset(&sess, 0, sizeof(sess));

// pesudo running stage
	for(int i=0; i < env.N_session; i++)
	{
		sess[i].session_exdat_rt=session_exdata_runtime_new(plug_mgr);
		sess[i].type=SESSION_TYPE_TCP;
	}

    for (int j = 0; j < env.N_per_session_pkt_cnt; j++)
    {
        plugin_manager_on_packet_input(plug_mgr, &pkt);

		plugin_manager_on_packet_output(plug_mgr, &pkt);

    }

    for(int i=0; i < env.N_session; i++)
	{
		session_exdata_runtime_free(sess[i].session_exdat_rt);
	}

//exit stage
	plugin_manager_exit(plug_mgr);
}


struct test_basic_ctx
{
	int called;
};


static void test_basic_on_session_ingress(int topic_id, const void *msg, void *plugin_env)
{
	struct session *sess=(struct session *)msg;
	struct session_plugin_env *env = (struct session_plugin_env *)plugin_env;
	EXPECT_TRUE(env!=NULL);
	if(msg)
	{
		EXPECT_EQ(session_exdata_set(sess, 2, sess), -1);// illegal set
		EXPECT_EQ(session_exdata_get(sess, 2), nullptr);// illegal get
		EXPECT_EQ(session_exdata_set(sess, env->basic_exdata_idx, sess), 0);
		env->basic_on_session_ingress_called+=1;
	}
	return;
}

static void test_basic_on_session_egress(int topic_id, const void *msg, void *plugin_env)
{
	struct session *sess=(struct session *)msg;
	struct session_plugin_env *env = (struct session_plugin_env *)plugin_env;
	EXPECT_TRUE(env!=NULL);

    EXPECT_EQ(session_exdata_set(sess, 2, sess), -1); // illegal set
    EXPECT_EQ(session_exdata_get(sess, 2), nullptr);  // illegal get
    env->basic_on_session_egress_called += 1;

    EXPECT_EQ(session_exdata_get(sess, env->basic_exdata_idx), sess);
	return;
}

static void test_basic_session_exdata_free(int idx, void *ex_ptr, void *arg)
{
	struct session_plugin_env *env = (struct session_plugin_env *)arg;
	EXPECT_EQ(env->basic_exdata_idx, idx);
	EXPECT_EQ(session_exdata_get((struct session *)ex_ptr, idx), nullptr);// illegal get in exdata_free callback
	EXPECT_EQ(session_exdata_set((struct session *)ex_ptr, idx, arg), -1);// illegal set in exdata_free callback
	env->basic_exdata_free_called+=1;
}

TEST(plugin_manager, session_plugin_on_intrinsic_ingress_egress) {

	struct stellar st={0};
	struct plugin_manager_schema *plug_mgr = plugin_manager_init(&st, NULL);
	whitebox_test_plugin_manager_intrisic_metadata(&st, plug_mgr);

	unsigned char ip_proto=6;
	struct session_plugin_env env;
	memset(&env, 0, sizeof(struct session_plugin_env));
	env.plug_mgr=plug_mgr;
	env.N_per_session_pkt_cnt=10;
	env.N_session=1;

	int plugin_id=stellar_plugin_register(&st, 0, NULL, NULL, &env);
	EXPECT_GE(plugin_id, 0);

	env.intrinsc_tcp_topic_id=stellar_mq_create_topic(&st, TOPIC_TCP_INPUT, NULL, NULL);
	EXPECT_GE(env.intrinsc_tcp_topic_id, 0);
	EXPECT_EQ(stellar_mq_subscribe(&st, env.intrinsc_tcp_topic_id, test_basic_on_session_ingress, plugin_id), 0);

	env.intrinsc_egress_topic_id=stellar_mq_create_topic(&st, TOPIC_TCP_OUTPUT, NULL, NULL);
	EXPECT_GE(env.intrinsc_egress_topic_id, 0);
	EXPECT_EQ(stellar_mq_subscribe(&st, env.intrinsc_egress_topic_id, test_basic_on_session_ingress, plugin_id), 0);// Intentional error

	EXPECT_EQ(stellar_mq_subscribe(&st, env.intrinsc_egress_topic_id, test_basic_on_session_egress, plugin_id), 0);

	env.basic_exdata_idx=stellar_exdata_new_index(&st, "SESSION_EXDATA", test_basic_session_exdata_free,&env);
	EXPECT_GE(env.basic_exdata_idx, 0);

	struct packet pkt={&st, TCP, ip_proto};


	struct session sess[env.N_session];

	for(int i=0; i < env.N_session; i++)
	{
		sess[i].session_exdat_rt=session_exdata_runtime_new(plug_mgr);
		sess[i].type=SESSION_TYPE_TCP;
	}

    for (int j = 0; j < env.N_per_session_pkt_cnt; j++)
    {
        plugin_manager_on_packet_input(plug_mgr, &pkt);

        for (int i = 0; i < env.N_session; i++)
        {
            stellar_mq_publish_message(&st, env.intrinsc_tcp_topic_id, &sess[i]);
            stellar_mq_publish_message(&st, env.intrinsc_egress_topic_id, &sess[i]);
        }

		plugin_manager_on_packet_output(plug_mgr, &pkt);
    }

    for(int i=0; i < env.N_session; i++)
	{
		session_exdata_runtime_free(sess[i].session_exdat_rt);
	}

	plugin_manager_exit(plug_mgr);

	EXPECT_EQ(env.basic_on_session_ingress_called, env.basic_on_session_egress_called);
	EXPECT_EQ(env.basic_on_session_ingress_called, env.N_session*env.N_per_session_pkt_cnt);
	EXPECT_EQ(env.basic_exdata_free_called, env.N_session);

}


static void test_mq_pub_on_session(int topic_id, const void *msg, void *plugin_env)
{
	struct session_plugin_env *env = (struct session_plugin_env *)plugin_env;
	struct test_basic_ctx *ctx=(struct test_basic_ctx *)msg;
	EXPECT_TRUE(env!=NULL);
	EXPECT_TRUE(ctx!=NULL);
    if (msg)
    {
        env->test_mq_pub_called += 1;
        ctx->called += 1;
        int *pub_msg = (int *)CALLOC(int, 1);
        *pub_msg = env->test_mq_pub_called;
        EXPECT_EQ(stellar_mq_publish_message(env->plug_mgr->st, env->test_mq_topic_id, pub_msg), 0);
    }
    return;
}

static void test_mq_on_sub_msg(int topic_id, const void *msg,void *plugin_env)
{
	struct session_plugin_env *env = (struct session_plugin_env *)plugin_env;
	struct test_basic_ctx *ctx=(struct test_basic_ctx *)msg;
	EXPECT_TRUE(env!=NULL);
	EXPECT_TRUE(ctx!=NULL);
	EXPECT_EQ(*(int *)msg, env->test_mq_pub_called);
	env->test_mq_sub_called+=1;
	ctx->called+=1;
	return;
}

static void test_session_msg_free(void *msg, void *msg_free_arg)
{
	struct session_plugin_env *env = (struct session_plugin_env *)msg_free_arg;
	if(msg)
	{
		EXPECT_EQ(env->test_mq_pub_called, *(int *)msg);
		env->test_mq_free_called+=1;
		FREE(msg);
	}
	return;
}

TEST(plugin_manager, DISABLED_session_plugin_ignore_on_ctx_new_sub_other_msg) {

	struct stellar st={0};
	struct plugin_manager_schema *plug_mgr = plugin_manager_init(&st, NULL);
	whitebox_test_plugin_manager_intrisic_metadata(&st, plug_mgr);

	unsigned char ip_proto=6;
	struct session_plugin_env env;
	memset(&env, 0, sizeof(struct session_plugin_env));
	env.plug_mgr=plug_mgr;
	env.N_per_session_pkt_cnt=10;
	env.N_session=10;

	env.test_mq_pub_plugin_id=stellar_plugin_register(&st, 0, NULL, NULL,&env);
	EXPECT_GE(env.test_mq_pub_plugin_id, 0);

	env.intrinsc_tcp_topic_id=stellar_mq_create_topic(&st, TOPIC_TCP_INPUT, NULL, NULL);
	EXPECT_GE(env.intrinsc_tcp_topic_id, 0);
	EXPECT_EQ(stellar_mq_subscribe(&st, env.intrinsc_tcp_topic_id, test_mq_pub_on_session, env.test_mq_pub_plugin_id), 0);

	env.test_mq_topic_id=stellar_mq_create_topic(&st, "SESSION_MQ_TOPIC", test_session_msg_free, &env);
	EXPECT_GE(env.test_mq_topic_id, 0);

	env.test_mq_sub_plugin_id=stellar_plugin_register(&st, 0, NULL, NULL, &env);
	EXPECT_GE(env.test_mq_sub_plugin_id, 0);
	EXPECT_EQ(stellar_mq_subscribe(&st, env.test_mq_topic_id, test_mq_on_sub_msg, env.test_mq_sub_plugin_id), 0);

	struct packet pkt={&st, TCP, ip_proto};


	struct session sess[env.N_session];

	for(int i=0; i < env.N_session; i++)
	{
		sess[i].session_exdat_rt=session_exdata_runtime_new(plug_mgr);
		sess[i].type=SESSION_TYPE_TCP;
	}

    for (int j = 0; j < env.N_per_session_pkt_cnt; j++)
    {
        plugin_manager_on_packet_input(plug_mgr, &pkt);

        for (int i = 0; i < env.N_session; i++)
        {
            stellar_mq_publish_message(&st, env.intrinsc_tcp_topic_id, &sess[i]);
        }

		plugin_manager_on_packet_output(plug_mgr, &pkt);

    }

    for(int i=0; i < env.N_session; i++)
	{
		session_exdata_runtime_free(sess[i].session_exdat_rt);
	}

	plugin_manager_exit(plug_mgr);

	EXPECT_EQ(env.test_mq_free_called, env.test_mq_pub_called);
	EXPECT_EQ(env.test_mq_sub_called, env.N_session*env.N_per_session_pkt_cnt);
}

struct test_overlimit_session_mq_ctx
{
	int pkt_cnt;
	int pub_cnt;
	int sub_cnt;
};

#if 0
static void *test_overlimit_pub_session_ctx_new(struct session *sess, void *plugin_env)
{
	struct test_overlimit_session_mq_ctx *ctx=CALLOC(struct test_overlimit_session_mq_ctx, 1);
	return ctx;
}

static void test_overlimit_pub_session_ctx_free(struct session *sess, void *session_ctx, void *plugin_env)
{
	struct session_plugin_env *env = (struct session_plugin_env *)plugin_env;
	struct test_overlimit_session_mq_ctx *ctx=(struct test_overlimit_session_mq_ctx *)session_ctx;
	EXPECT_EQ(ctx->pkt_cnt, env->N_per_session_pkt_cnt);
	FREE(ctx);
	return;
}

static void *test_overlimit_sub_session_ctx_new(struct session *sess, void *plugin_env)
{
	struct test_overlimit_session_mq_ctx *ctx=CALLOC(struct test_overlimit_session_mq_ctx, 1);
	return ctx;
}

static void test_overlimit_sub_session_ctx_free(struct session *sess, void *session_ctx, void *plugin_env)
{
	struct session_plugin_env *env = (struct session_plugin_env *)plugin_env;
	struct test_overlimit_session_mq_ctx *ctx=(struct test_overlimit_session_mq_ctx *)session_ctx;
	EXPECT_EQ(ctx->sub_cnt, (env->N_per_session_pkt_cnt*(MAX_MSG_PER_DISPATCH-1))); //minus intrinsic msg
	FREE(ctx);
	return;
}
#endif 

struct test_overlimit_msg
{
    struct session *sess;
	int called;
};

static void test_overlimit_pub_on_session(int topic_id, const void *msg, void *plugin_env)
{
	struct session *sess=(struct session *)msg;
	struct session_plugin_env *env = (struct session_plugin_env *)plugin_env;
	struct test_overlimit_session_mq_ctx *ctx=(struct test_overlimit_session_mq_ctx *)msg;
	EXPECT_TRUE(env!=NULL);
	EXPECT_TRUE(ctx!=NULL);
	struct test_overlimit_msg *pub_msg;
    if (msg)
    {
        env->test_mq_pub_called += 1;
        ctx->pkt_cnt += 1;
		for(int i=0; i < MAX_MSG_PER_DISPATCH*2; i++)
		{
			pub_msg = CALLOC(struct test_overlimit_msg, 1);
        	pub_msg->called = env->test_mq_pub_called;
			pub_msg->sess=sess;
			if(i<(MAX_MSG_PER_DISPATCH-1))// minus intrinsic msg
			{
        		EXPECT_EQ(stellar_mq_publish_message(env->plug_mgr->st, env->test_mq_topic_id, pub_msg), 0);
				ctx->pub_cnt+=1;
			}
			else
			{
        		EXPECT_EQ(stellar_mq_publish_message(env->plug_mgr->st, env->test_mq_topic_id, pub_msg), -1);
				FREE(pub_msg);
			}
		}
    }
    return;
}

static void test_overlimit_on_sub_msg(int topic_id, const void *msg, void *plugin_env)
{
	struct session_plugin_env *env = (struct session_plugin_env *)plugin_env;
	struct test_overlimit_msg *recv_msg=(struct test_overlimit_msg *)msg;
	struct test_overlimit_session_mq_ctx *ctx=(struct test_overlimit_session_mq_ctx *)msg;
	EXPECT_TRUE(env!=NULL);
	EXPECT_TRUE(ctx!=NULL);
	EXPECT_EQ(recv_msg->called, env->test_mq_pub_called);
	env->test_mq_sub_called+=1;
	ctx->sub_cnt+=1;
	return;
}

static void test_overlimit_session_msg_free(void *msg, void *msg_free_arg)
{
	struct session_plugin_env *env = (struct session_plugin_env *)msg_free_arg;
	struct test_overlimit_msg *recv_msg=(struct test_overlimit_msg *)msg;
	if(recv_msg)
	{
		EXPECT_EQ(stellar_mq_publish_message(env->plug_mgr->st, env->test_mq_topic_id, msg), -1);// illegal publish when msg_free
		EXPECT_EQ(env->test_mq_pub_called, recv_msg->called);
		env->test_mq_free_called+=1;
		FREE(msg);
	}
	return;
}

TEST(plugin_manager,DISABLED_session_plugin_pub_msg_overlimt) {

	struct stellar st={0};
	struct plugin_manager_schema *plug_mgr = plugin_manager_init(&st, NULL);
	whitebox_test_plugin_manager_intrisic_metadata(&st, plug_mgr);

	unsigned char ip_proto=6;
	struct session_plugin_env env;
	memset(&env, 0, sizeof(struct session_plugin_env));
	env.plug_mgr=plug_mgr;
	env.N_per_session_pkt_cnt=10;
	env.N_session=10;

	env.test_mq_pub_plugin_id=stellar_plugin_register(&st, 0, NULL, NULL, &env);
	EXPECT_GE(env.test_mq_pub_plugin_id, 0);

	env.intrinsc_tcp_topic_id=stellar_mq_create_topic(&st, TOPIC_TCP_INPUT, NULL, NULL);
	EXPECT_GE(env.intrinsc_tcp_topic_id, 0);
	EXPECT_EQ(stellar_mq_subscribe(&st, env.intrinsc_tcp_topic_id, test_overlimit_pub_on_session, env.test_mq_pub_plugin_id), 0);

	env.test_mq_topic_id=stellar_mq_create_topic(&st, "SESSION_MQ_TOPIC", test_overlimit_session_msg_free, &env);
	EXPECT_GE(env.test_mq_topic_id, 0);

	env.test_mq_sub_plugin_id=stellar_plugin_register(&st, 0, NULL, NULL,  &env);
	EXPECT_GE(env.test_mq_sub_plugin_id, 0);
	EXPECT_EQ(stellar_mq_subscribe(&st, env.test_mq_topic_id, test_overlimit_on_sub_msg, env.test_mq_sub_plugin_id), 0);

	struct packet pkt={&st, TCP, ip_proto};


	struct session sess[env.N_session];

	for(int i=0; i < env.N_session; i++)
	{
		sess[i].session_exdat_rt=session_exdata_runtime_new(plug_mgr);
		sess[i].type=SESSION_TYPE_TCP;
	}

    for (int j = 0; j < env.N_per_session_pkt_cnt; j++)
    {
        plugin_manager_on_packet_input(plug_mgr, &pkt);

        for (int i = 0; i < env.N_session; i++)
        {
            stellar_mq_publish_message(&st, env.intrinsc_tcp_topic_id, &sess[i]);
        }

		plugin_manager_on_packet_output(plug_mgr, &pkt);
    }

    for(int i=0; i < env.N_session; i++)
	{
		session_exdata_runtime_free(sess[i].session_exdat_rt);
	}

	plugin_manager_exit(plug_mgr);

	EXPECT_EQ(env.test_mq_pub_called,env.N_per_session_pkt_cnt*env.N_session);
	EXPECT_EQ(env.test_mq_free_called, env.N_session*env.N_per_session_pkt_cnt*(MAX_MSG_PER_DISPATCH-1));
	EXPECT_EQ(env.test_mq_sub_called, env.N_session*env.N_per_session_pkt_cnt*(MAX_MSG_PER_DISPATCH-1));

}


static void test_dettach_msg_free(void *msg, void *msg_free_arg)
{
	struct session_plugin_env *env = (struct session_plugin_env *)msg_free_arg;
	env->test_mq_free_called+=1;
	return;
}

#if 0
static void *test_dettach_session_ctx_new(struct session *sess, void *plugin_env)
{
	struct test_basic_ctx *ctx=CALLOC(struct test_basic_ctx, 1);
	struct session_plugin_env *env = (struct session_plugin_env *)plugin_env;

	EXPECT_EQ(stellar_mq_publish_message(env->plug_mgr->st, env->test_mq_topic_id, plugin_env), 0);// publish success, but won't be delivered to currnet plugin

	ctx->called+=1;

	EXPECT_EQ(stellar_mq_publish_message(env->plug_mgr->st, env->test_mq_topic_id, plugin_env), 0);// publish success, but won't be delivered to currnet plugin

	return ctx;
}

static void test_dettach_session_ctx_free(struct session *sess, void *session_ctx, void *plugin_env)
{
	struct session_plugin_env *env = (struct session_plugin_env *)plugin_env;
	env->basic_ctx_free_called+=1;
	struct test_basic_ctx *ctx=(struct test_basic_ctx *)session_ctx;
	EXPECT_EQ(sess->sess_pkt_cnt, 1);// first packet ingress, call ctx_free immediately
	EXPECT_EQ(ctx->called, 1);

	EXPECT_EQ(stellar_mq_publish_message(env->plug_mgr->st, env->test_mq_topic_id, plugin_env), 0);// publish success, but won't be delivered to currnet plugin

	FREE(ctx);
}
#endif

static void test_dettach_on_session(int topic_id, const void *msg, void *plugin_env)
{
	struct test_basic_ctx *ctx=(struct test_basic_ctx *)msg;
	struct session_plugin_env *env = (struct session_plugin_env *)plugin_env;

	EXPECT_EQ(topic_id, env->intrinsc_tcp_topic_id);

	ctx->called+=1;
}

TEST(plugin_manager, DISABLED_session_plugin_on_ctx_new_then_dettach) {

	struct stellar st={0};
	struct plugin_manager_schema *plug_mgr = plugin_manager_init(&st, NULL);
	whitebox_test_plugin_manager_intrisic_metadata(&st, plug_mgr);

	unsigned char ip_proto=6;
	struct session_plugin_env env;
	memset(&env, 0, sizeof(struct session_plugin_env));
	env.plug_mgr=plug_mgr;
	env.N_per_session_pkt_cnt=10;
	env.N_session=10;


	int plugin_id=stellar_plugin_register(&st, 0, NULL, NULL, &env);
	EXPECT_GE(plugin_id,0);

	env.intrinsc_tcp_topic_id=stellar_mq_create_topic(&st, TOPIC_TCP_INPUT, NULL, NULL);
	EXPECT_GE(env.intrinsc_tcp_topic_id, 0);
	EXPECT_EQ(stellar_mq_subscribe(&st, env.intrinsc_tcp_topic_id, test_dettach_on_session, plugin_id), 0);

	env.test_mq_topic_id=stellar_mq_create_topic(&st, "SESSION_MQ_TOPIC", test_dettach_msg_free, &env);
	EXPECT_GE(env.test_mq_topic_id, 0);

	EXPECT_EQ(stellar_mq_subscribe(&st, env.test_mq_topic_id, test_dettach_on_session, plugin_id), 0);

	struct packet pkt={&st, TCP, ip_proto};


	struct session sess[env.N_session];
	memset(&sess, 0, sizeof(sess));

	for(int i=0; i < env.N_session; i++)
	{
		sess[i].session_exdat_rt=session_exdata_runtime_new(plug_mgr);
		sess[i].type=SESSION_TYPE_TCP;
	}

    for (int j = 0; j < env.N_per_session_pkt_cnt; j++)
    {
        plugin_manager_on_packet_input(plug_mgr, &pkt);

        for (int i = 0; i < env.N_session; i++)
        {
			sess[i].sess_pkt_cnt+=1;
            stellar_mq_publish_message(&st, env.intrinsc_tcp_topic_id, &sess[i]);
        }

		plugin_manager_on_packet_output(plug_mgr, &pkt);
    }

    for(int i=0; i < env.N_session; i++)
	{
		session_exdata_runtime_free(sess[i].session_exdat_rt);
	}

	plugin_manager_exit(plug_mgr);

	EXPECT_EQ(env.basic_ctx_free_called,env.N_session);
	EXPECT_EQ(env.test_mq_free_called,env.N_session*3);
}

#if 0

static void *test_invalid_pub_msg_session_ctx_new(struct session *sess, void *plugin_env)
{
	struct test_basic_ctx *ctx=CALLOC(struct test_basic_ctx, 1);
	return ctx;
}

static void test_invalid_pub_msg_session_ctx_free(struct session *sess, void *session_ctx, void *plugin_env)
{
	struct session_plugin_env *env = (struct session_plugin_env *)plugin_env;
	env->basic_ctx_free_called+=1;
	struct test_basic_ctx *ctx=(struct test_basic_ctx *)session_ctx;
	EXPECT_EQ(ctx->called,(env->N_per_session_pkt_cnt));

	EXPECT_EQ(stellar_mq_publish_message(env->plug_mgr->st, env->test_mq_topic_id, ctx), -1);// illegal publish when session closing

	FREE(ctx);
}

#endif

static void test_invalid_pub_msg_on_session(int topic_id, const void *msg, void *plugin_env)
{
	struct test_basic_ctx *ctx=(struct test_basic_ctx *)msg;
	ctx->called+=1;
}

TEST(plugin_manager, DISABLED_session_plugin_pub_on_ctx_free) {

	struct stellar st={0};
	struct session_plugin_env env;

// pesudo init stage
	struct plugin_manager_schema *plug_mgr = plugin_manager_init(&st, NULL);
	whitebox_test_plugin_manager_intrisic_metadata(&st, plug_mgr);

// plugin manager register plugin

	int plugin_id=stellar_plugin_register(&st, 0, NULL, NULL, &env);
	EXPECT_GE(plugin_id,0);

	env.intrinsc_tcp_topic_id=stellar_mq_create_topic(&st, TOPIC_TCP_INPUT, NULL, NULL);
	EXPECT_GE(env.intrinsc_tcp_topic_id, 0);
	EXPECT_EQ(stellar_mq_subscribe(&st, env.intrinsc_tcp_topic_id, test_invalid_pub_msg_on_session, plugin_id), 0);

	env.test_mq_topic_id=stellar_mq_create_topic(&st, "SESSION_MQ_TOPIC", NULL, &env);
	EXPECT_GE(env.test_mq_topic_id, 0);

// pesudo packet and session

	memset(&env, 0, sizeof(struct session_plugin_env));
	env.plug_mgr=plug_mgr;
	env.N_per_session_pkt_cnt=10;
	env.N_session=10;

	struct packet pkt={&st, TCP, 6};

	struct session sess[env.N_session];
	memset(&sess, 0, sizeof(sess));

// pesudo running stage
	for(int i=0; i < env.N_session; i++)
	{
		sess[i].session_exdat_rt=session_exdata_runtime_new(plug_mgr);
		sess[i].type=SESSION_TYPE_TCP;
	}

    for (int j = 0; j < env.N_per_session_pkt_cnt; j++)
    {
        plugin_manager_on_packet_input(plug_mgr, &pkt);

        for (int i = 0; i < env.N_session; i++)
        {
			sess[i].sess_pkt_cnt+=1;
            stellar_mq_publish_message(&st, env.intrinsc_tcp_topic_id, &sess[i]);
        }

		plugin_manager_on_packet_output(plug_mgr, &pkt);
    }

    for(int i=0; i < env.N_session; i++)
	{
		session_exdata_runtime_free(sess[i].session_exdat_rt);
	}

// pesudo exit stage
	plugin_manager_exit(plug_mgr);

	EXPECT_EQ(env.basic_ctx_free_called,env.N_session);
}


struct test_session_closing_ctx
{
	int pkt_called;
	int session_free_called;
	int userdefine_on_msg_called;
};

#if 0
static void *test_session_closing_ctx_new(struct session *sess, void *plugin_env)
{
	struct test_session_closing_ctx *ctx=CALLOC(struct test_session_closing_ctx, 1);
	struct session_plugin_env *env = (struct session_plugin_env *)plugin_env;
	env->basic_ctx_new_called+=1;
	return ctx;
}

static void test_session_closing_ctx_free(struct session *sess, void *session_ctx, void *plugin_env)
{
	struct session_plugin_env *env = (struct session_plugin_env *)plugin_env;
	env->basic_ctx_free_called+=1;
	struct test_session_closing_ctx *ctx=(struct test_session_closing_ctx *)session_ctx;
	EXPECT_EQ(ctx->pkt_called,env->N_per_session_pkt_cnt);
	EXPECT_EQ(ctx->session_free_called,1);
	FREE(ctx);
}


static void test_session_closing_on_session_free(struct session *sess, void *per_session_ctx, void *plugin_env)
{
	struct test_session_closing_ctx *ctx=(struct test_session_closing_ctx *)per_session_ctx;
	struct session_plugin_env *env = (struct session_plugin_env *)plugin_env;
	if(sess->state==SESSION_STATE_CLOSING)
	{
		ctx->session_free_called+=1;
		stellar_mq_publish_message(env->plug_mgr->st, env->test_mq_topic_id, env);
		env->test_mq_pub_called+=1;
	}
}

#endif

static void test_session_closing_on_intrisic_msg( int topic_id, const void *msg,  void *plugin_env)
{
	struct test_session_closing_ctx *ctx=(struct test_session_closing_ctx *)msg;
	if(msg)ctx->pkt_called+=1;
}

static void test_session_closing_on_userdefine_msg(int topic_id, const void *msg, void *plugin_env)
{
	struct test_session_closing_ctx *ctx=(struct test_session_closing_ctx *)msg;
	struct session_plugin_env *env = (struct session_plugin_env *)plugin_env;
	ctx->userdefine_on_msg_called+=1;
	EXPECT_EQ(msg, plugin_env);
	env->test_mq_sub_called+=1;
}

TEST(plugin_manager, DISABLED_session_plugin_pub_msg_on_closing) {

	struct stellar st={0};
	struct session_plugin_env env;
	memset(&env, 0, sizeof(struct session_plugin_env));

// pesudo init stage
	struct plugin_manager_schema *plug_mgr = plugin_manager_init(&st, NULL);
	whitebox_test_plugin_manager_intrisic_metadata(&st, plug_mgr);

// plugin manager register plugin

	int plugin_id=stellar_plugin_register(&st, 0, NULL, NULL, &env);
	EXPECT_GE(plugin_id,0);

	env.intrinsc_tcp_topic_id=stellar_mq_create_topic(&st, TOPIC_TCP_INPUT, NULL, NULL);
	EXPECT_GE(env.intrinsc_tcp_topic_id, 0);
	EXPECT_EQ(stellar_mq_subscribe(&st, env.intrinsc_tcp_topic_id, test_session_closing_on_intrisic_msg, plugin_id), 0);

	env.test_mq_topic_id=stellar_mq_create_topic(&st, "SESSION_CLOSING_TOPIC", NULL, &env);
	EXPECT_GE(env.test_mq_topic_id, 0);
	EXPECT_EQ(stellar_mq_subscribe(&st, env.test_mq_topic_id, test_session_closing_on_userdefine_msg, plugin_id), 0);

// pesudo packet and session

	env.plug_mgr=plug_mgr;
	env.N_per_session_pkt_cnt=10;
	env.N_session=10;

	struct packet pkt={&st, TCP, 6};

	struct session sess[env.N_session];
	memset(&sess, 0, sizeof(sess));

// pesudo running stage
	for(int i=0; i < env.N_session; i++)
	{
		sess[i].state=SESSION_STATE_OPENING;
		sess[i].session_exdat_rt=session_exdata_runtime_new(plug_mgr);
		sess[i].type=SESSION_TYPE_TCP;
	}

    for (int j = 0; j < env.N_per_session_pkt_cnt; j++)
    {
        plugin_manager_on_packet_input(plug_mgr, &pkt);

        for (int i = 0; i < env.N_session; i++)
        {
			sess[i].sess_pkt_cnt+=1;
			sess[i].state=SESSION_STATE_ACTIVE;
            stellar_mq_publish_message(&st, env.intrinsc_tcp_topic_id, &sess[i]);
        }

		plugin_manager_on_packet_output(plug_mgr, &pkt);
    }

    for(int i=0; i < env.N_session; i++)
	{
		sess[i].state=SESSION_STATE_CLOSING;
		session_exdata_runtime_free(sess[i].session_exdat_rt);
	}

// pesudo exit stage
	plugin_manager_exit(plug_mgr);

	//EXPECT_EQ(env.basic_ctx_new_called,env.N_session);
	//EXPECT_EQ(env.basic_ctx_free_called,env.N_session);
	//EXPECT_EQ(env.test_mq_pub_called,env.N_session);
	//EXPECT_EQ(env.test_mq_sub_called,env.N_session);
}

struct test_session_called_ctx
{
	int called;
};

#if 0
static void *test_session_called_ctx_new(struct session *sess, void *plugin_env)
{
	struct test_session_called_ctx *ctx=CALLOC(struct test_session_called_ctx, 1);
	return ctx;
}

static void test_session_called_ctx_free(struct session *sess, void *session_ctx, void *plugin_env)
{
	FREE(session_ctx);
}
#endif
//test session topic is active
static void test_session_mq_topic_is_active_plugin_1_on_msg(int topic_id, const void *msg, void *plugin_env)
{
	struct session_plugin_env *env = (struct session_plugin_env *)plugin_env;	
	struct test_session_called_ctx *ctx=(struct test_session_called_ctx *)msg;
	env->plugin_id_1_called+=1;
	ctx->called+=1;
	EXPECT_EQ(env->intrinsc_tcp_topic_id, topic_id);

	return;
}

static void test_session_mq_topic_is_active_plugin_2_on_msg(int topic_id, const void *msg, void *plugin_env)
{
	struct session_plugin_env *env = (struct session_plugin_env *)plugin_env;	
	struct test_session_called_ctx *ctx=(struct test_session_called_ctx *)msg;
	env->plugin_id_2_called+=1;
	ctx->called+=1;
	EXPECT_EQ(env->intrinsc_tcp_topic_id, topic_id);


	if(ctx->called > env->N_per_session_pkt_cnt/2)
	{
	}
	return;
}

TEST(plugin_manager, DISABLED_test_session_mq_topic_is_active) {

	struct stellar st={0};
	struct session_plugin_env env;
	memset(&env, 0, sizeof(struct session_plugin_env));

// pesudo init stage
	struct plugin_manager_schema *plug_mgr = plugin_manager_init(&st, NULL);
	whitebox_test_plugin_manager_intrisic_metadata(&st, plug_mgr);

// plugin manager register plugin

	int plugin_id_1=stellar_plugin_register(&st, 0, NULL, NULL, &env);
	EXPECT_GE(plugin_id_1,0);

	int plugin_id_2=stellar_plugin_register(&st, 0, NULL, NULL, &env);
	EXPECT_GE(plugin_id_2,0);

	env.plugin_id_1=plugin_id_1;
	env.plugin_id_2=plugin_id_2;

	env.intrinsc_tcp_topic_id=stellar_mq_create_topic(&st, TOPIC_TCP_INPUT, NULL, NULL);
	EXPECT_GE(env.intrinsc_tcp_topic_id, 0);
	EXPECT_EQ(stellar_mq_subscribe(&st, env.intrinsc_tcp_topic_id, test_session_mq_topic_is_active_plugin_1_on_msg, plugin_id_1), 0);
	EXPECT_EQ(stellar_mq_subscribe(&st, env.intrinsc_tcp_topic_id, test_session_mq_topic_is_active_plugin_2_on_msg, plugin_id_2), 0);

// pesudo packet and session

	env.plug_mgr=plug_mgr;
	env.N_per_session_pkt_cnt=10;
	env.N_session=10;

	struct packet pkt={&st, TCP, 6};

	struct session sess[env.N_session];
	memset(&sess, 0, sizeof(sess));

// pesudo running stage
	for(int i=0; i < env.N_session; i++)
	{
		sess[i].state=SESSION_STATE_OPENING;
		sess[i].session_exdat_rt=session_exdata_runtime_new(plug_mgr);
		sess[i].type=SESSION_TYPE_TCP;
	}

    for (int j = 0; j < env.N_per_session_pkt_cnt; j++)
    {
        plugin_manager_on_packet_input(plug_mgr, &pkt);

        for (int i = 0; i < env.N_session; i++)
        {
			sess[i].sess_pkt_cnt+=1;
			sess[i].state=SESSION_STATE_ACTIVE;
            stellar_mq_publish_message(&st, env.intrinsc_tcp_topic_id, &sess[i]);
        }

		plugin_manager_on_packet_output(plug_mgr, &pkt);
    }

    for(int i=0; i < env.N_session; i++)
	{
		sess[i].state=SESSION_STATE_CLOSING;
		session_exdata_runtime_free(sess[i].session_exdat_rt);
	}

// pesudo exit stage
	plugin_manager_exit(plug_mgr);

	EXPECT_EQ(env.plugin_id_1_called,env.N_session*1);// per session called once, then ignore
	EXPECT_EQ(env.plugin_id_2_called,env.N_session*(env.N_per_session_pkt_cnt/2+1));// per session called one half, then ignore
}

//test dettach session
static void test_session_dettach_plugin_1_on_msg(int topic_id, const void *msg, void *plugin_env)
{
	struct session_plugin_env *env = (struct session_plugin_env *)plugin_env;	
	struct test_session_called_ctx *ctx=(struct test_session_called_ctx *)msg;
	env->plugin_id_1_called+=1;
	ctx->called+=1;
	EXPECT_EQ(env->intrinsc_tcp_topic_id, topic_id);
	return;
}

static void test_session_dettach_plugin_2_on_msg(int topic_id, const void *msg, void *plugin_env)
{
	struct session_plugin_env *env = (struct session_plugin_env *)plugin_env;	
	struct test_session_called_ctx *ctx=(struct test_session_called_ctx *)msg;
	env->plugin_id_2_called+=1;
	ctx->called+=1;
	EXPECT_EQ(env->intrinsc_tcp_topic_id, topic_id);


	if(ctx->called > env->N_per_session_pkt_cnt/2)
	{
	}

	return;
}

TEST(plugin_manager, DISABLED_test_session_dettach) {

	struct stellar st={0};
	struct session_plugin_env env;
	memset(&env, 0, sizeof(struct session_plugin_env));

// pesudo init stage
	struct plugin_manager_schema *plug_mgr = plugin_manager_init(&st, NULL);
	whitebox_test_plugin_manager_intrisic_metadata(&st, plug_mgr);

// plugin manager register plugin

	int plugin_id_1=stellar_plugin_register(&st, 0, NULL, NULL, &env);
	EXPECT_GE(plugin_id_1,0);

	int plugin_id_2=stellar_plugin_register(&st, 0, NULL, NULL, &env);
	EXPECT_GE(plugin_id_2,0);

	env.plugin_id_1=plugin_id_1;
	env.plugin_id_2=plugin_id_2;

	env.intrinsc_tcp_topic_id=stellar_mq_create_topic(&st, TOPIC_TCP_INPUT, NULL, NULL);
	EXPECT_GE(env.intrinsc_tcp_topic_id, 0);
	EXPECT_EQ(stellar_mq_subscribe(&st, env.intrinsc_tcp_topic_id, test_session_dettach_plugin_1_on_msg, plugin_id_1), 0);
	EXPECT_EQ(stellar_mq_subscribe(&st, env.intrinsc_tcp_topic_id, test_session_dettach_plugin_2_on_msg, plugin_id_2), 0);

// pesudo packet and session

	env.plug_mgr=plug_mgr;
	env.N_per_session_pkt_cnt=10;
	env.N_session=10;

	struct packet pkt={&st, TCP, 6};

	struct session sess[env.N_session];
	memset(&sess, 0, sizeof(sess));

// pesudo running stage
	for(int i=0; i < env.N_session; i++)
	{
		sess[i].state=SESSION_STATE_OPENING;
		sess[i].session_exdat_rt=session_exdata_runtime_new(plug_mgr);
		sess[i].type=SESSION_TYPE_TCP;
	}

    for (int j = 0; j < env.N_per_session_pkt_cnt; j++)
    {
        plugin_manager_on_packet_input(plug_mgr, &pkt);

        for (int i = 0; i < env.N_session; i++)
        {
			sess[i].sess_pkt_cnt+=1;
			sess[i].state=SESSION_STATE_ACTIVE;
            stellar_mq_publish_message(&st, env.intrinsc_tcp_topic_id, &sess[i]);
        }

		plugin_manager_on_packet_output(plug_mgr, &pkt);
    }

    for(int i=0; i < env.N_session; i++)
	{
		sess[i].state=SESSION_STATE_CLOSING;
		session_exdata_runtime_free(sess[i].session_exdat_rt);
	}

// pesudo exit stage
	plugin_manager_exit(plug_mgr);

	EXPECT_EQ(env.plugin_id_1_called,env.N_session*1);// per session called once, then ignore
	EXPECT_EQ(env.plugin_id_2_called,env.N_session*(env.N_per_session_pkt_cnt/2+1));// per session called one half, then ignore

}

#if 0

//test dettach session
static void test_session_mq_priority_plugin_1_on_msg(int topic_id, const void *msg, void *plugin_env)
{
	struct session_plugin_env *env = (struct session_plugin_env *)plugin_env;	
	env->plugin_id_1_called+=1;
	if(topic_id == env->intrinsc_tcp_topic_id)
	{
        struct session *sess = (struct session *)msg;
        struct test_session_called_ctx *ctx =
            (struct test_session_called_ctx *)session_exdata_get(sess, env->exdata_ctx_1_id);
        if (ctx == NULL)
        {
            ctx = CALLOC(struct test_session_called_ctx, 1);
            session_exdata_set(sess, env->exdata_ctx_1_id, ctx);
        }
		ctx->called+=1;
        EXPECT_EQ(ctx->called%3, 1);// intrinsc msg has high priority
		EXPECT_EQ(stellar_mq_publish_message_with_priority(env->plug_mgr->st, env->test_mq_topic_id, (void *)(long)env->plugin_id_1, STELLAR_MQ_PRIORITY_LOW), 0);
	}
	if(topic_id == env->test_mq_topic_id)
	{
		if(ctx->called%3 == 2)
		{
			EXPECT_EQ((int)(long)msg, env->plugin_id_2);
		}
		if(ctx->called%3 == 0)
		{
			EXPECT_EQ((int )(long)msg, env->plugin_id_1);
		}
	}
	return;
}

static void test_session_mq_priority_plugin_2_on_msg(int topic_id, const void *msg, void *plugin_env)
{
	struct session_plugin_env *env = (struct session_plugin_env *)plugin_env;	

	env->plugin_id_2_called+=1;

	if(topic_id == env->intrinsc_tcp_topic_id)
	{
        struct session *sess = (struct session *)msg;
        struct test_session_called_ctx *ctx =
            (struct test_session_called_ctx *)session_exdata_get(sess, env->exdata_ctx_2_id);
        if (ctx == NULL)
        {
            ctx = CALLOC(struct test_session_called_ctx, 1);
            session_exdata_set(sess, env->exdata_ctx_2_id, ctx);
        }
		ctx->called+=1;
        EXPECT_EQ(ctx->called % 3, 1);
        // publish msg has normal priority
		EXPECT_EQ(stellar_mq_publish_message(env->plug_mgr->st, env->test_mq_topic_id, (void *)(long)env->plugin_id_2), 0);
	}
	if(topic_id == env->test_mq_topic_id)
	{
		if(ctx->called%3 == 2)
		{
			EXPECT_EQ((int)(long)msg, env->plugin_id_2);
		}
		if(ctx->called%3 == 0)
		{
			EXPECT_EQ((int)(long)msg, env->plugin_id_1);
		}
	}	
	return;
}

TEST(plugin_manager, test_session_mq_priority) {

	struct stellar st={0};
	struct session_plugin_env env;
	memset(&env, 0, sizeof(struct session_plugin_env));

// pesudo init stage
	struct plugin_manager_schema *plug_mgr = plugin_manager_init(&st, NULL);
	whitebox_test_plugin_manager_intrisic_metadata(&st, plug_mgr);

// plugin manager register plugin

	int plugin_id_1=stellar_plugin_register(&st, 0,NULL, NULL, &env);
	EXPECT_GE(plugin_id_1,0);

	int plugin_id_2=stellar_plugin_register(&st, 0, NULL, NULL, &env);
	EXPECT_GE(plugin_id_2,0);

	env.plugin_id_1=plugin_id_1;
	env.plugin_id_2=plugin_id_2;

	env.exdata_ctx_1_id=stellar_exdata_new_index(&st, "SESSION_CTX_1", stellar_exdata_free_default, &env) ;
	env.exdata_ctx_2_id=stellar_exdata_new_index(&st, "SESSION_CTX_2", stellar_exdata_free_default, &env) ;

	env.intrinsc_tcp_topic_id=stellar_mq_create_topic(&st, TOPIC_TCP_INPUT, NULL, NULL);
	EXPECT_GE(env.intrinsc_tcp_topic_id, 0);
	EXPECT_EQ(stellar_mq_subscribe(&st, env.intrinsc_tcp_topic_id, test_session_mq_priority_plugin_1_on_msg, plugin_id_1), 0);
	EXPECT_EQ(stellar_mq_subscribe(&st, env.intrinsc_tcp_topic_id, test_session_mq_priority_plugin_2_on_msg, plugin_id_2), 0);

	env.test_mq_topic_id=stellar_mq_create_topic(&st, "SESSION_PRIORITY_TOPIC", NULL, &env);
	EXPECT_GE(env.test_mq_topic_id, 0);
	EXPECT_EQ(stellar_mq_subscribe(&st, env.test_mq_topic_id, test_session_mq_priority_plugin_1_on_msg, plugin_id_1), 0);
	EXPECT_EQ(stellar_mq_subscribe(&st, env.test_mq_topic_id, test_session_mq_priority_plugin_2_on_msg, plugin_id_2), 0);

// pesudo packet and session

	env.plug_mgr=plug_mgr;
	env.N_per_session_pkt_cnt=10;
	env.N_session=10;

	struct packet pkt={&st, TCP, 6};

	struct session sess[env.N_session];
	memset(&sess, 0, sizeof(sess));

// pesudo running stage
	for(int i=0; i < env.N_session; i++)
	{
		sess[i].state=SESSION_STATE_OPENING;
		sess[i].session_exdat_rt=session_exdata_runtime_new(plug_mgr);
		sess[i].type=SESSION_TYPE_TCP;
	}

    for (int j = 0; j < env.N_per_session_pkt_cnt; j++)
    {
        plugin_manager_on_packet_input(plug_mgr, &pkt);

        for (int i = 0; i < env.N_session; i++)
        {
			sess[i].sess_pkt_cnt+=1;
			sess[i].state=SESSION_STATE_ACTIVE;
            stellar_mq_publish_message(&st, env.intrinsc_tcp_topic_id, &sess[i]);
        }

		plugin_manager_on_packet_output(plug_mgr, &pkt);
    }

    for(int i=0; i < env.N_session; i++)
	{
		sess[i].state=SESSION_STATE_CLOSING;
		session_exdata_runtime_free(sess[i].session_exdat_rt);
	}

// pesudo exit stage
	plugin_manager_exit(plug_mgr);

	// each session publish TCP TOPIC per_session_pkt_cnt+1, and SESSION_PRIORITY_TOPIC 2*(msg per_session_pkt_cnt+1)
	EXPECT_EQ(env.plugin_id_1_called,env.N_session*((env.N_per_session_pkt_cnt)*3));
	EXPECT_EQ(env.plugin_id_2_called,env.N_session*((env.N_per_session_pkt_cnt)*3));

}

#endif

void test_session_exdata_free_pub_msg_exdata_free(int idx, void *ex_ptr, void *arg)
{
	struct session_plugin_env *env = (struct session_plugin_env *)arg;	
	EXPECT_EQ(stellar_mq_publish_message(env->plug_mgr->st, env->intrinsc_tcp_topic_id, arg), -1);	
	env->basic_exdata_free_called+=1;
}

static void test_session_exdata_free_pub_msg_on_session(int topic_id, const void *msg, void *plugin_env)
{
	struct session *sess=(struct session *)msg;
	struct session_plugin_env *env = (struct session_plugin_env *)plugin_env;		
	EXPECT_EQ(session_exdata_set(sess, env->basic_exdata_idx, sess), 0);
	if(msg)env->plugin_id_1_called+=1;
}

TEST(plugin_manager, session_exdata_free_pub_msg) {

	struct stellar st={0};
	struct session_plugin_env env;

// pesudo init stage
	struct plugin_manager_schema *plug_mgr = plugin_manager_init(&st, NULL);
	whitebox_test_plugin_manager_intrisic_metadata(&st, plug_mgr);

// plugin manager register plugin

	env.plugin_id_1=stellar_plugin_register(&st, 0, NULL, NULL, &env);
	EXPECT_GE(env.plugin_id_1,0);

	env.intrinsc_tcp_topic_id=stellar_mq_create_topic(&st, TOPIC_TCP_INPUT, NULL, NULL);
	EXPECT_GE(env.intrinsc_tcp_topic_id, 0);
	EXPECT_EQ(stellar_mq_subscribe(&st, env.intrinsc_tcp_topic_id, test_session_exdata_free_pub_msg_on_session, env.plugin_id_1), 0);

	env.basic_exdata_idx=stellar_exdata_new_index(&st, "BASIC_EXDATA", test_session_exdata_free_pub_msg_exdata_free, &env) ;
	EXPECT_GE(env.basic_exdata_idx, 0);

// pesudo packet and session

	memset(&env, 0, sizeof(struct session_plugin_env));
	env.plug_mgr=plug_mgr;
	env.N_per_session_pkt_cnt=10;
	env.N_session=10;

	struct packet pkt={&st, TCP, 6};

	struct session sess[env.N_session];
	memset(&sess, 0, sizeof(sess));

// pesudo running stage
	for(int i=0; i < env.N_session; i++)
	{
		sess[i].session_exdat_rt=session_exdata_runtime_new(plug_mgr);
		sess[i].type=SESSION_TYPE_TCP;
	}

    for (int j = 0; j < env.N_per_session_pkt_cnt; j++)
    {
        plugin_manager_on_packet_input(plug_mgr, &pkt);

        for (int i = 0; i < env.N_session; i++)
        {
			sess[i].sess_pkt_cnt+=1;
            stellar_mq_publish_message(&st, env.intrinsc_tcp_topic_id, &sess[i]);
        }

		plugin_manager_on_packet_output(plug_mgr, &pkt);
    }

    for(int i=0; i < env.N_session; i++)
	{
		session_exdata_runtime_free(sess[i].session_exdat_rt);
	}

// pesudo exit stage
	plugin_manager_exit(plug_mgr);

	EXPECT_EQ(env.basic_exdata_free_called,env.N_session);
	EXPECT_EQ(env.plugin_id_1_called,env.N_session*env.N_per_session_pkt_cnt);
}


/**********************************************
 * TEST PLUGIN MANAGER ON POLLING PLUGIN INIT *
 **********************************************/

int test_plugin_on_polling_func(void *plugin_env)
{
	return 1;
}

TEST(plugin_manager_init, polling_plugin_register) {
	struct stellar st={0};
	struct plugin_manager_schema *plug_mgr = plugin_manager_init(&st, NULL);
	st.plug_mgr=plug_mgr;
	whitebox_test_plugin_manager_intrisic_metadata(&st, plug_mgr);

	int plugin_id = stellar_polling_plugin_register(&st, test_plugin_on_polling_func, &st);
    {
        SCOPED_TRACE("White-box test, check stellar internal schema");
		EXPECT_TRUE(plugin_id>=0);
        struct registered_polling_plugin_schema *schema = (struct registered_polling_plugin_schema *)utarray_eltptr(
            plug_mgr->registered_polling_plugin_array, (unsigned int)(plugin_id));
        EXPECT_EQ(schema->on_polling, (void *)test_plugin_on_polling_func);
        EXPECT_EQ(schema->plugin_env, &st);
        EXPECT_EQ(utarray_len(plug_mgr->registered_polling_plugin_array), 1);
    }

    plugin_manager_exit(plug_mgr);
}

/**********************************************
 * TEST PLUGIN MANAGER ON POLLING PLUGIN RUNTIME *
 **********************************************/

struct polling_plugin_env
{
	struct plugin_manager_schema *plug_mgr;
	int N_polling;
	int return0_polling_called;
	int return1_polling_called;
};

int return1_plugin_on_polling(void *plugin_env)
{
	struct polling_plugin_env *env = (struct polling_plugin_env *)plugin_env;
	env->return1_polling_called+=1;
	return 1;
}

int return0_plugin_on_polling(void *plugin_env)
{
	struct polling_plugin_env *env = (struct polling_plugin_env *)plugin_env;
	env->return0_polling_called+=1;
	return 0;
}

TEST(plugin_manager, basic_polling_plugins) {

// pesudo init stage
	struct stellar st={0};
	struct plugin_manager_schema *plug_mgr = plugin_manager_init(&st, NULL);
	st.plug_mgr=plug_mgr;
	struct polling_plugin_env env;
	memset(&env, 0, sizeof(struct polling_plugin_env));
	env.plug_mgr=plug_mgr;
	whitebox_test_plugin_manager_intrisic_metadata(&st, plug_mgr);

// plugin manager register plugin
	int plugin_id = stellar_polling_plugin_register(&st, return0_plugin_on_polling, &env);
	EXPECT_TRUE(plugin_id>=0);
	plugin_id = stellar_polling_plugin_register(&st, return1_plugin_on_polling, &env);
	EXPECT_TRUE(plugin_id>=0);

// pesudo runtime stage

	env.plug_mgr=plug_mgr;
	env.N_polling=10;

	for(int i=0; i<env.N_polling; i++)
	{
		EXPECT_EQ(plugin_manager_on_polling(plug_mgr), 1);
	}

// pesudo exit stage
	plugin_manager_exit(plug_mgr);

	EXPECT_EQ(env.return0_polling_called,env.N_polling);
	EXPECT_EQ(env.return1_polling_called,env.N_polling);

}


/**********************************************
 * GTEST MAIN *
 **********************************************/

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