From ea38c8f1a202682d3bdbd3dcd4a2756d3aa5fefa Mon Sep 17 00:00:00 2001
From: Daijiro Fukuda <fukuda@clear-code.com>
Date: Tue, 11 Apr 2023 00:29:08 +0900
Subject: [PATCH] in_udp: add test of message_length_limit

Signed-off-by: Daijiro Fukuda <fukuda@clear-code.com>
---
 test/plugin/test_in_udp.rb | 28 ++++++++++++++++++++++++++++
 1 file changed, 28 insertions(+)

diff --git a/test/plugin/test_in_udp.rb b/test/plugin/test_in_udp.rb
index 08e7b8a5fe..dcc63e1f8f 100755
--- a/test/plugin/test_in_udp.rb
+++ b/test/plugin/test_in_udp.rb
@@ -265,4 +265,32 @@ def create_udp_socket(host, port)
       end
     end
   end
+
+  test 'message_length_limit' do
+    message_length_limit = 32
+    d = create_driver(base_config + %!
+      format none
+      message_length_limit #{message_length_limit}
+    !)
+    d.run(expect_records: 3) do
+      create_udp_socket('127.0.0.1', @port) do |u|
+        3.times do |i|
+          u.send("#{i}" * 40 + "\n", 0)
+        end
+      end
+    end
+
+    if Fluent.windows?
+      expected_records = []
+    else
+      expected_records = 3.times.collect do |i|
+        "#{i}" * message_length_limit
+      end
+    end
+    actual_records = d.events.collect do |event|
+      event[2]["message"]
+    end
+
+    assert_equal expected_records, actual_records
+  end
 end