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_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
  
  #Test regexp in when parsing the original email
  def test_multipart
   bounce = test_bounce('tt_bounce_25')
    assert bounce.bounced?
    assert_not_nil bounce.original_mail
  end
  
end