196 lines
6.7 KiB
C++
196 lines
6.7 KiB
C++
// Licensed to the Apache Software Foundation (ASF) under one
|
|
// or more contributor license agreements. See the NOTICE file
|
|
// distributed with this work for additional information
|
|
// regarding copyright ownership. The ASF licenses this file
|
|
// to you under the Apache License, Version 2.0 (the
|
|
// "License"); you may not use this file except in compliance
|
|
// with the License. You may obtain a copy of the License at
|
|
//
|
|
// http://www.apache.org/licenses/LICENSE-2.0
|
|
//
|
|
// Unless required by applicable law or agreed to in writing,
|
|
// software distributed under the License is distributed on an
|
|
// "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
|
|
// KIND, either express or implied. See the License for the
|
|
// specific language governing permissions and limitations
|
|
// under the License.
|
|
|
|
#include "exprs/timestamp_functions.h"
|
|
|
|
#include <gtest/gtest.h>
|
|
#include <boost/scoped_ptr.hpp>
|
|
|
|
#include "testutil/function_utils.h"
|
|
#include "udf/udf.h"
|
|
#include "udf/udf_internal.h"
|
|
#include "runtime/runtime_state.h"
|
|
#include "runtime/exec_env.h"
|
|
#include "runtime/test_env.h"
|
|
|
|
|
|
namespace doris {
|
|
class FunctionContextImpl;
|
|
|
|
class TimestampFunctionsTest : public testing::Test {
|
|
public:
|
|
TimestampFunctionsTest() { }
|
|
|
|
void SetUp() {
|
|
TQueryGlobals globals;
|
|
globals.__set_now_string("2019-08-06 01:38:57");
|
|
globals.__set_timestamp_ms(1565080737805);
|
|
globals.__set_time_zone("America/Los_Angeles");
|
|
state = new RuntimeState(globals);
|
|
utils = new FunctionUtils(state);
|
|
ctx = utils->get_fn_ctx();
|
|
}
|
|
|
|
void TearDown() {
|
|
delete state;
|
|
delete utils;
|
|
}
|
|
|
|
private:
|
|
RuntimeState* state = nullptr;
|
|
FunctionUtils* utils = nullptr;
|
|
FunctionContext* ctx = nullptr;
|
|
};
|
|
|
|
TEST_F(TimestampFunctionsTest, day_of_week_test) {
|
|
doris_udf::FunctionContext *context = new doris_udf::FunctionContext();
|
|
|
|
doris_udf::DateTimeVal tv;
|
|
//2001-02-03 12:34:56
|
|
tv.packed_time = 1830650338932162560L;
|
|
tv.type = TIME_DATETIME;
|
|
|
|
ASSERT_EQ(7, TimestampFunctions::day_of_week(context, tv).val);
|
|
delete context;
|
|
}
|
|
|
|
TEST_F(TimestampFunctionsTest, time_diff_test) {
|
|
DateTimeValue dt1(20190718120000);
|
|
dt1.set_type(TIME_DATETIME);
|
|
doris_udf::DateTimeVal tv1;
|
|
dt1.to_datetime_val(&tv1);
|
|
|
|
DateTimeValue dt2(20190718130102);
|
|
dt2.set_type(TIME_DATETIME);
|
|
doris_udf::DateTimeVal tv2;
|
|
dt2.to_datetime_val(&tv2);
|
|
|
|
ASSERT_EQ(-3662, TimestampFunctions::time_diff(ctx, tv1, tv2).val);
|
|
}
|
|
|
|
TEST_F(TimestampFunctionsTest, now) {
|
|
DateTimeVal now = TimestampFunctions::now(ctx);
|
|
DateTimeValue dt = DateTimeValue::from_datetime_val(now);
|
|
ASSERT_EQ(20190806013857, dt.to_int64());
|
|
}
|
|
|
|
TEST_F(TimestampFunctionsTest, from_unix) {
|
|
IntVal unixtimestamp(1565080737);
|
|
StringVal sval = TimestampFunctions::from_unix(ctx, unixtimestamp);
|
|
ASSERT_EQ("2019-08-06 01:38:57", std::string((char*) sval.ptr, sval.len));
|
|
|
|
IntVal unixtimestamp2(-123);
|
|
sval = TimestampFunctions::from_unix(ctx, unixtimestamp2);
|
|
ASSERT_TRUE(sval.is_null);
|
|
}
|
|
|
|
TEST_F(TimestampFunctionsTest, to_unix) {
|
|
DateTimeVal dt_val ;
|
|
dt_val.packed_time = 1847544683002068992;
|
|
dt_val.type = TIME_DATETIME;
|
|
ASSERT_EQ(1565080737, TimestampFunctions::to_unix(ctx).val);
|
|
ASSERT_EQ(1565080737, TimestampFunctions::to_unix(ctx, dt_val).val);
|
|
ASSERT_EQ(1565080737, TimestampFunctions::to_unix(ctx, StringVal("2019-08-06 01:38:57"), "%Y-%m-%d %H:%i:%S").val);
|
|
|
|
DateTimeValue dt_value;
|
|
dt_value.from_date_int64(99991230);
|
|
dt_value.to_datetime_val(&dt_val);
|
|
ASSERT_EQ(0, TimestampFunctions::to_unix(ctx, dt_val).val);
|
|
|
|
dt_value.from_date_int64(10000101);
|
|
dt_value.to_datetime_val(&dt_val);
|
|
ASSERT_EQ(0, TimestampFunctions::to_unix(ctx, dt_val).val);
|
|
}
|
|
|
|
TEST_F(TimestampFunctionsTest, curtime) {
|
|
ASSERT_EQ(3600 + 38*60 + 57, TimestampFunctions::curtime(ctx).val);
|
|
}
|
|
|
|
TEST_F(TimestampFunctionsTest, convert_tz_test) {
|
|
doris_udf::FunctionContext *context = new doris_udf::FunctionContext();
|
|
DateTimeValue dt1(20190806163857);
|
|
dt1.set_type(TIME_DATETIME);
|
|
doris_udf::DateTimeVal tv1;
|
|
dt1.to_datetime_val(&tv1);
|
|
DateTimeVal t = TimestampFunctions::convert_tz(context, tv1, StringVal("Asia/Shanghai"), StringVal("America/Los_Angeles"));
|
|
DateTimeValue dt2 = DateTimeValue::from_datetime_val(t);
|
|
ASSERT_EQ(20190806013857, dt2.to_int64());
|
|
|
|
t = TimestampFunctions::convert_tz(context, tv1, StringVal("CST"), StringVal("America/Los_Angeles"));
|
|
DateTimeValue dt3 = DateTimeValue::from_datetime_val(t);
|
|
ASSERT_EQ(20190806013857, dt3.to_int64());
|
|
delete context;
|
|
}
|
|
|
|
TEST_F(TimestampFunctionsTest, timestampdiff_test) {
|
|
doris_udf::FunctionContext *context = new doris_udf::FunctionContext();
|
|
DateTimeValue dt1(20120824000001);
|
|
doris_udf::DateTimeVal tv1;
|
|
dt1.to_datetime_val(&tv1);
|
|
DateTimeValue dt2(20120830000000);
|
|
doris_udf::DateTimeVal tv2;
|
|
dt2.to_datetime_val(&tv2);
|
|
|
|
//YEAR
|
|
ASSERT_EQ(0, TimestampFunctions::years_diff(context, tv2, tv1).val);
|
|
DateTimeValue dt_year(20100930000000);
|
|
doris_udf::DateTimeVal tv_year;
|
|
dt_year.to_datetime_val(&tv_year);
|
|
ASSERT_EQ(-1, TimestampFunctions::years_diff(context, tv_year, tv1).val);
|
|
//MONTH
|
|
ASSERT_EQ(0, TimestampFunctions::months_diff(context, tv2, tv1).val);
|
|
DateTimeValue dt3(20120924000000);
|
|
doris_udf::DateTimeVal tv3;
|
|
dt3.to_datetime_val(&tv3);
|
|
ASSERT_EQ(0, TimestampFunctions::months_diff(context, tv3, tv1).val);
|
|
DateTimeValue dt_month(20120631000000);
|
|
doris_udf::DateTimeVal tv_month;
|
|
dt_month.to_datetime_val(&tv_month);
|
|
ASSERT_EQ(-1, TimestampFunctions::months_diff(context, tv_month, tv1).val);
|
|
//WEEK
|
|
ASSERT_EQ(0, TimestampFunctions::weeks_diff(context, tv2, tv1).val);
|
|
//DAY
|
|
ASSERT_EQ(5, TimestampFunctions::days_diff(context, tv2, tv1).val);
|
|
DateTimeValue dt4(20120830000001);
|
|
doris_udf::DateTimeVal tv4;
|
|
dt4.to_datetime_val(&tv4);
|
|
ASSERT_EQ(6, TimestampFunctions::days_diff(context, tv4, tv1).val);
|
|
DateTimeValue dt5(20120901000001);
|
|
doris_udf::DateTimeVal tv5;
|
|
dt5.to_datetime_val(&tv5);
|
|
ASSERT_EQ(8, TimestampFunctions::days_diff(context, tv5, tv1).val);
|
|
|
|
DateTimeValue dt_day(20120823000005);
|
|
doris_udf::DateTimeVal tv_day;
|
|
dt_day.to_datetime_val(&tv_day);
|
|
ASSERT_EQ(0, TimestampFunctions::days_diff(context, tv_day, tv1).val);
|
|
|
|
//HOUR
|
|
ASSERT_EQ(143, TimestampFunctions::hours_diff(context, tv2, tv1).val);
|
|
//MINUTE
|
|
ASSERT_EQ(8639, TimestampFunctions::minutes_diff(context, tv2, tv1).val);
|
|
//SECOND
|
|
ASSERT_EQ(518399, TimestampFunctions::seconds_diff(context, tv2, tv1).val);
|
|
delete context;
|
|
}
|
|
|
|
}
|
|
int main(int argc, char** argv) {
|
|
::testing::InitGoogleTest(&argc, argv);
|
|
return RUN_ALL_TESTS();
|
|
}
|