tango-maat/test/maat_ex_data_gtest.cpp

#include "maat.h"
#include "log/log.h"
#include "maat_utils.h"
#include "json2iris.h"
#include "maat_command.h"
#include "maat_ex_data.h"
#include "maat_garbage_collection.h"

#include <gtest/gtest.h>
#include <limits.h>

const char *g_table_info_path = "./table_info.json";
const char *g_json_filename = "maat_json.json";
struct log_handle *g_logger = NULL;
struct maat *g_maat_inst = NULL;

struct user_info {
	char name[16];
	char ip_addr[32];
	int id;
};

static void
ex_data_new_cb(const char *table_name, int table_id, const char *key,
               const char *table_line, void **ad, long argl, void *argp)
{
	int *counter = (int *)argp;
	struct user_info *u = ALLOC(struct user_info, 1);

	int ret = sscanf(table_line, "%d\t%s\t%s", &(u->id), u->ip_addr, u->name);
	EXPECT_EQ(ret, 3);

	*ad = u;
	(*counter)++;
}

static void ex_data_free_cb(int table_id, void **ad, long argl, void *argp)
{
	struct user_info *u = (struct user_info *)(*ad);

    memset(u, 0, sizeof(struct user_info));
	free(u);
	*ad = NULL;
}

static void
ex_data_dup_cb(int table_id,	void **to, void **from, long argl, void *argp)
{
	struct user_info *u = (struct user_info *)(*from);
	*to = u;
}

TEST(EXDataRuntime, Update) {
    const char *table_name = "TEST_PLUGIN_EXDATA_TABLE";
	int table_id = maat_get_table_id(g_maat_inst, table_name);
	ASSERT_GT(table_id, 0);

    int ex_data_counter = 0;
    struct ex_data_runtime *ex_data_rt = ex_data_runtime_new(table_id, 3, g_logger);
    struct ex_container_schema *container_schema = ALLOC(struct ex_container_schema, 1);
    container_schema->custom_data_free = NULL;
    container_schema->table_id = table_id;
    container_schema->ex_schema.new_func = ex_data_new_cb;
    container_schema->ex_schema.free_func = ex_data_free_cb;
    container_schema->ex_schema.dup_func = ex_data_dup_cb;
    container_schema->ex_schema.argl = 0;
    container_schema->ex_schema.argp = &ex_data_counter;
    container_schema->set_flag = 1;

    ex_data_runtime_set_ex_container_schema(ex_data_rt, container_schema);

    const char *row1 = "101\t192.168.0.1\tmahuateng";
    const char *key1 = "192.168.0.1";
    size_t key1_len = strlen(key1);
    void *ex_data = ex_data_runtime_row2ex_data(ex_data_rt, table_name, row1,
                                                key1, key1_len);
    EXPECT_EQ(ex_data_counter, 1);

    struct ex_container *ex_container = ex_container_new(ex_data, NULL);
    int ret = ex_data_runtime_add_ex_container(ex_data_rt, key1, key1_len, ex_container);
    EXPECT_EQ(ret, 0);

    const char *row2 = "102\t192.168.0.2\tliyanhong";
    const char *key2 = "192.168.0.2";
    size_t key2_len = strlen(key2);
    ex_data = ex_data_runtime_row2ex_data(ex_data_rt, table_name, row2, key2, key2_len);
    ex_container = ex_container_new(ex_data, NULL);
    ret = ex_data_runtime_add_ex_container(ex_data_rt, key2, key2_len, ex_container);
    EXPECT_EQ(ret, 0);

    ex_data = ex_data_runtime_row2ex_data(ex_data_rt, table_name, row2, key2, key2_len);
    ex_container = ex_container_new(ex_data, NULL);
    ret = ex_data_runtime_add_ex_container(ex_data_rt, key2, key2_len, ex_container);
    EXPECT_EQ(ret, -1);
    ex_container_free(ex_data_rt, ex_container);

    ex_data_runtime_commit(ex_data_rt);

    void *res_data1 = ex_data_runtime_get_ex_data_by_key(ex_data_rt, key1, key1_len);
    EXPECT_TRUE(res_data1 != NULL);

    struct user_info *info1 = (struct user_info *)res_data1;
    EXPECT_EQ(0, strcmp(info1->name, "mahuateng"));
    EXPECT_EQ(info1->id, 101);

    ex_container = NULL;
    void *res_data2 = ex_data_runtime_get_ex_data_by_key(ex_data_rt, key2, key2_len);
    EXPECT_TRUE(res_data2 != NULL);

    struct user_info *info2 = (struct user_info *)res_data2;
    EXPECT_EQ(0, strcmp(info2->name, "liyanhong"));
    EXPECT_EQ(info2->id, 102);

    ex_data_runtime_del_ex_container(ex_data_rt, key1, key1_len);
    ex_data_runtime_commit(ex_data_rt);

    EXPECT_EQ(0, strcmp(info1->name, "mahuateng"));
    EXPECT_EQ(info1->id, 101);

    sleep(5);
    ex_data_runtime_garbage_collect_routine(ex_data_rt);

    EXPECT_EQ(0, strcmp(info2->name, "liyanhong"));
    EXPECT_EQ(info2->id, 102);

    ex_data_runtime_free(ex_data_rt);
    FREE(container_schema);
}

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

    g_logger = log_handle_create("./maat_ex_data_gtest.log", 0);
    
	struct maat_options *opts = maat_options_new();
    char json_path[PATH_MAX] = {0};
    snprintf(json_path, sizeof(json_path), "./%s", g_json_filename);
    maat_options_set_json_file(opts, json_path);

    g_maat_inst = maat_new(opts, g_table_info_path);
    maat_options_free(opts);
    EXPECT_TRUE(g_maat_inst != NULL);

	ret=RUN_ALL_TESTS();

    log_handle_destroy(g_logger);
    maat_free(g_maat_inst);
    g_maat_inst = NULL;
	return ret;
}