543 lines
		
	
	
		
			18 KiB
		
	
	
	
		
			C++
		
	
	
	
	
	
			
		
		
	
	
			543 lines
		
	
	
		
			18 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.
 | |
|  */
 | |
| 
 | |
| #include <gtest/gtest.h>
 | |
| #include "storage/access/mock_ob_ss_store.h"
 | |
| #include "storage/access/ob_multiple_scan_merge.h"
 | |
| #include "lib/allocator/ob_concurrent_fifo_allocator.h"
 | |
| #include "lib/allocator/page_arena.h"
 | |
| #include "storage/access/ob_table_scan_iterator.h"
 | |
| #include "mockcontainer/mock_ob_iterator.h"
 | |
| #include "mockcontainer/mock_misc_for_ps_test.h"
 | |
| 
 | |
| namespace oceanbase
 | |
| {
 | |
| using namespace common;
 | |
| using namespace storage;
 | |
| using namespace blocksstable;
 | |
| using namespace::testing;
 | |
| 
 | |
| namespace unittest
 | |
| {
 | |
| class ObMultipleScanMergeTest : public ::testing::Test
 | |
| {
 | |
| public:
 | |
|   const static int OUTPUT_COL_NUM = 4;
 | |
|   const static int ITERATOR_NUM = 4;
 | |
|   const static int ROWKEY_NUM = 1;
 | |
|   const static int CELL_NUM = 4;
 | |
|   const static int INVALID_ROWKEY_NUM = -1;
 | |
|   const static int INVALID_CELL_NUM = -1;
 | |
|   const static int INVALID_COL_INDEX = -1;
 | |
| 
 | |
| public:
 | |
|   ObMultipleScanMergeTest() : allocator_(ObModIds::OB_ST_TEMP) {}
 | |
|   void prepare_input(const ObIArray<const char *> &inputs, ObArray<ObMockStoreRowIterator *> &iters);
 | |
|   void prepare_store(const ObIArray<ObMockStoreRowIterator *> &iters);
 | |
|   void check_result(const char *output);
 | |
|   void SetUp()
 | |
|   {
 | |
|     for (int16_t i = 0; i < OUTPUT_COL_NUM; ++i) {
 | |
|       ASSERT_EQ(OB_SUCCESS, out_idxs_.push_back(i));
 | |
|     }
 | |
|     main_table_param_.reset();
 | |
|     ASSERT_EQ(OB_SUCCESS, main_table_param_.out_col_desc_param_.init());
 | |
|     main_table_param_.iter_param_.rowkey_cnt_ = ROWKEY_NUM;
 | |
|     main_table_param_.iter_param_.output_project_ = &out_idxs_;
 | |
|     main_table_param_.iter_param_.out_cols_ = &main_table_param_.out_col_desc_param_.get_col_descs();
 | |
|     ObColDesc col_desc;
 | |
|     col_desc.col_id_ = 1;
 | |
|     col_desc.col_type_ = ObObjMeta();
 | |
|     ASSERT_EQ(OB_SUCCESS, main_table_param_.out_col_desc_param_.push_back(col_desc));
 | |
| 
 | |
|     OB_STORE_CACHE.init(1,1,1,1,1, 10);
 | |
|     main_table_ctx_.allocator_ = &allocator_;
 | |
|     main_table_ctx_.stmt_allocator_ = &allocator_;
 | |
|     main_table_ctx_.query_flag_ = ObQueryFlag();
 | |
|     main_table_ctx_.query_flag_.query_stat_ = 1;
 | |
|     main_table_ctx_.store_ctx_ = &store_ctx_;
 | |
|     main_table_ctx_.is_inited_ = true;
 | |
|     range_.set_whole_range();
 | |
|   }
 | |
|   void TearDown()
 | |
|   {
 | |
|     out_idxs_.reset();
 | |
|     allocator_.clear();
 | |
|     iters_.reset();
 | |
|     for (int64_t i = 0; i < stores_.count(); ++i) {
 | |
|       delete stores_.at(i);
 | |
|     }
 | |
|     stores_.reset();
 | |
|     OB_STORE_CACHE.destroy();
 | |
|   }
 | |
| 
 | |
| public:
 | |
|   ObArenaAllocator allocator_;
 | |
|   ColIdxArray out_idxs_;
 | |
|   ObStoreCtx store_ctx_;
 | |
|   ObTableAccessParam main_table_param_;
 | |
|   ObTableAccessContext main_table_ctx_;
 | |
|   common::ObSEArray<ObIStore *, common::MAX_TABLE_CNT_IN_STORAGE> stores_;
 | |
|   ObStoreRange range_;
 | |
|   ObMultipleMerge::MergeIterators iters_;
 | |
| };
 | |
| 
 | |
| void ObMultipleScanMergeTest::prepare_input(const ObIArray<const char *> &input, ObArray<ObMockStoreRowIterator *> &iters)
 | |
| {
 | |
|   for (int i = 0; i < input.count(); ++i) {
 | |
|     ObMockStoreRowIterator *iter = new ObMockStoreRowIterator();
 | |
|     iter->from(input.at(i));
 | |
|     ASSERT_EQ(OB_SUCCESS, iters.push_back(iter));
 | |
|   }
 | |
| }
 | |
| 
 | |
| void ObMultipleScanMergeTest::check_result(const char *output)
 | |
| {
 | |
|   ObMockQueryRowIterator expect_iter;
 | |
|   expect_iter.from(output);
 | |
|   ObMultipleScanMerge merge_iter;
 | |
|   ASSERT_EQ(OB_SUCCESS, merge_iter.init(main_table_param_, main_table_ctx_, stores_));
 | |
|   ASSERT_EQ(OB_SUCCESS, merge_iter.open(range_));
 | |
|   ASSERT_TRUE(expect_iter.equals(merge_iter));
 | |
|   merge_iter.reset();
 | |
| }
 | |
| 
 | |
| void ObMultipleScanMergeTest::prepare_store(const ObIArray<ObMockStoreRowIterator *> &iters)
 | |
| {
 | |
|   stores_.reset();
 | |
|   for (int i = 0; i < iters.count(); ++i) {
 | |
|     MockObSSStore *store = new MockObSSStore();
 | |
|     EXPECT_CALL(*store, scan(_, _, _, _))
 | |
|           .WillOnce(DoAll(SetArgReferee<3>(static_cast<ObStoreRowIterator *>(iters.at(i))),
 | |
|                 Return(OB_SUCCESS)));
 | |
|     ASSERT_EQ(OB_SUCCESS, stores_.push_back(store));
 | |
|   }
 | |
| }
 | |
| 
 | |
| TEST_F(ObMultipleScanMergeTest, normal)
 | |
| {
 | |
|   ObArray<const char *> inputs;
 | |
|   ObArray<ObMockStoreRowIterator *> iters;
 | |
|   // first col is pk , others c1, c2, c3
 | |
|   const char *input1 =
 | |
|       "int int var int flag    \n"
 | |
|       "6   288 var6 666 EXIST  \n"
 | |
|       "7   381 var7 777 EXIST  \n"
 | |
|       "8   88  var8 888 EXIST  \n";
 | |
|   const char *input2 =
 | |
|       "int int var  int flag   \n"
 | |
|       "1   0   var1 111 EXIST  \n"
 | |
|       "999 0   var9 999 EXIST  \n";
 | |
|   const char *input3 =
 | |
|       "int int var  int flag   \n"
 | |
|       "2   281 NOP  NOP EXIST  \n"
 | |
|       "3   388 var3 333 EXIST  \n"
 | |
|       "4   481 NOP  NOP EXIST  \n"
 | |
|       "9   99  var9 999 EXIST  \n";
 | |
|   const char *input4 =
 | |
|       "int int var int flag   \n"
 | |
|       "1   100 NOP NOP EXIST  \n"
 | |
|       "2   288 NOP NOP EXIST  \n"
 | |
|       "3   381 NOP NOP EMPTY  \n"
 | |
|       "4   88  NOP NOP EXIST  \n";
 | |
|   const char *input5 =
 | |
|       "int int var  int flag  \n"
 | |
|       "1   99  NOP  NOP EXIST \n"
 | |
|       "2   88  NOP  NOP DELETE\n"
 | |
|       "4   488 var4 444 EXIST \n"
 | |
|       "5   100 var5 555 EXIST \n";
 | |
|   const char *expect =
 | |
|       "int int var  int flag  \n"
 | |
|       "1   99  var1 111 EXIST \n"
 | |
|       "3   388 var3 333 EXIST \n"
 | |
|       "4   488 var4 444 EXIST \n"
 | |
|       "5   100 var5 555 EXIST  \n"
 | |
|       "6   288 var6 666 EXIST  \n"
 | |
|       "7   381 var7 777 EXIST  \n"
 | |
|       "8   88  var8 888 EXIST  \n"
 | |
|       "9   99  var9 999 EXIST  \n"
 | |
|       "999 0   var9 999 EXIST \n";
 | |
| 
 | |
|   ASSERT_EQ(OB_SUCCESS, inputs.push_back(input1));
 | |
|   ASSERT_EQ(OB_SUCCESS, inputs.push_back(input2));
 | |
|   ASSERT_EQ(OB_SUCCESS, inputs.push_back(input3));
 | |
|   ASSERT_EQ(OB_SUCCESS, inputs.push_back(input4));
 | |
|   ASSERT_EQ(OB_SUCCESS, inputs.push_back(input5));
 | |
|   prepare_input(inputs, iters);
 | |
|   prepare_store(iters);
 | |
|   check_result(expect);
 | |
| 
 | |
| 
 | |
| }
 | |
| 
 | |
| TEST_F(ObMultipleScanMergeTest, normal_reverse)
 | |
| {
 | |
|   ObArray<const char *> inputs;
 | |
|   ObArray<ObMockStoreRowIterator *> iters;
 | |
|   // first col is pk , others c1, c2, c3
 | |
|   const char *input1 =
 | |
|       "int int var  int flag   \n"
 | |
|       "1   0   var1 111 EXIST  \n"
 | |
|       "999 0   var9 999 EXIST  \n";
 | |
|   const char *input2 =
 | |
|       "int int var  int flag   \n"
 | |
|       "2   281 NOP  NOP EXIST  \n"
 | |
|       "3   388 var3 333 EXIST  \n"
 | |
|       "4   481 NOP  NOP EXIST  \n";
 | |
|   const char *input3 =
 | |
|       "int int var int flag   \n"
 | |
|       "1   100 NOP NOP EXIST  \n"
 | |
|       "2   288 NOP NOP EXIST  \n"
 | |
|       "3   381 NOP NOP EMPTY  \n"
 | |
|       "4   88  NOP NOP EXIST  \n";
 | |
|   const char *input4 =
 | |
|       "int int var  int flag  \n"
 | |
|       "1   99  NOP  NOP EXIST \n"
 | |
|       "2   88  NOP  NOP DELETE\n"
 | |
|       "4   488 var4 444 EXIST \n";
 | |
| 
 | |
|   const char *expect =
 | |
|       "int int var  int flag  \n"
 | |
|       "1   99  var1 111 EXIST \n"
 | |
|       "3   388 var3 333 EXIST \n"
 | |
|       "4   488 var4 444 EXIST \n"
 | |
|       "999 0   var9 999 EXIST \n";
 | |
| 
 | |
|   ASSERT_EQ(OB_SUCCESS, inputs.push_back(input1));
 | |
|   ASSERT_EQ(OB_SUCCESS, inputs.push_back(input2));
 | |
|   ASSERT_EQ(OB_SUCCESS, inputs.push_back(input3));
 | |
|   ASSERT_EQ(OB_SUCCESS, inputs.push_back(input4));
 | |
|   prepare_input(inputs, iters);
 | |
|   prepare_store(iters);
 | |
|   check_result(expect);
 | |
| 
 | |
| }
 | |
| 
 | |
| 
 | |
| TEST_F(ObMultipleScanMergeTest, test_iterator_rowkey_not_equal)
 | |
| {
 | |
|   ObArray<const char *> inputs;
 | |
|   ObArray<ObMockStoreRowIterator *> iters;
 | |
|   // first col is pk , others c1, c2, c3
 | |
|   const char *input1 =
 | |
|       "int int var  int flag  \n"
 | |
|       "1   99  VAR   102 EXIST \n"
 | |
|       "2   88  VAR2  100 DELETE\n"
 | |
|       "10   488 var4 444 EXIST \n";
 | |
| 
 | |
|   const char *input2 =
 | |
|       "int int var int flag   \n"
 | |
|       "3   100 TEST 100 EXIST  \n"
 | |
|       "4   288 TEST2 101 EXIST  \n"
 | |
|       "5   381 TEST3 102 DELETE  \n"
 | |
|       "6   88  TEST4 103 EXIST  \n";
 | |
| 
 | |
|   const char *output =
 | |
|       "int int var  int flag  \n"
 | |
|       "1   99  VAR 102 EXIST \n"
 | |
|       "3   100 TEST 100 EXIST  \n"
 | |
|       "4   288 TEST2 101 EXIST  \n"
 | |
|       "6   88  TEST4 103 EXIST  \n"
 | |
|       "10   488 var4 444 EXIST \n";
 | |
|   ASSERT_EQ(OB_SUCCESS, inputs.push_back(input1));
 | |
|   ASSERT_EQ(OB_SUCCESS, inputs.push_back(input2));
 | |
|   prepare_input(inputs, iters);
 | |
|   prepare_store(iters);
 | |
|   check_result(output);
 | |
| }
 | |
| 
 | |
| TEST_F(ObMultipleScanMergeTest, test_iterator_rowkey_equal_exist)
 | |
| {
 | |
|   ObArray<const char *> inputs;
 | |
|   ObArray<ObMockStoreRowIterator *> iters;
 | |
|   // first col is pk , others c1, c2, c3
 | |
|   const char *input1 =
 | |
|       "int int var  int flag  \n"
 | |
|       "1   99  VAR   102 EXIST \n"
 | |
|       "2   88  VAR2  100 EXIST\n"
 | |
|       "10   488 var4 444 EXIST \n";
 | |
| 
 | |
|   const char *input2 =
 | |
|       "int int var int flag   \n"
 | |
|       "1   NOP NOP 100 EXIST  \n"
 | |
|       "2   288 TEST2 101 EXIST  \n"
 | |
|       "10   381 TEST3 NOP EXIST  \n";
 | |
| 
 | |
|   const char *output =
 | |
|       "int int var  int flag  \n"
 | |
|       "1   99  VAR 100 EXIST \n"
 | |
|       "2   288 TEST2 101 EXIST  \n"
 | |
|       "10   381 TEST3 444 EXIST \n";
 | |
|   ASSERT_EQ(OB_SUCCESS, inputs.push_back(input1));
 | |
|   ASSERT_EQ(OB_SUCCESS, inputs.push_back(input2));
 | |
|   prepare_input(inputs, iters);
 | |
|   prepare_store(iters);
 | |
|   check_result(output);
 | |
| }
 | |
| 
 | |
| TEST_F(ObMultipleScanMergeTest, test_iterator_rowkey_equal_not_exist)
 | |
| {
 | |
|   ObArray<const char *> inputs;
 | |
|   ObArray<ObMockStoreRowIterator *> iters;
 | |
|   // first col is pk , others c1, c2, c3
 | |
|   const char *input1 =
 | |
|       "int int var  int flag  \n"
 | |
|       "1   99  VAR   102 EXIST \n"
 | |
|       "2   88  VAR2  100 EXIST\n"
 | |
|       "10   488 var4 444 EXIST \n";
 | |
| 
 | |
|   const char *input2 =
 | |
|       "int int var int flag   \n"
 | |
|       "1   NOP NOP 100 DELETE  \n"
 | |
|       "2   288 TEST2 101 DELETE  \n"
 | |
|       "10   381 TEST3 NOP DELETE  \n";
 | |
| 
 | |
|   const char *output = NULL;
 | |
|   ASSERT_EQ(OB_SUCCESS, inputs.push_back(input1));
 | |
|   ASSERT_EQ(OB_SUCCESS, inputs.push_back(input2));
 | |
|   prepare_input(inputs, iters);
 | |
|   prepare_store(iters);
 | |
|   check_result(output);
 | |
| }
 | |
| 
 | |
| TEST_F(ObMultipleScanMergeTest, test_iterator_with_delete_row)
 | |
| {
 | |
|   ObMultipleScanMerge merge_iter;
 | |
|   ObArray<const char *> inputs;
 | |
|   ObArray<ObMockStoreRowIterator *> iters;
 | |
|   ObStoreRow *row = NULL;
 | |
|   // first col is pk , others c1, c2, c3
 | |
|   const char *input1 =
 | |
|       "int int var  int flag  \n"
 | |
|       "1   99  VAR   102 DELETE \n"
 | |
|       "2   88  VAR2  100 EXIST\n"
 | |
|       "10   488 var4 444 EXIST \n";
 | |
| 
 | |
|   ASSERT_EQ(OB_SUCCESS, inputs.push_back(input1));
 | |
|   prepare_input(inputs, iters);
 | |
|   prepare_store(iters);
 | |
|   ASSERT_EQ(OB_SUCCESS, merge_iter.init(main_table_param_, main_table_ctx_, stores_));
 | |
|   ASSERT_EQ(OB_SUCCESS, merge_iter.open(range_));
 | |
|   merge_iter.set_iter_del_row(true);
 | |
|   ASSERT_EQ(OB_SUCCESS, merge_iter.get_next_row(row));
 | |
|   ASSERT_EQ(OB_SUCCESS, merge_iter.get_next_row(row));
 | |
|   ASSERT_EQ(OB_SUCCESS, merge_iter.get_next_row(row));
 | |
|   ASSERT_EQ(OB_ITER_END, merge_iter.get_next_row(row));
 | |
| }
 | |
| 
 | |
| TEST_F(ObMultipleScanMergeTest, test_iterator_rowkey_equal_empty)
 | |
| {
 | |
|   ObArray<const char *> inputs;
 | |
|   ObArray<ObMockStoreRowIterator *> iters;
 | |
|   // first col is pk , others c1, c2, c3
 | |
|   const char *input1 = NULL;
 | |
|   const char *input2 = NULL;
 | |
|   const char *output = NULL;
 | |
|   ASSERT_EQ(OB_SUCCESS, inputs.push_back(input1));
 | |
|   ASSERT_EQ(OB_SUCCESS, inputs.push_back(input2));
 | |
|   prepare_input(inputs, iters);
 | |
|   prepare_store(iters);
 | |
|   check_result(output);
 | |
| }
 | |
| 
 | |
| TEST_F(ObMultipleScanMergeTest, test_estimate_cost)
 | |
| {
 | |
|   int ret = OB_SUCCESS;
 | |
|   ObQueryFlag query_flag;
 | |
|   const uint64_t table_id = combine_id(1, 3000);
 | |
|   share::schema::ObColDesc col;
 | |
|   ObArray<share::schema::ObColDesc> columns;
 | |
|   ObPartitionEst cost;
 | |
|   ObMultipleScanMerge merge_iter;
 | |
|   ObArray<ObEstRowCountRecord> records;
 | |
|   ret = merge_iter.estimate_cost(query_flag, table_id, range_, columns, stores_, cost, records);
 | |
|   ASSERT_EQ(OB_INVALID_ARGUMENT, ret);
 | |
|   ret = columns.push_back(col);
 | |
|   ASSERT_EQ(OB_SUCCESS, ret);
 | |
| 
 | |
|   ObPartitionEst cost_metric1;
 | |
|   ObPartitionEst cost_metric2;
 | |
|   cost_metric1.logical_row_count_ = 10;
 | |
|   cost_metric1.total_cost_ = 100;
 | |
|   MockObSSStore *store1 = new MockObSSStore();
 | |
|   //EXPECT_CALL(*store1, get_store_type())
 | |
|     //.WillRepeatedly(Return(MAJOR_SSSTORE));
 | |
|   EXPECT_CALL(*store1, estimate_scan_cost(_, _, _, _, _))
 | |
|     .WillRepeatedly(DoAll(SetArgReferee<4>(cost_metric1), Return(OB_SUCCESS)));
 | |
|   stores_.reset();
 | |
|   ret = stores_.push_back(store1);
 | |
|   ASSERT_EQ(OB_SUCCESS, ret);
 | |
|   cost_metric2.total_cost_ = 120;
 | |
|   cost_metric2.logical_row_count_ = 11;
 | |
|   MockObSSStore *store2 = new MockObSSStore();
 | |
|   //EXPECT_CALL(*store2, get_store_type())
 | |
|     //.WillRepeatedly(Return(MAJOR_SSSTORE));
 | |
|   EXPECT_CALL(*store2, estimate_scan_cost(_, _, _, _, _))
 | |
|     .WillRepeatedly(DoAll(SetArgReferee<4>(cost_metric2), Return(OB_SUCCESS)));
 | |
|   ret = stores_.push_back(store2);
 | |
|   ASSERT_EQ(OB_SUCCESS, ret);
 | |
|   ret = merge_iter.estimate_cost(query_flag, table_id, range_, columns, stores_, cost, records);
 | |
|   ASSERT_EQ(OB_SUCCESS, ret);
 | |
|   ASSERT_EQ(11, cost.logical_row_count_);
 | |
|   ASSERT_EQ(0, cost.physical_row_count_);
 | |
| }
 | |
| 
 | |
| TEST_F(ObMultipleScanMergeTest, test_reset)
 | |
| {
 | |
|   const char *input = "flag   int int var  int\n"
 | |
|                       "EXIST  4   488 var4 444\n";
 | |
| 
 | |
|   const char *expect = "flag   int int var   int\n"
 | |
|                        "EXIST  4   488 var4  444\n";
 | |
| 
 | |
|   ObMultipleScanMerge merge_iter;
 | |
|   ObArray<const char *> inputs;
 | |
|   ObArray<ObMockStoreRowIterator *> iters;
 | |
|   ASSERT_EQ(OB_SUCCESS, inputs.push_back(input));
 | |
|   prepare_input(inputs, iters);
 | |
|   prepare_store(iters);
 | |
|   check_result(expect);
 | |
| 
 | |
|   iters.reset();
 | |
|   for (int64_t i = 0; i < stores_.count(); ++i) {
 | |
|     delete stores_.at(i);
 | |
|   }
 | |
|   stores_.reset();
 | |
|   prepare_input(inputs, iters);
 | |
|   prepare_store(iters);
 | |
|   check_result(expect);
 | |
| }
 | |
| 
 | |
| TEST_F(ObMultipleScanMergeTest, test_init_twice)
 | |
| {
 | |
|   ObMultipleScanMerge merge_iter;
 | |
|   ASSERT_EQ(OB_SUCCESS, merge_iter.init(main_table_param_, main_table_ctx_, stores_));
 | |
|   ASSERT_EQ(OB_INIT_TWICE, merge_iter.init(main_table_param_, main_table_ctx_, stores_));
 | |
| }
 | |
| 
 | |
| // empty out idx_idxs is valid
 | |
| TEST_F(ObMultipleScanMergeTest, test_init_empty_out_col)
 | |
| {
 | |
|   ObMultipleScanMerge merge_iter;
 | |
|   ColIdxArray out_idxs;
 | |
|   main_table_param_.iter_param_.output_project_ = &out_idxs;
 | |
|   ASSERT_EQ(OB_SUCCESS, merge_iter.init(main_table_param_, main_table_ctx_, stores_));
 | |
|   main_table_param_.iter_param_.output_project_ = &out_idxs_;
 | |
| }
 | |
| 
 | |
| // out idx_idxs contains invalid col index
 | |
| TEST_F(ObMultipleScanMergeTest, test_init_invalid_out_col)
 | |
| {
 | |
|   const char *input = "flag   int int var  int\n"
 | |
|       "EXIST  4   488 var4 444\n";
 | |
| 
 | |
|   ColIdxArray out_idxs;
 | |
|   out_idxs.push_back(-1);
 | |
|   main_table_param_.iter_param_.output_project_ = &out_idxs;
 | |
|   ObArray<const char *> inputs;
 | |
|   ObArray<ObMockStoreRowIterator *> iters;
 | |
|   ASSERT_EQ(OB_SUCCESS, inputs.push_back(input));
 | |
|   prepare_input(inputs, iters);
 | |
|   prepare_store(iters);
 | |
|   ObMultipleScanMerge merge_iter;
 | |
|   ObStoreRow *row = NULL;
 | |
|   ASSERT_EQ(OB_SUCCESS, merge_iter.init(main_table_param_, main_table_ctx_, stores_));
 | |
|   ASSERT_EQ(OB_SUCCESS, merge_iter.open(range_));
 | |
|   ASSERT_EQ(OB_SUCCESS, merge_iter.get_next_row(row));
 | |
| }
 | |
| 
 | |
| /*
 | |
| TEST_F(ObMultipleScanMergeTest, test_default_value_row1)
 | |
| {
 | |
|   const char *input1 = "flag   int int var  int int\n"
 | |
|                        "EXIST  4   388 var4 NOP NOP\n";
 | |
|   const char *input2 = "flag   int int var  int int\n"
 | |
|                        "EXIST  4   488 var4 NOP NOP\n";
 | |
| 
 | |
|   ObMultipleScanMerge merge_iter;
 | |
|   ASSERT_EQ(OB_SUCCESS, merge_iter.init(main_table_param_, main_table_ctx_, stores_));
 | |
|   ObMockStoreRowIterator iter1;
 | |
|   ObMockStoreRowIterator iter2;
 | |
|   ASSERT_EQ(OB_SUCCESS, iter1.from(input1));
 | |
|   ASSERT_EQ(OB_SUCCESS, iter2.from(input2));
 | |
|   ASSERT_EQ(OB_SUCCESS, merge_iter.add_iterator(iter1));
 | |
|   ASSERT_EQ(OB_SUCCESS, merge_iter.add_iterator(iter2));
 | |
|   ASSERT_EQ(OB_SUCCESS, merge_iter.open(range_));
 | |
|   storage::ObStoreRow *row = NULL;
 | |
|   ASSERT_EQ(OB_ERR_UNEXPECTED, merge_iter.get_next_row(row));
 | |
| }
 | |
| 
 | |
| TEST_F(ObMultipleScanMergeTest, test_default_value_row2)
 | |
| {
 | |
|   const char *input1 = "flag   int int var  int int\n"
 | |
|                        "EXIST  4   388 var4 NOP NOP\n";
 | |
|   const char *input2 = "flag   int int var  int int\n"
 | |
|                        "EXIST  4   488 var4 NOP NOP\n";
 | |
|   const char *default_value_row = "flag  int  int\n"
 | |
|                                   "EXIST 1    3  \n";
 | |
|   ObMockStoreRowIterator default_value_iter;
 | |
|   const storage::ObStoreRow *row;
 | |
|   ASSERT_EQ(OB_SUCCESS, default_value_iter.from(default_value_row));
 | |
|   ASSERT_EQ(OB_SUCCESS, default_value_iter.get_next_row(row));
 | |
|   ObMockStoreRowIterator iter1;
 | |
|   ObMockStoreRowIterator iter2;
 | |
|   storage::ObStoreRow *r = NULL;
 | |
|   ASSERT_EQ(OB_SUCCESS, iter1.from(input1));
 | |
|   ASSERT_EQ(OB_SUCCESS, iter2.from(input2));
 | |
|   ObQueryFlag flag;
 | |
|   ObMultipleScanMerge merge_iter(allocator_, flag, CELL_NUM, ROWKEY_NUM, out_idxs_, row);
 | |
|   ASSERT_EQ(OB_SUCCESS, merge_iter.add_iterator(iter1));
 | |
|   ASSERT_EQ(OB_SUCCESS, merge_iter.add_iterator(iter2));
 | |
|   ASSERT_EQ(OB_SUCCESS, merge_iter.init());
 | |
|   ASSERT_EQ(OB_INVALID_ARGUMENT, merge_iter.get_next_row(r));
 | |
| }
 | |
| 
 | |
| TEST_F(ObMultipleScanMergeTest, test_default_value_row3)
 | |
| {
 | |
|   const char *input1 = "flag   int int var  int int\n"
 | |
|                        "EXIST  4   388 var4 NOP NOP\n";
 | |
|   const char *input2 = "flag   int int var  int int\n"
 | |
|                        "EXIST  4   488 var4 NOP NOP\n";
 | |
|   const char *default_value_row = "flag  int  int var  int int\n"
 | |
|                                   "EXIST 3    3   var4 4   NOP\n";
 | |
|   ObQueryFlag flag;
 | |
|   ObMockStoreRowIterator default_value_iter;
 | |
|   const storage::ObStoreRow *row;
 | |
|   ASSERT_EQ(OB_SUCCESS, default_value_iter.from(default_value_row));
 | |
|   ASSERT_EQ(OB_SUCCESS, default_value_iter.get_next_row(row));
 | |
|   ObMultipleScanMerge merge_iter(allocator_, flag, CELL_NUM, ROWKEY_NUM, out_idxs_, row);
 | |
|   ObMockStoreRowIterator iter1;
 | |
|   ObMockStoreRowIterator iter2;
 | |
|   storage::ObStoreRow *r = NULL;
 | |
|   ASSERT_EQ(OB_SUCCESS, iter1.from(input1));
 | |
|   ASSERT_EQ(OB_SUCCESS, iter2.from(input2));
 | |
|   ASSERT_EQ(OB_SUCCESS, merge_iter.add_iterator(iter1));
 | |
|   ASSERT_EQ(OB_SUCCESS, merge_iter.add_iterator(iter2));
 | |
|   ASSERT_EQ(OB_SUCCESS, merge_iter.init());
 | |
|   ASSERT_EQ(OB_ERR_UNEXPECTED, merge_iter.get_next_row(r));
 | |
| }
 | |
| */
 | |
| 
 | |
| } // namespace unittest
 | |
| } // namespace oceanbase
 | |
| 
 | |
| int main(int argc, char **argv)
 | |
| {
 | |
|   system("rm -f test_multiple_scan_merge.log");
 | |
|   OB_LOGGER.set_file_name("test_multiple_scan_merge.log");
 | |
|   OB_LOGGER.set_log_level("WARN");
 | |
|   CLOG_LOG(INFO, "begin unittest: test_multiple_scan_merge");
 | |
|   oceanbase::unittest::init_tenant_mgr();
 | |
|   ::testing::InitGoogleTest(&argc, argv);
 | |
|   return RUN_ALL_TESTS();
 | |
| }
 | 
