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

179 lines
3.8 KiB
Ruby
Raw Normal View History

require 'abstract_unit'
2009-02-27 22:25:45 -05:00
class GrandParent
include ActiveSupport::Callbacks
2009-02-27 22:25:45 -05:00
attr_reader :log, :action_name
def initialize(action_name)
@action_name, @log = action_name, []
end
2009-02-27 22:25:45 -05:00
define_callbacks :dispatch
2012-02-04 06:31:00 -05:00
set_callback :dispatch, :before, :before1, :before2, :if => proc {|c| c.action_name == "index" || c.action_name == "update" }
set_callback :dispatch, :after, :after1, :after2, :if => proc {|c| c.action_name == "update" || c.action_name == "delete" }
2009-02-27 22:25:45 -05:00
def before1
@log << "before1"
end
2009-02-27 22:25:45 -05:00
def before2
@log << "before2"
end
def after1
2009-02-27 22:25:45 -05:00
@log << "after1"
end
2009-02-27 22:25:45 -05:00
def after2
@log << "after2"
end
2009-02-27 22:25:45 -05:00
def dispatch
run_callbacks :dispatch do
2009-02-27 22:25:45 -05:00
@log << action_name
end
self
end
end
class Parent < GrandParent
2012-02-04 06:31:00 -05:00
skip_callback :dispatch, :before, :before2, :unless => proc {|c| c.action_name == "update" }
skip_callback :dispatch, :after, :after2, :unless => proc {|c| c.action_name == "delete" }
2009-02-27 22:25:45 -05:00
end
class Child < GrandParent
2012-02-04 06:31:00 -05:00
skip_callback :dispatch, :before, :before2, :unless => proc {|c| c.action_name == "update" }, :if => :state_open?
2009-02-27 22:25:45 -05:00
def state_open?
@state == :open
end
2009-02-27 22:25:45 -05:00
def initialize(action_name, state)
super(action_name)
@state = state
end
end
class EmptyParent
include ActiveSupport::Callbacks
def performed?
@performed ||= false
end
define_callbacks :dispatch
def perform!
@performed = true
end
def dispatch
run_callbacks :dispatch
self
end
end
class EmptyChild < EmptyParent
set_callback :dispatch, :before, :do_nothing
def do_nothing
end
end
2009-02-27 22:25:45 -05:00
class CountingParent
include ActiveSupport::Callbacks
attr_reader :count
define_callbacks :dispatch
def initialize
@count = 0
end
def count!
@count += 1
end
def dispatch
run_callbacks(:dispatch)
self
end
end
class CountingChild < CountingParent
end
2012-01-05 20:12:46 -05:00
class BasicCallbacksTest < ActiveSupport::TestCase
2009-02-27 22:25:45 -05:00
def setup
@index = GrandParent.new("index").dispatch
@update = GrandParent.new("update").dispatch
@delete = GrandParent.new("delete").dispatch
@unknown = GrandParent.new("unknown").dispatch
end
2012-02-04 06:31:00 -05:00
def test_basic_conditional_callback1
2009-02-27 22:25:45 -05:00
assert_equal %w(before1 before2 index), @index.log
end
2012-02-04 06:31:00 -05:00
def test_basic_conditional_callback2
2009-02-27 22:25:45 -05:00
assert_equal %w(before1 before2 update after2 after1), @update.log
end
2012-02-04 06:31:00 -05:00
def test_basic_conditional_callback3
2009-02-27 22:25:45 -05:00
assert_equal %w(delete after2 after1), @delete.log
end
end
2012-01-05 20:12:46 -05:00
class InheritedCallbacksTest < ActiveSupport::TestCase
2009-02-27 22:25:45 -05:00
def setup
@index = Parent.new("index").dispatch
@update = Parent.new("update").dispatch
@delete = Parent.new("delete").dispatch
@unknown = Parent.new("unknown").dispatch
end
2009-02-27 22:25:45 -05:00
def test_inherited_excluded
assert_equal %w(before1 index), @index.log
end
2009-02-27 22:25:45 -05:00
def test_inherited_not_excluded
assert_equal %w(before1 before2 update after1), @update.log
end
2009-02-27 22:25:45 -05:00
def test_partially_excluded
assert_equal %w(delete after2 after1), @delete.log
end
end
2012-01-05 20:12:46 -05:00
class InheritedCallbacksTest2 < ActiveSupport::TestCase
2009-02-27 22:25:45 -05:00
def setup
@update1 = Child.new("update", :open).dispatch
@update2 = Child.new("update", :closed).dispatch
end
2009-02-27 22:25:45 -05:00
def test_crazy_mix_on
assert_equal %w(before1 update after2 after1), @update1.log
end
2009-02-27 22:25:45 -05:00
def test_crazy_mix_off
assert_equal %w(before1 before2 update after2 after1), @update2.log
end
end
2012-01-05 20:12:46 -05:00
class DynamicInheritedCallbacks < ActiveSupport::TestCase
def test_callbacks_looks_to_the_superclass_before_running
child = EmptyChild.new.dispatch
assert !child.performed?
EmptyParent.set_callback :dispatch, :before, :perform!
child = EmptyChild.new.dispatch
assert child.performed?
end
def test_callbacks_should_be_performed_once_in_child_class
CountingParent.set_callback(:dispatch, :before) { count! }
child = CountingChild.new.dispatch
assert_equal 1, child.count
end
end