parserTests.cpp

/*
 * Copyright (c) 2019 TAOS Data, Inc. <jhtao@taosdata.com>
 *
 * This program is free software: you can use, redistribute, and/or modify
 * it under the terms of the GNU Affero General Public License, version 3
 * or later ("AGPL"), as published by the Free Software Foundation.
 *
 * This program is distributed in the hope that it will be useful, but WITHOUT
 * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
 * FITNESS FOR A PARTICULAR PURPOSE.
 *
 * You should have received a copy of the GNU Affero General Public License
 * along with this program. If not, see <http://www.gnu.org/licenses/>.
 */

#include <gtest/gtest.h>
#include <iostream>
#pragma GCC diagnostic ignored "-Wwrite-strings"

#pragma GCC diagnostic ignored "-Wunused-function"
#pragma GCC diagnostic ignored "-Wunused-variable"
#pragma GCC diagnostic ignored "-Wsign-compare"
#include "os.h"

#include "astGenerator.h"
#include "parserInt.h"
#include "taos.h"
#include "tdef.h"
#include "tvariant.h"

namespace {
void setSchema(SSchema* p, int32_t type, int32_t bytes, const char* name, int32_t colId) {
  p->colId = colId;
  p->bytes = bytes;
  p->type = type;
  strcpy(p->name, name);
}

void setTableMetaInfo(SQueryStmtInfo* pQueryInfo, SMetaReq *req) {
  pQueryInfo->numOfTables = 1;

  pQueryInfo->pTableMetaInfo = (STableMetaInfo**)calloc(1, POINTER_BYTES);
  STableMetaInfo* pTableMetaInfo = (STableMetaInfo*)calloc(1, sizeof(STableMetaInfo));
  pQueryInfo->pTableMetaInfo[0] = pTableMetaInfo;

  SName* name = (SName*)taosArrayGet(req->pTableName, 0);

  memcpy(&pTableMetaInfo->name, taosArrayGet(req->pTableName, 0), sizeof(SName));
  pTableMetaInfo->pTableMeta = (STableMeta*)calloc(1, sizeof(STableMeta) + 4 * sizeof(SSchema));
  strcpy(pTableMetaInfo->aliasName, name->tname);
  STableMeta* pTableMeta = pTableMetaInfo->pTableMeta;
  pTableMeta->tableType = TSDB_NORMAL_TABLE;
  pTableMeta->tableInfo.numOfColumns = 6;
  pTableMeta->tableInfo.rowSize = 28;
  pTableMeta->uid = 110;

  pTableMetaInfo->tagColList = (SArray*) taosArrayInit(4, POINTER_BYTES);

  SSchema* pSchema = pTableMetaInfo->pTableMeta->schema;
  setSchema(&pSchema[0], TSDB_DATA_TYPE_TIMESTAMP, 8, "ts", 0);
  setSchema(&pSchema[1], TSDB_DATA_TYPE_INT, 4, "a", 1);
  setSchema(&pSchema[2], TSDB_DATA_TYPE_DOUBLE, 8, "b", 2);
  setSchema(&pSchema[3], TSDB_DATA_TYPE_DOUBLE, 8, "col", 3);
  setSchema(&pSchema[4], TSDB_DATA_TYPE_BINARY, 12, "c", 4);
  setSchema(&pSchema[5], TSDB_DATA_TYPE_BINARY, 44, "d", 5);
}

void sqlCheck(const char* sql, bool valid) {
  SSqlInfo info1 = doGenerateAST(sql);
  ASSERT_EQ(info1.valid, true);

  char    msg[128] = {0};
  SMsgBuf buf;
  buf.len = 128;
  buf.buf = msg;

  SSqlNode* pNode = (SSqlNode*) taosArrayGetP(((SArray*)info1.list), 0);
  int32_t code = evaluateSqlNode(pNode, TSDB_TIME_PRECISION_NANO, &buf);
  ASSERT_EQ(code, 0);

  SMetaReq req = {0};
  int32_t  ret = qParserExtractRequestedMetaInfo(&info1, &req, msg, 128);
  ASSERT_EQ(ret, 0);
  ASSERT_EQ(taosArrayGetSize(req.pTableName), 1);

  SQueryStmtInfo* pQueryInfo = createQueryInfo();
  setTableMetaInfo(pQueryInfo, &req);

  SSqlNode* pSqlNode = (SSqlNode*)taosArrayGetP(info1.list, 0);
  ret = validateSqlNode(pSqlNode, pQueryInfo, &buf);

  if (valid) {
    ASSERT_EQ(ret, 0);
  } else {
    ASSERT_NE(ret, 0);
  }

  destroyQueryInfo(pQueryInfo);
  qParserClearupMetaRequestInfo(&req);
  destroySqlInfo(&info1);
}
}

