vanity/test/frameworks/rails/action_controller_test.rb

action_controller_test.rb

require "test_helper"

class UseVanityControllerTest < ActionController::TestCase
  tests UseVanityController

  def setup
    super
    metric :sugar_high
    new_ab_test :pie_or_cake do
      metrics :sugar_high
      alternatives :pie, :cake
      default :pie
    end

    # Class eval this instead of including in the controller to delay
    # execution until the request exists in the context of the test
    UseVanityController.class_eval do
      use_vanity :current_user
    end

    # Rails 3 configuration for cookies
    if ::Rails.respond_to?(:application)
      ::Rails.application.config.session_options[:domain] = '.foo.bar'
    end
  end

  def teardown
    super
  end

  def test_render_js_for_tests
    Vanity.playground.use_js!
    get :js
    assert_match /script.*v=pie_or_cake=.*script/m, @response.body
  end

  def test_working_render_js_for_tests
    Vanity.playground.use_js!
    get :working_js
    assert_match /script.*v=pie_or_cake=.*script/m, @response.body
  end

  def test_broken_render_js_for_tests
    Vanity.playground.use_js!
    get :broken_js
    assert_match /script.*v=pie_or_cake=.*script/m, @response.body
  end

  def test_render_model_js_for_tests
    Vanity.playground.use_js!
    get :model_js
    assert_match /script.*v=pie_or_cake=.*script/m, @response.body
  end

  def test_chooses_sets_alternatives_for_rails_tests
    experiment(:pie_or_cake).chooses(:pie)
    get :index
    assert_equal 'pie', @response.body

    experiment(:pie_or_cake).chooses(:cake)
    get :index
    assert_equal 'cake', @response.body
  end

  def test_adds_participant_to_experiment
    get :index
    assert_equal 1, experiment(:pie_or_cake).alternatives.map(&:participants).sum
  end

  def test_does_not_add_invalid_participant_to_experiment
    Vanity.playground.use_js!
    @request.user_agent = "Googlebot/2.1 ( http://www.google.com/bot.html)"
    get :index
    assert_equal 0, experiment(:pie_or_cake).alternatives.map(&:participants).sum
  end

  def test_vanity_cookie_is_persistent
    get :index
    cookie = @response["Set-Cookie"].to_s
    assert_match /vanity_id=[a-f0-9]{32};/, cookie
    expires = cookie[/expires=(.*)(;|$)/, 1]
    assert expires
    assert_in_delta Time.parse(expires), Time.now + 20 * 365 * 24 * 60 * 60, 1.day
  end

  def test_vanity_cookie_default_id
    get :index
    assert cookies["vanity_id"] =~ /^[a-f0-9]{32}$/
  end

  def test_vanity_cookie_retains_id
    @request.cookies["vanity_id"] = "from_last_time"
    get :index
    assert_equal "from_last_time",  cookies["vanity_id"]
  end

  def test_vanity_cookie_uses_configuration
    Vanity.configuration.cookie_name = "new_id"
    get :index
    assert cookies["new_id"] =~ /^[a-f0-9]{32}$/
  end

  def test_vanity_identity_set_from_cookie
    @request.cookies["vanity_id"] = "from_last_time"
    get :index
    assert_equal "from_last_time", @controller.send(:vanity_identity)
  end

  def test_vanity_identity_set_from_user
    @controller.current_user = stub("user", :id=>"user_id")
    get :index
    assert_equal "user_id", @controller.send(:vanity_identity)
  end

  def test_vanity_identity_with_null_user_falls_back_to_cookie
    @controller.current_user = stub("user", :id=>nil)
    get :index
    assert cookies["vanity_id"] =~ /^[a-f0-9]{32}$/
  end

  def test_vanity_identity_with_no_user_model
    UseVanityController.class_eval do
      use_vanity nil
    end
    @controller.current_user = Object.new
    get :index
    assert cookies["vanity_id"] =~ /^[a-f0-9]{32}$/
  end

  def test_vanity_identity_set_with_block
    UseVanityController.class_eval do
      attr_accessor :project_id
      use_vanity { |controller| controller.project_id }
    end
    @controller.project_id = "576"
    get :index
    assert_equal "576", @controller.send(:vanity_identity)
  end

  def test_vanity_identity_set_with_identity_paramater
    get :index, :_identity => "id_from_params"
    assert_equal "id_from_params", @controller.send(:vanity_identity)
  end

  def test_vanity_identity_prefers_block_over_symbol
    UseVanityController.class_eval do
      attr_accessor :project_id
      use_vanity(:current_user) { |controller| controller.project_id }
    end
    @controller.project_id = "576"
    @controller.current_user = stub(:id=>"user_id")

    get :index
    assert_equal "576", @controller.send(:vanity_identity)
  end

    def test_vanity_identity_prefers_parameter_over_cookie
    @request.cookies['vanity_id'] = "old_id"
    get :index, :_identity => "id_from_params"
    assert_equal "id_from_params", @controller.send(:vanity_identity)
    assert cookies['vanity_id'], "id_from_params"
  end

  def test_vanity_identity_prefers_cookie_over_object
    @request.cookies['vanity_id'] = "from_last_time"
    @controller.current_user = stub(:id=>"user_id")
    get :index
    assert_equal "from_last_time", @controller.send(:vanity_identity)
  end

  # query parameter filter

  def test_redirects_and_loses_vanity_query_parameter
    get :index, :foo=>"bar", :_vanity=>"567"
    assert_redirected_to "/use_vanity?foo=bar"
  end

  def test_sets_choices_from_vanity_query_parameter
    first = experiment(:pie_or_cake).alternatives.first
    fingerprint = experiment(:pie_or_cake).fingerprint(first)
    10.times do
      @controller = nil ; setup_controller_request_and_response
      get :index, :_vanity => fingerprint
      assert_equal experiment(:pie_or_cake).choose, experiment(:pie_or_cake).alternatives.first
      assert experiment(:pie_or_cake).showing?(first)
    end
  end

  def test_does_nothing_with_vanity_query_parameter_for_posts
    experiment(:pie_or_cake).chooses(experiment(:pie_or_cake).alternatives.last.value)
    first = experiment(:pie_or_cake).alternatives.first
    fingerprint = experiment(:pie_or_cake).fingerprint(first)
    post :index, :foo => "bar", :_vanity => fingerprint
    assert_response :success
    assert !experiment(:pie_or_cake).showing?(first)
  end

  def test_track_param_tracks_a_metric
    get :index, :_identity => "123", :_track => "sugar_high"
    assert_equal experiment(:pie_or_cake).alternatives[0].converted, 1
  end

  def test_cookie_domain_from_rails_configuration
    get :index
    assert_match /domain=.foo.bar/, @response["Set-Cookie"] if ::Rails.respond_to?(:application)
  end

end