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
        result =
          IOStreams::Record::Writer.file(file_name) do |io|
            inputs.each { |hash| io << hash }
            53534
          end
        assert_equal 53534, result
        result = File.read(file_name)
        assert_equal raw_csv_data, result
      end

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

        result = File.read(file_name)
        assert_equal raw_json_data, result
      end

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