diff options
Diffstat (limited to '')
-rw-r--r-- | fluent-bit/tests/internal/unit_sizes.c | 59 |
1 files changed, 59 insertions, 0 deletions
diff --git a/fluent-bit/tests/internal/unit_sizes.c b/fluent-bit/tests/internal/unit_sizes.c new file mode 100644 index 00000000..d009bafb --- /dev/null +++ b/fluent-bit/tests/internal/unit_sizes.c @@ -0,0 +1,59 @@ +/* -*- Mode: C; tab-width: 4; indent-tabs-mode: nil; c-basic-offset: 4 -*- */ + +#include <fluent-bit/flb_info.h> +#include <fluent-bit/flb_utils.h> + +#include "flb_tests_internal.h" + +#define to_bytes flb_utils_size_to_bytes + +void test_unit_sizes() +{ + int64_t KB = 1000; + int64_t MB = 1000 * KB; + int64_t GB = 1000 * MB; + + /* Bytes, no prefixes */ + TEST_CHECK(to_bytes("1") == 1); + TEST_CHECK(to_bytes("50") == 50); + TEST_CHECK(to_bytes("1000") == 1*KB); + + /* Decimal prefix: KB */ + TEST_CHECK(to_bytes("1KB") == KB); + TEST_CHECK(to_bytes("1K") == KB); + TEST_CHECK(to_bytes("1kB") == KB); + TEST_CHECK(to_bytes("1kb") == KB); + TEST_CHECK(to_bytes("1k") == KB); + + /* Decimal prefix: MB */ + TEST_CHECK(to_bytes("1MB") == MB); + TEST_CHECK(to_bytes("1M") == MB); + TEST_CHECK(to_bytes("1mB") == MB); + TEST_CHECK(to_bytes("1mb") == MB); + TEST_CHECK(to_bytes("1m") == MB); + TEST_CHECK(to_bytes("5m") == 5*MB); + + /* Decimal prefix: GB */ + TEST_CHECK(to_bytes("1GB") == GB); + TEST_CHECK(to_bytes("1G") == GB); + TEST_CHECK(to_bytes("1gB") == GB); + TEST_CHECK(to_bytes("1gb") == GB); + TEST_CHECK(to_bytes("1g") == GB); + TEST_CHECK(to_bytes("5g") == 5*GB); + TEST_CHECK(to_bytes("32g") == 32*GB); + + /* Invalid values */ + TEST_CHECK(to_bytes("aabb") == -1); + TEST_CHECK(to_bytes("") == -1); + + /* Invlid prefixes */ + TEST_CHECK(to_bytes("1kX") == -1); + TEST_CHECK(to_bytes("1kX") == -1); + TEST_CHECK(to_bytes("1MX") == -1); + TEST_CHECK(to_bytes("1GX") == -1); +} + +TEST_LIST = { + { "unit_sizes", test_unit_sizes }, + { 0 } +}; |