mirror of
https://github.com/mariadb-corporation/mariadb-columnstore-engine.git
synced 2025-04-23 07:05:36 +03:00
369 lines
11 KiB
C++
369 lines
11 KiB
C++
/* Copyright (C) 2014 InfiniDB, Inc.
|
|
|
|
This program is free software; you can redistribute it and/or
|
|
modify it under the terms of the GNU General Public License
|
|
as published by the Free Software Foundation; version 2 of
|
|
the License.
|
|
|
|
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. See the
|
|
GNU General Public License for more details.
|
|
|
|
You should have received a copy of the GNU General Public License
|
|
along with this program; if not, write to the Free Software
|
|
Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston,
|
|
MA 02110-1301, USA. */
|
|
|
|
/**
|
|
* $Id$
|
|
*/
|
|
|
|
#include <iostream>
|
|
#include <string>
|
|
using namespace std;
|
|
|
|
#include <cppunit/extensions/HelperMacros.h>
|
|
|
|
#include "dataconvert.h"
|
|
#include "funchelpers.h"
|
|
using namespace funcexp;
|
|
using namespace dataconvert;
|
|
#define MAX_DAY_NUMBER 3652424L
|
|
|
|
#include "functioncolumn.h"
|
|
#include "intervalcolumn.h"
|
|
using namespace execplan;
|
|
|
|
#include "timeextract.h"
|
|
|
|
class FuncExpTest : public CppUnit::TestFixture
|
|
{
|
|
CPPUNIT_TEST_SUITE(FuncExpTest);
|
|
|
|
CPPUNIT_TEST(fe_getnumbertest);
|
|
CPPUNIT_TEST(fe_strtodatetest);
|
|
CPPUNIT_TEST(fe_dateaddtest);
|
|
CPPUNIT_TEST(fe_daynametest);
|
|
CPPUNIT_TEST(fe_fromunixtimetest);
|
|
CPPUNIT_TEST_SUITE_END();
|
|
|
|
private:
|
|
public:
|
|
void setUp()
|
|
{
|
|
}
|
|
|
|
void tearDown()
|
|
{
|
|
}
|
|
|
|
void show_datetime_debugs(const string& nm, const DateTime& d1)
|
|
{
|
|
cout << nm << ".day = " << setw(10) << d1.day << endl;
|
|
cout << nm << ".month = " << setw(10) << d1.month << endl;
|
|
cout << nm << ".year = " << setw(10) << d1.year << endl;
|
|
cout << nm << ".hour = " << setw(10) << d1.hour << endl;
|
|
cout << nm << ".minute = " << setw(10) << d1.minute << endl;
|
|
cout << nm << ".second = " << setw(10) << d1.second << endl;
|
|
cout << nm << ".msecond = " << setw(10) << d1.msecond << endl;
|
|
}
|
|
|
|
void fe_strtodatetest()
|
|
{
|
|
struct DateCheck
|
|
{
|
|
const char* inputstr;
|
|
const char* formatstr;
|
|
DateTime date;
|
|
};
|
|
|
|
DateCheck date_tests[] = {
|
|
{"2009", "%Y", DateTime(2009, 0, 0, 0, 0, 0, 0)},
|
|
{" 2009", "%Y", DateTime(2009, 0, 0, 0, 0, 0, 0)},
|
|
{" 2009", " %Y", DateTime(2009, 0, 0, 0, 0, 0, 0)},
|
|
{"jan", "%b", DateTime(0, 1, 0, 0, 0, 0, 0)},
|
|
{"2009feb", "%Y%b", DateTime(2009, 2, 0, 0, 0, 0, 0)},
|
|
{" 2009 ApR", " %Y %b", DateTime(2009, 4, 0, 0, 0, 0, 0)},
|
|
{"200910", "%Y%m", DateTime(2009, 10, 0, 0, 0, 0, 0)},
|
|
{" 2009 10", " %Y %m", DateTime(2009, 10, 0, 0, 0, 0, 0)},
|
|
{"200910", "%Y%c", DateTime(2009, 10, 0, 0, 0, 0, 0)},
|
|
{" 2009 10", " %Y %c", DateTime(2009, 10, 0, 0, 0, 0, 0)},
|
|
{"01,5,2013", "%d,%m,%Y", DateTime(2013, 5, 1, 0, 0, 0, 0)},
|
|
{"01,5,2013", "%e,%m,%Y", DateTime(2013, 5, 1, 0, 0, 0, 0)},
|
|
{"7,5th,2012", "%m,%D,%Y", DateTime(2012, 7, 5, 0, 0, 0, 0)},
|
|
{"a09:30:17", "a%h:%i", DateTime(0, 0, 0, 9, 30, 0, 0)},
|
|
{"a09:30:17", "%h:%i", DateTime(0, 0, 0, 0, 0, 0, 0)},
|
|
{"9:30.170", "%h:%i.%f", DateTime(0, 0, 0, 9, 30, 0, 170000)},
|
|
{"178546 17", "%f %H", DateTime(0, 0, 0, 17, 0, 0, 178546)},
|
|
{"2009march415", "%Y%b", DateTime(2009, 3, 0, 0, 0, 0, 0)},
|
|
{" 2009 July", " %Y %b", DateTime(2009, 7, 0, 0, 0, 0, 0)},
|
|
{"09 pm", "%h %p", DateTime(0, 0, 0, 21, 0, 0, 0)},
|
|
{"09:13:14pm", "%r", DateTime(0, 0, 0, 9, 13, 14, 0)},
|
|
{"12:13:14am", "%r", DateTime(0, 0, 0, 12, 13, 14, 0)},
|
|
{"92", " %Y", DateTime(1992, 0, 0, 0, 0, 0, 0)},
|
|
{"9", " %Y", DateTime(2009, 0, 0, 0, 0, 0, 0)},
|
|
{"2013 31", "%Y %j", DateTime(2013, 1, 31, 0, 0, 0, 0)},
|
|
{"2013 1 3", "%Y %U %w", DateTime(2013, 1, 9, 0, 0, 0, 0)},
|
|
{"2013 1 3", "%Y %u %w", DateTime(2013, 1, 9, 0, 0, 0, 0)},
|
|
{"2007 1 3", "%Y %U %w", DateTime(2007, 1, 10, 0, 0, 0, 0)},
|
|
{"2007 1 3", "%Y %u %w", DateTime(2007, 1, 3, 0, 0, 0, 0)},
|
|
{"2013 1 3", "%X %V %w", DateTime(2013, 1, 9, 0, 0, 0, 0)},
|
|
{"2013 1 3", "%X %v %w", DateTime(2013, 1, 9, 0, 0, 0, 0)},
|
|
{"2007 1 3", "%X %V %w", DateTime(2007, 1, 10, 0, 0, 0, 0)},
|
|
{"2007 1 3", "%X %v %w", DateTime(2007, 1, 3, 0, 0, 0, 0)},
|
|
{"33:30.170", "%h:%i.%f", DateTime(0, 0, 0, 0, 0, 0, 0)},
|
|
{"01,25,2013", "%d,%m,%Y", DateTime(0, 0, 0, 0, 0, 0, 0)},
|
|
{"%09:30:17", "%%%h:%i:%s", DateTime(0, 0, 0, 9, 30, 17, 0)},
|
|
{"2009 foobar", "%Y", DateTime(2009, 0, 0, 0, 0, 0, 0)},
|
|
{"2009-12-31", "%h:%i:%s %p", DateTime(0, 0, 0, 0, 0, 0, 0)},
|
|
{"12:00:01 AM", "%h:%i:%s %p", DateTime(0, 0, 0, 0, 0, 1, 0)},
|
|
{"12:00:01 PM", "%h:%i:%s %p", DateTime(0, 0, 0, 12, 0, 1, 0)},
|
|
{"11:00:01 PM", "%h:%i:%s %p", DateTime(0, 0, 0, 23, 0, 1, 0)},
|
|
{"11:00:01 AM", "%h:%i:%s %p", DateTime(0, 0, 0, 11, 0, 1, 0)},
|
|
{"10000-01-02", "%Y:%m:%d", DateTime(0, 0, 0, 0, 0, 0, 0)},
|
|
};
|
|
|
|
for (unsigned i = 0; i < sizeof(date_tests) / sizeof(DateCheck); i++)
|
|
{
|
|
DateTime dt;
|
|
dt.year = 0;
|
|
dt.month = 0;
|
|
dt.day = 0;
|
|
dt.hour = 0;
|
|
dt.minute = 0;
|
|
dt.second = 0;
|
|
dt.msecond = 0;
|
|
|
|
TimeExtractor extractor;
|
|
int ret = extractor.extractTime(date_tests[i].inputstr, date_tests[i].formatstr, dt);
|
|
|
|
if (ret != 0)
|
|
cout << "Extractor reported error for " << date_tests[i].inputstr << "," << date_tests[i].formatstr
|
|
<< endl;
|
|
|
|
bool check =
|
|
((*(reinterpret_cast<uint64_t*>(&date_tests[i].date))) == ((*(reinterpret_cast<uint64_t*>(&dt)))));
|
|
|
|
if (!check)
|
|
{
|
|
printf("For input \"%s\", format \"%s\", check 0x%016lx vs 0x%016lx\n", date_tests[i].inputstr,
|
|
date_tests[i].formatstr, *(reinterpret_cast<uint64_t*>(&dt)),
|
|
*(reinterpret_cast<uint64_t*>(&date_tests[i].date)));
|
|
show_datetime_debugs("check", dt);
|
|
show_datetime_debugs("ref", date_tests[i].date);
|
|
}
|
|
|
|
CPPUNIT_ASSERT(check);
|
|
}
|
|
}
|
|
|
|
void fe_getnumbertest()
|
|
{
|
|
const int MAX_VALS = 5;
|
|
struct Check
|
|
{
|
|
const char* str;
|
|
int numvals;
|
|
int vals[MAX_VALS];
|
|
};
|
|
|
|
Check tests[] = {
|
|
{" ", 0, {}},
|
|
{"0", 1, {0}},
|
|
{"0.0", 2, {0, 0}},
|
|
{"0:0.0", 3, {0, 0, 0}},
|
|
{"-0", 1, {0}},
|
|
{"0000-000", 1, {0}},
|
|
{"-2", 1, {-2}},
|
|
{"2223", 1, {2223}},
|
|
{"37", 1, {37}},
|
|
{"393", 1, {393}},
|
|
{"39-3", 1, {39}},
|
|
{"17-", 1, {17}},
|
|
{"10.2303", 2, {10, 2303}},
|
|
{"20:10.2303", 3, {20, 10, 2303}},
|
|
{"17:37", 2, {17, 37}},
|
|
};
|
|
|
|
int a1[MAX_VALS] = {0xdeadbeef};
|
|
|
|
for (unsigned i = 0; i < sizeof(tests) / sizeof(Check); i++)
|
|
{
|
|
int rc1 = helpers::getNumbers(tests[i].str, a1, execplan::OP_ADD);
|
|
|
|
cout << "For input \"" << tests[i].str << "\", numbers = " << rc1 << ",[";
|
|
|
|
bool check = true;
|
|
|
|
if (rc1 != tests[i].numvals)
|
|
check = false;
|
|
|
|
for (int j = 0; j < rc1; ++j)
|
|
{
|
|
if (j > 0)
|
|
cout << ',';
|
|
|
|
cout << a1[j];
|
|
|
|
if (a1[j] != tests[i].vals[j])
|
|
{
|
|
check = false;
|
|
}
|
|
}
|
|
|
|
cout << ']' << endl;
|
|
|
|
CPPUNIT_ASSERT(check);
|
|
}
|
|
}
|
|
|
|
void fe_dateaddtest()
|
|
{
|
|
struct DateCheck
|
|
{
|
|
const char* intervalstr;
|
|
IntervalColumn::interval_type unit;
|
|
OpType funcType;
|
|
DateTime date;
|
|
DateTime ref;
|
|
};
|
|
|
|
DateCheck date_tests[] = {
|
|
{"111111:22", IntervalColumn::INTERVAL_YEAR_MONTH, execplan::OP_ADD,
|
|
DateTime(1997, 01, 01, 0, 0, 0, 0), DateTime(0, 0, 0, 0, 0, 0, 0)},
|
|
{"-30", IntervalColumn::INTERVAL_DAY, execplan::OP_ADD, DateTime(2013, 3, 1, 0, 0, 0, 0),
|
|
DateTime(2013, 1, 30, 0, 0, 0, 0)},
|
|
};
|
|
|
|
for (unsigned i = 0; i < sizeof(date_tests) / sizeof(DateCheck); i++)
|
|
{
|
|
DateTime dt;
|
|
|
|
uint64_t timeval = *(reinterpret_cast<uint64_t*>(&date_tests[i].date));
|
|
uint64_t value = helpers::dateAdd(timeval, date_tests[i].intervalstr, date_tests[i].unit, false,
|
|
date_tests[i].funcType);
|
|
|
|
*(reinterpret_cast<uint64_t*>(&dt)) = value;
|
|
|
|
bool check = (value == *(reinterpret_cast<uint64_t*>(&date_tests[i].ref)));
|
|
|
|
if (!check)
|
|
{
|
|
printf("For interval \"%s\", check 0x%016lx vs 0x%016lx\n", date_tests[i].intervalstr,
|
|
*(reinterpret_cast<uint64_t*>(&dt)), *(reinterpret_cast<uint64_t*>(&date_tests[i].ref)));
|
|
show_datetime_debugs("check", dt);
|
|
show_datetime_debugs("ref", date_tests[i].ref);
|
|
}
|
|
|
|
CPPUNIT_ASSERT(check);
|
|
}
|
|
}
|
|
|
|
void fe_daynametest()
|
|
{
|
|
struct DateCheck
|
|
{
|
|
DateTime date;
|
|
const char* dayname;
|
|
};
|
|
|
|
DateCheck date_tests[] = {
|
|
{DateTime(1997, 01, 01, 0, 0, 0, 0), "Wednesday"},
|
|
};
|
|
|
|
for (unsigned i = 0; i < sizeof(date_tests) / sizeof(DateCheck); i++)
|
|
{
|
|
boost::gregorian::date d(date_tests[i].date.year, date_tests[i].date.month, date_tests[i].date.day);
|
|
uint32_t dayofweek = helpers::calc_mysql_weekday(date_tests[i].date.year, date_tests[i].date.month,
|
|
date_tests[i].date.day, false);
|
|
|
|
bool check = (strcmp(helpers::weekdayFullNames[dayofweek].c_str(), date_tests[i].dayname) == 0);
|
|
|
|
if (!check)
|
|
{
|
|
printf("For date %s, check %s vs %s\n", to_simple_string(d).c_str(),
|
|
helpers::weekdayFullNames[dayofweek].c_str(), date_tests[i].dayname);
|
|
}
|
|
|
|
CPPUNIT_ASSERT(check);
|
|
}
|
|
}
|
|
|
|
DateTime getDateTime(int64_t val)
|
|
{
|
|
if (val < 0 || val > helpers::TIMESTAMP_MAX_VALUE)
|
|
return 0;
|
|
|
|
DateTime dt;
|
|
|
|
struct tm tmp_tm;
|
|
time_t tmp_t = (time_t)val;
|
|
localtime_r(&tmp_t, &tmp_tm);
|
|
|
|
// to->neg=0;
|
|
dt.year = (int64_t)((tmp_tm.tm_year + 1900) % 10000);
|
|
dt.month = (int64_t)tmp_tm.tm_mon + 1;
|
|
dt.day = (int64_t)tmp_tm.tm_mday;
|
|
dt.hour = (int64_t)tmp_tm.tm_hour;
|
|
dt.minute = (int64_t)tmp_tm.tm_min;
|
|
dt.second = (int64_t)tmp_tm.tm_sec;
|
|
dt.msecond = 0;
|
|
return dt;
|
|
}
|
|
|
|
void fe_fromunixtimetest()
|
|
{
|
|
struct DateCheck
|
|
{
|
|
int64_t ts;
|
|
const char* formatstr;
|
|
const char* refstr;
|
|
};
|
|
|
|
DateCheck date_tests[] = {
|
|
{-26, "%H:%i:%s", "NULL"},
|
|
{26, "%H:%i:%s", "18:00:26"},
|
|
{0, "%H:%i:%s", "18:00:00"},
|
|
};
|
|
|
|
for (unsigned i = 0; i < sizeof(date_tests) / sizeof(DateCheck); i++)
|
|
{
|
|
bool check = false;
|
|
string ret;
|
|
DateTime dt = getDateTime(date_tests[i].ts);
|
|
|
|
if (*reinterpret_cast<int64_t*>(&dt) == 0)
|
|
{
|
|
check = (strcmp(date_tests[i].refstr, "NULL") == 0);
|
|
}
|
|
else
|
|
{
|
|
ret = helpers::IDB_date_format(dt, date_tests[i].formatstr);
|
|
check = (strcmp(date_tests[i].refstr, ret.c_str()) == 0);
|
|
}
|
|
|
|
if (!check)
|
|
{
|
|
printf("For timestamp %ld, check %s vs %s\n", date_tests[i].ts, ret.c_str(), date_tests[i].refstr);
|
|
}
|
|
|
|
CPPUNIT_ASSERT(check);
|
|
}
|
|
}
|
|
};
|
|
|
|
CPPUNIT_TEST_SUITE_REGISTRATION(FuncExpTest);
|
|
|
|
#include <cppunit/extensions/TestFactoryRegistry.h>
|
|
#include <cppunit/ui/text/TestRunner.h>
|
|
|
|
int main(int argc, char** argv)
|
|
{
|
|
CppUnit::TextUi::TestRunner runner;
|
|
CppUnit::TestFactoryRegistry& registry = CppUnit::TestFactoryRegistry::getRegistry();
|
|
runner.addTest(registry.makeTest());
|
|
bool wasSuccessful = runner.run("", false);
|
|
return (wasSuccessful ? 0 : 1);
|
|
}
|