1
0
Fork 0
mirror of https://github.com/rails/rails.git synced 2022-11-09 12:12:34 -05:00
rails--rails/railties/test/application/current_attributes_integration_test.rb

88 lines
2.1 KiB
Ruby

# frozen_string_literal: true
require "isolation/abstract_unit"
require "rack/test"
class CurrentAttributesIntegrationTest < ActiveSupport::TestCase
include ActiveSupport::Testing::Isolation
include Rack::Test::Methods
setup do
build_app
app_file "app/models/current.rb", <<-RUBY
class Current < ActiveSupport::CurrentAttributes
attribute :customer
resets { Time.zone = "UTC" }
def customer=(customer)
super
Time.zone = customer&.time_zone
end
end
RUBY
app_file "app/models/customer.rb", <<-RUBY
class Customer < Struct.new(:name)
def time_zone
"Copenhagen"
end
end
RUBY
app_file "config/routes.rb", <<-RUBY
Rails.application.routes.draw do
get "/customers/:action", controller: :customers
end
RUBY
app_file "app/controllers/customers_controller.rb", <<-RUBY
class CustomersController < ApplicationController
layout false
def set_current_customer
Current.customer = Customer.new("david")
render :index
end
def set_no_customer
render :index
end
end
RUBY
app_file "app/views/customers/index.html.erb", <<-RUBY
<%= Current.customer&.name || 'noone' %>,<%= Time.zone.name %>
RUBY
require "#{app_path}/config/environment"
end
teardown :teardown_app
test "current customer is assigned and cleared" do
get "/customers/set_current_customer"
assert_equal 200, last_response.status
assert_match(/david,Copenhagen/, last_response.body)
get "/customers/set_no_customer"
assert_equal 200, last_response.status
assert_match(/noone,UTC/, last_response.body)
end
test "resets after execution" do
assert_nil Current.customer
assert_equal "UTC", Time.zone.name
Rails.application.executor.wrap do
Current.customer = Customer.new("david")
assert_equal "david", Current.customer.name
assert_equal "Copenhagen", Time.zone.name
end
assert_nil Current.customer
assert_equal "UTC", Time.zone.name
end
end