feature:新增register测试用例,修复插入metric slot位置错误bug

1 files changed, 91 insertions, 0 deletions

diff --git a/test/src/gtest_fieldstat.cpp b/test/src/gtest_fieldstat.cpp
index 9850d7f..4bd6c3a 100644
--- a/test/src/gtest_fieldstat.cpp
+++ b/test/src/gtest_fieldstat.cpp
@@ -46,6 +46,97 @@ TEST(FeildStatAPI, FieldStatSetPrometheusOutput)
 	EXPECT_EQ(1, g_prometheus_endpoint_instance.fs_instance_cnt);
 }
 
+TEST(FeildStatAPI, FieldStatSetLineProtocolServer)
+{
+	struct fieldstat_instance *instance = NULL;	
+	int ret_set_line_protocol_server = 0;
+
+	instance = fieldstat_instance_create("test");
+	EXPECT_STREQ("test", instance->name);
+
+	ret_set_line_protocol_server = fieldstat_set_line_protocol_server(instance, "127.0.0.1", 8001);
+
+	EXPECT_EQ(0, ret_set_line_protocol_server);
+	EXPECT_EQ(1, instance->line_protocol_output_enable);
+	EXPECT_NE(-1, instance->line_protocol_socket);
+}
+
+TEST(FeildStatAPI, FieldStatSetLocalOutput)
+{
+	struct fieldstat_instance *instance = NULL;
+	int ret_set_local_output  = 0;
+	instance = fieldstat_instance_create("test");
+	EXPECT_STREQ("test", instance->name);
+
+	ret_set_local_output = fieldstat_set_local_output(instance,"test.fs","default");
+	EXPECT_EQ(0, ret_set_local_output);
+
+	EXPECT_STREQ("test.fs", instance->local_output_filename);
+	EXPECT_STREQ("default", instance->local_output_format);
+	EXPECT_EQ(1, instance->local_output_enable);
+}
+
+int fieldstat_background_thread_disable(struct fieldstat_instance *instance);
+
+TEST(FeildStatAPI, FieldStatBackgroundThreadDisable)
+{
+	struct fieldstat_instance *instance = NULL;
+	int ret_background_thread_disable  = 0;
+	instance = fieldstat_instance_create("test");
+	EXPECT_STREQ("test", instance->name);
+
+	ret_background_thread_disable = fieldstat_background_thread_disable(instance);
+	EXPECT_EQ(0, ret_background_thread_disable);
+	EXPECT_EQ(1, instance->background_thread_disable);
+}
+
+TEST(FeildStatAPI, FieldStatSetOutputInterval)
+{
+	struct fieldstat_instance *instance = NULL;
+	int ret_set_output_interval = 0;
+
+	instance = fieldstat_instance_create("test");
+	EXPECT_STREQ("test", instance->name);
+
+	ret_set_output_interval = fieldstat_set_output_interval(instance, 1);
+	EXPECT_EQ(0, ret_set_output_interval);
+	EXPECT_EQ(1, instance->output_interval_s);
+}
+
+TEST(FeildStatAPI, FieldStatRegister)
+{
+	struct fieldstat_instance * instance = NULL;
+	int metric_id = -1;
+	struct metric_t **metric_in_block = NULL;
+	const char *tag_key[] = {"dest_ip", "src_ip"};
+	const char *tag_value[] = {"127.0.0.1", "127.0.0.1"};
+
+	instance = fieldstat_instance_create("test");
+	EXPECT_STREQ("test", instance->name);
+
+	metric_id = fieldstat_register(instance, FIELD_TYPE_COUNTER, "Traffic_bytes",NULL,NULL,0);
+	EXPECT_EQ(0, metric_id);
+	EXPECT_EQ(1, instance->metric_cnt);
+	EXPECT_NE(nullptr, instance->metric_block_list[0]);
+	metric_in_block = instance->metric_block_list[0];
+	EXPECT_NE(nullptr, metric_in_block[0]);
+
+	metric_id = fieldstat_register(instance, FIELD_TYPE_GAUGE, "Active_sessions", NULL, NULL, 0);
+	EXPECT_EQ(1, metric_id);
+	EXPECT_EQ(2, instance->metric_cnt);
+	EXPECT_NE(nullptr, metric_in_block[1]);
+
+	metric_id = fieldstat_register(instance, FIELD_TYPE_COUNTER, "New_sessions",tag_key, tag_value, sizeof(tag_key)/sizeof(tag_key[0]));
+	EXPECT_EQ(2, metric_id);
+	EXPECT_EQ(3, instance->metric_cnt);
+	EXPECT_NE(nullptr, metric_in_block[2]);
+
+}
+
+
+
+
+
 int main(int argc, char *argv[]) 
 {
 	testing::InitGoogleTest(&argc, argv);