//TEST(testCase, validateAST_test) {
//  SSqlInfo info1 = doGenerateAST("select a a1111, a+b + 22, tbname from `t.1abc` where ts<now+2h and `col` < 20 + 99");
//  ASSERT_EQ(info1.valid, true);
//
//  char    msg[128] = {0};
//  SMsgBuf buf;
//  buf.len = 128;
//  buf.buf = msg;
//
//  SSqlNode* pNode = (SSqlNode*) taosArrayGetP(((SArray*)info1.list), 0);
//  int32_t code = evaluateSqlNode(pNode, TSDB_TIME_PRECISION_NANO, &buf);
//  ASSERT_EQ(code, 0);
//
//  SMetaReq req = {0};
//  int32_t  ret = qParserExtractRequestedMetaInfo(&info1, &req, msg, 128);
//  ASSERT_EQ(ret, 0);
//  ASSERT_EQ(taosArrayGetSize(req.pTableName), 1);
//
//  SQueryStmtInfo* pQueryInfo = createQueryInfo();
//  setTableMetaInfo(pQueryInfo, &req);
//
//  SSqlNode* pSqlNode = (SSqlNode*)taosArrayGetP(info1.list, 0);
//  ret = validateSqlNode(pSqlNode, pQueryInfo, &buf);
//
//  SArray* pExprList = pQueryInfo->exprList;
//  ASSERT_EQ(taosArrayGetSize(pExprList), 3);
//
//  SExprInfo* p1 = (SExprInfo*) taosArrayGetP(pExprList, 0);
//  ASSERT_EQ(p1->base.pColumns->info.colId, 110);
//  ASSERT_EQ(p1->base.numOfParams, 0);
//  ASSERT_EQ(p1->base.resSchema.type, TSDB_DATA_TYPE_INT);
//  ASSERT_STRCASEEQ(p1->base.resSchema.name, "a1111");
//  ASSERT_STRCASEEQ(p1->base.pColumns->name, "t.1abc.a");
//  ASSERT_EQ(p1->base.pColumns->info.colId, 1);
//  ASSERT_EQ(p1->base.pColumns->flag, TSDB_COL_NORMAL);
//  ASSERT_STRCASEEQ(p1->base.token, "a");
//
//  ASSERT_EQ(taosArrayGetSize(pExprList), 3);
//
//  SExprInfo* p2 = (SExprInfo*) taosArrayGetP(pExprList, 1);
//  ASSERT_EQ(p2->base.pColumns->uid, 0);
//  ASSERT_EQ(p2->base.numOfParams, 1);  // it is the serialized binary string of expression.
//  ASSERT_EQ(p2->base.resSchema.type, TSDB_DATA_TYPE_DOUBLE);
//  ASSERT_STRCASEEQ(p2->base.resSchema.name, "a+b + 22");
//
////  ASSERT_STRCASEEQ(p2->base.colInfo.name, "t.1abc.a");
////  ASSERT_EQ(p1->base.colInfo.colId, 1);
////  ASSERT_EQ(p1->base.colInfo.flag, TSDB_COL_NORMAL);
//  ASSERT_STRCASEEQ(p2->base.token, "a+b + 22");
//
//  ASSERT_EQ(taosArrayGetSize(pQueryInfo->colList), 3);
//  ASSERT_EQ(pQueryInfo->fieldsInfo.numOfOutput, 3);
//
//  destroyQueryInfo(pQueryInfo);
//  qParserClearupMetaRequestInfo(&req);
//  destroySqlInfo(&info1);
//}
//
//TEST(testCase, function_Test) {
//  SSqlInfo info1 = doGenerateAST("select count(a) from `t.1abc`");
//  ASSERT_EQ(info1.valid, true);
//
//  char    msg[128] = {0};
//  SMsgBuf buf;
//  buf.len = 128;
//  buf.buf = msg;
//
//  SSqlNode* pNode = (SSqlNode*) taosArrayGetP(((SArray*)info1.list), 0);
//  int32_t code = evaluateSqlNode(pNode, TSDB_TIME_PRECISION_NANO, &buf);
//  ASSERT_EQ(code, 0);
//
//  SMetaReq req = {0};
//  int32_t  ret = qParserExtractRequestedMetaInfo(&info1, &req, msg, 128);
//  ASSERT_EQ(ret, 0);
//  ASSERT_EQ(taosArrayGetSize(req.pTableName), 1);
//
//  SQueryStmtInfo* pQueryInfo = createQueryInfo();
//  setTableMetaInfo(pQueryInfo, &req);
//
//  SSqlNode* pSqlNode = (SSqlNode*)taosArrayGetP(info1.list, 0);
//  ret = validateSqlNode(pSqlNode, pQueryInfo, &buf);
//
//  SArray* pExprList = pQueryInfo->exprList;
//  ASSERT_EQ(taosArrayGetSize(pExprList), 1);
//
//  SExprInfo* p1 = (SExprInfo*) taosArrayGetP(pExprList, 0);
//  ASSERT_EQ(p1->base.pColumns->uid, 110);
//  ASSERT_EQ(p1->base.numOfParams, 0);
//  ASSERT_EQ(p1->base.resSchema.type, TSDB_DATA_TYPE_BIGINT);
//  ASSERT_STRCASEEQ(p1->base.resSchema.name, "count(a)");
//  ASSERT_STRCASEEQ(p1->base.pColumns->name, "t.1abc.a");
//  ASSERT_EQ(p1->base.pColumns->info.colId, 1);
//  ASSERT_EQ(p1->base.pColumns->flag, TSDB_COL_NORMAL);
//  ASSERT_STRCASEEQ(p1->base.token, "count(a)");
//  ASSERT_EQ(p1->base.interBytes, 8);
//
//  ASSERT_EQ(taosArrayGetSize(pQueryInfo->colList), 2);
//  ASSERT_EQ(pQueryInfo->fieldsInfo.numOfOutput, 1);
//
//  destroyQueryInfo(pQueryInfo);
//  qParserClearupMetaRequestInfo(&req);
//  destroySqlInfo(&info1);
//}
//
//TEST(testCase, function_Test2) {
//  SSqlInfo info1 = doGenerateAST("select count(a) abc from `t.1abc`");
//  ASSERT_EQ(info1.valid, true);
//
//  char    msg[128] = {0};
//  SMsgBuf buf;
//  buf.len = 128;
//  buf.buf = msg;
//
//  SSqlNode* pNode = (SSqlNode*) taosArrayGetP(((SArray*)info1.list), 0);
//  int32_t code = evaluateSqlNode(pNode, TSDB_TIME_PRECISION_NANO, &buf);
//  ASSERT_EQ(code, 0);
//
//  SMetaReq req = {0};
//  int32_t  ret = qParserExtractRequestedMetaInfo(&info1, &req, msg, 128);
//  ASSERT_EQ(ret, 0);
//  ASSERT_EQ(taosArrayGetSize(req.pTableName), 1);
//
//  SQueryStmtInfo* pQueryInfo = createQueryInfo();
//  setTableMetaInfo(pQueryInfo, &req);
//
//  SSqlNode* pSqlNode = (SSqlNode*)taosArrayGetP(info1.list, 0);
//  ret = validateSqlNode(pSqlNode, pQueryInfo, &buf);
//
//  SArray* pExprList = pQueryInfo->exprList;
//  ASSERT_EQ(taosArrayGetSize(pExprList), 1);
//
//  SExprInfo* p1 = (SExprInfo*) taosArrayGetP(pExprList, 0);
//  ASSERT_EQ(p1->base.pColumns->uid, 110);
//  ASSERT_EQ(p1->base.numOfParams, 0);
//  ASSERT_EQ(p1->base.resSchema.type, TSDB_DATA_TYPE_BIGINT);
//  ASSERT_STRCASEEQ(p1->base.resSchema.name, "abc");
//  ASSERT_STRCASEEQ(p1->base.pColumns->name, "t.1abc.a");
//  ASSERT_EQ(p1->base.pColumns->info.colId, 1);
//  ASSERT_EQ(p1->base.pColumns->flag, TSDB_COL_NORMAL);
//  ASSERT_STRCASEEQ(p1->base.token, "count(a)");
//  ASSERT_EQ(p1->base.interBytes, 8);
//
//  ASSERT_EQ(taosArrayGetSize(pQueryInfo->colList), 2);
//  ASSERT_EQ(pQueryInfo->fieldsInfo.numOfOutput, 1);
//
//  destroyQueryInfo(pQueryInfo);
//  qParserClearupMetaRequestInfo(&req);
//  destroySqlInfo(&info1);
//}
//
//TEST(testCase, function_Test3) {
//  SSqlInfo info1 = doGenerateAST("select first(*) from `t.1abc`");
//  ASSERT_EQ(info1.valid, true);
//
//  char    msg[128] = {0};
//  SMsgBuf buf;
//  buf.len = 128;
//  buf.buf = msg;
//
//  SSqlNode* pNode = (SSqlNode*) taosArrayGetP(((SArray*)info1.list), 0);
//  int32_t code = evaluateSqlNode(pNode, TSDB_TIME_PRECISION_NANO, &buf);
//  ASSERT_EQ(code, 0);
//
//  SMetaReq req = {0};
//  int32_t  ret = qParserExtractRequestedMetaInfo(&info1, &req, msg, 128);
//  ASSERT_EQ(ret, 0);
//  ASSERT_EQ(taosArrayGetSize(req.pTableName), 1);
//
//  SQueryStmtInfo* pQueryInfo = createQueryInfo();
//  setTableMetaInfo(pQueryInfo, &req);
//
//  SSqlNode* pSqlNode = (SSqlNode*)taosArrayGetP(info1.list, 0);
//  ret = validateSqlNode(pSqlNode, pQueryInfo, &buf);
//
//  SArray* pExprList = pQueryInfo->exprList;
//  ASSERT_EQ(taosArrayGetSize(pExprList), 4);
//
//  SExprInfo* p1 = (SExprInfo*) taosArrayGetP(pExprList, 0);
//  ASSERT_EQ(p1->base.pColumns->uid, 110);
//  ASSERT_EQ(p1->base.numOfParams, 0);
//  ASSERT_EQ(p1->base.resSchema.type, TSDB_DATA_TYPE_TIMESTAMP);
//  ASSERT_STRCASEEQ(p1->base.resSchema.name, "first(ts)");
//  ASSERT_STRCASEEQ(p1->base.pColumns->name, "t.1abc.ts");
//  ASSERT_EQ(p1->base.pColumns->info.colId, 0);
//  ASSERT_EQ(p1->base.pColumns->flag, TSDB_COL_NORMAL);
//  ASSERT_STRCASEEQ(p1->base.token, "first(ts)");
//  ASSERT_EQ(p1->base.interBytes, 24);
//
//  ASSERT_EQ(pQueryInfo->fieldsInfo.numOfOutput, 4);
//
//  destroyQueryInfo(pQueryInfo);
//  qParserClearupMetaRequestInfo(&req);
//  destroySqlInfo(&info1);
//}
//
//TEST(testCase, function_Test4) {
//  SSqlInfo info1 = doGenerateAST("select _block_dist() as a1 from `t.1abc`");
//  ASSERT_EQ(info1.valid, true);
//
//  char    msg[128] = {0};
//  SMsgBuf buf;
//  buf.len = 128;
//  buf.buf = msg;
//
//  SSqlNode* pNode = (SSqlNode*) taosArrayGetP(((SArray*)info1.list), 0);
//  int32_t code = evaluateSqlNode(pNode, TSDB_TIME_PRECISION_NANO, &buf);
//  ASSERT_EQ(code, 0);
//
//  SMetaReq req = {0};
//  int32_t  ret = qParserExtractRequestedMetaInfo(&info1, &req, msg, 128);
//  ASSERT_EQ(ret, 0);
//  ASSERT_EQ(taosArrayGetSize(req.pTableName), 1);
//
//  SQueryStmtInfo* pQueryInfo = createQueryInfo();
//  setTableMetaInfo(pQueryInfo, &req);
//
//  SSqlNode* pSqlNode = (SSqlNode*)taosArrayGetP(info1.list, 0);
//  ret = validateSqlNode(pSqlNode, pQueryInfo, &buf);
//
//  SArray* pExprList = pQueryInfo->exprList;
//  ASSERT_EQ(taosArrayGetSize(pExprList), 1);
//
//  SExprInfo* p1 = (SExprInfo*) taosArrayGetP(pExprList, 0);
//  ASSERT_EQ(p1->base.pColumns->uid, 110);
//  ASSERT_EQ(p1->base.numOfParams, 1);
//  ASSERT_EQ(p1->base.resSchema.type, TSDB_DATA_TYPE_BINARY);
//  ASSERT_STRCASEEQ(p1->base.resSchema.name, "a1");
////  ASSERT_STRCASEEQ(p1->base.colInfo.name, "t.1abc.ts");
////  ASSERT_EQ(p1->base.colInfo.colId, 0);
//  ASSERT_EQ(p1->base.pColumns->flag, TSDB_COL_NORMAL);
//  ASSERT_STRCASEEQ(p1->base.token, "_block_dist()");
//  ASSERT_EQ(p1->base.interBytes, 0);
//
//  ASSERT_EQ(taosArrayGetSize(pQueryInfo->colList), 1);
//  ASSERT_EQ(pQueryInfo->fieldsInfo.numOfOutput, 1);
//
//  destroyQueryInfo(pQueryInfo);
//  qParserClearupMetaRequestInfo(&req);
//  destroySqlInfo(&info1);
//}

