From 99b424b181d527a3ce368b199dcf8be0a142d70d Mon Sep 17 00:00:00 2001
From: slzhou <shenglian_zhou@163.com>
Date: Thu, 18 May 2023 16:26:20 +0800
Subject: [PATCH] fix: add test case

---
 tests/script/output.txt                      |  6 ++
 tests/script/sh/l2norm2.c                    | 71 ++++++++++++++++++++
 tests/script/tsim/parser/join_manyblocks.sim |  8 +--
 3 files changed, 81 insertions(+), 4 deletions(-)
 create mode 100644 tests/script/output.txt
 create mode 100644 tests/script/sh/l2norm2.c

diff --git a/tests/script/output.txt b/tests/script/output.txt
new file mode 100644
index 0000000000..0c7c386bbc
--- /dev/null
+++ b/tests/script/output.txt
@@ -0,0 +1,6 @@
+[03/30 08:08:52.140115] ERROR: failed to connect native (null):6030, code: 0x8000000b, reason: Unable to establish connection
+[03/30 08:08:52.140136] ERROR: insert test process failed
+[03/30 08:09:02.860424] ERROR: failed to connect native (null):6030, code: 0x8000000b, reason: Unable to establish connection
+[03/30 08:09:02.860445] ERROR: insert test process failed
+[03/30 08:10:57.082603] ERROR: failed to connect native (null):6030, code: 0x8000000b, reason: Unable to establish connection
+[03/30 08:10:57.082626] ERROR: insert test process failed
diff --git a/tests/script/sh/l2norm2.c b/tests/script/sh/l2norm2.c
new file mode 100644
index 0000000000..0739f3a23b
--- /dev/null
+++ b/tests/script/sh/l2norm2.c
@@ -0,0 +1,71 @@
+#include <string.h>
+#include <stdlib.h>
+#include <stdio.h>
+#include <math.h>
+
+#include "taosudf.h"
+
+DLL_EXPORT int32_t l2norm2_init() {
+  return 0;
+}
+
+DLL_EXPORT int32_t l2norm2_destroy() {
+  return 0;
+}
+
+DLL_EXPORT int32_t l2norm2_start(SUdfInterBuf *buf) {
+  *(int64_t*)(buf->buf) = 0;
+  buf->bufLen = sizeof(double);
+  buf->numOfResult = 1;
+  return 0;
+}
+
+DLL_EXPORT int32_t l2norm2(SUdfDataBlock* block, SUdfInterBuf *interBuf, SUdfInterBuf *newInterBuf) {
+  double sumSquares = *(double*)interBuf->buf;
+  int8_t numNotNull = 0;
+  for (int32_t i = 0; i < block->numOfCols; ++i) {
+    SUdfColumn* col = block->udfCols[i];
+    if (!(col->colMeta.type == TSDB_DATA_TYPE_INT || 
+          col->colMeta.type == TSDB_DATA_TYPE_DOUBLE)) {
+      return TSDB_CODE_UDF_INVALID_INPUT;
+    }
+  }
+  for (int32_t i = 0; i < block->numOfCols; ++i) {
+    for (int32_t j = 0; j < block->numOfRows; ++j) {
+      SUdfColumn* col = block->udfCols[i];
+      if (udfColDataIsNull(col, j)) {
+        continue;
+      }
+      switch (col->colMeta.type) {
+        case TSDB_DATA_TYPE_INT: {
+          char* cell = udfColDataGetData(col, j);
+          int32_t num = *(int32_t*)cell;
+          sumSquares += (double)num * num;
+          break;
+        }
+        case TSDB_DATA_TYPE_DOUBLE: {
+          char* cell = udfColDataGetData(col, j);
+          double num = *(double*)cell;
+          sumSquares += num * num;
+          break;
+        }
+        default: 
+          break;
+      }
+      ++numNotNull;
+    }
+  }
+
+  *(double*)(newInterBuf->buf) = sumSquares;
+  newInterBuf->bufLen = sizeof(double);
+  newInterBuf->numOfResult = 1;
+  return 0;
+}
+
+DLL_EXPORT int32_t l2norm2_finish(SUdfInterBuf* buf, SUdfInterBuf *resultData) {
+  double sumSquares = *(double*)(buf->buf);
+  *(double*)(resultData->buf) = sumSquares;
+  resultData->bufLen = sizeof(double);
+  resultData->numOfResult = 1;
+  return 0;
+}
diff --git a/tests/script/tsim/parser/join_manyblocks.sim b/tests/script/tsim/parser/join_manyblocks.sim
index a40a75f50c..7fd0df21b3 100644
--- a/tests/script/tsim/parser/join_manyblocks.sim
+++ b/tests/script/tsim/parser/join_manyblocks.sim
@@ -6,8 +6,8 @@ sql connect
 $dbPrefix = join_m_db
 $tbPrefix = join_tb
 $mtPrefix = join_mt
-$tbNum = 3
-$rowNum = 2000
+$tbNum = 20
+$rowNum = 200
 $totalNum = $tbNum * $rowNum
 
 print =============== join_manyBlocks.sim
@@ -78,8 +78,8 @@ print ==============> td-3313
 sql select join_mt0.ts,join_mt0.ts,join_mt0.t1 from join_mt0, join_mt1 where join_mt0.ts=join_mt1.ts and join_mt0.t1=join_mt1.t1;
 
 print $row
-if $row != 6000 then
-  print expect 6000, actual: $row
+if $row != 4000 then
+  print expect 4000, actual: $row
   return -1
 endi