1 files changed, 82 insertions, 0 deletions
diff --git a/fluent-bit/tests/internal/error_reporter.c b/fluent-bit/tests/internal/error_reporter.c
new file mode 100644
index 00000000..9ab84475
--- /dev/null
+++ b/fluent-bit/tests/internal/error_reporter.c
@@ -0,0 +1,82 @@
+/* -*- Mode: C; tab-width: 4; indent-tabs-mode: nil; c-basic-offset: 4 -*- */
+#include <stdio.h>
+#include <stdlib.h>
+#include <unistd.h>
+#include <time.h>
+#include <monkey/mk_core/mk_list.h>
+
+#include <fluent-bit/flb_mem.h>
+#include <fluent-bit/aws/flb_aws_error_reporter.h>
+
+#include "flb_tests_internal.h"
+
+const char* file_path = "/tmp/error.log";
+const char* error_message_1 = "[engine] scheduler could not start";
+const char* error_message_2 = "[engine] scheduler could not stop";
+
+
+void test_flb_aws_error_reporter_create() {
+
+    setenv(STATUS_MESSAGE_FILE_PATH_ENV, file_path, 1);
+    struct flb_aws_error_reporter *error_reporter = flb_aws_error_reporter_create();
+    TEST_CHECK((void*) error_reporter != NULL);
+    TEST_CHECK((void*)error_reporter->file_path != NULL);
+    TEST_CHECK(flb_sds_cmp(error_reporter->file_path, file_path, strlen(file_path)) == 0);
+    TEST_CHECK(error_reporter->ttl == STATUS_MESSAGE_TTL_DEFAULT);
+    TEST_CHECK(error_reporter->max_size == STATUS_MESSAGE_MAX_BYTE_LENGTH_DEFAULT);
+    flb_aws_error_reporter_destroy(error_reporter);
+}
+
+void test_flb_aws_error_reporter_write() {
+
+    int size;
+    char error_message_3[1041];
+    for (int i = 0; i < 1040; i++) {
+        error_message_3[i] = 'a';
+    }
+    error_message_3[1040] = '\0';
+
+    setenv(STATUS_MESSAGE_FILE_PATH_ENV, file_path, 1);
+
+    struct flb_aws_error_reporter *error_reporter = flb_aws_error_reporter_create();
+    flb_aws_error_reporter_write(error_reporter, error_message_1);
+    size = mk_list_size(&error_reporter->messages);
+    TEST_CHECK(size == 1);
+    /* message same with the latest one will be combined*/
+    flb_aws_error_reporter_write(error_reporter, error_message_1);
+    size = mk_list_size(&error_reporter->messages);
+    TEST_CHECK(size == 1);
+
+    /* message different of the latest one will be combined*/
+    flb_aws_error_reporter_write(error_reporter, error_message_2);
+    size = mk_list_size(&error_reporter->messages);
+    TEST_CHECK(size == 2);
+
+    /* message larger than max size limit*/
+    flb_aws_error_reporter_write(error_reporter, error_message_3);
+    size = mk_list_size(&error_reporter->messages);
+    TEST_CHECK(size == 1);
+
+    flb_aws_error_reporter_destroy(error_reporter);
+}
+
+void test_flb_aws_error_reporter_clean() {
+
+    setenv(STATUS_MESSAGE_FILE_PATH_ENV, file_path, 1);
+    struct flb_aws_error_reporter *error_reporter = flb_aws_error_reporter_create();
+    flb_aws_error_reporter_write(error_reporter, error_message_1);
+    time_t start = time(NULL);
+    while (time(NULL) - start <= error_reporter->ttl) {
+        flb_aws_error_reporter_clean(error_reporter);
+    }
+    TEST_CHECK(mk_list_size(&error_reporter->messages) == 0);
+
+    flb_aws_error_reporter_destroy(error_reporter);
+}
+
+TEST_LIST = {
+    { "test_flb_aws_error_reporter_create", test_flb_aws_error_reporter_create},
+    {"test_flb_aws_error_reporter_write", test_flb_aws_error_reporter_write},
+    {"test_flb_aws_error_reporter_clean", test_flb_aws_error_reporter_clean},
+    { 0 }
+};
+\ No newline at end of file