oceanbase/unittest/sql/engine/aggregate/test_hash_groupby2.cpp
obdev b6773084c6 [FEAT MERGE] impl vectorization 2.0
Co-authored-by: Naynahs <cfzy002@126.com>
Co-authored-by: hwx65 <1780011298@qq.com>
Co-authored-by: oceanoverflow <oceanoverflow@gmail.com>
2023-12-22 03:43:22 +00:00

135 lines
4.1 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 SQL_ENGINE
#define USING_LOG_PREFIX COMMON
#include <iterator>
#include <gtest/gtest.h>
#include "../test_op_engine.h"
#include "../ob_test_config.h"
#include <vector>
#include <string>
using namespace ::oceanbase::sql;
namespace test
{
class TestHashGroupByVec : public TestOpEngine
{
public:
TestHashGroupByVec();
virtual ~TestHashGroupByVec();
virtual void SetUp();
virtual void TearDown();
private:
// disallow copy
DISALLOW_COPY_AND_ASSIGN(TestHashGroupByVec);
protected:
// function members
protected:
// data members
};
TestHashGroupByVec::TestHashGroupByVec()
{
std::string schema_filename = ObTestOpConfig::get_instance().test_filename_prefix_ + ".schema";
strcpy(schema_file_path_, schema_filename.c_str());
}
TestHashGroupByVec::~TestHashGroupByVec()
{}
void TestHashGroupByVec::SetUp()
{
TestOpEngine::SetUp();
}
void TestHashGroupByVec::TearDown()
{
destroy();
}
TEST_F(TestHashGroupByVec, basic_test)
{
std::string test_file_path = ObTestOpConfig::get_instance().test_filename_prefix_ + ".test";
int ret = basic_random_test(test_file_path);
EXPECT_EQ(ret, 0);
}
// TEST_F(TestHashGroupByVec, basic_test2)
// {
// int ret = OB_SUCCESS;
// std::string test_file_path = ObTestOpConfig::get_instance().test_filename_prefix_ + ".test";
// if(OB_FAIL(basic_random_test_output_to_file(test_file_path, true))) {
// LOG_ERROR("Some error occur in running vectorization 2.0 operator", K(ret));
// } else if (OB_FAIL(basic_random_test_output_to_file(test_file_path, false))) {
// LOG_ERROR("Some error occur in running original operator", K(ret));
// }
// EXPECT_EQ(ret, 0);
// }
// TEST_F(TestHashGroupByVec, your_own_test)
// {
// std::string test_file_path = ObTestOpConfig::get_instance().test_filename_prefix_ + ".test";
// std::ifstream if_tests(test_file_path);
// if (if_tests.is_open() == false) { return; }
// std::string line;
// while (std::getline(if_tests, line)) {
// // handle query
// if (line.size() <= 0) continue;
// if (line.at(0) == '#') continue;
// ObOperator *root = NULL;
// ObExecutor exector;
// if (OB_FAIL(get_tested_op_from_string(line, false, root, exector))) {
// LOG_WARN("generate tested op fail, sql: ", K(line.data()));
// } else {
// int round = 1;
// const int64_t max_row_cnt = 256;
// const ObBatchRows *child_brs = nullptr;
// LOG_INFO("============== Final output ===============", K(round));
// while (!root->brs_.end_) {
// if (OB_FAIL(root->get_next_batch(max_row_cnt, child_brs))) {
// LOG_ERROR("root op fail to get_next_batch data", K(original_root));
// break;
// }
// }
// }
// }
// }
} // namespace test
int main(int argc, char **argv)
{
ObTestOpConfig::get_instance().test_filename_prefix_ = "test_hash_groupby2";
for (int i = 1; i < argc; i++) {
if (strcmp(argv[i], "-bg") == 0) {
ObTestOpConfig::get_instance().test_filename_prefix_ += "_bg";
ObTestOpConfig::get_instance().run_in_background_ = true;
}
}
ObTestOpConfig::get_instance().init();
system(("rm -f " + ObTestOpConfig::get_instance().test_filename_prefix_ + ".log").data());
system(("rm -f " + ObTestOpConfig::get_instance().test_filename_prefix_ + ".log.*").data());
oceanbase::common::ObClockGenerator::init();
observer::ObReqTimeGuard req_timeinfo_guard;
OB_LOGGER.set_log_level("INFO");
OB_LOGGER.set_file_name((ObTestOpConfig::get_instance().test_filename_prefix_ + ".log").data(), true);
init_sql_factories();
::testing::InitGoogleTest(&argc, argv);
return RUN_ALL_TESTS();
}