@@ -685,27 +685,4 @@ TEST_F(RealtimePrimaryKeyReaderTest, TestFactoryFailureClosesReaders) {
685685 ASSERT_EQ (factory_failure_close_count, 1 );
686686}
687687
688- TEST_F (RealtimePrimaryKeyReaderTest, TestQueryReaderClose) {
689- std::vector<DataField> value_fields = {DataField (0 , arrow::field (" k0" , arrow::int32 ())),
690- DataField (1 , arrow::field (" v0" , arrow::int32 ()))};
691- std::shared_ptr<arrow::Schema> value_schema =
692- DataField::ConvertDataFieldsToArrowSchema (value_fields);
693- std::shared_ptr<arrow::Schema> key_schema = arrow::schema ({value_schema->field (0 )});
694- std::shared_ptr<arrow::Schema> transport_schema = MakeTransportSchema (value_schema->fields ());
695- std::shared_ptr<arrow::DataType> transport_type = arrow::struct_ (transport_schema->fields ());
696- std::shared_ptr<arrow::Array> transport_array =
697- arrow::ipc::internal::json::ArrayFromJSON (transport_type, R"( [[0, 10, 0, 1, 100]])" )
698- .ValueOrDie ();
699-
700- int32_t close_count = 0 ;
701- auto batch_reader = std::make_unique<TrackingBatchReader>(
702- std::make_unique<MockFileBatchReader>(transport_array, transport_type, 1 ), &close_count);
703- ASSERT_OK_AND_ASSIGN (std::unique_ptr<KeyValueRecordReader> reader,
704- CreateRealtimePrimaryKeyQueryReaderForTest (
705- std::move (batch_reader), transport_schema, OffsetRange (0 , 1 ),
706- key_schema, value_schema, pool_));
707- reader->Close ();
708- ASSERT_EQ (close_count, 1 );
709- }
710-
711688} // namespace paimon::test
0 commit comments