#include <gtest/gtest.h>

#include "policy.h"
#include "raw_packet.h"

unsigned char data1[] = {
	0x00, 0x00, 0x00, 0x00, 0x00, 0x04, 0xa4, 0xc6, 0x4f, 0x3b, 0xb3, 0x9a, 0x81, 0x00, 0x66, 0x58, 0x81, 0x00, 0x61, 0xf9, 0x08, 0x00, 0x45, 0xb8, 0x00, 0x94,
	0xe8, 0x58, 0x00, 0x00, 0xff, 0x04, 0x11, 0x48, 0x45, 0x43, 0x23, 0x92, 0x29, 0xca, 0x2e, 0x6e, 0x45, 0xb8, 0x00, 0x80, 0x00, 0x01, 0x00, 0x00, 0xfe, 0x11,
	0xde, 0x84, 0x0a, 0x0a, 0x64, 0x19, 0x0a, 0x0a, 0x65, 0x02, 0xf3, 0x9f, 0x42, 0x68, 0x00, 0x6c, 0x4b, 0x9a, 0x00, 0x02, 0x00, 0x00, 0x04, 0x73, 0x6c, 0x10,
	0x00, 0x00, 0x00, 0x00, 0x00, 0x01, 0x00, 0x00, 0xab, 0xcd, 0xab, 0xcd, 0xab, 0xcd, 0xab, 0xcd, 0xab, 0xcd, 0xab, 0xcd, 0xab, 0xcd, 0xab, 0xcd, 0xab, 0xcd,
	0xab, 0xcd, 0xab, 0xcd, 0xab, 0xcd, 0xab, 0xcd, 0xab, 0xcd, 0xab, 0xcd, 0xab, 0xcd, 0xab, 0xcd, 0xab, 0xcd, 0xab, 0xcd, 0xab, 0xcd, 0xab, 0xcd, 0xab, 0xcd,
	0xab, 0xcd, 0xab, 0xcd, 0xab, 0xcd, 0xab, 0xcd, 0xab, 0xcd, 0xab, 0xcd, 0xab, 0xcd, 0xab, 0xcd, 0xab, 0xcd, 0xab, 0xcd, 0xab, 0xcd, 0xab, 0xcd, 0xab, 0xcd,
	0xab, 0xcd, 0xab, 0xcd, 0xab, 0xcd, 0xab, 0xcd, 0xab, 0xcd, 0xab, 0xcd, 0xab, 0xcd};

TEST(POLICY, SELECTED_CHAINING_LIFE_CYCLE)
{
	struct selected_chaining *chaining = NULL;

	chaining = selected_chaining_create(128);
	EXPECT_TRUE(chaining != nullptr);

	selected_chaining_destory(chaining);
}

TEST(POLICY, POLICY_ENFORCER_LIFE_CYCLE)
{
	struct raw_pkt_parser *parser = raw_packet_parser_create(LAYER_TYPE_ALL, 8);
	EXPECT_TRUE(parser != nullptr);
	const void *payload = raw_packet_parser_parse(parser, (const void *)data1, sizeof(data1));
	EXPECT_TRUE(payload != nullptr);
	EXPECT_TRUE((char *)payload - (char *)&data1 == 70);

	const char *profile = "./test_resource/sce.conf";
	struct policy_enforcer *enforcer = policy_enforcer_create("SCE", profile, 8, NULL);
	EXPECT_TRUE(enforcer != nullptr);
	EXPECT_TRUE(policy_enforcer_register(enforcer) == 0);

	int policy_id = 2;
	int dir_is_internal = 1;
	struct selected_chaining *chaining = policy_enforce_select_chaining(enforcer, parser, policy_id, dir_is_internal);
	EXPECT_TRUE(chaining != nullptr);
	selected_chaining_dump(chaining);
	selected_chaining_bref(chaining);

	selected_chaining_destory(chaining);
	policy_enforcer_destory(enforcer);
	raw_packet_parser_destory(parser);
}

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