require "minitest/autorun"
require "email_reply_trimmer"

class TestEmailReplyTrimmer < Minitest::Test

  EMAILS  = Dir["test/emails/*.txt"].map  { |path| File.basename(path) }
  TRIMMED = Dir["test/trimmed/*.txt"].map { |path| File.basename(path) }
  ELIDED  = Dir["test/elided/*.txt"].map { |path| File.basename(path) }

  def test_all_emails_have_a_matching_reply
    assert_equal(EMAILS, TRIMMED, "Files in /emails and /trimmed folders should match 1-to-1.")
  end

  def test_all_emails_have_a_matching_elided
    assert_equal(EMAILS, ELIDED, "Files in /emails and /elided folders should match 1-to-1.")
  end

  def test_normalize_line_endings_email_has_windows_line_endings
    assert_match(/\r\n/, File.read("test/emails/normalize_line_endings.txt"))
  end

  EMAILS.each do |filename|
    name = File.basename(filename, ".txt")
    define_method("test_#{name}") do
      assert_equal(trim(filename), trimmed(filename), "[TRIMMED] EMAIL: #{filename}")
      assert_equal(elide(filename), elided(filename), "[ELIDED] EMAIL: #{filename}")
    end
  end

  EMBEDDED_EMAILS = %w{
    email_headers_1 email_headers_2 email_headers_3 email_headers_4
    embedded_email_10 embedded_email_german_3 embedded_email_spanish_2
    forwarded_message
  }

  EMBEDDED_EMAILS.each do |name|
    filename = "#{name}.txt"
    define_method("test_embedded_extraction_for_#{name}") do
      e, b = extract_embedded_email(filename)
      assert_equal(e, embedded(filename), "[EMBEDDED] EMAIL: #{filename}")
      assert_equal(b, before(filename), "[BEFORE] EMAIL: #{filename}")
    end
  end

  def trim(filename)
    EmailReplyTrimmer.trim(email(filename))
  end

  def elide(filename)
    EmailReplyTrimmer.trim(email(filename), true)[1]
  end

  def extract_embedded_email(filename)
    EmailReplyTrimmer.extract_embedded_email(email(filename))
  end

  def email(filename)
    File.read("test/emails/#{filename}").strip
  end

  def trimmed(filename)
    File.read("test/trimmed/#{filename}").strip
  end

  def elided(filename)
    File.read("test/elided/#{filename}").strip
  end

  def embedded(filename)
    File.read("test/embedded/#{filename}").strip
  end

  def before(filename)
    File.read("test/before/#{filename}").strip
  end

end
