446 lines
		
	
	
		
			23 KiB
		
	
	
	
		
			C++
		
	
	
	
	
	
			
		
		
	
	
			446 lines
		
	
	
		
			23 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.
 | 
						|
 */
 | 
						|
 | 
						|
#define USING_LOG_PREFIX RPC_TEST
 | 
						|
 | 
						|
#include <gtest/gtest.h>
 | 
						|
#include <fstream>
 | 
						|
#include "lib/timezone/ob_time_convert.h"
 | 
						|
#include "rpc/obrpc/ob_rpc_packet.h"
 | 
						|
#include "rpc/obmysql/ob_mysql_util.h"
 | 
						|
 | 
						|
using namespace oceanbase::common;
 | 
						|
using namespace oceanbase::obmysql;
 | 
						|
 | 
						|
class TestObMySQLUtil : public ::testing::Test {
 | 
						|
public:
 | 
						|
  TestObMySQLUtil()
 | 
						|
  {}
 | 
						|
 | 
						|
  virtual void SetUp()
 | 
						|
  {}
 | 
						|
 | 
						|
  virtual void TearDown()
 | 
						|
  {}
 | 
						|
 | 
						|
protected:
 | 
						|
};
 | 
						|
 | 
						|
#define PREPEND_ZEROS(char_size, offset, src_str, result_str) \
 | 
						|
  {                                                           \
 | 
						|
    memset(buf, 0, 1024);                                     \
 | 
						|
    memcpy(buf, src_str, char_size);                          \
 | 
						|
    ObMySQLUtil::prepend_zeros(buf, char_size, offset);       \
 | 
						|
    EXPECT_TRUE(0 == strcmp(buf, result_str));                \
 | 
						|
  }
 | 
						|
 | 
						|
#define INT_CELL_STR(val, is_unsigned, obj_type, type, zerofill, zflength, ERR)                                     \
 | 
						|
  {                                                                                                                 \
 | 
						|
    memset(buf, 0, 1024);                                                                                           \
 | 
						|
    pos = 0;                                                                                                        \
 | 
						|
    of_result << "value = " << val << ", is_unsigned = " << is_unsigned << ", zerofill = " << zerofill              \
 | 
						|
              << ", len = " << zflength << std::endl;                                                               \
 | 
						|
    ASSERT_EQ(ERR, ObMySQLUtil::int_cell_str(buf, len, val, obj_type, is_unsigned, type, pos, zerofill, zflength)); \
 | 
						|
    if (OB_SUCCESS == ERR) {                                                                                        \
 | 
						|
      of_result << buf << std::endl;                                                                                \
 | 
						|
    } else {                                                                                                        \
 | 
						|
      of_result << "err" << std::endl;                                                                              \
 | 
						|
    }                                                                                                               \
 | 
						|
  }
 | 
						|
 | 
						|
#define FLOAT_CELL_STR(val, type, scale, zerofill, zflength, ERR)                                             \
 | 
						|
  {                                                                                                           \
 | 
						|
    memset(buf, 0, 1024);                                                                                     \
 | 
						|
    pos = 0;                                                                                                  \
 | 
						|
    of_result << "value=" << val << ",scale=" << scale << ",zerofill=" << zerofill << ",zflength" << zflength \
 | 
						|
              << std::endl;                                                                                   \
 | 
						|
    ASSERT_EQ(ERR, ObMySQLUtil::float_cell_str(buf, len, val, type, pos, scale, zerofill, zflength));         \
 | 
						|
    if (OB_SUCCESS == ERR) {                                                                                  \
 | 
						|
      of_result << buf << std::endl;                                                                          \
 | 
						|
    } else {                                                                                                  \
 | 
						|
      of_result << "err" << std::endl;                                                                        \
 | 
						|
    }                                                                                                         \
 | 
						|
  }
 | 
						|
 | 
						|
#define DOUBLE_CELL_STR(val, type, scale, zerofill, zflength, ERR)                                            \
 | 
						|
  {                                                                                                           \
 | 
						|
    memset(buf, 0, 1024);                                                                                     \
 | 
						|
    pos = 0;                                                                                                  \
 | 
						|
    of_result << "value=" << val << ",scale=" << scale << ",zerofill=" << zerofill << ",zflength" << zflength \
 | 
						|
              << std::endl;                                                                                   \
 | 
						|
    ASSERT_EQ(ERR, ObMySQLUtil::double_cell_str(buf, len, val, type, pos, scale, zerofill, zflength));        \
 | 
						|
    if (OB_SUCCESS == ERR) {                                                                                  \
 | 
						|
      of_result << buf << std::endl;                                                                          \
 | 
						|
    } else {                                                                                                  \
 | 
						|
      of_result << "err" << std::endl;                                                                        \
 | 
						|
    }                                                                                                         \
 | 
						|
  }
 | 
						|
 | 
						|
