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 }
+};