require "test_helper"

class IPTCTest < Minitest::Test
  def test_iptc_string
    iptc = IPTCR.parse("\x1c\x00\x00\x00\x05Hello".b)
    assert iptc.is_a? IPTCR::IPTC
    assert_equal 1, iptc.fields.size
    assert_equal "Hello".b, iptc.fields.first.value
  end

  def test_iptc_string_respects_length
    iptc = IPTCR.parse("\x1c\x00\x00\x00\x05Hello\x1c\x00\x00\x07\x00Goodbye".b, length: 10)
    assert iptc.is_a? IPTCR::IPTC
    assert_equal 1, iptc.fields.size
    assert_equal "Hello".b, iptc.fields.first.value
  end

  def test_iptc_io
    iptc = IPTCR.parse(StringIO.new("\x1c\x00\x00\x00\x05Hello"))
    assert iptc.is_a? IPTCR::IPTC
    assert_equal 1, iptc.fields.size
    assert_equal "Hello".b, iptc.fields.first.value
  end

  def test_iptc_io_respects_length
    iptc = IPTCR.parse(StringIO.new("\x1c\x00\x00\x00\x05Hello\x1c\x00\x00\x07\x00Goodbye".b), length: 10)
    assert iptc.is_a? IPTCR::IPTC
    assert_equal 1, iptc.fields.size
    assert_equal "Hello".b, iptc.fields.first.value
  end

  def test_bad_marker
    assert_raises(IPTCR::Malformed) do
      IPTCR.parse("\x1d\x00\x00\x00\x05Hello".b)
    end

    assert_raises(IPTCR::Malformed) do
      IPTCR.parse("\x1c\x00\x00\x00\x05Hello\x1d".b)
    end
  end

  def test_iptc_unknown_record
    iptc = IPTCR.parse("\x1c\x00\x00\x00\x05Hello".b)
    assert_equal(1, iptc.fields.size)
    refute iptc.fields.first.record?
    refute iptc.fields.first.dataset?
    assert_equal "Hello".b, iptc.fields.first.value
  end

  def test_iptc_unknown_dataset
    iptc = IPTCR.parse("\x1c\x01\x01\x00\x05Hello".b)
    assert_equal(1, iptc.fields.size)
    assert iptc.fields.first.record?
    refute iptc.fields.first.dataset?
    assert_equal "Hello".b, iptc.fields.first.value
  end

  def test_iptc_int8u
    iptc = IPTCR.parse("\x1c\x03\x37\x00\x02\x01".b)
    assert_equal({"SupplementalType" => 1}, iptc.to_hash)
  end

  def test_iptc_int16u
    iptc = IPTCR.parse("\x1c\x01\x00\x00\x02\x01\x02".b)
    assert_equal({"EnvelopeRecordVersion" => 513}, iptc.to_hash)
  end

  def test_iptc_int32u
    iptc = IPTCR.parse("\x1c\x03\x6e\x00\x04\x01\x02\x04\x08".b)
    assert_equal({"DataCompressionMethod" => 134480385}, iptc.to_hash)
  end

  def test_iptc_string
    iptc = IPTCR.parse("\x1c\x01\x1e\x00\x05Hello".b)
    assert_equal({"ServiceIdentifier" => "Hello".b}, iptc.to_hash)
  end

  def test_iptc_encoded_string
    iptc = IPTCR.parse("\x1c\x01\x5a\x00\x03\x1b\x25\x47\x1c\x02\x05\x00\x03\xE2\x98\x83".b)
    assert_equal(2, iptc.fields.size)
    assert_equal("\e%G".b, iptc["CodedCharacterSet"])
    assert_equal(Encoding::UTF_8, iptc["ObjectName"].encoding)
    assert_equal("☃".force_encoding("UTF-8"), iptc["ObjectName"])
  end

  def test_iptc_default_encoding_string
    iptc = IPTCR.parse("\x1c\x02\x05\x00\x01\x80".b, default_encoding: "windows-1252")
    assert_equal(Encoding::Windows_1252, iptc["ObjectName"].encoding)
    assert_equal("€", iptc["ObjectName"].encode("utf-8"))
  end

  def test_iptc_string_list
    iptc = IPTCR.parse("\x1c\x01\x05\x00\x05Hello\x1c\x01\x05\x00\x05World".b)
    assert_equal(2, iptc.fields.size)
    assert_equal(["Hello", "World"], iptc["Destination"])
  end
end