#define DATETIME_CELL_STR(val_str, type, ERR)                                                      \
 | 
						|
  {                                                                                                \
 | 
						|
    memset(buf, 0, 1024);                                                                          \
 | 
						|
    pos = 0;                                                                                       \
 | 
						|
    of_result << "val = " << val_str << std::endl;                                                 \
 | 
						|
    int64_t val = 0;                                                                               \
 | 
						|
    ObTimeConvertCtx cvrt_ctx(&tz_info, true);                                                     \
 | 
						|
    const ObDataTypeCastParams dtc_params(&tz_info);                                               \
 | 
						|
    ASSERT_EQ(ERR, ObTimeConverter::str_to_datetime(val_str, cvrt_ctx, val, &scale));              \
 | 
						|
    if (ERR == OB_SUCCESS) {                                                                       \
 | 
						|
      ASSERT_EQ(ERR, ObMySQLUtil::datetime_cell_str(buf, len, val, type, pos, dtc_params, scale)); \
 | 
						|
      if (OB_SUCCESS == ERR) {                                                                     \
 | 
						|
        of_result << buf << std::endl;                                                             \
 | 
						|
      } else {                                                                                     \
 | 
						|
        of_result << "err" << std::endl;                                                           \
 | 
						|
      }                                                                                            \
 | 
						|
    } else {                                                                                       \
 | 
						|
      of_result << "invalid datetime" << std::endl;                                                \
 | 
						|
    }                                                                                              \
 | 
						|
  }
 | 
						|
 | 
						|
#define DATE_CELL_STR(val_str, type, ERR)                                   \
 | 
						|
  {                                                                         \
 | 
						|
    memset(buf, 0, 1024);                                                   \
 | 
						|
    pos = 0;                                                                \
 | 
						|
    of_result << "val = " << val_str << std::endl;                          \
 | 
						|
    int32_t val = 0;                                                        \
 | 
						|
    ASSERT_EQ(ERR, ObTimeConverter::str_to_date(val_str, val));             \
 | 
						|
    if (ERR == OB_SUCCESS) {                                                \
 | 
						|
      ASSERT_EQ(ERR, ObMySQLUtil::date_cell_str(buf, len, val, type, pos)); \
 | 
						|
      if (OB_SUCCESS == ERR) {                                              \
 | 
						|
        of_result << buf << " " << std::endl;                               \
 | 
						|
      } else {                                                              \
 | 
						|
        of_result << "err" << std::endl;                                    \
 | 
						|
      }                                                                     \
 | 
						|
    } else {                                                                \
 | 
						|
      of_result << "invalid date" << std::endl;                             \
 | 
						|
    }                                                                       \
 | 
						|
  }
 | 
						|
 | 
						|
#define TIME_CELL_STR(val_str, type, scale, ERR)                                   \
 | 
						|
  {                                                                                \
 | 
						|
    memset(buf, 0, 1024);                                                          \
 | 
						|
    pos = 0;                                                                       \
 | 
						|
    of_result << "val = " << val_str << "scale = " << scale << std::endl;          \
 | 
						|
    int64_t val = 0;                                                               \
 | 
						|
    ASSERT_EQ(ERR, ObTimeConverter::str_to_time(val_str, val, &scale));            \
 | 
						|
    if (OB_SUCCESS == ERR) {                                                       \
 | 
						|
      ASSERT_EQ(ERR, ObMySQLUtil::time_cell_str(buf, len, val, type, pos, scale)); \
 | 
						|
      if (OB_SUCCESS == ERR) {                                                     \
 | 
						|
        of_result << buf << std::endl;                                             \
 | 
						|
      } else {                                                                     \
 | 
						|
        of_result << "err" << std::endl;                                           \
 | 
						|
      }                                                                            \
 | 
						|
    } else {                                                                       \
 | 
						|
      of_result << "invalid time" << std::endl;                                    \
 | 
						|
    }                                                                              \
 | 
						|
  }
 | 
						|
 | 
						|
