Adding upstream version 1.44.3.upstream/1.44.3upstream

Signed-off-by: Daniel Baumann <daniel.baumann@progress-linux.org>

1 files changed, 172 insertions, 0 deletions

diff --git a/fluent-bit/lib/cmetrics/tests/summary.c b/fluent-bit/lib/cmetrics/tests/summary.c
new file mode 100644
index 00000000..99ffe579
--- /dev/null
+++ b/fluent-bit/lib/cmetrics/tests/summary.c
@@ -0,0 +1,172 @@
+/* -*- 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_summary.h>
+#include <cmetrics/cmt_encode_prometheus.h>
+#include <cmetrics/cmt_encode_text.h>
+
+#include "cmt_tests.h"
+
+static void prometheus_encode_test(struct cmt *cmt)
+{
+    cfl_sds_t buf;
+
+    buf = cmt_encode_prometheus_create(cmt, CMT_FALSE);
+    printf("\n%s\n", buf);
+    cmt_encode_prometheus_destroy(buf);
+
+    /* encode to all possible formats */
+    cmt_test_encode_all(cmt);
+}
+
+void test_set_defaults()
+{
+    double sum;
+    uint64_t count;
+    uint64_t ts;
+    double q[6];
+    double r[6];
+    struct cmt *cmt;
+    struct cmt_summary *s;
+
+    cmt_initialize();
+
+    /* Timestamp */
+    ts = cfl_time_now();
+
+    /* CMetrics context */
+    cmt = cmt_create();
+    TEST_CHECK(cmt != NULL);
+
+    /* set quantiles, no labels */
+    q[0] = 0.1;
+    q[1] = 0.2;
+    q[2] = 0.3;
+    q[3] = 0.4;
+    q[4] = 0.5;
+    q[5] = 1.0;
+
+    r[0] = 1;
+    r[1] = 2;
+    r[2] = 3;
+    r[3] = 4;
+    r[4] = 5;
+    r[5] = 6;
+
+    /* Create a gauge metric type */
+    s = cmt_summary_create(cmt,
+                           "k8s", "network", "load", "Network load",
+                           6, q,
+                           1, (char *[]) {"my_label"});
+    TEST_CHECK(s != NULL);
+
+    count = 10;
+    sum   = 51.612894511314444;
+
+    /* no quantiles, no labels */
+    cmt_summary_set_default(s, ts, NULL, sum, count, 0, NULL);
+    prometheus_encode_test(cmt);
+
+    cmt_summary_set_default(s, ts, r, sum, count, 0, NULL);
+    prometheus_encode_test(cmt);
+
+    /* static label: register static label for the context */
+    cmt_label_add(cmt, "static", "test");
+    prometheus_encode_test(cmt);
+
+    cmt_destroy(cmt);
+}
+
+/* ref: https://github.com/fluent/fluent-bit/issues/5894 */
+void fluentbit_bug_5894()
+{
+    double sum;
+    uint64_t count;
+    uint64_t ts;
+    double q[6];
+    double r[6];
+    struct cmt *cmt;
+    struct cmt_summary *s;
+
+    cmt_initialize();
+
+    /* Timestamp */
+    ts = cfl_time_now();
+
+    /* CMetrics context */
+    cmt = cmt_create();
+    TEST_CHECK(cmt != NULL);
+
+    /* set quantiles, no labels */
+    q[0] = 0.1;
+    q[1] = 0.2;
+    q[2] = 0.3;
+    q[3] = 0.4;
+    q[4] = 0.5;
+    q[5] = 1.0;
+
+    r[0] = 1;
+    r[1] = 2;
+    r[2] = 3;
+    r[3] = 4;
+    r[4] = 5;
+    r[5] = 6;
+
+    /* Create a gauge metric type */
+    s = cmt_summary_create(cmt,
+                           "spring", "kafka_listener", "seconds", "Kafka Listener Timer",
+                           6, q,
+                           3, (char *[]) {"exception", "name", "result"});
+    TEST_CHECK(s != NULL);
+
+    /* no quantiles, labels */
+    sum = 0.0;
+    count = 1;
+
+    cmt_summary_set_default(s, ts, NULL, sum, count,
+                            3, (char *[]) {"ListenerExecutionFailedException",
+                                           "org.springframework.kafka.KafkaListenerEndpointContainer#0-0",
+                                           "failure"});
+
+    /* no quantiles, labels */
+    sum = 0.1;
+    count = 2;
+    cmt_summary_set_default(s, ts, NULL, sum, count,
+                            3, (char *[]) {"none",
+                                          "org.springframework.kafka.KafkaListenerEndpointContainer#0-0",
+                                          "success"});
+
+    /* quantiles, labels */
+    sum = 0.2;
+    count = 3;
+    cmt_summary_set_default(s, ts, r, sum, count,
+                            3, (char *[]) {"extra test",
+                                           "org.springframework.kafka.KafkaListenerEndpointContainer#0-0",
+                                           "success"});
+
+    prometheus_encode_test(cmt);
+    cmt_destroy(cmt);
+}
+
+TEST_LIST = {
+    {"set_defaults"      , test_set_defaults},
+    {"fluentbit_bug_5894", fluentbit_bug_5894},
+    { 0 }
+};