1 files changed, 154 insertions, 0 deletions
diff --git a/src/fluent-bit/tests/internal/log.c b/src/fluent-bit/tests/internal/log.c
new file mode 100644
index 000000000..662df98ec
--- /dev/null
+++ b/src/fluent-bit/tests/internal/log.c
@@ -0,0 +1,154 @@
+#include <fluent-bit/flb_sds.h>
+#include <fluent-bit/flb_mem.h>
+#include <fluent-bit/flb_log.h>
+
+#include "flb_tests_internal.h"
+
+#define TIMEOUT             5
+#define TEST_RECORD_01      "this is a test message"
+#define TEST_RECORD_01_SIZE sizeof(TEST_RECORD_01) - 1
+
+#define TEST_RECORD_02      "other type of message"
+#define TEST_RECORD_02_SIZE sizeof(TEST_RECORD_02) - 1
+
+static int check_interval(int timeout, int *interval)
+{
+    if (!TEST_CHECK( (*interval >= timeout - 1) && *interval <= timeout) ) {
+        TEST_MSG("interval error. got=%d expect=%d-%d", *interval, timeout -1 ,timeout);
+        return -1;
+    }
+
+    return 0;
+}
+
+static int update_and_check_interval(int timeout, int ret, int *interval)
+{
+    int ret_val;
+
+    if (ret == FLB_TRUE) {
+        *interval += 1;
+        return 0;
+    }
+
+    /* false means timeout. check interval. */
+    ret_val = check_interval(timeout, interval);
+    *interval = 0; /* reset interval */
+
+    return ret_val;
+}
+
+static void cache_basic_timeout()
+{
+    int i;
+    int ret;
+    int ret_1;
+    int ret_2;
+    int timeout = 5;
+    int interval1 = 0;
+    int interval2 = 0;
+    struct flb_log_cache *cache;
+    struct flb_log_cache_entry *entry;
+
+    printf("\n");
+
+    cache = flb_log_cache_create(10, 0);
+    TEST_CHECK(cache == NULL);
+
+    cache = flb_log_cache_create(5, 4);
+    TEST_CHECK(cache != NULL);
+
+    /* cache must be empty */
+    entry = flb_log_cache_exists(cache, TEST_RECORD_01, TEST_RECORD_01_SIZE);
+    TEST_CHECK(entry == NULL);
+
+    /* upon trying to check for a suppress and if not found, it must be added */
+    ret_1 = flb_log_cache_check_suppress(cache, TEST_RECORD_01, TEST_RECORD_01_SIZE);
+    TEST_CHECK(ret_1 == FLB_FALSE);
+
+    /* double check that it was added */
+    entry = flb_log_cache_exists(cache, TEST_RECORD_01, TEST_RECORD_01_SIZE);
+    TEST_CHECK(entry != NULL);
+
+    printf("------------------------\n");
+
+    /* reset */
+    flb_log_cache_destroy(cache);
+
+    /* create a new cache */
+    cache = flb_log_cache_create(timeout, 4);
+    TEST_CHECK(cache != NULL);
+
+    ret_1 = flb_log_cache_check_suppress(cache, TEST_RECORD_01, TEST_RECORD_01_SIZE);
+    ret_2 = flb_log_cache_check_suppress(cache, TEST_RECORD_02, TEST_RECORD_02_SIZE);
+    TEST_CHECK(ret_1 == FLB_FALSE);
+    TEST_CHECK(ret_2 == FLB_FALSE);
+    sleep(1);
+    interval1++;
+    interval2++;
+
+    for (i = 1; i < 10; i++) {
+        ret_1 = flb_log_cache_check_suppress(cache, TEST_RECORD_01, TEST_RECORD_01_SIZE);
+        ret = update_and_check_interval(timeout, ret_1, &interval1);
+        if (!TEST_CHECK(ret == 0)) {
+            TEST_MSG("update_and_check_interval for TEST_RECORD_01 failed. i=%d", i);
+        }
+
+        ret_2 = flb_log_cache_check_suppress(cache, TEST_RECORD_02, TEST_RECORD_02_SIZE);
+        ret = update_and_check_interval(timeout, ret_2, &interval2);
+        if (!TEST_CHECK(ret == 0)) {
+            TEST_MSG("update_and_check_interval for TEST_RECORD_02 failed. i=%d", i);
+        }
+
+        sleep(1);
+    }
+    ret_1 = flb_log_cache_check_suppress(cache, TEST_RECORD_01, TEST_RECORD_01_SIZE);
+    ret = update_and_check_interval(timeout, ret_1, &interval1);
+    TEST_CHECK(ret == 0);
+
+    ret_2 = flb_log_cache_check_suppress(cache, TEST_RECORD_02, TEST_RECORD_02_SIZE);
+    ret = update_and_check_interval(timeout, ret_2, &interval2);
+    TEST_CHECK(ret == 0);
+
+    flb_log_cache_destroy(cache);
+}
+
+static void cache_one_slot()
+{
+    int i;
+    int ret_1;
+    int ret_2;
+    struct flb_log_cache *cache;
+
+    printf("\n");
+
+    cache = flb_log_cache_create(2, 1);
+    TEST_CHECK(cache != NULL);
+
+    for (i = 0; i < 10; i++) {
+
+        if (i == 0) {
+            ret_1 = flb_log_cache_check_suppress(cache, TEST_RECORD_01, TEST_RECORD_01_SIZE);
+            TEST_CHECK(ret_1 == FLB_FALSE);
+
+            ret_1 = flb_log_cache_check_suppress(cache, TEST_RECORD_01, TEST_RECORD_01_SIZE);
+            TEST_CHECK(ret_1 == FLB_TRUE);
+        }
+        else {
+            ret_2 = flb_log_cache_check_suppress(cache, TEST_RECORD_02, TEST_RECORD_02_SIZE);
+            ret_1 = flb_log_cache_check_suppress(cache, TEST_RECORD_01, TEST_RECORD_01_SIZE);
+
+            TEST_CHECK(ret_1 == FLB_FALSE);
+            TEST_CHECK(ret_2 == FLB_FALSE);
+        }
+
+        sleep(1);
+    }
+
+    flb_log_cache_destroy(cache);
+}
+
+TEST_LIST = {
+    { "cache_basic_timeout" , cache_basic_timeout },
+    { "cache_one_slot"      , cache_one_slot      },
+    { 0 }
+};