#define YEAR_CELL_STR(yval, type, ERR)                                      \
 | 
						|
  {                                                                         \
 | 
						|
    memset(buf, 0, 1024);                                                   \
 | 
						|
    pos = 0;                                                                \
 | 
						|
    of_result << "val = " << yval << std::endl;                             \
 | 
						|
    uint8_t val = 0;                                                        \
 | 
						|
    ASSERT_EQ(ERR, ObTimeConverter::int_to_year(yval, val));                \
 | 
						|
    if (OB_SUCCESS == ERR) {                                                \
 | 
						|
      ASSERT_EQ(ERR, ObMySQLUtil::year_cell_str(buf, len, val, type, pos)); \
 | 
						|
      if (OB_SUCCESS == ERR) {                                              \
 | 
						|
        of_result << buf << std::endl;                                      \
 | 
						|
      } else {                                                              \
 | 
						|
        of_result << "err" << std::endl;                                    \
 | 
						|
      }                                                                     \
 | 
						|
    } else {                                                                \
 | 
						|
      of_result << "invalid year" << std::endl;                             \
 | 
						|
    }                                                                       \
 | 
						|
  }
 | 
						|
 | 
						|
#define VARCHAR_CELL_STR(str, str_len, ERR)                                   \
 | 
						|
  {                                                                           \
 | 
						|
    memset(buf, 0, 1024);                                                     \
 | 
						|
    pos = 0;                                                                  \
 | 
						|
    of_result << "val = ";                                                    \
 | 
						|
    for (int i = 0; i < str_len; i++) {                                       \
 | 
						|
      of_result << str[i];                                                    \
 | 
						|
    }                                                                         \
 | 
						|
    of_result << std::endl;                                                   \
 | 
						|
    of_result << "str_len = " << str_len << std::endl;                        \
 | 
						|
    ObString val;                                                             \
 | 
						|
    val.assign_ptr(str, str_len);                                             \
 | 
						|
    ASSERT_EQ(ERR, ObMySQLUtil::varchar_cell_str(buf, len, val, false, pos)); \
 | 
						|
    if (OB_SUCCESS == ERR) {                                                  \
 | 
						|
      of_result << buf << std::endl;                                          \
 | 
						|
    } else {                                                                  \
 | 
						|
      of_result << "err" << std::endl;                                        \
 | 
						|
    }                                                                         \
 | 
						|
  }
 | 
						|
 | 
						|
#define NUMBER_CELL_STR(val, scale, zerofill, zflen, ERR)                                                             \
 | 
						|
  {                                                                                                                   \
 | 
						|
    memset(buf, 0, 1024);                                                                                             \
 | 
						|
    pos = 0;                                                                                                          \
 | 
						|
    of_result << "val" << val.format() << ", scale = " << scale << ", zerofill=" << zerofill << ", zflen = " << zflen \
 | 
						|
              << std::endl;                                                                                           \
 | 
						|
    ASSERT_EQ(OB_SUCCESS, ObMySQLUtil::number_cell_str(buf, len, val, pos, scale, zerofill, zflen));                  \
 | 
						|
    if (OB_SUCCESS == ERR) {                                                                                          \
 | 
						|
      of_result << buf << std::endl;                                                                                  \
 | 
						|
    } else {                                                                                                          \
 | 
						|
      of_result << "err" << std::endl;                                                                                \
 | 
						|
    }                                                                                                                 \
 | 
						|
  }
 | 
						|
 | 
						|
TEST_F(TestObMySQLUtil, TestPrependZero)
 | 
						|
{
 | 
						|
  char buf[1024];
 | 
						|
  // test func prepend_zeros
 | 
						|
  // char_size, offset, src_str, result_str
 | 
						|
  PREPEND_ZEROS(5, 3, "hello", "000hello");
 | 
						|
  PREPEND_ZEROS(5, 0, "hello", "hello");
 | 
						|
  PREPEND_ZEROS(5, 1, "hello", "0hello");
 | 
						|
  PREPEND_ZEROS(5, 2, "hello", "00hello");
 | 
						|
  PREPEND_ZEROS(0, 2, "hello", "00");
 | 
						|
  PREPEND_ZEROS(0, 5, "hello", "00000");
 | 
						|
  //
 | 
						|
  LOG_INFO("buf", K(ObString(buf)));
 | 
						|
}
 | 
						|
 | 
						|