//TEST(testCase, function_Test5) {
//  //todo  select concat(concat(a, b), concat(b, a)) from `t.1abc`;
//
//  SSqlInfo info1 = doGenerateAST("select sum(a) + avg(b)  as a1 from `t.1abc`");
//  ASSERT_EQ(info1.valid, true);
//
//  char    msg[128] = {0};
//  SMsgBuf buf;
//  buf.len = 128;
//  buf.buf = msg;
//
//  SSqlNode* pNode = (SSqlNode*) taosArrayGetP(((SArray*)info1.list), 0);
//  int32_t code = evaluateSqlNode(pNode, TSDB_TIME_PRECISION_NANO, &buf);
//  ASSERT_EQ(code, 0);
//
//  SMetaReq req = {0};
//  int32_t  ret = qParserExtractRequestedMetaInfo(&info1, &req, msg, 128);
//  ASSERT_EQ(ret, 0);
//  ASSERT_EQ(taosArrayGetSize(req.pTableName), 1);
//
//  SQueryStmtInfo* pQueryInfo = createQueryInfo();
//  setTableMetaInfo(pQueryInfo, &req);
//
//  SSqlNode* pSqlNode = (SSqlNode*)taosArrayGetP(info1.list, 0);
//  ret = validateSqlNode(pSqlNode, pQueryInfo, &buf);
//  ASSERT_EQ(ret, 0);
//
//  SArray* pExprList = pQueryInfo->exprList;
//  ASSERT_EQ(taosArrayGetSize(pExprList), 3);
//
//  SExprInfo* p1 = (SExprInfo*) taosArrayGetP(pExprList, 0);
//  ASSERT_EQ(p1->base.numOfCols, 2);
//  ASSERT_EQ(p1->base.pColumns->uid, 110);
//
//  ASSERT_EQ(p1->base.numOfParams, 1);
//  ASSERT_EQ(p1->base.resSchema.type, TSDB_DATA_TYPE_DOUBLE);
//  ASSERT_STRCASEEQ(p1->base.resSchema.name, "a1");
//
//  ASSERT_EQ(p1->base.pColumns->flag, TSDB_COL_NORMAL);
//  ASSERT_STRCASEEQ(p1->base.token, "sum(a) + avg(b)");
//  ASSERT_EQ(p1->base.interBytes, 0);
//
//  ASSERT_EQ(taosArrayGetSize(pQueryInfo->colList), 3);
//  ASSERT_EQ(pQueryInfo->fieldsInfo.numOfOutput, 1);
//
//  destroyQueryInfo(pQueryInfo);
//  qParserClearupMetaRequestInfo(&req);
//  destroySqlInfo(&info1);
//}

