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