Contents

require_relative 'test_helper'
require 'csv'

class RecordWriterTest < Minitest::Test
  describe IOStreams::Record::Writer do
    let :csv_file_name do
      File.join(File.dirname(__FILE__), 'files', 'test.csv')
    end

    let :json_file_name do
      File.join(File.dirname(__FILE__), 'files', 'test.json')
    end

    let :raw_csv_data do
      File.read(csv_file_name)
    end

    let :raw_json_data do
      File.read(json_file_name)
    end

    let :csv_rows do
      CSV.read(csv_file_name)
    end

    let :inputs do
      rows   = csv_rows.dup
      header = rows.shift
      rows.collect { |row| header.zip(row).to_h }
    end

    let :temp_file do
      Tempfile.new('iostreams')
    end

    let :file_name do
      temp_file.path
    end

    after do
      temp_file.delete
    end

    describe '#<<' do
      it 'file' do
        IOStreams::Record::Writer.file(file_name) do |io|
          inputs.each { |hash| io << hash }
        end
        result = File.read(file_name)
        assert_equal raw_csv_data, result
      end

      it 'json file' do
        IOStreams::Record::Writer.file(file_name, file_name: 'abc.json') do |io|
          inputs.each { |hash| io << hash }
        end
        result = File.read(file_name)
        assert_equal raw_json_data, result
      end

      it 'stream' do
        io_string = StringIO.new
        IOStreams::Line::Writer.stream(io_string) do |io|
          IOStreams::Record::Writer.stream(io) do |stream|
            inputs.each { |row| stream << row }
          end
        end
        assert_equal raw_csv_data, io_string.string
      end
    end

  end
end

Version data entries

9 entries across 9 versions & 1 rubygems

Version	Path
iostreams-1.1.0	test/record_writer_test.rb
iostreams-1.0.0	test/record_writer_test.rb
iostreams-1.0.0.beta7	test/record_writer_test.rb
iostreams-1.0.0.beta6	test/record_writer_test.rb
iostreams-1.0.0.beta5	test/record_writer_test.rb
iostreams-1.0.0.beta4	test/record_writer_test.rb
iostreams-1.0.0.beta3	test/record_writer_test.rb
iostreams-1.0.0.beta2	test/record_writer_test.rb
iostreams-1.0.0.beta	test/record_writer_test.rb