diff options
Diffstat (limited to 'fluent-bit/lib/cmetrics/tests/gauge.c')
| -rw-r--r-- | fluent-bit/lib/cmetrics/tests/gauge.c | 174 |
1 files changed, 174 insertions, 0 deletions
diff --git a/fluent-bit/lib/cmetrics/tests/gauge.c b/fluent-bit/lib/cmetrics/tests/gauge.c new file mode 100644 index 00000000..3c219601 --- /dev/null +++ b/fluent-bit/lib/cmetrics/tests/gauge.c @@ -0,0 +1,174 @@ +/* -*- Mode: C; tab-width: 4; indent-tabs-mode: nil; c-basic-offset: 4 -*- */ + +/* CMetrics + * ======== + * Copyright 2021-2022 The CMetrics Authors + * + * Licensed under the Apache License, Version 2.0 (the "License"); + * you may not use this file except in compliance with the License. + * You may obtain a copy of the License at + * + * http://www.apache.org/licenses/LICENSE-2.0 + * + * Unless required by applicable law or agreed to in writing, software + * distributed under the License is distributed on an "AS IS" BASIS, + * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. + * See the License for the specific language governing permissions and + * limitations under the License. + */ + +#include <cmetrics/cmetrics.h> +#include <cmetrics/cmt_gauge.h> +#include <cmetrics/cmt_encode_prometheus.h> + +#include "cmt_tests.h" + +void test_gauge() +{ + int ret; + double val; + uint64_t ts; + struct cmt *cmt; + struct cmt_gauge *g; + + cmt_initialize(); + + cmt = cmt_create(); + TEST_CHECK(cmt != NULL); + + /* Create a gauge metric type */ + g = cmt_gauge_create(cmt, "kubernetes", "network", "load", "Network load", 0, NULL); + TEST_CHECK(g != NULL); + + /* Timestamp */ + ts = cfl_time_now(); + + /* Default value */ + ret = cmt_gauge_get_val(g, 0, NULL, &val); + TEST_CHECK(val == 0.0); + + /* Set a value of two */ + cmt_gauge_set(g, ts, 2.0, 0, NULL); + ret = cmt_gauge_get_val(g, 0, NULL, &val); + TEST_CHECK(ret == 0); + TEST_CHECK(val == 2.0); + + /* Increment one */ + cmt_gauge_inc(g, ts, 0, NULL); + ret = cmt_gauge_get_val(g, 0, NULL, &val); + TEST_CHECK(ret == 0); + TEST_CHECK(val == 3.0); + + /* Substract 2 */ + ret = cmt_gauge_sub(g, ts, 2, 0, NULL); + TEST_CHECK(ret == 0); + + ret = cmt_gauge_get_val(g, 0, NULL, &val); + TEST_CHECK(ret == 0); + TEST_CHECK(val == 1.0); + + /* Decrement by one */ + ret = cmt_gauge_dec(g, ts, 0, NULL); + TEST_CHECK(ret == 0); + + ret = cmt_gauge_get_val(g, 0, NULL, &val); + TEST_CHECK(ret == 0); + TEST_CHECK(val == 0.0); + + cmt_destroy(cmt); +} + +void test_labels() +{ + int ret; + double val; + uint64_t ts; + cfl_sds_t prom; + struct cmt *cmt; + struct cmt_gauge *g; + + cmt_initialize(); + + cmt = cmt_create(); + TEST_CHECK(cmt != NULL); + + /* Create a counter metric type */ + g = cmt_gauge_create(cmt, "kubernetes", "network", "load", "Network load", + 2, (char *[]) {"hostname", "app"}); + TEST_CHECK(g != NULL); + + /* Timestamp */ + ts = cfl_time_now(); + + /* + * Test 1: hash zero (no labels) + * ----------------------------- + */ + + /* Default value for hash zero */ + ret = cmt_gauge_get_val(g, 0, NULL, &val); + TEST_CHECK(ret == -1); + + /* Increment hash zero by 1 */ + ret = cmt_gauge_inc(g, ts, 0, NULL); + TEST_CHECK(ret == 0); + + /* Check the new value */ + ret = cmt_gauge_get_val(g, 0, NULL, &val); + TEST_CHECK(ret == 0); + TEST_CHECK(val == 1.0); + + /* Add two */ + ret = cmt_gauge_add(g, ts, 2, 0, NULL); + TEST_CHECK(ret == 0); + + /* Check that hash zero val is 3.0 */ + ret = cmt_gauge_get_val(g, 0, NULL, &val); + TEST_CHECK(ret == 0); + TEST_CHECK(val == 3.0); + + /* + * Test 2: custom labels + * --------------------- + */ + + /* Increment custom metric */ + ret = cmt_gauge_inc(g, ts, 2, (char *[]) {"localhost", "cmetrics"}); + TEST_CHECK(ret == 0); + + /* Check ret = 1 */ + ret = cmt_gauge_get_val(g, 2, (char *[]) {"localhost", "cmetrics"}, &val); + TEST_CHECK(ret == 0); + TEST_CHECK(val == 1.000); + + /* Add 10 to another metric using a different second label */ + ret = cmt_gauge_add(g, ts, 10, 2, (char *[]) {"localhost", "test"}); + TEST_CHECK(ret == 0); + + /* Validate the value */ + ret = cmt_gauge_get_val(g, 2, (char *[]) {"localhost", "test"}, &val); + TEST_CHECK(ret == 0); + TEST_CHECK(val == 10.00); + + /* Substract two */ + ret = cmt_gauge_sub(g, ts, 2.5, 2, (char *[]) {"localhost", "test"}); + TEST_CHECK(ret == 0); + + /* Validate the value */ + ret = cmt_gauge_get_val(g, 2, (char *[]) {"localhost", "test"}, &val); + TEST_CHECK(ret == 0); + TEST_CHECK(val == 7.50); + + printf("\n"); + prom = cmt_encode_prometheus_create(cmt, CMT_TRUE); + printf("%s\n", prom); + cmt_encode_prometheus_destroy(prom); + + cmt_destroy(cmt); +} + +TEST_LIST = { + {"basic" , test_gauge}, + {"labels", test_labels}, + { 0 } +}; |