/**
 * Copyright (c) 2021 OceanBase
 * OceanBase CE is licensed under Mulan PubL v2.
 * You can use this software according to the terms and conditions of the Mulan PubL v2.
 * You may obtain a copy of Mulan PubL v2 at:
 *          http://license.coscl.org.cn/MulanPubL-2.0
 * THIS SOFTWARE IS PROVIDED ON AN "AS IS" BASIS, WITHOUT WARRANTIES OF ANY KIND,
 * EITHER EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO NON-INFRINGEMENT,
 * MERCHANTABILITY OR FIT FOR A PARTICULAR PURPOSE.
 * See the Mulan PubL v2 for more details.
 */

#include <gtest/gtest.h>
#include "sql/engine/expr/ob_expr_concat_ws.h"
#include "ob_expr_test_utils.h"
#include <sstream>
#include<string>
using namespace oceanbase::common;
using namespace oceanbase::sql;

class TestAllocator: public ObIAllocator
{
public:
	TestAllocator(): label_(ObModIds::TEST) {}
  virtual ~TestAllocator() {}
  void *alloc(const int64_t sz) { UNUSED(sz); return NULL; }
  void free(void *p) { UNUSED(p); }
  void freed(const int64_t sz) { UNUSED(sz); }
  void set_label(const char *label) {label_ = label;};
private:
  const char *label_;
};

class ObExprConcatWsTest : public ::testing::Test
{
public:
  ObExprConcatWsTest();
  virtual ~ObExprConcatWsTest();
  virtual void SetUp();
  virtual void TearDown();
private:
  // disallow copy
  ObExprConcatWsTest(const ObExprConcatWsTest &other);
  ObExprConcatWsTest& operator=(const ObExprConcatWsTest &other);
private:
  // data members
};
ObExprConcatWsTest::ObExprConcatWsTest()
{
}

ObExprConcatWsTest::~ObExprConcatWsTest()
{
}

void ObExprConcatWsTest::SetUp()
{
}

void ObExprConcatWsTest::TearDown()
{
}
#define TN(num,rtype, res,...) EXPECT_RESULTN(concatws, &buf, calc_resultN, num,rtype, res,  __VA_ARGS__)

#define FN(num,...) EXPECT_FAIL_RESULTN(concatws, &buf, calc_resultN, num,  __VA_ARGS__)

#define EXCEPT_RESULTNN(obj, str_buf, func, num, params) \
{\
	ObObj res;\
  int err = obj.func(res, params, num, str_buf); \
  EXPECT_TRUE(OB_SUCCESS == err); \
} while(0)

#define EXCEPT_NULL_RESULT1(obj,str_buf,func,t1,v1,t2,v2,t3,v3)\
{\
	ObObj param[3];\
	ObObj res;\
	param[2].set_null();\
	param[1].set_null();\
	param[0].set_##t1(v1);\
	param[3].set_##t2(v2);\
	param[4].set_##t3(v3);\
	int ret = obj.func(res,param,5,str_buf);\
	_OB_LOG(INFO, "res=%s", to_cstring(res)); \
	ASSERT_EQ(OB_SUCCESS, ret);   \
}while(0)
#define EXCEPT_NULL_RESULT2(obj,str_buf,func,rt,rv)\
{\
	ObObj param[2];\
	ObObj res;\
	param[0].set_null();\
	param[1].set_null();\
	int ret = obj.func(res,param,2,str_buf);\
	_OB_LOG(INFO, "res=%s", to_cstring(res)); \
	ASSERT_EQ(OB_SUCCESS, ret);   \
}while(0)
#define EXCEPT_NULL_RESULT3(obj,str_buf,func,rt,rv)\
{\
	ObObj param[1];\
	ObObj res;\
	param[0].set_null();\
	int ret = obj.func(res,param,1,str_buf);\
	ASSERT_EQ(OB_INVALID_ARGUMENT, ret);   \
}while(0)
#define EXCEPT_NULL_RESULT4(obj,str_buf,func,t1,v1,t2,v2,t3,v3)\
{\
	ObObj param[3];\
	ObObj res;\
	param[0].set_##t1(v1);\
	param[1].set_##t2(v2);\
	param[2].set_##t3(v3);\
	int ret = obj.func(res,param,5,str_buf);\
	ASSERT_EQ(OB_NOT_INIT, ret);   \
}while(0)
#define EXCEPT_NULL_RESULT5(obj,str_buf,func,t1,v1,t2,v2,t3,v3)\
{\
	ObObj param[3];\
	ObObj res;\
	param[0].set_##t1(v1);\
	param[1].set_##t2(v2);\
	param[2].set_##t3(v3);\
	int ret = obj.func(res,param,5,str_buf);\
	ASSERT_EQ(OB_ALLOCATE_MEMORY_FAILED, ret);   \
}while(0)
#define EXCEPT_MAXLEN_RESULT(obj,str_buf,func,num, params)\
{\
	ObObj res;\
	int ret = obj.func(res,params,num,str_buf);\
	ASSERT_EQ(OB_SIZE_OVERFLOW, ret);   \
}while(0)