TEST_F(TestObMySQLUtil, serialize_test)
 | 
						|
{
 | 
						|
  const char* tmp_file = "test_mysql_util.tmp";
 | 
						|
  const char* result_file = "test_mysql_util.result";
 | 
						|
  std::ofstream of_result(tmp_file);
 | 
						|
  ASSERT_TRUE(of_result.is_open());
 | 
						|
  char buf[1024];
 | 
						|
  char bitmap[1024];
 | 
						|
  memset(buf, 0, sizeof(buf));
 | 
						|
  memset(bitmap, 0, sizeof(bitmap));
 | 
						|
  int64_t len = 1024;
 | 
						|
  int64_t pos = 0;
 | 
						|
  // test func int_cell_str
 | 
						|
  int ret = OB_SUCCESS;
 | 
						|
  of_result << "***********INT_TEST***********" << std::endl;
 | 
						|
  INT_CELL_STR(10, true, ObUInt64Type, TEXT, true, 5, ret);
 | 
						|
  INT_CELL_STR(100, true, ObUInt64Type, TEXT, true, 5, ret);
 | 
						|
  INT_CELL_STR(100, false, ObIntType, TEXT, true, 5, ret);
 | 
						|
  INT_CELL_STR(100, true, ObUInt64Type, TEXT, false, 5, ret);
 | 
						|
  INT_CELL_STR(100, false, ObIntType, TEXT, false, 5, ret);
 | 
						|
  INT_CELL_STR(-100, false, ObIntType, TEXT, false, 5, ret);
 | 
						|
  INT_CELL_STR(-100, true, ObUInt64Type, TEXT, false, 5, ret);
 | 
						|
  INT_CELL_STR(-1, false, ObIntType, TEXT, true, 5, ret);
 | 
						|
  INT_CELL_STR(0, false, ObIntType, TEXT, false, 5, ret);
 | 
						|
  INT_CELL_STR(18446744073709551615ULL, false, ObIntType, TEXT, false, 5, ret);
 | 
						|
  INT_CELL_STR(18446744073709551615ULL, true, ObUInt64Type, TEXT, false, 5, ret);
 | 
						|
  INT_CELL_STR(18446744073709551615ULL, false, ObIntType, TEXT, true, 5, ret);
 | 
						|
  INT_CELL_STR(18446744073709551615ULL, true, ObUInt64Type, TEXT, true, 20, ret);
 | 
						|
  INT_CELL_STR(18446744073709551615ULL, true, ObUInt64Type, TEXT, true, 21, ret);
 | 
						|
  INT_CELL_STR(9223372036854775807LL, true, ObUInt64Type, TEXT, true, 20, ret);
 | 
						|
  INT_CELL_STR(9223372036854775807LL, false, ObIntType, TEXT, true, 20, ret);
 | 
						|
  INT_CELL_STR(9223372036854775807LL, false, ObIntType, TEXT, false, 20, ret);
 | 
						|
  INT_CELL_STR(-9223372036854775807LL, false, ObIntType, TEXT, false, 20, ret);
 | 
						|
  INT_CELL_STR(-9223372036854775807LL, true, ObUInt64Type, TEXT, false, 20, ret);
 | 
						|
  INT_CELL_STR(-9223372036854775807LL, false, ObIntType, TEXT, true, 20, ret);
 | 
						|
  INT_CELL_STR(-9223372036854775807LL, true, ObUInt64Type, TEXT, true, 20, ret);
 | 
						|
  INT_CELL_STR(0, false, ObIntType, TEXT, false, 10000, ret);
 | 
						|
 | 
						|
  // error
 | 
						|
  ret = OB_SIZE_OVERFLOW;
 | 
						|
  INT_CELL_STR(0, false, ObIntType, TEXT, true, 10000, ret);
 | 
						|
 | 
						|
  // test func float_cell_str
 | 
						|
  ret = OB_SUCCESS;
 | 
						|
  of_result << std::endl;
 | 
						|
  of_result << "***********FLOAT_TEST***********" << std::endl;
 | 
						|
  FLOAT_CELL_STR(1.5, TEXT, 1, true, 5, ret);
 | 
						|
  FLOAT_CELL_STR(1.5, TEXT, 1, true, 5, ret);
 | 
						|
  FLOAT_CELL_STR(1.5, TEXT, 10, true, 10, ret);
 | 
						|
  FLOAT_CELL_STR(1.50, TEXT, 10, true, 10, ret);
 | 
						|
  float fval = 1.0;
 | 
						|
  FLOAT_CELL_STR(fval, TEXT, 10, true, 10, ret);
 | 
						|
  FLOAT_CELL_STR(fval, TEXT, 20, true, 10, ret);
 | 
						|
  FLOAT_CELL_STR(fval, TEXT, 20, true, 20, ret);
 | 
						|
  FLOAT_CELL_STR(fval, TEXT, 10, true, 20, ret);
 | 
						|
  double dval = 1e-15;
 | 
						|
  of_result << std::endl;
 | 
						|
  of_result << "***********DOUBLE_TEST***********" << std::endl;
 | 
						|
  DOUBLE_CELL_STR(dval, TEXT, 10, true, 20, ret);
 | 
						|
  DOUBLE_CELL_STR(dval, TEXT, 20, true, 20, ret);
 | 
						|
  dval = 1e-16;
 | 
						|
  DOUBLE_CELL_STR(dval, TEXT, 10, true, 20, ret);
 | 
						|
  DOUBLE_CELL_STR(dval, TEXT, 20, true, 20, ret);
 | 
						|
  DOUBLE_CELL_STR(0, TEXT, 20, true, 20, ret);
 | 
						|
  DOUBLE_CELL_STR(1.555555555, TEXT, 10, true, 15, ret);
 | 
						|
  DOUBLE_CELL_STR(1.0000000005, TEXT, 5, true, 15, ret);
 | 
						|
  dval = 1.23e+12;
 | 
						|
  DOUBLE_CELL_STR(dval, TEXT, 5, true, 15, ret);
 | 
						|
  DOUBLE_CELL_STR(dval, TEXT, 5, true, 10, ret);
 | 
						|
  DOUBLE_CELL_STR(dval, TEXT, 0, true, 5, ret);
 | 
						|
  DOUBLE_CELL_STR(dval, TEXT, 0, false, 20, ret);
 | 
						|
  DOUBLE_CELL_STR(dval, TEXT, 3, true, 20, ret);
 | 
						|
  dval = 1.23333333333333333333e+10;
 | 
						|
  DOUBLE_CELL_STR(dval, TEXT, 5, false, 10, ret);
 | 
						|
  DOUBLE_CELL_STR(dval, TEXT, 5, true, 30, ret);
 | 
						|
  DOUBLE_CELL_STR(dval, TEXT, 20, true, 10, ret);
 | 
						|
  DOUBLE_CELL_STR(dval, TEXT, 20, true, 30, ret);
 | 
						|
  DOUBLE_CELL_STR(dval, TEXT, 20, false, 40, ret);
 | 
						|
  DOUBLE_CELL_STR(dval, TEXT, 20, true, 40, ret);
 | 
						|
 | 
						|
  // test func datetime_cell_str
 | 
						|
  ret = OB_SUCCESS;
 | 
						|
  ObTimeZoneInfo tz_info;
 | 
						|
  int16_t scale = 0;
 | 
						|
  of_result << std::endl;
 | 
						|
  of_result << "***********DATETIME_TEST***********" << std::endl;
 | 
						|
  DATETIME_CELL_STR("2011-01-01 12:01:01", TEXT, ret);
 | 
						|
  DATETIME_CELL_STR("2011-01-01 12:01:01.555", TEXT, ret);
 | 
						|
  DATETIME_CELL_STR("99991231235959", TEXT, ret);
 | 
						|
  DATETIME_CELL_STR("99991231235959.999999", TEXT, ret);
 | 
						|
  DATETIME_CELL_STR("0000-00-00 00:00:00", TEXT, ret);
 | 
						|
  DATETIME_CELL_STR("1000-01-01 00:00:00", TEXT, ret);
 | 
						|
  DATETIME_CELL_STR("2011-01-01 12:01:01.555555", TEXT, ret);
 | 
						|
  DATETIME_CELL_STR("2011-01-01 23:01:01.5", TEXT, ret);
 | 
						|
  DATETIME_CELL_STR("2011-12-01 12:01:01.55", TEXT, ret);
 | 
						|
 | 
						|
  // error
 | 
						|
  ret = OB_INVALID_DATE_VALUE;
 | 
						|
  DATETIME_CELL_STR("0000-00-00 00:00:00.555", TEXT, ret);
 | 
						|
  DATETIME_CELL_STR("2011-13-01 12:01:01.55", TEXT, ret);
 | 
						|
  ret = OB_DATETIME_FUNCTION_OVERFLOW;
 | 
						|
  DATETIME_CELL_STR("99991231235959.9999999", TEXT, ret);
 | 
						|
 | 
						|
  // test func date_cell_str  length=10 -> '\n'
 | 
						|
  ret = OB_SUCCESS;
 | 
						|
  of_result << std::endl;
 | 
						|
  of_result << "***********DATE_TEST***********" << std::endl;
 | 
						|
  DATE_CELL_STR("2011-01-01", TEXT, ret);
 | 
						|
  DATE_CELL_STR("2011-01-01", TEXT, ret);
 | 
						|
  DATE_CELL_STR("99991231", TEXT, ret);
 | 
						|
  DATE_CELL_STR("99990131", TEXT, ret);
 | 
						|
  DATE_CELL_STR("0000-00-00", TEXT, ret);
 | 
						|
  DATE_CELL_STR("1000-01-01", TEXT, ret);
 | 
						|
  DATE_CELL_STR("2011-01-01", TEXT, ret);
 | 
						|
  DATE_CELL_STR("2018-12-01", TEXT, ret);
 | 
						|
  DATE_CELL_STR("2011-12-01", TEXT, ret);
 | 
						|
 | 
						|
  // error
 | 
						|
  ret = OB_INVALID_DATE_VALUE;
 | 
						|
  DATE_CELL_STR("10000-00-00", TEXT, ret);
 | 
						|
  DATE_CELL_STR("99990132", TEXT, ret);
 | 
						|
  DATE_CELL_STR("2011-13-01", TEXT, ret);
 | 
						|
 | 
						|
  // test func time_cell_str
 | 
						|
  ret = OB_SUCCESS;
 | 
						|
  of_result << std::endl;
 | 
						|
  of_result << "***********TIME_TEST***********" << std::endl;
 | 
						|
  TIME_CELL_STR("23:00:00", TEXT, scale, ret);
 | 
						|
  TIME_CELL_STR("00:00:00", TEXT, scale, ret);
 | 
						|
  TIME_CELL_STR("12:00:00.1233", TEXT, scale, ret);
 | 
						|
  TIME_CELL_STR("12:00:00.123323232", TEXT, scale, ret);
 | 
						|
  TIME_CELL_STR("-838:59:59", TEXT, scale, ret);
 | 
						|
  TIME_CELL_STR("838:59:59", TEXT, scale, ret);
 | 
						|
 | 
						|
  // err
 | 
						|
  ret = OB_ERR_TRUNCATED_WRONG_VALUE;
 | 
						|
  TIME_CELL_STR("838:59:59.1212", TEXT, scale, ret);
 | 
						|
  TIME_CELL_STR("-838:59:59.1212", TEXT, scale, ret);
 | 
						|
  TIME_CELL_STR("839:59:59.9999999", TEXT, scale, ret);
 | 
						|
  TIME_CELL_STR("-839:59:59.9999999", TEXT, scale, ret);
 | 
						|
 | 
						|
  // test func year_cell_str
 | 
						|
  ret = OB_SUCCESS;
 | 
						|
  of_result << std::endl;
 | 
						|
  of_result << "***********YEAR_TEST***********" << std::endl;
 | 
						|
  YEAR_CELL_STR(2012, TEXT, ret);
 | 
						|
  YEAR_CELL_STR(0, TEXT, ret);
 | 
						|
  YEAR_CELL_STR(1901, TEXT, ret);
 | 
						|
  YEAR_CELL_STR(2155, TEXT, ret);
 | 
						|
  YEAR_CELL_STR(12, TEXT, ret);
 | 
						|
  YEAR_CELL_STR(69, TEXT, ret);
 | 
						|
  YEAR_CELL_STR(70, TEXT, ret);
 | 
						|
 | 
						|
  // err
 | 
						|
  ret = OB_DATA_OUT_OF_RANGE;
 | 
						|
  YEAR_CELL_STR(-1, TEXT, ret);
 | 
						|
 | 
						|
  // test func varchar_cell_str
 | 
						|
  ret = OB_SUCCESS;
 | 
						|
  of_result << std::endl;
 | 
						|
  of_result << "***********VARCHAR_TEST***********" << std::endl;
 | 
						|
  VARCHAR_CELL_STR("", 0, ret);
 | 
						|
  VARCHAR_CELL_STR("123", 3, ret);
 | 
						|
  VARCHAR_CELL_STR("adafa", 5, ret);
 | 
						|
  VARCHAR_CELL_STR("123abc", 6, ret);
 | 
						|
  char src_buf[2048];
 | 
						|
  memset(src_buf, 0, 2048);
 | 
						|
  for (int i = 0; i < 2048; i++) {
 | 
						|
    src_buf[i] = 'o';
 | 
						|
  }
 | 
						|
  // err
 | 
						|
  ret = OB_SIZE_OVERFLOW;
 | 
						|
  VARCHAR_CELL_STR(src_buf, 2048, ret);
 | 
						|
  VARCHAR_CELL_STR(src_buf, 1124, ret);
 | 
						|
  VARCHAR_CELL_STR(src_buf, 1024, ret);
 | 
						|
  VARCHAR_CELL_STR(src_buf, 1022, ret);
 | 
						|
  ret = OB_SUCCESS;
 | 
						|
  VARCHAR_CELL_STR(src_buf, 1020, ret);
 | 
						|
 | 
						|
  // test func number_cell_str
 | 
						|
  ret = OB_SUCCESS;
 | 
						|
  of_result << std::endl;
 | 
						|
  of_result << "***********NUMBER_TEST***********" << std::endl;
 | 
						|
  ObMalloc allocator;
 | 
						|
  number::ObNumber nmb;
 | 
						|
  nmb.from("123", allocator);
 | 
						|
  scale = 0;
 | 
						|
  NUMBER_CELL_STR(nmb, scale, true, 20, ret);
 | 
						|
  scale = 10;
 | 
						|
  NUMBER_CELL_STR(nmb, scale, true, 20, ret);
 | 
						|
  NUMBER_CELL_STR(nmb, scale, false, 20, ret);
 | 
						|
  scale = 20;
 | 
						|
  NUMBER_CELL_STR(nmb, scale, true, 20, ret);
 | 
						|
  number::ObNumber nmb1;
 | 
						|
  nmb1.from("", allocator);
 | 
						|
  scale = 0;
 | 
						|
  NUMBER_CELL_STR(nmb1, scale, true, 20, ret);
 | 
						|
  NUMBER_CELL_STR(nmb1, scale, false, 20, ret);
 | 
						|
  scale = 10;
 | 
						|
  NUMBER_CELL_STR(nmb1, scale, true, 20, ret);
 | 
						|
  scale = 20;
 | 
						|
  NUMBER_CELL_STR(nmb1, scale, true, 20, ret);
 | 
						|
  number::ObNumber nmb2;
 | 
						|
  nmb2.from("122323.23123123123123123131311", allocator);
 | 
						|
  scale = 0;
 | 
						|
  NUMBER_CELL_STR(nmb2, scale, true, 20, ret);
 | 
						|
  NUMBER_CELL_STR(nmb2, scale, false, 20, ret);
 | 
						|
  scale = 10;
 | 
						|
  NUMBER_CELL_STR(nmb2, scale, true, 20, ret);
 | 
						|
  scale = 20;
 | 
						|
  NUMBER_CELL_STR(nmb2, scale, true, 20, ret);
 | 
						|
 | 
						|
  std::ifstream if_result(tmp_file);
 | 
						|
  ASSERT_TRUE(if_result.is_open());
 | 
						|
  std::istream_iterator<std::string> it_result(if_result);
 | 
						|
  std::ifstream if_expected(result_file);
 | 
						|
  ASSERT_TRUE(if_expected.is_open());
 | 
						|
  std::istream_iterator<std::string> it_expected(if_expected);
 | 
						|
  ASSERT_TRUE(std::equal(it_result, std::istream_iterator<std::string>(), it_expected));
 | 
						|
  std::remove(tmp_file);
 | 
						|
  LOG_INFO("buf", K(ObString(buf)));
 | 
						|
}
 | 
						|
 | 
						|
int main(int argc, char* argv[])
 | 
						|
{
 | 
						|
  system("rm -rf test_mysql_util.log");
 | 
						|
  OB_LOGGER.set_log_level("INFO");
 | 
						|
  OB_LOGGER.set_file_name("test_mysql_util.log", true);
 | 
						|
  ::testing::InitGoogleTest(&argc, argv);
 | 
						|
  return RUN_ALL_TESTS();
 | 
						|
}
 |