ENV['RACK_ENV'] = 'test'

begin
  require 'rack'
rescue LoadError
  require 'rubygems'
  require 'rack'
end

require 'contest'
require 'sinatra'
require 'erubis'
require 'haml'
require 'rack/test'

begin
  require 'redgreen'
rescue LoadError
end

$LOAD_PATH.unshift('./lib')

require 'sinatra/content_for2'

Sinatra::Base.set :environment, :test

module Sinatra
  class Base
    set :environment, :test
    helpers ContentFor2
  end
end

class Test::Unit::TestCase
  include Rack::Test::Methods

  class << self
    alias_method :it, :test
  end

  def mock_app(base=Sinatra::Base, &block)
    @app = Sinatra.new(base, &block)
  end

  def app
    @app
  end
end

class ContentForTest < Test::Unit::TestCase
  context 'using erubis' do
    def erubis_app(view)
      mock_app {
        layout { '<% yield_content :foo %>' }
        get('/') { erb view } 
      }
    end

    it 'renders blocks declared with the same key you use when rendering' do
      erubis_app '<% content_for :foo do %>foo<% end %>'

      get '/'
      assert last_response.ok?
      assert_equal 'foo', last_response.body
    end

    it 'does not render a block with a different key' do
      erubis_app '<% content_for :bar do %>bar<% end %>'

      get '/'
      assert last_response.ok?
      assert_equal '', last_response.body
    end

    it 'renders multiple blocks with the same key' do
      erubis_app <<-erb_snippet
        <% content_for :foo do %>foo<% end %>
        <% content_for :foo do %>bar<% end %>
        <% content_for :baz do %>WON'T RENDER ME<% end %>
        <% content_for :foo do %>baz<% end %>
      erb_snippet

      get '/'
      assert last_response.ok?
      assert_equal 'foobarbaz', last_response.body
    end

    it 'passes values to the blocks' do
      mock_app {
        layout { '<% yield_content :foo, 1, 2 %>' }
        get('/') { erb '<% content_for :foo do |a, b| %><i><%= a %></i> <%= b %><% end %>' }
      }

      get '/'
      assert last_response.ok?
      assert_equal '<i>1</i> 2', last_response.body
    end
  end

  context 'with haml' do
    def haml_app(view)
      mock_app {
        layout { '= yield_content :foo' }
        get('/') { haml view } 
      }
    end

    it 'renders blocks declared with the same key you use when rendering' do
      haml_app <<-haml_end
- content_for :foo do
  foo
haml_end

      get '/'
      assert last_response.ok?
      assert_equal "foo\n", last_response.body
    end

    it 'does not render a block with a different key' do
      haml_app <<-haml_end
- content_for :bar do
  bar
haml_end

      get '/'
      assert last_response.ok?
      assert_equal "\n", last_response.body
    end

    it 'renders multiple blocks with the same key' do
      haml_app <<-haml_end
- content_for :foo do
  foo
- content_for :foo do
  bar
- content_for :baz do
  WON'T RENDER ME
- content_for :foo do
  baz
haml_end

      get '/'
      assert last_response.ok?
      assert_equal "foo\nbar\nbaz\n", last_response.body
    end

    it 'passes values to the blocks' do
      mock_app {
        layout { '= yield_content :foo, 1, 2' }
        get('/') { 
          haml <<-haml_end
- content_for :foo do |a, b|
  %i= a
  =b
haml_end
        }
      }

      get '/'
      assert last_response.ok?
      assert_equal "<i>1</i>\n2\n", last_response.body
    end
  end

  context 'content_for?' do
    class TestApp
      include Sinatra::ContentFor2
    end

    setup do
      @helper = TestApp.new 
    end

    context 'defined content' do
      setup do
        @helper.instance_variable_set(:@content_blocks, {:test => ["block"]})
      end

      it 'returns true' do
        assert_equal @helper.content_for?(:test), true
      end
    end

    context 'undefined content' do
      setup do
        @helper.instance_variable_set(:@content_blocks, {:test => []})
      end

      it 'returns false' do
        assert_equal @helper.content_for?(:test), false
      end
    end
  end
end