#define TNN(num,params) EXCEPT_RESULTNN(concatws, &buf, calc_resultN, num, params)
#define ENR1(t1,v1,t2,v2,t3,v3) EXCEPT_NULL_RESULT1(concatws, &buf, calc_resultN,t1,v1,t2,v2,t3,v3)
#define ENR2(rt,rv) EXCEPT_NULL_RESULT2(concatws, &buf, calc_resultN, rt,rv)
#define ENR3(rt,rv) EXCEPT_NULL_RESULT3(concatws, &buf, calc_resultN, rt,rv)
#define ENR4(buf,t1,v1,t2,v2,t3,v3) EXCEPT_NULL_RESULT4(concatws,buf, calc_resultN,t1,v1,t2,v2,t3,v3)
#define ENR5(buf,t1,v1,t2,v2,t3,v3) EXCEPT_NULL_RESULT5(concatws,buf, calc_resultN,t1,v1,t2,v2,t3,v3)
#define EMR(num,params) EXCEPT_MAXLEN_RESULT(concatws,&buf,calc_resultN,num, params)
TEST_F(ObExprConcatWsTest, basic_test)
{
	ObExprConcatWs concatws;
	DefaultPageAllocator buf;

  //  case 1. concat_ws(',','1');
  TN(2, varchar, "1", varchar, ",",varchar, "1");
  //  case 2. concat_ws(',','1','2');
  TN(3, varchar, "1,2", varchar,",",varchar, "1", varchar, "2");
  //  case 3. concat_ws(',','','');
  TN(3, varchar, ",", varchar,",",varchar, "", varchar, "");
  //  case 4. concat_ws('dffd','221','2','3');
  TN(4, varchar, "221dffd2dffd3", varchar,"dffd",varchar, "221", varchar, "2",varchar,"3");
  //  case 5. concat_ws(',','','2333','3');
  TN(4, varchar, ",2333,3", varchar,",",varchar, "", varchar, "2333",varchar,"3");
  //  case 6. concat_ws('','','','2','3');
  TN(5, varchar, "#23", varchar,"",varchar,"#",varchar, "", varchar, "2",varchar,"3");
  //  case 7. concat_ws('#','','','2','324343','');
  TN(6, varchar, "#,,2,324343,", varchar,",",varchar,"#",varchar, "", varchar, "2",varchar,"324343",varchar,"");
  //  case 8. concat_ws('sdsada',null,null,'3234sdda','4324sawsd');
  ENR1(varchar,"sdsada",varchar,"3234sdda",varchar,"4324sawsd");
  //  case 9. concat_ws('',null,null,'43534534','fdgsawsd');
  ENR1(varchar,"",varchar,"43534534",varchar,"fdgsawsd");
  //  case 10. concat_ws('',null,null,'','');
  ENR1(varchar,"",varchar,"",varchar,"");
  //  case 11. concat_ws('sds',null,null,'242%4','^7dfs');
  ENR1(varchar,"sds",varchar,"242%4",varchar,"^7dfs");
  //  case 12. concat_ws(null,null)
  ENR2(varchar,"");
  //	case 13. testing big size
  ObObj params[1000];
  for(int32_t idx = 0 ; idx < 1000 ; idx++) {
    params[idx].set_varchar("dsfdsfsfsdffsfdsfwererewrewrewddddddwererxdsseeefr343242sas**");
  }
  TNN(1000,params);
  //	case 14. testing the first parameter is null for big size
  params[0].set_null();
  TNN(1000,params);
}

TEST_F(ObExprConcatWsTest, special_test)
{
	TestAllocator tbuf;
	ObExprConcatWs concatws;
	DefaultPageAllocator buf;
	//  case 1. concat_ws('1');
	FN(1, varchar, "1");
	//  case 2. concat_ws(1,'1','2');
	FN(3,int,1,varchar, "1", varchar, "2");
	//  case 3. concat_ws('1saas','222',1);
	FN(3,varchar,"1saas", varchar, "222",int,1);
	//  case 4. concat_ws('abcdf','1',3.0,1);
	FN(4, varchar,"abcdf",varchar, "1", double, 3.0,int,1);
	//  case 5. concat_ws(',','1','2',1,2);
	FN(5,varchar,",",varchar, "1", varchar, "2",int,1,int, 2);
	//	case 6. concat_ws('ds','dfd343','433453') test when buf is NULL
	ENR4(NULL,varchar,"ds",varchar,"fdfds",varchar,"433453");
	//	case 7. concat_ws('ds','dfd343','433453') test when tbuf can not allocator memory
  ENR5(&tbuf,varchar,"ds",varchar,"fdfds",varchar,"433453");
  //  case 8. concat_ws(null)
  ENR3(varchar,"");
  // case 9. testing max length
  ObObj params[10000];
	for(int32_t idx = 0 ; idx < 10000 ; idx++) {
		params[idx].set_varchar("dsfdsfsfsdffsfdsfwererewrewrewddddddwererxdsseeefr343242sas**");
	}
	EMR(10000,params);
}
int main(int argc, char **argv)
{
	  oceanbase::common::ObLogger::get_logger().set_log_level("DEBUG");
	  ::testing::InitGoogleTest(&argc,argv);
	  return RUN_ALL_TESTS();
}