rebase be ut

Author: zhangstar333

be/test/vec/exec/format/parquet/parquet_reader_test.cpp

@@ -212,8 +212,7 @@ TEST_F(ParquetReaderTest, uuid_varbinary) {
     RuntimeState runtime_state((TQueryGlobals()));
     runtime_state.set_desc_tbl(desc_tbl);
 
-    st = p_reader->init_reader(column_names, nullptr, {}, nullptr, nullptr, nullptr, nullptr,
-                               nullptr);
+    st = p_reader->init_reader(column_names, {}, nullptr, nullptr, nullptr, nullptr, nullptr);
     EXPECT_TRUE(st.ok()) << st;
     std::unordered_map<std::string, std::tuple<std::string, const SlotDescriptor*>>
             partition_columns;
@@ -283,8 +282,7 @@ TEST_F(ParquetReaderTest, varbinary_varbinary) {
     RuntimeState runtime_state((TQueryGlobals()));
     runtime_state.set_desc_tbl(desc_tbl);
 
-    st = p_reader->init_reader(column_names, nullptr, {}, nullptr, nullptr, nullptr, nullptr,
-                               nullptr);
+    st = p_reader->init_reader(column_names, {}, nullptr, nullptr, nullptr, nullptr, nullptr);
     EXPECT_TRUE(st.ok()) << st;
     std::unordered_map<std::string, std::tuple<std::string, const SlotDescriptor*>>
             partition_columns;
@@ -356,8 +354,7 @@ TEST_F(ParquetReaderTest, varbinary_string) {
     RuntimeState runtime_state((TQueryGlobals()));
     runtime_state.set_desc_tbl(desc_tbl);
 
-    st = p_reader->init_reader(column_names, nullptr, {}, nullptr, nullptr, nullptr, nullptr,
-                               nullptr);
+    st = p_reader->init_reader(column_names, {}, nullptr, nullptr, nullptr, nullptr, nullptr);
     EXPECT_TRUE(st.ok()) << st;
     std::unordered_map<std::string, std::tuple<std::string, const SlotDescriptor*>>
             partition_columns;
@@ -429,8 +426,7 @@ TEST_F(ParquetReaderTest, varbinary_string2) {
     RuntimeState runtime_state((TQueryGlobals()));
     runtime_state.set_desc_tbl(desc_tbl);
 
-    st = p_reader->init_reader(column_names, nullptr, {}, nullptr, nullptr, nullptr, nullptr,
-                               nullptr);
+    st = p_reader->init_reader(column_names, {}, nullptr, nullptr, nullptr, nullptr, nullptr);
     EXPECT_TRUE(st.ok()) << st;
     std::unordered_map<std::string, std::tuple<std::string, const SlotDescriptor*>>
             partition_columns;
@@ -461,76 +457,5 @@ TEST_F(ParquetReaderTest, varbinary_string2) {
     EXPECT_EQ(StringView(string_column->get_data_at(2)).dump_hex(), "X'00'");
 }
 
-static ParquetReader* create_parquet_reader(TFileScanRangeParams& scan_params,
-                                            std::vector<std::string> table_column_names,
-                                            std::vector<TPrimitiveType::type> types) {
-    TDescriptorTable t_desc_table;
-    TTableDescriptor t_table_desc;
-
-    create_table_desc(t_desc_table, t_table_desc, table_column_names, types);
-    DescriptorTbl* desc_tbl;
-    ObjectPool obj_pool;
-    static_cast<void>(DescriptorTbl::create(&obj_pool, t_desc_table, &desc_tbl));
-
-    auto slot_descs = desc_tbl->get_tuple_descriptor(0)->slots();
-    auto local_fs = io::global_local_filesystem();
-    io::FileReaderSPtr reader;
-    static_cast<void>(local_fs->open_file(
-            "./be/test/exec/test_data/parquet_scanner/type-decoder.parquet", &reader));
-
-    cctz::time_zone ctz;
-    TimezoneUtils::find_cctz_time_zone(TimezoneUtils::default_time_zone, ctz);
-    std::vector<std::string> column_names;
-    for (int i = 0; i < slot_descs.size(); i++) {
-        column_names.push_back(slot_descs[i]->col_name());
-    }
-    TFileRangeDesc scan_range;
-    {
-        scan_range.start_offset = 0;
-        scan_range.size = 1000;
-    }
-    auto p_reader =
-            new ParquetReader(nullptr, scan_params, scan_range, 992, &ctz, nullptr, nullptr);
-    p_reader->set_file_reader(reader);
-    return p_reader;
-}
-
-TEST_F(ParquetReaderTest, use_column_name) {
-    std::vector<std::string> table_column_names = {"boolean_col", "tinyint_col", "smallint_col",
-                                                   "int_col",     "bigint_col",  "float_col",
-                                                   "double_col"};
-    std::vector<TPrimitiveType::type> table_column_types = {
-            TPrimitiveType::BOOLEAN, TPrimitiveType::TINYINT, TPrimitiveType::SMALLINT,
-            TPrimitiveType::INT,     TPrimitiveType::BIGINT,  TPrimitiveType::FLOAT,
-            TPrimitiveType::DOUBLE};
-    TFileScanRangeParams scan_params;
-
-    auto p_reader = create_parquet_reader(scan_params, table_column_names, table_column_types);
-    std::unordered_map<std::string, ColumnValueRangeType> colname_to_value_range;
-    colname_to_value_range.emplace("boolean_col", ColumnValueRange<TYPE_BOOLEAN>("boolean_col"));
-    colname_to_value_range.emplace("tinyint_col", ColumnValueRange<TYPE_TINYINT>("tinyint_col"));
-    colname_to_value_range.emplace("smallint_col", ColumnValueRange<TYPE_SMALLINT>("smallint_col"));
-    colname_to_value_range.emplace("int_col", ColumnValueRange<TYPE_INT>("int_col"));
-
-    static_cast<void>(p_reader->init_reader(
-            table_column_names, &colname_to_value_range, {}, nullptr, nullptr, nullptr, nullptr,
-            nullptr, TableSchemaChangeHelper::ConstNode::get_instance(), false));
-
-    std::vector<std::string> read_columns_ans = {"tinyint_col", "smallint_col", "int_col",
-                                                 "bigint_col",  "boolean_col",  "float_col",
-                                                 "double_col"};
-    EXPECT_EQ(p_reader->_read_file_columns, read_columns_ans);
-
-    std::vector<std::string> miss_columns_ans = {};
-    EXPECT_EQ(p_reader->_missing_cols, miss_columns_ans);
-    std::vector<std::string> colname_to_value_range_names_ans = {"tinyint_col", "smallint_col",
-                                                                 "int_col", "boolean_col"};
-    for (auto col : colname_to_value_range_names_ans) {
-        EXPECT_TRUE(p_reader->_colname_to_value_range->contains(col));
-    }
-    EXPECT_EQ(p_reader->_colname_to_value_range->size(), colname_to_value_range_names_ans.size());
-    delete p_reader;
-}
-
 } // namespace vectorized
 } // namespace doris