require 'test_helper'
require 'tmail/mailbox'
require 'fileutils'

class MailboxTester < Test::Unit::TestCase
  include FileUtils

  MAILBOX = '_mh'
  N = 5

  def setup
    rm_rf(MAILBOX, :verbose => false)
    mkdir(MAILBOX, :verbose => false)
    N.downto(1) do |i|
      File.open( "#{MAILBOX}/#{i}", 'w' ) {|f|
          f.puts 'From: aamine'
          f.puts 'To: aamine@loveruby.net'
          f.puts "Subject: #{i}"
          f.puts ''
          f.puts 'body'
      }
    end
    @n = N

    @ld = TMail::MhMailbox.new( MAILBOX )
  end

  def make_mails_older( diff )
    Dir.entries( MAILBOX ).collect {|n| "#{MAILBOX}/#{n}" }.each do |path|
      if File.file? path then
        t = File.mtime(path) - diff
        File.utime t, t, path
      end
    end
  end

  def teardown
    rm_rf(MAILBOX, :verbose => false)
  end

  def test_s_new
    ld = TMail::MhMailbox.new( MAILBOX )
    assert_instance_of TMail::MhMailbox, ld
  end

  def test_each_port
    dir = File.expand_path(MAILBOX)
    c = 0
    n = 0
    TMail::MhMailbox.new( MAILBOX ).each_port do |port|
      assert_kind_of TMail::FilePort, port
      assert_equal dir, File.dirname(port.filename)
      assert_match(/\A\d+\z/, File.basename(port.filename))
      nn = File.basename(port.filename).to_i
      assert nn > n
      n = nn
      c += 1
    end
    assert_equal N, c
  end

  def test_reverse_each_port
    dir = File.expand_path(MAILBOX)
    c = 0
    n = 100000
    TMail::MhMailbox.new( MAILBOX ).reverse_each_port do |port|
      assert_kind_of TMail::FilePort, port
      assert_equal dir, File.dirname(port.filename)
      assert_match(/\A\d+\z/, File.basename(port.filename))
      nn = File.basename(port.filename).to_i
      assert nn < n
      n = nn
      c += 1
    end
    assert_equal N, c
  end

  def test_new_port
    port = @ld.new_port
    assert_kind_of TMail::FilePort, port
    assert_equal File.expand_path('.') + '/' + MAILBOX,
                 File.dirname(port.filename)
    assert_equal( (N+1).to_s, File.basename(port.filename) )
    
    create port
  end

  def create( port )
    port.wopen {|f|
      f.puts 'From: aamine'
      f.puts 'To: aamine@loveruby.net'
      f.puts "Subject: #{@n + 1}"
      f.puts ''
      f.puts 'body'
    }
    @n += 1
  end

  def test_each_new_port
    make_mails_older 5
        
    c = 0
    @ld.each_new_port do |port|
      assert_kind_of TMail::FilePort, port
      c += 1
    end
    assert_equal @n, c

    t = Time.now - 2
    create @ld.new_port
    c = 0
    @ld.each_new_port( t ) do |port|
      assert_kind_of TMail::FilePort, port
      c += 1
    end
    assert_equal 1, c

    make_mails_older 5
    c = 0
    @ld.each_new_port do |port|
      assert_kind_of TMail::FilePort, port
      c += 1
    end
    assert_equal 0, c
  end
  
  def test_unix_mbox_fromaddr_method
    p = TMail::FilePort.new("#{File.dirname(__FILE__)}/fixtures/mailbox")
    assert_equal(TMail::UNIXMbox.fromaddr(p), "mikel@return_path.com")
  end
  
  def test_unix_mbox_fromaddr_method_missing_return_path
    p = TMail::FilePort.new("#{File.dirname(__FILE__)}/fixtures/mailbox_without_return_path")
    assert_equal(TMail::UNIXMbox.fromaddr(p), "mikel@from_address.com")
  end
  
  def test_unix_mbox_fromaddr_method_missing_from_address
    p = TMail::FilePort.new("#{File.dirname(__FILE__)}/fixtures/mailbox_without_from")
    assert_equal(TMail::UNIXMbox.fromaddr(p), "mike@envelope_sender.com.au")
  end
  
  def test_unix_mbox_from_addr_method_missing_all_from_fields_in_the_email
    p = TMail::FilePort.new("#{File.dirname(__FILE__)}/fixtures/mailbox_without_any_from_or_sender")
    assert_equal(TMail::UNIXMbox.fromaddr(p), "nobody")
  end
  
  def test_opening_mailbox_to_read_does_not_raise_IO_error
    mailbox = TMail::UNIXMbox.new("#{File.dirname(__FILE__)}/fixtures/mailbox", nil, true)
    assert_nothing_raised do
      mailbox.each_port do |port| 
        TMail::Mail.new(port) 
      end
    end
  end
  
  def test_reading_correct_number_of_emails_from_a_mailbox
    mailbox = TMail::UNIXMbox.new("#{File.dirname(__FILE__)}/fixtures/mailbox", nil, true)
    @emails = []
    mailbox.each_port { |m| @emails << TMail::Mail.new(m) }
    assert_equal(4, @emails.length)
  end

  def test_truncating_a_mailbox_to_zero_if_it_is_opened_with_readonly_false
    filename = "#{File.dirname(__FILE__)}/fixtures/mailbox"
    FileUtils.copy(filename, "#{filename}_test")
    filename = "#{filename}_test"
    mailbox = TMail::UNIXMbox.new(filename, nil, false)
    @emails = []
    mailbox.each_port { |m| @emails << TMail::Mail.new(m) }
    assert_equal(4, @emails.length)
    assert_equal('', File.read(filename))
    File.delete(filename)
  end
  
  def test_fromline2time_should_return_nil_on_out_of_range_dates
    filename = "#{File.dirname(__FILE__)}/fixtures/mailbox"
    FileUtils.copy(filename, "#{filename}_test")
    filename = "#{filename}_test"
    line = "From mikel mikel March 24 01:02:03 1900"
    mailbox = TMail::UNIXMbox.new(filename, nil, false)
    assert_equal(nil, mailbox.send(:fromline2time, line)) 
  end
  
end