TEST(testCase, function_Test10) {
//  sqlCheck("select c from `t.1abc`", true);
//  sqlCheck("select length(c) from `t.1abc`", true);
  sqlCheck("select sum(length(a+b)) from `t.1abc`", false);
//  sqlCheck("select sum(sum(a+b)) from `t.1abc`", false);
}

TEST(testCase, function_Test6) {
  SSqlInfo info1 = doGenerateAST("select sum(a+b) as a1, first(b*a), count(b+b), count(1), count(42.1) from `t.1abc` interval(10s, 1s)");
  ASSERT_EQ(info1.valid, true);

  char    msg[128] = {0};
  SMsgBuf buf;
  buf.len = 128;
  buf.buf = msg;

  SSqlNode* pNode = (SSqlNode*) taosArrayGetP(((SArray*)info1.list), 0);
  int32_t code = evaluateSqlNode(pNode, TSDB_TIME_PRECISION_NANO, &buf);
  ASSERT_EQ(code, 0);

  SMetaReq req = {0};
  int32_t  ret = qParserExtractRequestedMetaInfo(&info1, &req, msg, 128);
  ASSERT_EQ(ret, 0);
  ASSERT_EQ(taosArrayGetSize(req.pTableName), 1);

  SQueryStmtInfo* pQueryInfo = createQueryInfo();
  setTableMetaInfo(pQueryInfo, &req);

  SSqlNode* pSqlNode = (SSqlNode*)taosArrayGetP(info1.list, 0);
  ret = validateSqlNode(pSqlNode, pQueryInfo, &buf);
  ASSERT_EQ(ret, 0);

  SArray* pExprList = pQueryInfo->exprList;
  ASSERT_EQ(taosArrayGetSize(pExprList), 5);

  SExprInfo* p1 = (SExprInfo*) taosArrayGetP(pExprList, 0);
  ASSERT_EQ(p1->base.pColumns->uid, 110);
  ASSERT_EQ(p1->base.numOfParams, 0);
  ASSERT_EQ(p1->base.resSchema.type, TSDB_DATA_TYPE_DOUBLE);
  ASSERT_STRCASEEQ(p1->base.resSchema.name, "a1");
  ASSERT_EQ(p1->base.pColumns->flag, TSDB_COL_NORMAL);
  ASSERT_STRCASEEQ(p1->base.token, "sum(a+b)");
  ASSERT_EQ(p1->base.interBytes, 16);
  ASSERT_EQ(p1->pExpr->nodeType, TEXPR_UNARYEXPR_NODE);
  ASSERT_EQ(p1->pExpr->_function.functionId, FUNCTION_SUM);
  ASSERT_TRUE(p1->pExpr->_node.pRight == NULL);

  tExprNode* pParam = p1->pExpr->_node.pLeft;

  ASSERT_EQ(pParam->nodeType, TEXPR_BINARYEXPR_NODE);
  ASSERT_EQ(pParam->_node.optr, TSDB_BINARY_OP_ADD);
  ASSERT_EQ(pParam->_node.pLeft->nodeType, TEXPR_COL_NODE);
  ASSERT_EQ(pParam->_node.pRight->nodeType, TEXPR_COL_NODE);

  ASSERT_EQ(taosArrayGetSize(pQueryInfo->colList), 3);
  ASSERT_EQ(pQueryInfo->fieldsInfo.numOfOutput, 5);

  SExprInfo* p2 = (SExprInfo*) taosArrayGetP(pExprList, 1);
  ASSERT_EQ(p2->base.pColumns->uid, 110);
  ASSERT_EQ(p2->base.numOfParams, 0);
  ASSERT_EQ(p2->base.resSchema.type, TSDB_DATA_TYPE_DOUBLE);
  ASSERT_STRCASEEQ(p2->base.resSchema.name, "first(b*a)");
  ASSERT_EQ(p2->base.pColumns->flag, TSDB_COL_NORMAL);
  ASSERT_STRCASEEQ(p2->base.token, "first(b*a)");
  ASSERT_EQ(p2->base.interBytes, 24);
  ASSERT_EQ(p2->pExpr->nodeType, TEXPR_UNARYEXPR_NODE);
  ASSERT_EQ(p2->pExpr->_function.functionId, FUNCTION_FIRST);
  ASSERT_TRUE(p2->pExpr->_node.pRight == NULL);

  destroyQueryInfo(pQueryInfo);
  qParserClearupMetaRequestInfo(&req);
  destroySqlInfo(&info1);
}

