1 files changed, 181 insertions, 0 deletions
diff --git a/fluent-bit/lib/chunkio/tests/memfs.c b/fluent-bit/lib/chunkio/tests/memfs.c
new file mode 100644
index 00000000..0c79f009
--- /dev/null
+++ b/fluent-bit/lib/chunkio/tests/memfs.c
@@ -0,0 +1,181 @@
+/* -*- Mode: C; tab-width: 4; indent-tabs-mode: nil; c-basic-offset: 4 -*- */
+
+/*  Chunk I/O
+ *  =========
+ *  Copyright 2018 Eduardo Silva <eduardo@monkey.io>
+ *
+ *  Licensed under the Apache License, Version 2.0 (the "License");
+ *  you may not use this file except in compliance with the License.
+ *  You may obtain a copy of the License at
+ *
+ *      http://www.apache.org/licenses/LICENSE-2.0
+ *
+ *  Unless required by applicable law or agreed to in writing, software
+ *  distributed under the License is distributed on an "AS IS" BASIS,
+ *  WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ *  See the License for the specific language governing permissions and
+ *  limitations under the License.
+ */
+
+#include <sys/stat.h>
+#include <fcntl.h>
+
+#include <chunkio/chunkio.h>
+#include <chunkio/cio_log.h>
+#include <chunkio/cio_scan.h>
+#include <chunkio/cio_file.h>
+#include <chunkio/cio_memfs.h>
+#include <chunkio/cio_meta.h>
+#include <chunkio/cio_stream.h>
+#include <chunkio/cio_utils.h>
+
+#include "cio_tests_internal.h"
+
+#define CIO_FILE_400KB    CIO_TESTS_DATA_PATH "/data/400kb.txt"
+#define CIO_FILE_400KB_SIZE 409600
+
+/* Logging callback, once called it just turn on the log_check flag */
+static int log_cb(struct cio_ctx *ctx, int level, const char *file, int line,
+                  char *str)
+{
+    (void) ctx;
+
+    printf("[cio-test-fs] %-60s => %s:%i\n",  str, file, line);
+    return 0;
+}
+
+/* Read a file into the buffer at most 'size' bytes. Return bytes read */
+static int read_file(const char *file, char *buf, size_t size)
+{
+    char *p = buf;
+    size_t total = 0;
+    ssize_t nb;
+
+    int fd = open(file, O_RDONLY);
+    if (fd == -1)
+        return -1;
+
+    while (1) {
+        nb = read(fd, p, size);
+        if (nb == 0)
+            break;
+        if (nb < 0) {
+            close(fd);
+            return -1;
+        }
+        p += nb;
+        size -= nb;
+        total += nb;
+    }
+    close(fd);
+    return total;
+}
+
+/* Test API generating files to the file system and then scanning them back */
+static void test_memfs_write()
+{
+    int i;
+    int err;
+    int ret;
+    int n_files = 100;
+    int flags;
+    char *in_data;
+    size_t in_size;
+    char tmp[255];
+    struct cio_ctx *ctx;
+    struct cio_stream *stream;
+    struct cio_chunk *chunk;
+    struct cio_chunk **carr;
+    struct cio_options cio_opts;
+
+    /* Dummy break line for clarity on acutest output */
+    printf("\n");
+
+    flags = CIO_CHECKSUM;
+
+    cio_options_init(&cio_opts);
+    cio_opts.flags = flags;
+
+    cio_opts.log_cb = log_cb;
+    cio_opts.flags = flags;
+
+    /* Create main context */
+    ctx = cio_create(&cio_opts);
+    TEST_CHECK(ctx != NULL);
+
+    /* Try to create a file with an invalid stream */
+    chunk = cio_chunk_open(ctx, NULL, "invalid", 0, 0, &err);
+    TEST_CHECK(chunk == NULL);
+
+    /* Check invalid stream */
+    stream = cio_stream_create(ctx, "", CIO_STORE_MEM);
+    TEST_CHECK(stream == NULL);
+
+    /* Another invalid name */
+    stream = cio_stream_create(ctx, "/", CIO_STORE_MEM);
+    TEST_CHECK(stream == NULL);
+
+    /* Create valid stream */
+    stream = cio_stream_create(ctx, "test-write", CIO_STORE_MEM);
+    TEST_CHECK(stream != NULL);
+
+    /*
+     * Load sample data file and with the same content through multiple write
+     * operations generating other files.
+     */
+    in_size = CIO_FILE_400KB_SIZE;
+    in_data = malloc(in_size);
+    if (!in_data) {
+        perror("calloc");
+        exit(EXIT_FAILURE);
+    }
+
+    ret = read_file(CIO_FILE_400KB, in_data, in_size);
+    if (ret == -1) {
+        cio_destroy(ctx);
+        exit(EXIT_FAILURE);
+    }
+
+    /* Number of test files to create */
+    n_files = 100;
+
+    /* Allocate files array */
+    carr = calloc(1, sizeof(struct cio_chunk) * n_files);
+    if (!carr) {
+        perror("calloc");
+        exit(EXIT_FAILURE);
+    }
+
+
+    for (i = 0; i < n_files; i++) {
+        snprintf(tmp, sizeof(tmp), "api-test-%04i.txt", i);
+        carr[i] = cio_chunk_open(ctx, stream, tmp, CIO_OPEN, 1000000, &err);
+
+        if (carr[i] == NULL) {
+            continue;
+        }
+
+        cio_chunk_write(carr[i], in_data, in_size);
+        cio_chunk_write(carr[i], in_data, in_size);
+
+        /* update metadata */
+        //cio_meta_write(farr[i], tmp, len);
+
+        /* continue appending data to content area */
+        cio_chunk_write(carr[i], in_data, in_size);
+        cio_chunk_write(carr[i], in_data, in_size);
+        cio_chunk_write(carr[i], in_data, in_size);
+    }
+
+    /* Release file data and destroy context */
+    free(carr);
+    free(in_data);
+
+    cio_scan_dump(ctx);
+    cio_destroy(ctx);
+}
+
+TEST_LIST = {
+    {"memfs_write",   test_memfs_write},
+    { 0 }
+};