2017-07-09 08:06:36 -04:00
|
|
|
# frozen_string_literal: true
|
2017-07-10 09:39:13 -04:00
|
|
|
|
2017-05-26 14:00:27 -04:00
|
|
|
require "abstract_unit"
|
|
|
|
|
|
|
|
class CurrentAttributesTest < ActiveSupport::TestCase
|
2019-01-27 05:18:40 -05:00
|
|
|
Person = Struct.new(:id, :name, :time_zone)
|
2017-05-26 14:00:27 -04:00
|
|
|
|
|
|
|
class Current < ActiveSupport::CurrentAttributes
|
|
|
|
attribute :world, :account, :person, :request
|
|
|
|
delegate :time_zone, to: :person
|
|
|
|
|
2019-08-01 03:41:26 -04:00
|
|
|
before_reset { Session.previous = person&.id }
|
2019-01-27 05:18:40 -05:00
|
|
|
|
|
|
|
resets do
|
|
|
|
Time.zone = "UTC"
|
|
|
|
Session.current = nil
|
|
|
|
end
|
2017-05-26 14:00:27 -04:00
|
|
|
|
|
|
|
def account=(account)
|
|
|
|
super
|
2019-08-01 03:41:26 -04:00
|
|
|
self.person = Person.new(1, "#{account}'s person")
|
2017-05-26 14:00:27 -04:00
|
|
|
end
|
|
|
|
|
|
|
|
def person=(person)
|
|
|
|
super
|
2019-08-01 03:41:26 -04:00
|
|
|
Time.zone = person&.time_zone
|
|
|
|
Session.current = person&.id
|
2017-05-26 14:00:27 -04:00
|
|
|
end
|
|
|
|
|
|
|
|
def request
|
|
|
|
"#{super} something"
|
|
|
|
end
|
|
|
|
|
|
|
|
def intro
|
|
|
|
"#{person.name}, in #{time_zone}"
|
|
|
|
end
|
|
|
|
end
|
|
|
|
|
2019-01-27 05:18:40 -05:00
|
|
|
class Session < ActiveSupport::CurrentAttributes
|
|
|
|
attribute :current, :previous
|
|
|
|
end
|
|
|
|
|
2017-05-29 18:58:00 -04:00
|
|
|
setup do
|
|
|
|
@original_time_zone = Time.zone
|
|
|
|
Current.reset
|
2019-01-27 05:18:40 -05:00
|
|
|
Session.reset
|
2017-05-29 18:58:00 -04:00
|
|
|
end
|
|
|
|
|
|
|
|
teardown do
|
|
|
|
Time.zone = @original_time_zone
|
|
|
|
end
|
2017-05-26 14:00:27 -04:00
|
|
|
|
|
|
|
test "read and write attribute" do
|
|
|
|
Current.world = "world/1"
|
|
|
|
assert_equal "world/1", Current.world
|
|
|
|
end
|
|
|
|
|
|
|
|
test "read overwritten attribute method" do
|
|
|
|
Current.request = "request/1"
|
|
|
|
assert_equal "request/1 something", Current.request
|
|
|
|
end
|
|
|
|
|
|
|
|
test "set attribute via overwritten method" do
|
|
|
|
Current.account = "account/1"
|
|
|
|
assert_equal "account/1", Current.account
|
2019-08-01 03:41:26 -04:00
|
|
|
assert_equal "account/1's person", Current.person.name
|
2017-05-26 14:00:27 -04:00
|
|
|
end
|
|
|
|
|
|
|
|
test "set auxiliary class via overwritten method" do
|
2019-01-27 05:18:40 -05:00
|
|
|
Current.person = Person.new(42, "David", "Central Time (US & Canada)")
|
2017-05-26 14:00:27 -04:00
|
|
|
assert_equal "Central Time (US & Canada)", Time.zone.name
|
2019-01-27 05:18:40 -05:00
|
|
|
assert_equal 42, Session.current
|
2017-05-26 14:00:27 -04:00
|
|
|
end
|
|
|
|
|
2019-01-27 05:18:40 -05:00
|
|
|
test "resets auxiliary classes via callback" do
|
|
|
|
Current.person = Person.new(42, "David", "Central Time (US & Canada)")
|
2017-05-26 14:00:27 -04:00
|
|
|
assert_equal "Central Time (US & Canada)", Time.zone.name
|
|
|
|
|
|
|
|
Current.reset
|
|
|
|
assert_equal "UTC", Time.zone.name
|
2019-01-27 05:18:40 -05:00
|
|
|
assert_nil Session.current
|
|
|
|
end
|
|
|
|
|
|
|
|
test "set auxiliary class based on current attributes via before callback" do
|
|
|
|
Current.person = Person.new(42, "David", "Central Time (US & Canada)")
|
|
|
|
assert_nil Session.previous
|
|
|
|
assert_equal 42, Session.current
|
|
|
|
|
|
|
|
Current.reset
|
|
|
|
assert_equal 42, Session.previous
|
|
|
|
assert_nil Session.current
|
2017-05-26 14:00:27 -04:00
|
|
|
end
|
|
|
|
|
|
|
|
test "set attribute only via scope" do
|
|
|
|
Current.world = "world/1"
|
|
|
|
|
|
|
|
Current.set(world: "world/2") do
|
|
|
|
assert_equal "world/2", Current.world
|
|
|
|
end
|
|
|
|
|
|
|
|
assert_equal "world/1", Current.world
|
|
|
|
end
|
|
|
|
|
|
|
|
test "set multiple attributes" do
|
|
|
|
Current.world = "world/1"
|
|
|
|
Current.account = "account/1"
|
|
|
|
|
|
|
|
Current.set(world: "world/2", account: "account/2") do
|
|
|
|
assert_equal "world/2", Current.world
|
|
|
|
assert_equal "account/2", Current.account
|
|
|
|
end
|
|
|
|
|
|
|
|
assert_equal "world/1", Current.world
|
|
|
|
assert_equal "account/1", Current.account
|
|
|
|
end
|
|
|
|
|
|
|
|
test "delegation" do
|
2019-01-27 05:18:40 -05:00
|
|
|
Current.person = Person.new(42, "David", "Central Time (US & Canada)")
|
2017-05-26 14:00:27 -04:00
|
|
|
assert_equal "Central Time (US & Canada)", Current.time_zone
|
|
|
|
assert_equal "Central Time (US & Canada)", Current.instance.time_zone
|
|
|
|
end
|
|
|
|
|
|
|
|
test "all methods forward to the instance" do
|
2019-01-27 05:18:40 -05:00
|
|
|
Current.person = Person.new(42, "David", "Central Time (US & Canada)")
|
2017-05-26 14:00:27 -04:00
|
|
|
assert_equal "David, in Central Time (US & Canada)", Current.intro
|
|
|
|
assert_equal "David, in Central Time (US & Canada)", Current.instance.intro
|
|
|
|
end
|
|
|
|
end
|