tests: internal: timeout: added coroutine hang test case (#4605)

Signed-off-by: Leonardo Alminana <[email protected]>

tests/internal/CMakeLists.txt

@@ -22,6 +22,7 @@ set(UNIT_TESTS_FILES
   flb_time.c
   multiline.c
   typecast.c
+  timeout.c
   )
 
 if (NOT WIN32)

tests/internal/timeout.c

@@ -0,0 +1,99 @@
+/* -*- Mode: C; tab-width: 4; indent-tabs-mode: nil; c-basic-offset: 4 -*- */
+
+/*  Fluent Bit
+ *  ==========
+ *  Copyright (C) 2015-2016 Treasure Data Inc.
+ *
+ *  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 <fluent-bit.h>
+#include <stdlib.h>
+#include <sys/stat.h>
+#include <sys/types.h>
+#include <fcntl.h>
+#include <string.h>
+#include "flb_tests_internal.h"
+
+#define ELAPSED_TIME_LIMIT 9
+
+void flb_test_timeout_coroutine_recovery()
+{
+    int        output_instance_id;
+    int        input_instance_id;
+    time_t     elapsed_time;
+    time_t     start_time;
+    time_t     stop_time;
+    flb_ctx_t *ctx;
+    int64_t    ret;
+
+    ctx = flb_create();
+
+    TEST_CHECK(flb_service_set(ctx, "Flush", "0.5",
+                                    "Grace", "10",
+                                    NULL) == 0);
+
+    ret = flb_service_set(ctx,
+                          "Log_Level", "info",
+                          NULL);
+
+    TEST_CHECK_(ret == 0, "setting service options");
+
+    input_instance_id = flb_input(ctx, (char *) "dummy", NULL);
+    TEST_CHECK(input_instance_id >= 0);
+
+    TEST_CHECK(flb_input_set(ctx, input_instance_id,
+                             "samples", "1",
+                             "rate"   , "1",
+                             NULL) == 0);
+
+    output_instance_id = flb_output(ctx, (char *) "tcp", NULL);
+    TEST_CHECK(output_instance_id >= 0);
+    TEST_CHECK(flb_output_set(ctx, output_instance_id,
+                              "match", "*",
+                              "retry_limit", "no_retries",
+                              "host", "35.243.247.233",
+                              "port", "54321",
+                              "net.keepalive", "off",
+                              "net.connect_timeout", "5s",
+                              NULL) == 0);
+
+    /* Start test */
+    /* Start the engine */
+    ret = flb_start(ctx);
+    TEST_CHECK_(ret == 0, "starting engine");
+
+    sleep(10);
+
+    start_time = time(NULL);
+
+    ret = flb_stop(ctx);
+    TEST_CHECK_(ret == 0, "stopping engine");
+
+    stop_time = time(NULL);
+
+    if (ctx) {
+        flb_destroy(ctx);
+    }
+
+    elapsed_time = stop_time - start_time;
+
+    TEST_CHECK_(ELAPSED_TIME_LIMIT >= elapsed_time,
+                "for hung coroutines");
+}
+
+/* Test list */
+TEST_LIST = {
+    {"timeout_coroutine_recovery", flb_test_timeout_coroutine_recovery},
+    {NULL, NULL}
+};