 0d0b11c0cc
			
		
	
	0d0b11c0cc
	
	
	
		
			
			Co-authored-by: oceanoverflow <oceanoverflow@gmail.com> Co-authored-by: hezuojiao <hezuojiao@gmail.com> Co-authored-by: Monk-Liu <1152761042@qq.com>
		
			
				
	
	
		
			135 lines
		
	
	
		
			4.1 KiB
		
	
	
	
		
			C++
		
	
	
	
	
	
			
		
		
	
	
			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();
 | |
| } |