require_relative "test_helper"

class DelimitedWriterTest < Minitest::Test
  describe IOStreams::Line::Writer do
    let :file_name do
      File.join(File.dirname(__FILE__), "files", "text.txt")
    end

    let :raw do
      File.read(file_name)
    end

    let :lines do
      raw.lines.map(&:strip)
    end

    describe "#<<" do
      it "file" do
        temp_file = Tempfile.new("rocket_job")
        file_name = temp_file.to_path
        IOStreams::Line::Writer.file(file_name) do |io|
          lines.each { |line| io << line }
        end
        result = File.read(file_name)
        assert_equal raw, result
      end

      it "stream" do
        io_string = StringIO.new
        IOStreams::Line::Writer.stream(io_string) do |io|
          lines.each { |line| io << line }
        end
        assert_equal raw, io_string.string
      end
    end

    describe ".write" do
      it "returns byte count" do
        io_string = StringIO.new
        count     = 0
        IOStreams::Line::Writer.stream(io_string) do |io|
          lines.each { |line| count += io.write(line) }
        end
        assert_equal raw, io_string.string
        assert_equal raw.size, count
      end
    end
  end
end