TEST(testCase, function_Test7) {
  SSqlInfo info1 = doGenerateAST("select count(a+b),count(1) from `t.1abc` interval(10s, 1s)");
  ASSERT_EQ(info1.valid, true);

  char    msg[128] = {0};
  SMsgBuf buf;
  buf.len = 128;
  buf.buf = msg;

  SSqlNode* pNode = (SSqlNode*) taosArrayGetP(((SArray*)info1.list), 0);
  int32_t code = evaluateSqlNode(pNode, TSDB_TIME_PRECISION_NANO, &buf);
  ASSERT_EQ(code, 0);

  SMetaReq req = {0};
  int32_t  ret = qParserExtractRequestedMetaInfo(&info1, &req, msg, 128);
  ASSERT_EQ(ret, 0);
  ASSERT_EQ(taosArrayGetSize(req.pTableName), 1);

  SQueryStmtInfo* pQueryInfo = createQueryInfo();
  setTableMetaInfo(pQueryInfo, &req);

  SSqlNode* pSqlNode = (SSqlNode*)taosArrayGetP(info1.list, 0);
  ret = validateSqlNode(pSqlNode, pQueryInfo, &buf);
  ASSERT_EQ(ret, 0);

  SArray* pExprList = pQueryInfo->exprList;
  ASSERT_EQ(taosArrayGetSize(pExprList), 2);

  SExprInfo* p1 = (SExprInfo*) taosArrayGetP(pExprList, 0);
  ASSERT_EQ(p1->base.pColumns->uid, 110);
  ASSERT_EQ(p1->base.numOfParams, 0);
  ASSERT_EQ(p1->base.resSchema.type, TSDB_DATA_TYPE_BIGINT);
  ASSERT_STRCASEEQ(p1->base.resSchema.name, "count(a+b)");
  ASSERT_EQ(p1->base.pColumns->flag, TSDB_COL_NORMAL);
  ASSERT_STRCASEEQ(p1->base.token, "count(a+b)");
  ASSERT_EQ(p1->base.interBytes, 8);
  ASSERT_EQ(p1->pExpr->nodeType, TEXPR_UNARYEXPR_NODE);
  ASSERT_EQ(p1->pExpr->_function.functionId, FUNCTION_COUNT);
  ASSERT_TRUE(p1->pExpr->_node.pRight == NULL);

  tExprNode* pParam = p1->pExpr->_node.pLeft;

  ASSERT_EQ(pParam->nodeType, TEXPR_BINARYEXPR_NODE);
  ASSERT_EQ(pParam->_node.optr, TSDB_BINARY_OP_ADD);
  ASSERT_EQ(pParam->_node.pLeft->nodeType, TEXPR_COL_NODE);
  ASSERT_EQ(pParam->_node.pRight->nodeType, TEXPR_COL_NODE);

  ASSERT_EQ(taosArrayGetSize(pQueryInfo->colList), 3);
  ASSERT_EQ(pQueryInfo->fieldsInfo.numOfOutput, 2);

  destroyQueryInfo(pQueryInfo);
  qParserClearupMetaRequestInfo(&req);
  destroySqlInfo(&info1);
}

