[Test] Add large TCP request test

1 files changed, 44 insertions, 0 deletions

diff --git a/test/functional/lua/tcp.lua b/test/functional/lua/tcp.lua
index a95497086..925a18913 100644
--- a/test/functional/lua/tcp.lua
+++ b/test/functional/lua/tcp.lua
@@ -59,6 +59,44 @@ local function http_simple_tcp_ssl_symbol(task)
   })
 end
 
+local function http_large_tcp_ssl_symbol(task)
+  local data = {}
+
+  local function ssl_get_cb(err, rep, conn)
+    logger.errx(task, 'ssl_get_cb: got reply: %s, error: %s, conn: %s', rep, err, conn)
+    task:insert_result('TCP_SSL_LARGE_2', 1.0)
+  end
+  local function ssl_read_post_cb(err, conn)
+    logger.errx(task, 'ssl_large_read_post_cb: write done: error: %s, conn: %s', err, conn)
+    conn:add_read(ssl_get_cb)
+  end
+  local function ssl_read_cb(err, rep, conn)
+    logger.errx(task, 'ssl_large_read_cb: got reply: %s, error: %s, conn: %s', rep, err, conn)
+    conn:add_write(ssl_read_post_cb, data)
+    task:insert_result('TCP_SSL_LARGE', 1.0)
+  end
+
+  if task:get_queue_id() == 'SSL Large TCP request' then
+    logger.errx(task, 'ssl_large_tcp_symbol: begin')
+    for i = 1,10000000 do
+      data[i] = 'test\n'
+    end
+
+    rspamd_tcp:request({
+      task = task,
+      callback = ssl_read_cb,
+      host = '127.0.0.1',
+      data = data,
+      read = true,
+      ssl = true,
+      ssl_noverify = true,
+      port = 14433,
+    })
+  else
+    logger.errx(task, 'ssl_large_tcp_symbol: skip')
+  end
+end
+
 local function http_simple_tcp_symbol(task)
   logger.errx(task, 'connect_sync, before')
 
@@ -212,6 +250,12 @@ rspamd_config:register_symbol({
   no_squeeze = true
 })
 rspamd_config:register_symbol({
+  name = 'LARGE_TCP_ASYNC_SSL_TEST',
+  score = 1.0,
+  callback = http_large_tcp_ssl_symbol,
+  no_squeeze = true
+})
+rspamd_config:register_symbol({
   name = 'SIMPLE_TCP_TEST',
   score = 1.0,
   callback = http_simple_tcp_symbol,