1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
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 }
};
|