TEST(testCase, function_Test8) {
  SSqlInfo info1 = doGenerateAST("select top(a*b / 99, 20) from `t.1abc` interval(10s, 1s)");
  ASSERT_EQ(info1.valid, true);

  char    msg[128] = {0};
  SMsgBuf buf;
  buf.len = 128;
  buf.buf = msg;

  SSqlNode* pNode = (SSqlNode*) taosArrayGetP(((SArray*)info1.list), 0);
  int32_t code = evaluateSqlNode(pNode, TSDB_TIME_PRECISION_NANO, &buf);
  ASSERT_EQ(code, 0);

  SMetaReq req = {0};
  int32_t  ret = qParserExtractRequestedMetaInfo(&info1, &req, msg, 128);
  ASSERT_EQ(ret, 0);
  ASSERT_EQ(taosArrayGetSize(req.pTableName), 1);

  SQueryStmtInfo* pQueryInfo = createQueryInfo();
  setTableMetaInfo(pQueryInfo, &req);

  SSqlNode* pSqlNode = (SSqlNode*)taosArrayGetP(info1.list, 0);
  ret = validateSqlNode(pSqlNode, pQueryInfo, &buf);
  ASSERT_EQ(ret, 0);

  SArray* pExprList = pQueryInfo->exprList;
  ASSERT_EQ(taosArrayGetSize(pExprList), 2);

  SExprInfo* p1 = (SExprInfo*) taosArrayGetP(pExprList, 1);
  ASSERT_EQ(p1->base.pColumns->uid, 110);
  ASSERT_EQ(p1->base.numOfParams, 1);
  ASSERT_EQ(p1->base.resSchema.type, TSDB_DATA_TYPE_DOUBLE);
  ASSERT_STRCASEEQ(p1->base.resSchema.name, "top(a*b / 99, 20)");
  ASSERT_EQ(p1->base.pColumns->flag, TSDB_COL_NORMAL);
  ASSERT_STRCASEEQ(p1->base.token, "top(a*b / 99, 20)");
  ASSERT_EQ(p1->base.interBytes, 16);

  ASSERT_EQ(p1->pExpr->nodeType, TEXPR_UNARYEXPR_NODE);
  ASSERT_EQ(p1->pExpr->_function.functionId, FUNCTION_TOP);
  ASSERT_TRUE(p1->pExpr->_node.pRight == NULL);

  tExprNode* pParam = p1->pExpr->_node.pLeft;

  ASSERT_EQ(pParam->nodeType, TEXPR_BINARYEXPR_NODE);
  ASSERT_EQ(pParam->_node.optr, TSDB_BINARY_OP_DIVIDE);
  ASSERT_EQ(pParam->_node.pLeft->nodeType, TEXPR_BINARYEXPR_NODE);
  ASSERT_EQ(pParam->_node.pRight->nodeType, TEXPR_VALUE_NODE);

  ASSERT_EQ(taosArrayGetSize(pQueryInfo->colList), 3);
  ASSERT_EQ(pQueryInfo->fieldsInfo.numOfOutput, 2);

  destroyQueryInfo(pQueryInfo);
  qParserClearupMetaRequestInfo(&req);
  destroySqlInfo(&info1);
}

