Ticket #6514: caching_patch_tests.rb

require "fileutils"
require File.dirname(__FILE__) + "/../abstract_unit"

CACHE_DIR = 'test_cache'
# Don't change '/../temp/' cavalierly or you might hoze something you don't want hozed
FILE_STORE_PATH = File.join(File.dirname(__FILE__), '/../temp/', CACHE_DIR)
ActionController::Base.perform_caching = true

class MakeBelieveController < ActionController::Base
  caches_page :index, :moved, :target
  
  def index
    @var = "before render"
    render :text => "index"
    @var = "after render"
  end
  
  def moved
    redirect_to :action => "target"
  end
  
  def target
    render :text => "target"
  end
end

class PageCachingFixOneTest < Test::Unit::TestCase
  def setup
    @controller = MakeBelieveController.new
    @request, @response = ActionController::TestRequest.new, ActionController::TestResponse.new
    ActionController::Routing::Routes.draw do |map|
      map.index "/index", :controller => "make_believe", :action => "index"
    end
    FileUtils.mkdir_p(FILE_STORE_PATH)
  end
  
  def teardown
    FileUtils.rm_rf(File.dirname(FILE_STORE_PATH))
  end
  
  def test_index_and_reasons_for_failure
    get :index
    # Woops! Looks like there's something wrong.
    assert_response :error
    # Because @var == "after render", the problem cannot be inside the method
    # but with something that occurs after it. 
    assert_equal assigns["var"], "after render"
    # I figured that must be the after_filter generated by caches_page.
    # I did some poking around and discovered that the class variable
    # ActionController::Base.page_cache_directory doesn't get set in the testing
    # environment because RAILS_ROOT isn't set there either. I'm throwing these next
    # assertions in to back me up.
    assert !defined?(RAILS_ROOT)
    # @@page_cache_directory is a class variable of ActionController::Base,
    # to my surprise.
    assert_equal ActionController::Base.page_cache_directory, ""
  end
end

# Let's try this again!
class PageCachingFixTwoTest < Test::Unit::TestCase
  def setup
    @controller = MakeBelieveController.new
    @request, @response = ActionController::TestRequest.new, ActionController::TestResponse.new
    ActionController::Routing::Routes.draw do |map|
      map.index "/index", :controller => "make_believe", :action => "index"
      map.redirect_me "/moved", :controller => "make_believe", :action => "moved"
      map.target "/target", :controller => "make_believe", :action => "target"
    end
    FileUtils.mkdir_p(FILE_STORE_PATH)
    # This time, let's make sure to set the page_cache_directory.
    ActionController::Base.page_cache_directory = FILE_STORE_PATH
  end
  
  def teardown
    FileUtils.rm_rf(File.dirname(FILE_STORE_PATH))
  end
  
  def test_index
    get :index
    # No errors this time.
    assert_response :success
    # Is the file there though? Yep.
    assert File.exists?(FILE_STORE_PATH + "/index.html")
  end
  
  def test_target
    get :target
    assert_response :success
    assert File.exists?(FILE_STORE_PATH + "/target.html")
  end
  
  # Now let's see what happens with a redirect.
  def test_moved
    get :moved
    assert_response :redirect
    # The page still gets cached and every subsequent request for that URI will
    # return the content of /moved.html instead of performing a redirect.
    # This is not actually the case in the test environment [Surprise!] but it is
    # in the production environment.
    assert File.exists?(FILE_STORE_PATH + "/moved.html")
  end
end

class PageCachingFixThreeTest < Test::Unit::TestCase
  def setup
    @controller = MakeBelieveController.new
    @request, @response = ActionController::TestRequest.new, ActionController::TestResponse.new
    ActionController::Routing::Routes.draw do |map|
      map.index "/index", :controller => "make_believe", :action => "index"
      map.redirect_me "/moved", :controller => "make_believe", :action => "moved"
      map.target "/target", :controller => "make_believe", :action => "target"
    end
    FileUtils.mkdir_p(FILE_STORE_PATH)
    # Still make sure to set the page_cache_directory.
    ActionController::Base.page_cache_directory = FILE_STORE_PATH
    # And also change that we redefine caching_allowed
    ActionController::Caching::Pages.module_eval { def caching_allowed(); !request.post? && response.headers['Status'] && response.headers['Status'].to_i == 200; end}
  end
  
  def teardown
    # Restore that bad method because you never know who is going to be called first!
    ActionController::Caching::Pages.module_eval { def caching_allowed(); !request.post? && response.headers['Status'] && response.headers['Status'].to_i < 400; end}
    FileUtils.rm_rf(File.dirname(FILE_STORE_PATH))
  end
  
  def test_index
    get :index
    # No errors this time.
    assert_response :success
    # Is the file there though? Yep.
    assert File.exists?(FILE_STORE_PATH + "/index.html")
  end
  
  def test_target
    get :target
    assert_response :success
    assert File.exists?(FILE_STORE_PATH + "/target.html")
  end
  
  # This time it doesn't cache the file! Hooray!
  def test_moved
    get :moved
    assert_response :redirect
    assert !File.exists?(FILE_STORE_PATH + "/moved.html")
  end
end