require "abstract_unit" class CurrentAttributesTest < ActiveSupport::TestCase Person = Struct.new(:name, :time_zone) class Current < ActiveSupport::CurrentAttributes attribute :world, :account, :person, :request delegate :time_zone, to: :person resets { Time.zone = "UTC" } def account=(account) super self.person = "#{account}'s person" end def person=(person) super Time.zone = person.try(:time_zone) end def request "#{super} something" end def intro "#{person.name}, in #{time_zone}" end end setup { Current.reset } 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 assert_equal "account/1's person", Current.person end test "set auxiliary class via overwritten method" do Current.person = Person.new("David", "Central Time (US & Canada)") assert_equal "Central Time (US & Canada)", Time.zone.name end test "resets auxiliary class via callback" do Current.person = Person.new("David", "Central Time (US & Canada)") assert_equal "Central Time (US & Canada)", Time.zone.name Current.reset assert_equal "UTC", Time.zone.name 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 Current.person = Person.new("David", "Central Time (US & Canada)") 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 Current.person = Person.new("David", "Central Time (US & Canada)") assert_equal "David, in Central Time (US & Canada)", Current.intro assert_equal "David, in Central Time (US & Canada)", Current.instance.intro end end