test

Signed-off-by: Jiao Mingye <mxdzs0612@gmail.com>

be/src/exec/pipeline/exchange/exchange_sink_operator.h

@@ -84,6 +84,9 @@ class ExchangeSinkOperator final : public Operator {
     // Return the physical bytes of attachment.
     int64_t construct_brpc_attachment(const PTransmitChunkParamsPtr& _chunk_request, butil::IOBuf& attachment);
 
+    // For UT Only
+    TPartitionType::type get_part_type() { return _part_type; }
+
 private:
     bool _is_large_chunk(size_t sz) const {
         // ref olap_scan_node.cpp release_large_columns

be/test/exec/pipeline/exchange_sink_operator_test.cpp

@@ -104,6 +104,8 @@ TEST_F(ExchangeSinkOperatorTest, test_push_random_scale) {
     exchange_sink_operator_factory->set_runtime_state(_runtime_state);
     auto exchange_sink_operator =
             std::static_pointer_cast<ExchangeSinkOperator>(exchange_sink_operator_factory->create(1, 0));
+    ASSERT_EQ(exchange_sink_operator->get_part_type(), TPartitionType::RANDOM_SCALE);
+
     auto chunk = std::make_shared<Chunk>();
     EXPECT_OK(exchange_sink_operator->push_chunk(_runtime_state, chunk));
 }