165 lines
7.4 KiB
C++
165 lines
7.4 KiB
C++
/**
|
|
* 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 <math.h>
|
|
#include <string.h>
|
|
#include "share/ob_define.h"
|
|
#include "common/object/ob_obj_type.h"
|
|
#include "common/expression/ob_expr_obj.h"
|
|
#include <gtest/gtest.h>
|
|
#include "sql/engine/expr/ob_expr_nvl.h"
|
|
|
|
using namespace oceanbase::common;
|
|
using namespace oceanbase::sql;
|
|
|
|
class ObExprNvlTest : public ::testing::Test {
|
|
public:
|
|
ObExprNvlTest();
|
|
virtual ~ObExprNvlTest();
|
|
virtual void SetUp();
|
|
virtual void TearDown();
|
|
|
|
private:
|
|
// disallow copy
|
|
ObExprNvlTest(const ObExprNvlTest& other);
|
|
ObExprNvlTest& operator=(const ObExprNvlTest& other);
|
|
|
|
protected:
|
|
// data members
|
|
};
|
|
|
|
ObExprNvlTest::ObExprNvlTest()
|
|
{}
|
|
|
|
ObExprNvlTest::~ObExprNvlTest()
|
|
{}
|
|
|
|
void ObExprNvlTest::SetUp()
|
|
{}
|
|
|
|
void ObExprNvlTest::TearDown()
|
|
{}
|
|
#define EV 0.00001
|
|
#define NVL_EXPECT(funop, str_buf, func, type1, v1, type2, v2, restype, res) \
|
|
{ \
|
|
ObObj t1; \
|
|
ObObj t2; \
|
|
ObObj vres; \
|
|
funop op; \
|
|
op.set_result_type(restype); \
|
|
t1.set_##type1(v1); \
|
|
t2.set_##type2(v2); \
|
|
int err = op.func(vres, t1, t2, str_buf); \
|
|
if (OB_SUCCESS != err) { \
|
|
ASSERT_EQ(res, err); \
|
|
} else { \
|
|
switch (vres.get_type()) { \
|
|
case ObIntType: \
|
|
ASSERT_EQ(res, vres.get_int()); \
|
|
break; \
|
|
case ObFloatType: \
|
|
ASSERT_TRUE(fabsf(res - vres.get_float() < EV)); \
|
|
break; \
|
|
case ObDoubleType: \
|
|
ASSERT_TRUE(fabs(res - vres.get_double()) < EV); \
|
|
break; \
|
|
case ObMaxType: \
|
|
ASSERT_EQ(res, vres.get_type()); \
|
|
break; \
|
|
case ObNullType: \
|
|
ASSERT_EQ(res, vres.get_type()); \
|
|
default: \
|
|
ASSERT_TRUE(vres.is_null()); \
|
|
break; \
|
|
} \
|
|
} \
|
|
} \
|
|
while (0)
|
|
|
|
#define NVL_EXPECT_VARCHAR(funop, str_buf, func, type1, v1, type2, v2, res) \
|
|
{ \
|
|
ObObj t1; \
|
|
ObObj t2; \
|
|
ObObj vres; \
|
|
funop op; \
|
|
op.set_result_type(ObVarcharType); \
|
|
t1.set_##type1(v1); \
|
|
t2.set_##type2(v2); \
|
|
int err = op.func(vres, t1, t2, str_buf); \
|
|
if (OB_SUCCESS != err) { \
|
|
} else { \
|
|
ASSERT_EQ(ObVarcharType, vres.get_type()); \
|
|
ASSERT_EQ(strlen(res), vres.get_varchar().length()); \
|
|
ASSERT_EQ(0, memcmp(res, vres.get_varchar().ptr(), strlen(res))); \
|
|
} \
|
|
} \
|
|
while (0)
|
|
|
|
#define R(t1, v1, t2, v2, restype, res) NVL_EXPECT(ObExprNvl, &buf, calc_result2, t1, v1, t2, v2, restype, res)
|
|
#define RV(t1, v1, t2, v2, res) NVL_EXPECT_VARCHAR(ObExprNvl, &buf, calc_result2, t1, v1, t2, v2, res)
|
|
|
|
TEST_F(ObExprNvlTest, result_test)
|
|
{
|
|
// num with other
|
|
ObExprStringBuf buf;
|
|
R(int, 1, varchar, "hello", varchar, 1);
|
|
R(int, 1, int, 3, ObIntType, 1);
|
|
R(int, 1, float, float(3.2), ObDoubleType, 1.0);
|
|
R(int, 1, double, 3.3, ObDoubleType, 1.0);
|
|
R(int, 1, null, 0, ObIntType, 1);
|
|
|
|
R(float, float(1.23), varchar, "hello", varchar, 1.23);
|
|
R(float, float(1.23), int, 3, ObFloatType, 1.230000);
|
|
R(float, float(1.23), float, float(3.2), ObFloatType, 1.230000);
|
|
R(float, float(1.23), double, 3.3, ObDoubleType, 1.230000);
|
|
R(float, float(1.23), null, 0, ObFloatType, 1.230000);
|
|
|
|
R(double, 1.23, varchar, "hello", ObDoubleType, 1.230000);
|
|
R(double, 1.23, int, 3, ObDoubleType, 1.230000);
|
|
R(double, 1.23, float, float(3.2), ObDoubleType, 1.230000);
|
|
R(double, 1.23, double, 3.3, ObDoubleType, 1.230000);
|
|
R(double, 1.23, null, 0, ObDoubleType, 1.230000);
|
|
|
|
// invalid type op
|
|
R(ctime, 1, int, 2, ObNullType, OB_ERR_INVALID_TYPE_FOR_OP);
|
|
R(mtime, 1, int, 2, ObNullType, OB_ERR_INVALID_TYPE_FOR_OP);
|
|
R(ext, 1, int, 2, ObNullType, OB_ERR_INVALID_TYPE_FOR_OP);
|
|
|
|
// varchar with other
|
|
RV(varchar, "hello", int, 2, "hello");
|
|
RV(varchar, "hello", float, float(2), "hello");
|
|
RV(varchar, "hello", double, 2, "hello");
|
|
RV(varchar, "hello", varchar, "2", "hello");
|
|
RV(varchar, "hello", null, 0, "hello");
|
|
RV(varchar, "", varchar, "hello", "");
|
|
RV(varchar, "", double, 2.2, "");
|
|
RV(varchar, "", int, 4, "");
|
|
RV(varchar, "", float, float(2.3), "");
|
|
// R(varchar, "", null, 0, ObNullType, 0);
|
|
|
|
// null with other
|
|
R(null, 0, int, 2, ObIntType, 2);
|
|
R(null, 0, float, float(2.2), ObFloatType, 2.20000);
|
|
R(null, 0, double, 2.3, ObDoubleType, 2.30000);
|
|
RV(null, 0, varchar, "hello", "hello");
|
|
R(null, 0, ctime, 2, ObIntType, OB_ERR_INVALID_TYPE_FOR_OP);
|
|
R(null, 0, mtime, 2, ObIntType, OB_ERR_INVALID_TYPE_FOR_OP);
|
|
R(null, 0, ext, 2, ObIntType, OB_ERR_INVALID_TYPE_FOR_OP);
|
|
R(null, 0, null, 0, ObNullType, 0);
|
|
}
|
|
|
|
int main(int argc, char** argv)
|
|
{
|
|
::testing::InitGoogleTest(&argc, argv);
|
|
return RUN_ALL_TESTS();
|
|
}
|