require_relative 'test_helper'
require 'csv'

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

    let :raw_csv_data do
      File.read(csv_file_name)
    end

    let :csv_rows do
      CSV.read(csv_file_name)
    end

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

    let :file_name do
      temp_file.path
    end

    after do
      temp_file.delete
    end

    describe '.stream' do
      it 'file' do
        IOStreams::Row::Writer.file(file_name) do |io|
          csv_rows.each { |array| io << array }
        end
        result = ::File.read(file_name)
        assert_equal raw_csv_data, result
      end

      it 'streams' do
        io_string = StringIO.new
        IOStreams::Line::Writer.stream(io_string) do |io|
          IOStreams::Row::Writer.stream(io) do |stream|
            csv_rows.each { |array| stream << array }
          end
        end
        assert_equal raw_csv_data, io_string.string
      end
    end

  end
end