sinatra/test/delegator_test.rb

class DelegatorTest < Test::Unit::TestCase
  class Mirror
    attr_reader :last_call

    Sinatra::Delegator.private_instance_methods.each do |method|
      define_method(method) do |*a, &b|
        method_missing(method, *a, &b)
      end
    end

    def method_missing(*a, &b)
      @last_call = [*a.map(&:to_s)]
      @last_call << b if b
    end
  end

  def self.delegates(name)
    it "delegates #{name}" do
      m = mirror { send name }
      assert_equal m.last_call, [name.to_s]
    end

    it "delegates #{name} with arguments" do
      m = mirror { send name, "foo", "bar" }
      assert_equal m.last_call, [name.to_s, "foo", "bar"]
    end

    it "delegates #{name} with block" do
      block = proc { }
      m = mirror { send(name, &block) }
      assert_equal m.last_call, [name.to_s, block]
    end
  end

  setup do
    @target_was = Sinatra::Delegator.target
  end

  def teardown
    Sinatra::Delegator.target = @target_was
  end

  def delegation_app(&block)
    mock_app { Sinatra::Delegator.target = self }
    delegate(&block)
  end

  def mirror(&block)
    mirror = Mirror.new
    Sinatra::Delegator.target = mirror
    delegate(&block)
  end

  def delegate(&block)
    assert Sinatra::Delegator.target != Sinatra::Application
    Object.new.extend(Sinatra::Delegator).instance_eval(&block) if block
    Sinatra::Delegator.target
  end

  def target
    Sinatra::Delegator.target
  end

  it 'defaults to Sinatra::Application as target' do
    assert_equal Sinatra::Delegator.target, Sinatra::Application
  end

  %w[get put post delete options patch].each do |verb|
    it "delegates #{verb} correctly" do
      delegation_app do
        send verb, '/hello' do
          'Hello World'
        end
      end

      request = Rack::MockRequest.new(@app)
      response = request.request(verb.upcase, '/hello', {})
      assert response.ok?
      assert_equal 'Hello World', response.body
    end
  end

  it "delegates head correctly" do
    delegation_app do
      head '/hello' do
        response['X-Hello'] = 'World!'
        'remove me'
      end
    end

    request = Rack::MockRequest.new(@app)
    response = request.request('HEAD', '/hello', {})
    assert response.ok?
    assert_equal 'World!', response['X-Hello']
    assert_equal '', response.body
  end

  it "registers extensions with the delegation target" do
    app, mixin = mirror, Module.new
    Sinatra.register mixin
    assert_equal app.last_call, ["register", mixin.to_s ]
  end

  it "registers helpers with the delegation target" do
    app, mixin = mirror, Module.new
    Sinatra.helpers mixin
    assert_equal app.last_call, ["helpers", mixin.to_s ]
  end

  it "registers helpers with the delegation target" do
    app, mixin = mirror, Module.new
    Sinatra.use mixin
    assert_equal app.last_call, ["use", mixin.to_s ]
  end

  it "should work with method_missing proxies for options" do
    mixin = Module.new do
      def method_missing(method, *args, &block)
        return super unless method.to_sym == :options
        {:some => :option}
      end
    end

    app = mirror
    def app.options(arg, *rest) "yay" end

    a = b = c = nil
    delegate do
      extend mixin
      a, b, c = options, options(1), options(1, 2)
    end

    assert_equal({:some => :option}, a)
    assert_equal("yay", b)
    assert_equal("yay", c)
  end

  it "should work with method_missing proxies for methods arity > 0" do
    mixin = Module.new do
      def method_missing(method, *args, &block)
        return super unless method.to_sym == :options
        {:some => :option}
      end
    end

    app = mirror
    def app.options(arg) "yay" end

    a = b = c = nil
    delegate do
      extend mixin
      a, b, c = options, options(1), options(1, 2)
    end

    assert_equal({:some => :option}, a)
    assert_equal("yay", b)
    assert_equal({:some => :option}, c)
  end

  it "should not raise a NameError for methods not handled by a proxy" do
    app = mirror
    def app.options(arg) "yay" end
    assert_raises(ArgumentError) { delegate { options }}
  end

  delegates 'get'
  delegates 'patch'
  delegates 'put'
  delegates 'post'
  delegates 'delete'
  delegates 'head'
  delegates 'options'
  delegates 'template'
  delegates 'layout'
  delegates 'before'
  delegates 'after'
  delegates 'error'
  delegates 'not_found'
  delegates 'configure'
  delegates 'set'
  delegates 'mime_type'
  delegates 'enable'
  delegates 'disable'
  delegates 'use'
  delegates 'development?'
  delegates 'test?'
  delegates 'production?'
  delegates 'helpers'
  delegates 'settings'
end