# Licensed to the Apache Software Foundation (ASF) under one
# or more contributor license agreements.  See the NOTICE file
# distributed with this work for additional information
# regarding copyright ownership.  The ASF licenses this file
# to you under the Apache License, Version 2.0 (the
# "License"); you may not use this file except in compliance
# with the License.  You may obtain a copy of the License at
#
#   http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing,
# software distributed under the License is distributed on an
# "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
# KIND, either express or implied.  See the License for the
# specific language governing permissions and limitations
# under the License.

module RawRecordsTableTests
  def test_2_arrays
    raw_record_batches = [
      [
        [true, nil, "Ruby"],
        [nil, 0, "GLib"],
        [false, 2 ** 8 - 1, nil],
      ],
      [
        [nil, 10, "A"],
        [true, 20, "B"],
        [false, nil, "C"],
        [nil, 40, nil],
      ]
    ]
    raw_records = raw_record_batches.inject do |all_records, record_batch|
      all_records + record_batch
    end
    schema = [
      {name: :column0, type: :boolean},
      {name: :column1, type: :uint8},
      {name: :column2, type: :string},
    ]
    record_batches = raw_record_batches.collect do |record_batch|
      Arrow::RecordBatch.new(schema, record_batch)
    end
    table = Arrow::Table.new(schema, record_batches)
    assert_equal(raw_records, actual_records(table))
  end
end

class EachRawRecordTableTest < Test::Unit::TestCase
  include RawRecordsTableTests

  def actual_records(target)
    target.each_raw_record.to_a
  end
end

class RawRecordsTableTest < Test::Unit::TestCase
  include RawRecordsTableTests

  def actual_records(target)
    target.raw_records
  end
end
