require 'testing_env'
require 'patch'

class PatchTests < Test::Unit::TestCase
  def test_create_simple
    patch = Patch.create(:p2)
    assert_kind_of ExternalPatch, patch
    assert patch.external?
    assert_equal :p2, patch.strip
  end

  def test_create_io
    patch = Patch.create(:p0, StringIO.new("foo"))
    assert_kind_of IOPatch, patch
    assert !patch.external?
    assert_equal :p0, patch.strip
  end

  def test_create_io_without_strip
    patch = Patch.create(StringIO.new("foo"))
    assert_kind_of IOPatch, patch
    assert_equal :p1, patch.strip
  end

  def test_create_string
    patch = Patch.create(:p0, "foo")
    assert_kind_of IOPatch, patch
    assert_equal :p0, patch.strip
  end

  def test_create_string_without_strip
    patch = Patch.create("foo")
    assert_kind_of IOPatch, patch
    assert_equal :p1, patch.strip
  end

  def test_create_DATA
    patch = Patch.create(:p0, :DATA)
    assert_kind_of IOPatch, patch
    assert_equal :p0, patch.strip
  end

  def test_create_DATA_without_strip
    patch = Patch.create(:DATA)
    assert_kind_of IOPatch, patch
    assert_equal :p1, patch.strip
  end

  def test_raises_for_unknown_values
    assert_raises(ArgumentError) { Patch.create(Object.new) }
  end
end

class LegacyPatchTests < Test::Unit::TestCase
  def test_patch_single_string
    patches = Patch.normalize_legacy_patches("http://example.com/patch.diff")
    assert_equal 1, patches.length
    assert_equal :p1, patches.first.strip
  end

  def test_patch_array
    patches = Patch.normalize_legacy_patches(
      %w{http://example.com/patch1.diff http://example.com/patch2.diff}
    )

    assert_equal 2, patches.length
    assert_equal :p1, patches[0].strip
    assert_equal :p1, patches[1].strip
  end

  def test_p0_hash_to_string
    patches = Patch.normalize_legacy_patches(
      :p0 => "http://example.com/patch.diff"
    )

    assert_equal 1, patches.length
    assert_equal :p0, patches.first.strip
  end

  def test_p1_hash_to_string
    patches = Patch.normalize_legacy_patches(
      :p1 => "http://example.com/patch.diff"
    )

    assert_equal 1, patches.length
    assert_equal :p1, patches.first.strip
  end

  def test_mixed_hash_to_strings
    patches = Patch.normalize_legacy_patches(
      :p1 => "http://example.com/patch1.diff",
      :p0 => "http://example.com/patch0.diff"
    )
    assert_equal 2, patches.length
    assert_equal 1, patches.select { |p| p.strip == :p0 }.length
    assert_equal 1, patches.select { |p| p.strip == :p1 }.length
  end

  def test_mixed_hash_to_arrays
    patches = Patch.normalize_legacy_patches(
      :p1 => ["http://example.com/patch10.diff",
              "http://example.com/patch11.diff"],
      :p0 => ["http://example.com/patch00.diff",
              "http://example.com/patch01.diff"]
    )

    assert_equal 4, patches.length
    assert_equal 2, patches.select { |p| p.strip == :p0 }.length
    assert_equal 2, patches.select { |p| p.strip == :p1 }.length
  end

  def test_nil
    assert_empty Patch.normalize_legacy_patches(nil)
  end
end