require File.dirname(__FILE__) + '/test_helper.rb'

class BounceEmailTest < Test::Unit::TestCase

  def test_bounce_type_hard_fail
    bounce = test_bounce('tt_bounce_01')
    assert_equal '5.1.2', bounce.code, "Code should return 5.1.2, returns #{bounce.code}"
    assert_equal BounceEmail::TYPE_HARD_FAIL, bounce.type
  end

  #  Specific tests
  def test_unrouteable_mail_domain
    bounce = test_bounce('tt_bounce_01')
    assert_equal '5.1.2', bounce.code, "Code should return 5.1.2, returns #{bounce.code}"

    bounce = test_bounce('tt_bounce_02')
    assert_equal '5.1.2', bounce.code, "Code should return 5.1.2, returns #{bounce.code}"
  end

  def test_set_5_0_status
    bounce = test_bounce('tt_bounce_03')
    assert_equal '5.0.0', bounce.code, "Code should return 5.0.0, returns #{bounce.code}"

    bounce = test_bounce('tt_bounce_04')
    assert_equal '5.0.0', bounce.code, "Code should return 5.0.0, returns #{bounce.code}"

    bounce = test_bounce('tt_bounce_05')
    assert_equal '5.0.0', bounce.code, "Code should return 5.0.0, returns #{bounce.code}"
  end

  def test_rota_dnsbl # TODO make this more general (match DNSBL only?)
    bounce = test_bounce('tt_bounce_06')
    assert_equal '5.7.1', bounce.code, "Code should return 5.7.1, returns #{bounce.code}"
  end

  # this test email suggests the library fails on this email;
  # mail.part[0] includes a specific status code (5.1.1 User unknown)
  # but the library tests mail.part[1], which returns the general code (5.0.0)
  # either the test email is not a good example, or the parsing could be improved
  def test_user_unknown
    bounce = test_bounce('tt_bounce_07')
    assert_equal '5.0.0', bounce.code
  end

  def test_permanent_failure
    bounce = test_bounce('tt_bounce_08')
    assert_equal '5.3.2', bounce.code

    bounce = test_bounce('tt_bounce_09')
    assert_equal '5.3.2', bounce.code
  end

  def test_bounce_type_soft_fail
    bounce = test_bounce('tt_bounce_10')
    assert_equal '4.0.0', bounce.code, "Code should return 4.0.0, returns #{bounce.code}"
    assert_equal BounceEmail::TYPE_SOFT_FAIL, bounce.type
  end

  # Added because kept getting errors with malformed bounce messages
  def test_malformed_bounce
    bounce = test_bounce('malformed_bounce_01')
    assert_equal '5.1.1', bounce.code
  end

  # Added because kept getting errors with unknown code messages
  def test_unknown_code
    bounce = test_bounce('unknown_code_bounce_01')
    assert bounce.bounced?
    assert_equal 'unknown', bounce.code
    assert_equal BounceEmail::TYPE_HARD_FAIL, bounce.type
    assert_equal 'unknown', bounce.reason
  end

  # test all other files
  def test_all_bounces
    path = File.join(File.dirname(__FILE__), 'bounces')
    Dir[path + "/*.txt"].map do |file|
      bounce = BounceEmail::Mail.new Mail.read(file)
      assert bounce.bounced?, "#{file} failed"
    end
  end

  def test_all_non_bounces
    path = File.join(File.dirname(__FILE__), 'non_bounces')
    Dir[path + "/*.txt"].map do |file|
      non_bounce = BounceEmail::Mail.new Mail.read(file)
      assert !non_bounce.bounced?, "#{file} failed"
    end
  end

  def test_does_not_fail_if_subject_is_nil
    bounce = BounceEmail::Mail.new load_email('no_subject')
    assert !bounce.bounced?
  end

  def test_mail_methods_fallback
    bounce = test_bounce('tt_bounce_10')
    assert bounce.body
    assert bounce.date
  end

  #Test mutlipart message from exchange
  def test_multipart
    bounce = test_bounce('tt_bounce_24')
    assert bounce.bounced?
    assert_equal BounceEmail::TYPE_HARD_FAIL, bounce.type
    assert_not_nil bounce.original_mail
  end

  def test_original_message_with_multipart_mails
    multipart_mails = %w(05 07 10 11 13 15 16 23 24)
    multipart_mails.map do |file|
      mail = File.join(File.dirname(__FILE__), 'bounces', "tt_bounce_#{file}.txt")
      bounce = BounceEmail::Mail.new Mail.read(mail)
      assert_not_nil bounce.original_mail
      assert_not_nil bounce.original_mail.message_id
      assert_not_nil bounce.original_mail.to
      assert_not_nil bounce.original_mail.from
    end
  end

  def test_original_message_with_multipart_mails_without_to_field
    multipart_mails = %w(03 04)
    multipart_mails.map do |file|
      mail = File.join(File.dirname(__FILE__), 'bounces', "tt_bounce_#{file}.txt")
      bounce = BounceEmail::Mail.new Mail.read(mail)
      assert_not_nil bounce.original_mail
      assert_not_nil bounce.original_mail.message_id
      assert_equal [], bounce.original_mail.to
      assert_not_nil bounce.original_mail.from
    end
  end

  def test_original_message_without_inline_original_message
    bounce = test_bounce('tt_bounce_01')
    assert_nil bounce.original_mail
  end

  def test_original_message_with_inline_original_message
    mails_with_inline_original_message = %w(06 08 09 12_soft 14 17 18 19 20 21 22 25)
    mails_with_inline_original_message.map do |file|
      mail = File.join(File.dirname(__FILE__), 'bounces', "tt_bounce_#{file}.txt")
      bounce = BounceEmail::Mail.new Mail.read(mail)
      assert_not_nil bounce.original_mail
      assert_not_nil bounce.original_mail.message_id
      assert_not_nil bounce.original_mail.to
      assert_not_nil bounce.original_mail.from
    end
  end

  def test_original_message_with_subject
    bounce = test_bounce('tt_bounce_04')
    assert_not_nil bounce.original_mail.subject
  end

  def test_original_message_with_bounced_gmail
    bounce = test_bounce('undeliverable_gmail')
    assert_not_nil bounce.original_mail
    assert_not_nil bounce.original_mail.message_id
    assert_not_nil bounce.original_mail.to
    assert_not_nil bounce.original_mail.from
    assert_not_nil bounce.original_mail.subject
  end
  private

  def load_email(name, prefix = 'fixtures')
    default_extention = '.txt'
    name << default_extention unless name.end_with?(default_extention)

    path = File.join(File.dirname(__FILE__), prefix, name)
    Mail.read(path)
  end
end