summaryrefslogtreecommitdiffstats
path: root/src/fluent-bit/tests/internal/gelf.c
blob: 0a3b21e68d0b02c586fe7a8f86d96cf6b29c41b1 (plain)
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
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
/* -*- Mode: C; tab-width: 4; indent-tabs-mode: nil; c-basic-offset: 4 -*- */

#include <fluent-bit/flb_info.h>
#include <fluent-bit/flb_pack.h>
#include <fluent-bit/flb_sds.h>
#include <fluent-bit/flb_time.h>
#include <msgpack.h>

#include "flb_tests_internal.h"
#include <string.h>

#define EXPECTED_OUT \
    "{\"version\":\"1.1\", \"short_message\":\"true, 2019, str\", \"_t2\":\"false\", \"timestamp\":337647600.1234}"

void test_gelf_pack()
{
    msgpack_sbuffer mp_sbuf;
    msgpack_packer mp_pck;
    struct flb_time ts;
    struct flb_gelf_fields fields = {0};
    flb_sds_t out;

    /* Pack sample msgpack */
    msgpack_sbuffer_init(&mp_sbuf);
    msgpack_packer_init(&mp_pck, &mp_sbuf, msgpack_sbuffer_write);


    ts.tm.tv_sec = 337647600;
    ts.tm.tv_nsec = 1234111111;

    msgpack_pack_map(&mp_pck, 2);
    msgpack_pack_str(&mp_pck, 2);
    msgpack_pack_str_body(&mp_pck, "t1", 2);
    msgpack_pack_array(&mp_pck, 3);
    msgpack_pack_true(&mp_pck);
    msgpack_pack_uint64(&mp_pck, 2019);
    msgpack_pack_str(&mp_pck, 3);
    msgpack_pack_str_body(&mp_pck, "str", 3);
    msgpack_pack_str(&mp_pck, 2);
    msgpack_pack_str_body(&mp_pck, "t2", 2);
    msgpack_pack_false(&mp_pck);

    fields.short_message_key = flb_sds_create("t1");
    out = flb_msgpack_raw_to_gelf(mp_sbuf.data, mp_sbuf.size, &ts, &fields);
    TEST_CHECK(out != NULL);

    TEST_CHECK(strcmp(out, EXPECTED_OUT) == 0);
    flb_sds_destroy(out);
    flb_sds_destroy(fields.short_message_key);
    msgpack_sbuffer_destroy(&mp_sbuf);
}

#define EXPECTED_OUT_MSEC \
    "{\"version\":\"1.1\", \"short_message\":\"true, 2019, str\", \"_t2\":\"false\", \"timestamp\":337647600.012}"

/* https://github.com/fluent/fluent-bit/issues/3727 */
void test_gelf_pack_msec()
{
    msgpack_sbuffer mp_sbuf;
    msgpack_packer mp_pck;
    struct flb_time ts;
    struct flb_gelf_fields fields = {0};
    flb_sds_t out;

    /* Pack sample msgpack */
    msgpack_sbuffer_init(&mp_sbuf);
    msgpack_packer_init(&mp_pck, &mp_sbuf, msgpack_sbuffer_write);


    ts.tm.tv_sec =  337647600;
    ts.tm.tv_nsec =  12341111; /* 12.34msec */

    msgpack_pack_map(&mp_pck, 2);
    msgpack_pack_str(&mp_pck, 2);
    msgpack_pack_str_body(&mp_pck, "t1", 2);
    msgpack_pack_array(&mp_pck, 3);
    msgpack_pack_true(&mp_pck);
    msgpack_pack_uint64(&mp_pck, 2019);
    msgpack_pack_str(&mp_pck, 3);
    msgpack_pack_str_body(&mp_pck, "str", 3);
    msgpack_pack_str(&mp_pck, 2);
    msgpack_pack_str_body(&mp_pck, "t2", 2);
    msgpack_pack_false(&mp_pck);

    fields.short_message_key = flb_sds_create("t1");
    out = flb_msgpack_raw_to_gelf(mp_sbuf.data, mp_sbuf.size, &ts, &fields);
    TEST_CHECK(out != NULL);

    if(!TEST_CHECK(strcmp(out, EXPECTED_OUT_MSEC) == 0)) {
        TEST_MSG("out=%s", out);
    }
    flb_sds_destroy(out);
    flb_sds_destroy(fields.short_message_key);
    msgpack_sbuffer_destroy(&mp_sbuf);
}

TEST_LIST = {
    {"gelf_pack",      test_gelf_pack},
    {"gelf_pack_msec", test_gelf_pack_msec},
    { 0 }
};