tests: runtime: Add test cases for confirming escaped backslash

Signed-off-by: Hiroshi Hatake <hiroshi@chronosphere.io>

Author: cosmo0920

tests/runtime/out_cloudwatch.c

@@ -439,6 +439,85 @@ void flb_test_cloudwatch_error_put_retention_policy(void)
     flb_destroy(ctx);
 }
 
+void flb_test_cloudwatch_create_stream_escapes_json(void)
+{
+    struct log_stream stream;
+    flb_sds_t body;
+    char *expected;
+    int ret;
+
+    memset(&stream, 0, sizeof(struct log_stream));
+    stream.group = flb_sds_create("fluent");
+    stream.name = flb_sds_create("systemd-fsck@dev-disk-by\\x2dlabel-BOOT.service");
+    TEST_CHECK(stream.group != NULL);
+    TEST_CHECK(stream.name != NULL);
+
+    if (stream.group && stream.name) {
+        body = flb_cloudwatch_create_log_stream_body(&stream);
+        TEST_CHECK(body != NULL);
+
+        if (body) {
+            expected = "{\"logGroupName\":\"fluent\","
+                       "\"logStreamName\":\"systemd-fsck@dev-disk-by\\\\x2dlabel-BOOT.service\"}";
+            ret = strcmp(body, expected);
+            TEST_CHECK(ret == 0);
+            flb_sds_destroy(body);
+        }
+    }
+
+    flb_sds_destroy(stream.group);
+    flb_sds_destroy(stream.name);
+}
+
+void flb_test_cloudwatch_put_events_escapes_stream_name(void)
+{
+    struct flb_cloudwatch ctx;
+    struct log_stream stream;
+    struct cw_flush buf;
+    char out_buf[512];
+    char *expected_stream_name;
+    char *expected;
+    int offset;
+    int ret;
+
+    memset(&ctx, 0, sizeof(struct flb_cloudwatch));
+    memset(&stream, 0, sizeof(struct log_stream));
+    memset(&buf, 0, sizeof(struct cw_flush));
+
+    stream.group = flb_sds_create("fluent");
+    stream.name = flb_sds_create("systemd-fsck@dev-disk-by\\x2dlabel-BOOT.service");
+    TEST_CHECK(stream.group != NULL);
+    TEST_CHECK(stream.name != NULL);
+
+    if (stream.group && stream.name) {
+        offset = 0;
+        buf.out_buf = out_buf;
+        buf.out_buf_size = sizeof(out_buf);
+        buf.current_stream = &stream;
+
+        ret = flb_cloudwatch_init_put_payload(&ctx, &buf, &stream, &offset);
+        TEST_CHECK(ret == 0);
+
+        if (ret == 0) {
+            expected = "{\"logGroupName\":\"fluent\","
+                       "\"logStreamName\":\"systemd-fsck@dev-disk-by\\\\x2dlabel-BOOT.service\","
+                       "\"logEvents\":[";
+            TEST_CHECK(offset == strlen(expected));
+            TEST_CHECK(strncmp(out_buf, expected, offset) == 0);
+        }
+
+        expected_stream_name = "systemd-fsck@dev-disk-by\\\\x2dlabel-BOOT.service";
+        reset_flush_buf(&ctx, &buf);
+        TEST_CHECK(buf.data_size == PUT_LOG_EVENTS_HEADER_LEN +
+                                    PUT_LOG_EVENTS_FOOTER_LEN +
+                                    strlen("fluent") +
+                                    strlen(expected_stream_name));
+    }
+
+    flb_sds_destroy(stream.group);
+    flb_sds_destroy(stream.name);
+}
+
 /* Helper function to create a large JSON message of specified size */
 static char* create_large_json_message(size_t target_size)
 {
@@ -591,6 +670,8 @@ TEST_LIST = {
     {"put_retention_policy_success", flb_test_cloudwatch_put_retention_policy_success },
     {"already_exists_create_group_put_retention_policy", flb_test_cloudwatch_already_exists_create_group_put_retention_policy },
     {"error_put_retention_policy", flb_test_cloudwatch_error_put_retention_policy },
+    {"create_stream_escapes_json", flb_test_cloudwatch_create_stream_escapes_json },
+    {"put_events_escapes_stream_name", flb_test_cloudwatch_put_events_escapes_stream_name },
     {"event_size_at_limit", flb_test_cloudwatch_event_size_at_limit },
     {"event_size_over_limit", flb_test_cloudwatch_event_size_over_limit },
     {"event_truncation_with_backslash", flb_test_cloudwatch_event_truncation_with_backslash },