TEST(testCase, invalid_sql_Test) {
  char    msg[128] = {0};
  SMsgBuf buf;
  buf.len = 128;
  buf.buf = msg;

  SSqlInfo info1 = doGenerateAST("select count(k) from `t.1abc` interval(10s, 1s)");
  ASSERT_EQ(info1.valid, true);

  SSqlNode* pNode = (SSqlNode*) taosArrayGetP(((SArray*)info1.list), 0);
  int32_t code = evaluateSqlNode(pNode, TSDB_TIME_PRECISION_NANO, &buf);
  ASSERT_EQ(code, 0);

  SMetaReq req = {0};
  int32_t  ret = qParserExtractRequestedMetaInfo(&info1, &req, msg, 128);
  ASSERT_EQ(ret, 0);
  ASSERT_EQ(taosArrayGetSize(req.pTableName), 1);

  SQueryStmtInfo* pQueryInfo = createQueryInfo();
  setTableMetaInfo(pQueryInfo, &req);

  SSqlNode* pSqlNode = (SSqlNode*)taosArrayGetP(info1.list, 0);
  ret = validateSqlNode(pSqlNode, pQueryInfo, &buf);
  ASSERT_NE(ret, 0);

  destroyQueryInfo(pQueryInfo);
  qParserClearupMetaRequestInfo(&req);
  destroySqlInfo(&info1);
//===============================================================================================================
  info1 = doGenerateAST("select top(a*b, ABC) from `t.1abc` interval(10s, 1s)");
  ASSERT_EQ(info1.valid, true);

  pNode = (SSqlNode*) taosArrayGetP(((SArray*)info1.list), 0);
  code = evaluateSqlNode(pNode, TSDB_TIME_PRECISION_NANO, &buf);
  ASSERT_EQ(code, 0);

  ret = qParserExtractRequestedMetaInfo(&info1, &req, msg, 128);
  ASSERT_EQ(ret, 0);
  ASSERT_EQ(taosArrayGetSize(req.pTableName), 1);

  pQueryInfo = createQueryInfo();
  setTableMetaInfo(pQueryInfo, &req);

  pSqlNode = (SSqlNode*)taosArrayGetP(info1.list, 0);
  ret = validateSqlNode(pSqlNode, pQueryInfo, &buf);
  ASSERT_NE(ret, 0);

  destroyQueryInfo(pQueryInfo);
  qParserClearupMetaRequestInfo(&req);
  destroySqlInfo(&info1);
}