2010-03-18 17:59:13 -04:00
|
|
|
require 'abstract_unit'
|
2009-10-12 23:15:43 -04:00
|
|
|
require 'test/unit'
|
2008-01-18 21:44:45 -05:00
|
|
|
|
2009-10-12 23:15:43 -04:00
|
|
|
module CallbacksTest
|
2010-12-16 22:04:20 -05:00
|
|
|
class Phone
|
|
|
|
include ActiveSupport::Callbacks
|
|
|
|
define_callbacks :save, :rescuable => true
|
|
|
|
|
|
|
|
set_callback :save, :before, :before_save1
|
|
|
|
set_callback :save, :after, :after_save1
|
|
|
|
|
|
|
|
def before_save1; self.history << :before; end
|
|
|
|
def after_save1; self.history << :after; end
|
|
|
|
|
|
|
|
def save
|
2011-01-09 13:15:05 -05:00
|
|
|
run_callbacks :save do
|
2010-12-16 22:04:20 -05:00
|
|
|
raise 'boom'
|
|
|
|
end
|
|
|
|
end
|
|
|
|
|
|
|
|
def history
|
|
|
|
@history ||= []
|
|
|
|
end
|
|
|
|
end
|
|
|
|
|
2009-10-12 23:15:43 -04:00
|
|
|
class Record
|
|
|
|
include ActiveSupport::Callbacks
|
2008-01-18 21:44:45 -05:00
|
|
|
|
2009-10-12 23:15:43 -04:00
|
|
|
define_callbacks :save
|
2008-01-18 21:44:45 -05:00
|
|
|
|
2009-10-12 23:15:43 -04:00
|
|
|
def self.before_save(*filters, &blk)
|
|
|
|
set_callback(:save, :before, *filters, &blk)
|
|
|
|
end
|
|
|
|
|
|
|
|
def self.after_save(*filters, &blk)
|
|
|
|
set_callback(:save, :after, *filters, &blk)
|
|
|
|
end
|
|
|
|
|
|
|
|
class << self
|
|
|
|
def callback_symbol(callback_method)
|
|
|
|
method_name = :"#{callback_method}_method"
|
|
|
|
define_method(method_name) do
|
|
|
|
history << [callback_method, :symbol]
|
|
|
|
end
|
|
|
|
method_name
|
|
|
|
end
|
|
|
|
|
|
|
|
def callback_string(callback_method)
|
|
|
|
"history << [#{callback_method.to_sym.inspect}, :string]"
|
|
|
|
end
|
|
|
|
|
|
|
|
def callback_proc(callback_method)
|
|
|
|
Proc.new { |model| model.history << [callback_method, :proc] }
|
|
|
|
end
|
|
|
|
|
|
|
|
def callback_object(callback_method)
|
|
|
|
klass = Class.new
|
|
|
|
klass.send(:define_method, callback_method) do |model|
|
|
|
|
model.history << [:"#{callback_method}_save", :object]
|
|
|
|
end
|
|
|
|
klass.new
|
2008-01-18 21:44:45 -05:00
|
|
|
end
|
|
|
|
end
|
|
|
|
|
2009-10-12 23:15:43 -04:00
|
|
|
def history
|
|
|
|
@history ||= []
|
|
|
|
end
|
|
|
|
end
|
|
|
|
|
|
|
|
class Person < Record
|
|
|
|
[:before_save, :after_save].each do |callback_method|
|
|
|
|
callback_method_sym = callback_method.to_sym
|
|
|
|
send(callback_method, callback_symbol(callback_method_sym))
|
|
|
|
send(callback_method, callback_string(callback_method_sym))
|
|
|
|
send(callback_method, callback_proc(callback_method_sym))
|
|
|
|
send(callback_method, callback_object(callback_method_sym.to_s.gsub(/_save/, '')))
|
|
|
|
send(callback_method) { |model| model.history << [callback_method_sym, :block] }
|
2008-01-18 21:44:45 -05:00
|
|
|
end
|
|
|
|
|
2009-10-12 23:15:43 -04:00
|
|
|
def save
|
|
|
|
run_callbacks :save
|
2008-01-18 21:44:45 -05:00
|
|
|
end
|
2009-10-12 23:15:43 -04:00
|
|
|
end
|
|
|
|
|
|
|
|
class PersonSkipper < Person
|
|
|
|
skip_callback :save, :before, :before_save_method, :if => :yes
|
|
|
|
skip_callback :save, :after, :before_save_method, :unless => :yes
|
|
|
|
skip_callback :save, :after, :before_save_method, :if => :no
|
|
|
|
skip_callback :save, :before, :before_save_method, :unless => :no
|
|
|
|
def yes; true; end
|
|
|
|
def no; false; end
|
|
|
|
end
|
|
|
|
|
|
|
|
class ParentController
|
|
|
|
include ActiveSupport::Callbacks
|
|
|
|
|
|
|
|
define_callbacks :dispatch
|
|
|
|
|
|
|
|
set_callback :dispatch, :before, :log, :per_key => {:unless => proc {|c| c.action_name == :index || c.action_name == :show }}
|
|
|
|
set_callback :dispatch, :after, :log2
|
2008-01-18 21:44:45 -05:00
|
|
|
|
2009-10-12 23:15:43 -04:00
|
|
|
attr_reader :action_name, :logger
|
|
|
|
def initialize(action_name)
|
|
|
|
@action_name, @logger = action_name, []
|
|
|
|
end
|
|
|
|
|
|
|
|
def log
|
|
|
|
@logger << action_name
|
|
|
|
end
|
|
|
|
|
|
|
|
def log2
|
|
|
|
@logger << action_name
|
|
|
|
end
|
|
|
|
|
|
|
|
def dispatch
|
|
|
|
run_callbacks :dispatch, action_name do
|
|
|
|
@logger << "Done"
|
2008-01-18 21:44:45 -05:00
|
|
|
end
|
2009-10-12 23:15:43 -04:00
|
|
|
self
|
2008-01-18 21:44:45 -05:00
|
|
|
end
|
|
|
|
end
|
|
|
|
|
2009-10-12 23:15:43 -04:00
|
|
|
class Child < ParentController
|
|
|
|
skip_callback :dispatch, :before, :log, :per_key => {:if => proc {|c| c.action_name == :update} }
|
|
|
|
skip_callback :dispatch, :after, :log2
|
2008-01-18 21:44:45 -05:00
|
|
|
end
|
|
|
|
|
2009-10-12 23:15:43 -04:00
|
|
|
class OneTimeCompile < Record
|
|
|
|
@@starts_true, @@starts_false = true, false
|
|
|
|
|
|
|
|
def initialize
|
|
|
|
super
|
|
|
|
end
|
|
|
|
|
|
|
|
before_save Proc.new {|r| r.history << [:before_save, :starts_true, :if] }, :per_key => {:if => :starts_true}
|
|
|
|
before_save Proc.new {|r| r.history << [:before_save, :starts_false, :if] }, :per_key => {:if => :starts_false}
|
|
|
|
before_save Proc.new {|r| r.history << [:before_save, :starts_true, :unless] }, :per_key => {:unless => :starts_true}
|
|
|
|
before_save Proc.new {|r| r.history << [:before_save, :starts_false, :unless] }, :per_key => {:unless => :starts_false}
|
|
|
|
|
|
|
|
def starts_true
|
|
|
|
if @@starts_true
|
|
|
|
@@starts_true = false
|
|
|
|
return true
|
|
|
|
end
|
|
|
|
@@starts_true
|
|
|
|
end
|
|
|
|
|
|
|
|
def starts_false
|
|
|
|
unless @@starts_false
|
|
|
|
@@starts_false = true
|
|
|
|
return false
|
|
|
|
end
|
|
|
|
@@starts_false
|
|
|
|
end
|
|
|
|
|
|
|
|
def save
|
|
|
|
run_callbacks :save, :action
|
|
|
|
end
|
2008-01-18 21:44:45 -05:00
|
|
|
end
|
|
|
|
|
2009-10-12 23:15:43 -04:00
|
|
|
class OneTimeCompileTest < Test::Unit::TestCase
|
|
|
|
def test_optimized_first_compile
|
|
|
|
around = OneTimeCompile.new
|
|
|
|
around.save
|
|
|
|
assert_equal [
|
|
|
|
[:before_save, :starts_true, :if],
|
|
|
|
[:before_save, :starts_true, :unless]
|
|
|
|
], around.history
|
|
|
|
end
|
2008-01-18 21:44:45 -05:00
|
|
|
end
|
|
|
|
|
2010-09-25 18:31:43 -04:00
|
|
|
class AfterSaveConditionalPerson < Record
|
|
|
|
after_save Proc.new { |r| r.history << [:after_save, :string1] }
|
|
|
|
after_save Proc.new { |r| r.history << [:after_save, :string2] }
|
|
|
|
def save
|
|
|
|
run_callbacks :save
|
|
|
|
end
|
|
|
|
end
|
|
|
|
|
|
|
|
class AfterSaveConditionalPersonCallbackTest < Test::Unit::TestCase
|
|
|
|
def test_after_save_runs_in_the_reverse_order
|
|
|
|
person = AfterSaveConditionalPerson.new
|
|
|
|
person.save
|
|
|
|
assert_equal [
|
|
|
|
[:after_save, :string2],
|
|
|
|
[:after_save, :string1]
|
|
|
|
], person.history
|
|
|
|
end
|
|
|
|
end
|
|
|
|
|
|
|
|
|
|
|
|
|
2009-10-12 23:15:43 -04:00
|
|
|
class ConditionalPerson < Record
|
|
|
|
# proc
|
|
|
|
before_save Proc.new { |r| r.history << [:before_save, :proc] }, :if => Proc.new { |r| true }
|
|
|
|
before_save Proc.new { |r| r.history << "b00m" }, :if => Proc.new { |r| false }
|
|
|
|
before_save Proc.new { |r| r.history << [:before_save, :proc] }, :unless => Proc.new { |r| false }
|
|
|
|
before_save Proc.new { |r| r.history << "b00m" }, :unless => Proc.new { |r| true }
|
|
|
|
# symbol
|
|
|
|
before_save Proc.new { |r| r.history << [:before_save, :symbol] }, :if => :yes
|
|
|
|
before_save Proc.new { |r| r.history << "b00m" }, :if => :no
|
|
|
|
before_save Proc.new { |r| r.history << [:before_save, :symbol] }, :unless => :no
|
|
|
|
before_save Proc.new { |r| r.history << "b00m" }, :unless => :yes
|
|
|
|
# string
|
|
|
|
before_save Proc.new { |r| r.history << [:before_save, :string] }, :if => 'yes'
|
|
|
|
before_save Proc.new { |r| r.history << "b00m" }, :if => 'no'
|
|
|
|
before_save Proc.new { |r| r.history << [:before_save, :string] }, :unless => 'no'
|
|
|
|
before_save Proc.new { |r| r.history << "b00m" }, :unless => 'yes'
|
|
|
|
# Combined if and unless
|
|
|
|
before_save Proc.new { |r| r.history << [:before_save, :combined_symbol] }, :if => :yes, :unless => :no
|
|
|
|
before_save Proc.new { |r| r.history << "b00m" }, :if => :yes, :unless => :yes
|
|
|
|
|
|
|
|
def yes; true; end
|
|
|
|
def other_yes; true; end
|
|
|
|
def no; false; end
|
|
|
|
def other_no; false; end
|
|
|
|
|
|
|
|
def save
|
|
|
|
run_callbacks :save
|
|
|
|
end
|
2008-01-18 21:44:45 -05:00
|
|
|
end
|
|
|
|
|
2009-10-12 23:15:43 -04:00
|
|
|
class CleanPerson < ConditionalPerson
|
|
|
|
reset_callbacks :save
|
2008-01-18 21:44:45 -05:00
|
|
|
end
|
|
|
|
|
2009-10-12 23:15:43 -04:00
|
|
|
class MySuper
|
|
|
|
include ActiveSupport::Callbacks
|
|
|
|
define_callbacks :save
|
2008-01-18 21:44:45 -05:00
|
|
|
end
|
2008-03-18 13:56:05 -04:00
|
|
|
|
2009-10-12 23:15:43 -04:00
|
|
|
class AroundPerson < MySuper
|
|
|
|
attr_reader :history
|
|
|
|
|
|
|
|
set_callback :save, :before, :nope, :if => :no
|
|
|
|
set_callback :save, :before, :nope, :unless => :yes
|
|
|
|
set_callback :save, :after, :tweedle
|
|
|
|
set_callback :save, :before, "tweedle_dee"
|
|
|
|
set_callback :save, :before, proc {|m| m.history << "yup" }
|
|
|
|
set_callback :save, :before, :nope, :if => proc { false }
|
|
|
|
set_callback :save, :before, :nope, :unless => proc { true }
|
|
|
|
set_callback :save, :before, :yup, :if => proc { true }
|
|
|
|
set_callback :save, :before, :yup, :unless => proc { false }
|
|
|
|
set_callback :save, :around, :tweedle_dum
|
|
|
|
set_callback :save, :around, :w0tyes, :if => :yes
|
|
|
|
set_callback :save, :around, :w0tno, :if => :no
|
|
|
|
set_callback :save, :around, :tweedle_deedle
|
|
|
|
|
|
|
|
def no; false; end
|
|
|
|
def yes; true; end
|
|
|
|
|
|
|
|
def nope
|
|
|
|
@history << "boom"
|
|
|
|
end
|
|
|
|
|
|
|
|
def yup
|
|
|
|
@history << "yup"
|
|
|
|
end
|
|
|
|
|
|
|
|
def w0tyes
|
|
|
|
@history << "w0tyes before"
|
|
|
|
yield
|
|
|
|
@history << "w0tyes after"
|
|
|
|
end
|
|
|
|
|
|
|
|
def w0tno
|
|
|
|
@history << "boom"
|
|
|
|
yield
|
|
|
|
end
|
|
|
|
|
|
|
|
def tweedle_dee
|
|
|
|
@history << "tweedle dee"
|
|
|
|
end
|
|
|
|
|
|
|
|
def tweedle_dum
|
|
|
|
@history << "tweedle dum pre"
|
|
|
|
yield
|
|
|
|
@history << "tweedle dum post"
|
|
|
|
end
|
|
|
|
|
|
|
|
def tweedle
|
|
|
|
@history << "tweedle"
|
|
|
|
end
|
|
|
|
|
|
|
|
def tweedle_deedle
|
|
|
|
@history << "tweedle deedle pre"
|
|
|
|
yield
|
|
|
|
@history << "tweedle deedle post"
|
|
|
|
end
|
|
|
|
|
|
|
|
def initialize
|
|
|
|
@history = []
|
|
|
|
end
|
2008-04-18 00:30:01 -04:00
|
|
|
|
2009-10-12 23:15:43 -04:00
|
|
|
def save
|
|
|
|
run_callbacks :save do
|
|
|
|
@history << "running"
|
|
|
|
end
|
|
|
|
end
|
2008-03-18 13:56:05 -04:00
|
|
|
end
|
2011-03-07 08:07:46 -05:00
|
|
|
|
|
|
|
class AroundPersonResult < MySuper
|
|
|
|
attr_reader :result
|
|
|
|
|
2011-03-07 08:29:38 -05:00
|
|
|
set_callback :save, :after, :tweedle_1
|
2011-03-07 08:07:46 -05:00
|
|
|
set_callback :save, :around, :tweedle_dum
|
2011-03-07 08:29:38 -05:00
|
|
|
set_callback :save, :after, :tweedle_2
|
2011-03-07 08:07:46 -05:00
|
|
|
|
|
|
|
def tweedle_dum
|
|
|
|
@result = yield
|
|
|
|
end
|
2011-03-07 08:29:38 -05:00
|
|
|
|
|
|
|
def tweedle_1
|
|
|
|
:tweedle_1
|
|
|
|
end
|
2011-03-07 08:07:46 -05:00
|
|
|
|
2011-03-07 08:29:38 -05:00
|
|
|
def tweedle_2
|
|
|
|
:tweedle_2
|
|
|
|
end
|
|
|
|
|
2011-03-07 08:07:46 -05:00
|
|
|
def save
|
|
|
|
run_callbacks :save do
|
|
|
|
:running
|
|
|
|
end
|
|
|
|
end
|
|
|
|
end
|
2008-03-18 13:56:05 -04:00
|
|
|
|
2009-10-12 23:15:43 -04:00
|
|
|
class HyphenatedCallbacks
|
|
|
|
include ActiveSupport::Callbacks
|
|
|
|
define_callbacks :save
|
|
|
|
attr_reader :stuff
|
|
|
|
|
2010-01-31 00:27:24 -05:00
|
|
|
set_callback :save, :before, :action, :per_key => {:if => :yes}
|
2009-10-12 23:15:43 -04:00
|
|
|
|
|
|
|
def yes() true end
|
|
|
|
|
2010-01-31 00:27:24 -05:00
|
|
|
def action
|
|
|
|
@stuff = "ACTION"
|
2009-10-12 23:15:43 -04:00
|
|
|
end
|
|
|
|
|
|
|
|
def save
|
|
|
|
run_callbacks :save, "hyphen-ated" do
|
|
|
|
@stuff
|
|
|
|
end
|
|
|
|
end
|
|
|
|
end
|
|
|
|
|
|
|
|
class AroundCallbacksTest < Test::Unit::TestCase
|
|
|
|
def test_save_around
|
|
|
|
around = AroundPerson.new
|
|
|
|
around.save
|
|
|
|
assert_equal [
|
|
|
|
"tweedle dee",
|
|
|
|
"yup", "yup",
|
|
|
|
"tweedle dum pre",
|
|
|
|
"w0tyes before",
|
|
|
|
"tweedle deedle pre",
|
|
|
|
"running",
|
|
|
|
"tweedle deedle post",
|
|
|
|
"w0tyes after",
|
|
|
|
"tweedle dum post",
|
|
|
|
"tweedle"
|
|
|
|
], around.history
|
|
|
|
end
|
|
|
|
end
|
2011-03-07 08:07:46 -05:00
|
|
|
|
|
|
|
class AroundCallbackResultTest < Test::Unit::TestCase
|
|
|
|
def test_save_around
|
|
|
|
around = AroundPersonResult.new
|
|
|
|
around.save
|
|
|
|
assert_equal :running, around.result
|
|
|
|
end
|
|
|
|
end
|
2009-10-12 23:15:43 -04:00
|
|
|
|
|
|
|
class SkipCallbacksTest < Test::Unit::TestCase
|
|
|
|
def test_skip_person
|
|
|
|
person = PersonSkipper.new
|
|
|
|
assert_equal [], person.history
|
|
|
|
person.save
|
|
|
|
assert_equal [
|
|
|
|
[:before_save, :string],
|
|
|
|
[:before_save, :proc],
|
|
|
|
[:before_save, :object],
|
|
|
|
[:before_save, :block],
|
|
|
|
[:after_save, :block],
|
|
|
|
[:after_save, :object],
|
|
|
|
[:after_save, :proc],
|
|
|
|
[:after_save, :string],
|
|
|
|
[:after_save, :symbol]
|
|
|
|
], person.history
|
|
|
|
end
|
|
|
|
end
|
|
|
|
|
|
|
|
class CallbacksTest < Test::Unit::TestCase
|
2010-12-16 22:04:20 -05:00
|
|
|
def test_save_phone
|
|
|
|
phone = Phone.new
|
|
|
|
assert_raise RuntimeError do
|
|
|
|
phone.save
|
|
|
|
end
|
|
|
|
assert_equal [:before, :after], phone.history
|
|
|
|
end
|
|
|
|
|
2009-10-12 23:15:43 -04:00
|
|
|
def test_save_person
|
|
|
|
person = Person.new
|
|
|
|
assert_equal [], person.history
|
|
|
|
person.save
|
|
|
|
assert_equal [
|
|
|
|
[:before_save, :symbol],
|
|
|
|
[:before_save, :string],
|
|
|
|
[:before_save, :proc],
|
|
|
|
[:before_save, :object],
|
|
|
|
[:before_save, :block],
|
|
|
|
[:after_save, :block],
|
|
|
|
[:after_save, :object],
|
|
|
|
[:after_save, :proc],
|
|
|
|
[:after_save, :string],
|
|
|
|
[:after_save, :symbol]
|
|
|
|
], person.history
|
|
|
|
end
|
|
|
|
end
|
|
|
|
|
|
|
|
class ConditionalCallbackTest < Test::Unit::TestCase
|
|
|
|
def test_save_conditional_person
|
|
|
|
person = ConditionalPerson.new
|
|
|
|
person.save
|
|
|
|
assert_equal [
|
|
|
|
[:before_save, :proc],
|
|
|
|
[:before_save, :proc],
|
|
|
|
[:before_save, :symbol],
|
|
|
|
[:before_save, :symbol],
|
|
|
|
[:before_save, :string],
|
|
|
|
[:before_save, :string],
|
|
|
|
[:before_save, :combined_symbol],
|
|
|
|
], person.history
|
|
|
|
end
|
|
|
|
end
|
|
|
|
|
2010-09-25 18:31:43 -04:00
|
|
|
|
|
|
|
|
2009-10-12 23:15:43 -04:00
|
|
|
class ResetCallbackTest < Test::Unit::TestCase
|
|
|
|
def test_save_conditional_person
|
|
|
|
person = CleanPerson.new
|
|
|
|
person.save
|
|
|
|
assert_equal [], person.history
|
|
|
|
end
|
|
|
|
end
|
|
|
|
|
|
|
|
class CallbackTerminator
|
|
|
|
include ActiveSupport::Callbacks
|
|
|
|
|
|
|
|
define_callbacks :save, :terminator => "result == :halt"
|
|
|
|
|
|
|
|
set_callback :save, :before, :first
|
|
|
|
set_callback :save, :before, :second
|
|
|
|
set_callback :save, :around, :around_it
|
|
|
|
set_callback :save, :before, :third
|
|
|
|
set_callback :save, :after, :first
|
|
|
|
set_callback :save, :around, :around_it
|
|
|
|
set_callback :save, :after, :second
|
|
|
|
set_callback :save, :around, :around_it
|
|
|
|
set_callback :save, :after, :third
|
|
|
|
|
|
|
|
|
|
|
|
attr_reader :history, :saved
|
|
|
|
def initialize
|
|
|
|
@history = []
|
|
|
|
end
|
|
|
|
|
|
|
|
def around_it
|
|
|
|
@history << "around1"
|
|
|
|
yield
|
|
|
|
@history << "around2"
|
|
|
|
end
|
|
|
|
|
|
|
|
def first
|
|
|
|
@history << "first"
|
|
|
|
end
|
|
|
|
|
|
|
|
def second
|
|
|
|
@history << "second"
|
|
|
|
:halt
|
|
|
|
end
|
|
|
|
|
|
|
|
def third
|
|
|
|
@history << "third"
|
|
|
|
end
|
|
|
|
|
|
|
|
def save
|
|
|
|
run_callbacks :save do
|
|
|
|
@saved = true
|
|
|
|
end
|
|
|
|
end
|
|
|
|
end
|
|
|
|
|
|
|
|
class CallbackObject
|
|
|
|
def before(caller)
|
|
|
|
caller.record << "before"
|
|
|
|
end
|
|
|
|
|
|
|
|
def before_save(caller)
|
|
|
|
caller.record << "before save"
|
|
|
|
end
|
|
|
|
|
|
|
|
def around(caller)
|
|
|
|
caller.record << "around before"
|
|
|
|
yield
|
|
|
|
caller.record << "around after"
|
|
|
|
end
|
|
|
|
end
|
|
|
|
|
|
|
|
class UsingObjectBefore
|
|
|
|
include ActiveSupport::Callbacks
|
|
|
|
|
|
|
|
define_callbacks :save
|
|
|
|
set_callback :save, :before, CallbackObject.new
|
|
|
|
|
|
|
|
attr_accessor :record
|
|
|
|
def initialize
|
|
|
|
@record = []
|
|
|
|
end
|
|
|
|
|
|
|
|
def save
|
|
|
|
run_callbacks :save do
|
|
|
|
@record << "yielded"
|
|
|
|
end
|
|
|
|
end
|
2008-03-18 13:56:05 -04:00
|
|
|
end
|
2008-04-18 00:30:01 -04:00
|
|
|
|
2009-10-12 23:15:43 -04:00
|
|
|
class UsingObjectAround
|
|
|
|
include ActiveSupport::Callbacks
|
2008-04-18 00:30:01 -04:00
|
|
|
|
2009-10-12 23:15:43 -04:00
|
|
|
define_callbacks :save
|
|
|
|
set_callback :save, :around, CallbackObject.new
|
|
|
|
|
|
|
|
attr_accessor :record
|
|
|
|
def initialize
|
|
|
|
@record = []
|
|
|
|
end
|
|
|
|
|
|
|
|
def save
|
|
|
|
run_callbacks :save do
|
|
|
|
@record << "yielded"
|
|
|
|
end
|
|
|
|
end
|
2008-04-18 00:30:01 -04:00
|
|
|
end
|
|
|
|
|
2009-10-12 23:15:43 -04:00
|
|
|
class CustomScopeObject
|
|
|
|
include ActiveSupport::Callbacks
|
|
|
|
|
|
|
|
define_callbacks :save, :scope => [:kind, :name]
|
|
|
|
set_callback :save, :before, CallbackObject.new
|
|
|
|
|
|
|
|
attr_accessor :record
|
|
|
|
def initialize
|
|
|
|
@record = []
|
|
|
|
end
|
|
|
|
|
|
|
|
def save
|
|
|
|
run_callbacks :save do
|
|
|
|
@record << "yielded"
|
|
|
|
"CallbackResult"
|
|
|
|
end
|
|
|
|
end
|
2008-04-18 00:30:01 -04:00
|
|
|
end
|
|
|
|
|
2009-10-12 23:15:43 -04:00
|
|
|
class UsingObjectTest < Test::Unit::TestCase
|
|
|
|
def test_before_object
|
|
|
|
u = UsingObjectBefore.new
|
|
|
|
u.save
|
|
|
|
assert_equal ["before", "yielded"], u.record
|
|
|
|
end
|
|
|
|
|
|
|
|
def test_around_object
|
|
|
|
u = UsingObjectAround.new
|
|
|
|
u.save
|
|
|
|
assert_equal ["around before", "yielded", "around after"], u.record
|
|
|
|
end
|
|
|
|
|
|
|
|
def test_customized_object
|
|
|
|
u = CustomScopeObject.new
|
|
|
|
u.save
|
|
|
|
assert_equal ["before save", "yielded"], u.record
|
|
|
|
end
|
|
|
|
|
|
|
|
def test_block_result_is_returned
|
|
|
|
u = CustomScopeObject.new
|
|
|
|
assert_equal "CallbackResult", u.save
|
|
|
|
end
|
2008-04-18 00:30:01 -04:00
|
|
|
end
|
|
|
|
|
2009-10-12 23:15:43 -04:00
|
|
|
class CallbackTerminatorTest < Test::Unit::TestCase
|
|
|
|
def test_termination
|
|
|
|
terminator = CallbackTerminator.new
|
|
|
|
terminator.save
|
|
|
|
assert_equal ["first", "second", "third", "second", "first"], terminator.history
|
|
|
|
end
|
|
|
|
|
|
|
|
def test_block_never_called_if_terminated
|
|
|
|
obj = CallbackTerminator.new
|
|
|
|
obj.save
|
|
|
|
assert !obj.saved
|
|
|
|
end
|
2008-04-18 00:30:01 -04:00
|
|
|
end
|
|
|
|
|
2009-10-12 23:15:43 -04:00
|
|
|
class HyphenatedKeyTest < Test::Unit::TestCase
|
|
|
|
def test_save
|
|
|
|
obj = HyphenatedCallbacks.new
|
|
|
|
obj.save
|
2010-05-19 15:14:51 -04:00
|
|
|
assert_equal "ACTION", obj.stuff
|
2009-10-12 23:15:43 -04:00
|
|
|
end
|
2008-04-18 00:30:01 -04:00
|
|
|
end
|
2010-08-05 18:09:23 -04:00
|
|
|
|
|
|
|
class WriterSkipper < Person
|
|
|
|
attr_accessor :age
|
|
|
|
skip_callback :save, :before, :before_save_method, :if => lambda {self.age > 21}
|
|
|
|
end
|
|
|
|
|
|
|
|
class WriterCallbacksTest < Test::Unit::TestCase
|
|
|
|
def test_skip_writer
|
|
|
|
writer = WriterSkipper.new
|
|
|
|
writer.age = 18
|
|
|
|
assert_equal [], writer.history
|
|
|
|
writer.save
|
|
|
|
assert_equal [
|
|
|
|
[:before_save, :symbol],
|
|
|
|
[:before_save, :string],
|
|
|
|
[:before_save, :proc],
|
|
|
|
[:before_save, :object],
|
|
|
|
[:before_save, :block],
|
|
|
|
[:after_save, :block],
|
|
|
|
[:after_save, :object],
|
|
|
|
[:after_save, :proc],
|
|
|
|
[:after_save, :string],
|
|
|
|
[:after_save, :symbol]
|
|
|
|
], writer.history
|
|
|
|
end
|
|
|
|
end
|
2010-12-16 22:04:20 -05:00
|
|
|
|
2008-04-18 00:30:01 -04:00
|
|
|
end
|