135 lines
4.4 KiB
C++
135 lines
4.4 KiB
C++
/*
|
|
* 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 "querynodes.h"
|
|
|
|
using namespace std;
|
|
|
|
static EDealRes rewriterTest(SNode** pNode, void* pContext) {
|
|
EDealRes* pRes = (EDealRes*)pContext;
|
|
if (QUERY_NODE_OPERATOR == nodeType(*pNode)) {
|
|
SOperatorNode* pOp = (SOperatorNode*)(*pNode);
|
|
if (QUERY_NODE_VALUE != nodeType(pOp->pLeft) || QUERY_NODE_VALUE != nodeType(pOp->pRight)) {
|
|
*pRes = DEAL_RES_ERROR;
|
|
}
|
|
SValueNode* pVal = NULL;
|
|
int32_t code = nodesMakeNode(QUERY_NODE_VALUE, (SNode**)&pVal);
|
|
string tmp = to_string(stoi(((SValueNode*)(pOp->pLeft))->literal) + stoi(((SValueNode*)(pOp->pRight))->literal));
|
|
pVal->literal = taosStrdup(tmp.c_str());
|
|
nodesDestroyNode(*pNode);
|
|
*pNode = (SNode*)pVal;
|
|
}
|
|
return DEAL_RES_CONTINUE;
|
|
}
|
|
|
|
TEST(NodesTest, traverseTest) {
|
|
SNode* pRoot = NULL;
|
|
int32_t code = nodesMakeNode(QUERY_NODE_OPERATOR,(SNode**)&pRoot);
|
|
ASSERT_EQ(code, TSDB_CODE_SUCCESS);
|
|
SOperatorNode* pOp = (SOperatorNode*)pRoot;
|
|
SOperatorNode* pLeft = NULL;
|
|
ASSERT_EQ(TSDB_CODE_SUCCESS, nodesMakeNode(QUERY_NODE_OPERATOR, (SNode**)&pLeft));
|
|
ASSERT_EQ(TSDB_CODE_SUCCESS, nodesMakeNode(QUERY_NODE_VALUE, &pLeft->pLeft));
|
|
((SValueNode*)(pLeft->pLeft))->literal = taosStrdup("10");
|
|
ASSERT_EQ(TSDB_CODE_SUCCESS, nodesMakeNode(QUERY_NODE_VALUE, &pLeft->pRight));
|
|
((SValueNode*)(pLeft->pRight))->literal = taosStrdup("5");
|
|
pOp->pLeft = (SNode*)pLeft;
|
|
ASSERT_EQ(TSDB_CODE_SUCCESS, nodesMakeNode(QUERY_NODE_VALUE, &pOp->pRight));
|
|
((SValueNode*)(pOp->pRight))->literal = taosStrdup("3");
|
|
|
|
EXPECT_EQ(nodeType(pRoot), QUERY_NODE_OPERATOR);
|
|
EDealRes res = DEAL_RES_CONTINUE;
|
|
nodesRewriteExprPostOrder(&pRoot, rewriterTest, &res);
|
|
EXPECT_EQ(res, DEAL_RES_CONTINUE);
|
|
EXPECT_EQ(nodeType(pRoot), QUERY_NODE_VALUE);
|
|
EXPECT_EQ(string(((SValueNode*)pRoot)->literal), "18");
|
|
nodesDestroyNode(pRoot);
|
|
}
|
|
|
|
int32_t compareValueNode(SNode* pNode1, SNode* pNode2) {
|
|
SValueNode* p1 = (SValueNode*)pNode1;
|
|
SValueNode* p2 = (SValueNode*)pNode2;
|
|
|
|
if (p1->datum.i < p2->datum.i)
|
|
return -1;
|
|
else if (p1->datum.i > p2->datum.i)
|
|
return 1;
|
|
else
|
|
return 0;
|
|
}
|
|
|
|
void assert_sort_result(SNodeList* pList) {
|
|
SNode* pNode;
|
|
int32_t i = 0;
|
|
FOREACH(pNode, pList) {
|
|
SValueNode* p = (SValueNode*)pNode;
|
|
ASSERT_EQ(p->datum.i, i++);
|
|
}
|
|
SListCell* pCell = pList->pHead;
|
|
ASSERT_TRUE(pCell->pPrev == NULL);
|
|
ASSERT_TRUE(pList->pTail->pNext == NULL);
|
|
int32_t len = 1;
|
|
while (pCell) {
|
|
if (pCell->pNext) {
|
|
ASSERT_TRUE(pCell->pNext->pPrev == pCell);
|
|
}
|
|
pCell = pCell->pNext;
|
|
if (pCell) len++;
|
|
}
|
|
ASSERT_EQ(len, pList->length);
|
|
}
|
|
|
|
TEST(NodesTest, sort) {
|
|
SValueNode *vn1 = NULL;
|
|
ASSERT_EQ(TSDB_CODE_SUCCESS, nodesMakeNode(QUERY_NODE_VALUE, (SNode**)&vn1));
|
|
vn1->datum.i = 4;
|
|
|
|
SValueNode *vn2 = NULL;
|
|
ASSERT_EQ(TSDB_CODE_SUCCESS, nodesMakeNode(QUERY_NODE_VALUE, (SNode**)&vn2));
|
|
vn2->datum.i = 3;
|
|
|
|
SValueNode *vn3 = NULL;
|
|
ASSERT_EQ(TSDB_CODE_SUCCESS, nodesMakeNode(QUERY_NODE_VALUE, (SNode**)&vn3));
|
|
vn3->datum.i = 2;
|
|
|
|
SValueNode *vn4 = NULL;
|
|
ASSERT_EQ(TSDB_CODE_SUCCESS, nodesMakeNode(QUERY_NODE_VALUE, (SNode**)&vn4));
|
|
vn4->datum.i = 1;
|
|
|
|
SValueNode *vn5 = NULL;
|
|
ASSERT_EQ(TSDB_CODE_SUCCESS, nodesMakeNode(QUERY_NODE_VALUE, (SNode**)&vn5));
|
|
vn5->datum.i = 0;
|
|
|
|
SNodeList* l = NULL;
|
|
ASSERT_EQ(TSDB_CODE_SUCCESS, nodesListMakeAppend(&l, (SNode*)vn1));
|
|
ASSERT_EQ(TSDB_CODE_SUCCESS, nodesListMakeAppend(&l, (SNode*)vn2));
|
|
ASSERT_EQ(TSDB_CODE_SUCCESS, nodesListMakeAppend(&l, (SNode*)vn3));
|
|
ASSERT_EQ(TSDB_CODE_SUCCESS, nodesListMakeAppend(&l, (SNode*)vn4));
|
|
ASSERT_EQ(TSDB_CODE_SUCCESS, nodesListMakeAppend(&l, (SNode*)vn5));
|
|
|
|
nodesSortList(&l, compareValueNode);
|
|
|
|
assert_sort_result(l);
|
|
|
|
nodesDestroyList(l);
|
|
}
|
|
|
|
int main(int argc, char* argv[]) {
|
|
testing::InitGoogleTest(&argc, argv);
|
|
return RUN_ALL_TESTS();
|
|
}
|