require_relative 'test_helper'

class EncodeWriterTest < Minitest::Test
  describe IOStreams::Encode::Writer do
    let :bad_data do
      [
        "New M\xE9xico,NE".b,
        'good line',
        "New M\xE9xico,\x07SF".b
      ].join("\n").encode('BINARY')
    end

    let :cleansed_data do
      bad_data.gsub("\xE9".b, '?')
    end

    let :stripped_data do
      cleansed_data.gsub("\x07", '')
    end

    describe '#<<' do
      it 'file' do
        temp_file = Tempfile.new('rocket_job')
        file_name = temp_file.to_path
        IOStreams::Encode::Writer.file(file_name, encoding: 'ASCII-8BIT') do |io|
          io << bad_data
        end
        result = File.read(file_name, mode: 'rb')
        assert_equal bad_data, result
      end

      it 'stream' do
        io = StringIO.new(''.b)
        IOStreams::Encode::Writer.stream(io, encoding: 'ASCII-8BIT') do |encoded|
          encoded << bad_data
        end
        assert_equal 'ASCII-8BIT', io.string.encoding.to_s
        assert_equal bad_data, io.string
      end

      it 'stream as utf-8' do
        io = StringIO.new('')
        assert_raises Encoding::UndefinedConversionError do
          IOStreams::Encode::Writer.stream(io, encoding: 'UTF-8') do |encoded|
            encoded << bad_data
          end
        end
      end

      it 'stream as utf-8 with replacement' do
        io = StringIO.new('')
        IOStreams::Encode::Writer.stream(io, encoding: 'UTF-8', replace: '?') do |encoded|
          encoded << bad_data
        end
        assert_equal 'UTF-8', io.string.encoding.to_s
        assert_equal cleansed_data, io.string
      end

      it 'stream as utf-8 with replacement and printable cleansing' do
        io = StringIO.new('')
        IOStreams::Encode::Writer.stream(io, encoding: 'UTF-8', replace: '?', cleaner: :printable) do |encoded|
          encoded << bad_data
        end
        assert_equal 'UTF-8', io.string.encoding.to_s
        assert_equal stripped_data, io.string
      end
    end

    describe '.write' do
      it 'returns byte count' do
        io_string = StringIO.new(''.b)
        count     = 0
        IOStreams::Encode::Writer.stream(io_string, encoding: 'ASCII-8BIT') do |io|
          count += io.write(bad_data)
        end
        assert_equal bad_data, io_string.string
        assert_equal bad_data.size, count
      end
    end

  end
end