timeout_increaser_test.rb

require 'minitest/autorun'
require_relative '../../lib/timeout_increaser'

class FakeImapConnection 
  attr_accessor :timeout
  def initialize
    @timeout = 100
  end

  def reconnect
  end
end

module Timeout
  class Error < StandardError; end
end

class FakeAttachmentSaver
  def retrieve_message_body(tracked_msg, uid)
    raise Timeout::Error
  end
end

class TimeoutIncreaserTest < MiniTest::Unit::TestCase
  def setup 
    @imap_connection = FakeImapConnection.new
    @uid = 'uid'
  end

  def test_connection_retry 
    tracked_msg = 'msg'
    attacher = MiniTest::Mock.new
    attacher.expect :retrieve_message_body, true, [tracked_msg, @uid]
    TimeoutIncreaser.increase(attacher, 
                              @imap_connection, tracked_msg, @uid)
    attacher.verify
  end

  def test_process_attachment_if_retry_failed
    attacher = FakeAttachmentSaver.new
    tracked_msg = MiniTest::Mock.new
    tracked_msg.expect :attachments_processed!, true 
    TimeoutIncreaser.increase(attacher, @imap_connection, 
                              tracked_msg, @uid)
    tracked_msg.verify
  end
end