| 1004 | } |
| 1005 | |
| 1006 | void TestGetRecordBatchSize(const IpcWriteOptions& options, |
| 1007 | std::shared_ptr<RecordBatch> batch) { |
| 1008 | io::MockOutputStream mock; |
| 1009 | ipc::IpcPayload payload; |
| 1010 | int32_t mock_metadata_length = -1; |
| 1011 | int64_t mock_body_length = -1; |
| 1012 | int64_t size = -1; |
| 1013 | ASSERT_OK(WriteRecordBatch(*batch, 0, &mock, &mock_metadata_length, &mock_body_length, |
| 1014 | options)); |
| 1015 | ASSERT_OK(GetRecordBatchPayload(*batch, options, &payload)); |
| 1016 | int64_t payload_size = GetPayloadSize(payload, options); |
| 1017 | ASSERT_OK(GetRecordBatchSize(*batch, options, &size)); |
| 1018 | ASSERT_EQ(mock.GetExtentBytesWritten(), size); |
| 1019 | ASSERT_EQ(mock.GetExtentBytesWritten(), payload_size); |
| 1020 | } |
| 1021 | |
| 1022 | TEST_F(TestWriteRecordBatch, IntegerGetRecordBatchSize) { |
| 1023 | std::shared_ptr<RecordBatch> batch; |
no test coverage detected