Method TestWithBatchFactory

cpp/src/arrow/c/bridge_test.cc:4412–4455

  template <typename BatchFactory>
  void TestWithBatchFactory(BatchFactory&& factory) {
    std::shared_ptr<Device> device = std::make_shared<MyDevice>(1);
    auto mm = device->default_memory_manager();

    std::shared_ptr<RecordBatch> batch;
    struct ArrowDeviceArray c_array {};
    struct ArrowSchema c_schema {};
    ArrayExportGuard array_guard(&c_array.array);
    SchemaExportGuard schema_guard(&c_schema);

    auto orig_bytes = pool_->bytes_allocated();
    ASSERT_OK_AND_ASSIGN(batch, ToResult(factory()));
    ASSERT_OK(ExportSchema(*batch->schema(), &c_schema));
    ASSERT_OK_AND_ASSIGN(auto sync, mm->MakeDeviceSyncEvent());
    ASSERT_OK(ExportDeviceRecordBatch(*batch, sync, &c_array));

    auto new_bytes = pool_->bytes_allocated();
    batch.reset();
    ASSERT_EQ(pool_->bytes_allocated(), new_bytes);
    ASSERT_OK_AND_ASSIGN(batch,
                         ImportDeviceRecordBatch(&c_array, &c_schema, DeviceMapper));
    ASSERT_OK(batch->ValidateFull());
    ASSERT_TRUE(ArrowSchemaIsReleased(&c_schema));
    ASSERT_TRUE(ArrowArrayIsReleased(&c_array.array));

    // Re-export and re-import, now both at once
    ASSERT_OK(ExportDeviceRecordBatch(*batch, sync, &c_array, &c_schema));
    batch.reset();
    ASSERT_OK_AND_ASSIGN(batch,
                         ImportDeviceRecordBatch(&c_array, &c_schema, DeviceMapper));
    ASSERT_OK(batch->ValidateFull());
    ASSERT_TRUE(ArrowSchemaIsReleased(&c_schema));
    ASSERT_TRUE(ArrowArrayIsReleased(&c_array.array));

    // Check value of imported record batch
    {
      std::shared_ptr<RecordBatch> expected;
      ASSERT_OK_AND_ASSIGN(expected, ToResult(factory()));
      AssertSchemaEqual(*expected->schema(), *batch->schema(), /*check_metadata=*/true);
      AssertBatchesEqual(*expected, *batch);
    }
    batch.reset();
    ASSERT_EQ(pool_->bytes_allocated(), orig_bytes);
  }

  void TestWithJSON(const std::shared_ptr<MemoryManager>& mm,
                    std::shared_ptr<DataType> type, const char* json) {