vendor/rails/activerecord/test/lifecycle_test.rb

   1 require 'abstract_unit'
   2 require 'fixtures/topic'
   3 require 'fixtures/developer'
   4 require 'fixtures/reply'
   5
   6 class Topic; def after_find() end end
   7 class Developer; def after_find() end end
   8 class SpecialDeveloper < Developer; end
   9
  10 class TopicManualObserver
  11   include Singleton
  12
  13   attr_reader :action, :object, :callbacks
  14
  15   def initialize
  16     Topic.add_observer(self)
  17     @callbacks = []
  18   end
  19
  20   def update(callback_method, object)
  21     @callbacks << { "callback_method" => callback_method, "object" => object }
  22   end
  23
  24   def has_been_notified?
  25     !@callbacks.empty?
  26   end
  27 end
  28
  29 class TopicaObserver < ActiveRecord::Observer
  30   observe :topic
  31
  32   attr_reader :topic
  33
  34   def after_find(topic)
  35     @topic = topic
  36   end
  37 end
  38
  39 class TopicObserver < ActiveRecord::Observer
  40   attr_reader :topic
  41
  42   def after_find(topic)
  43     @topic = topic
  44   end
  45 end
  46
  47 class MultiObserver < ActiveRecord::Observer
  48   attr_reader :record
  49
  50   def self.observed_class() [ Topic, Developer ] end
  51
  52   cattr_reader :last_inherited
  53   @@last_inherited = nil
  54
  55   def observed_class_inherited_with_testing(subclass)
  56     observed_class_inherited_without_testing(subclass)
  57     @@last_inherited = subclass
  58   end
  59
  60   alias_method_chain :observed_class_inherited, :testing
  61
  62   def after_find(record)
  63     @record = record
  64   end
  65 end
  66
  67 class LifecycleTest < Test::Unit::TestCase
  68   fixtures :topics, :developers
  69
  70   def test_before_destroy
  71     assert_equal 2, Topic.count
  72     Topic.find(1).destroy
  73     assert_equal 0, Topic.count
  74   end
  75
  76   def test_after_save
  77     ActiveRecord::Base.observers = :topic_manual_observer
  78     ActiveRecord::Base.instantiate_observers
  79
  80     topic = Topic.find(1)
  81     topic.title = "hello"
  82     topic.save
  83
  84     assert TopicManualObserver.instance.has_been_notified?
  85     assert_equal :after_save, TopicManualObserver.instance.callbacks.last["callback_method"]
  86   end
  87
  88   def test_observer_update_on_save
  89     ActiveRecord::Base.observers = TopicManualObserver
  90     ActiveRecord::Base.instantiate_observers
  91
  92     topic = Topic.find(1)
  93     assert TopicManualObserver.instance.has_been_notified?
  94     assert_equal :after_find, TopicManualObserver.instance.callbacks.first["callback_method"]
  95   end
  96
  97   def test_auto_observer
  98     topic_observer = TopicaObserver.instance
  99
 100     topic = Topic.find(1)
 101     assert_equal topic.title, topic_observer.topic.title
 102   end
 103
 104   def test_inferred_auto_observer
 105     topic_observer = TopicObserver.instance
 106
 107     topic = Topic.find(1)
 108     assert_equal topic.title, topic_observer.topic.title
 109   end
 110
 111   def test_observing_two_classes
 112     multi_observer = MultiObserver.instance
 113
 114     topic = Topic.find(1)
 115     assert_equal topic.title, multi_observer.record.title
 116
 117     developer = Developer.find(1)
 118     assert_equal developer.name, multi_observer.record.name
 119   end
 120
 121   def test_observing_subclasses
 122     multi_observer = MultiObserver.instance
 123
 124     developer = SpecialDeveloper.find(1)
 125     assert_equal developer.name, multi_observer.record.name
 126
 127     klass = Class.new(Developer)
 128     assert_equal klass, multi_observer.last_inherited
 129
 130     developer = klass.find(1)
 131     assert_equal developer.name, multi_observer.record.name
 132   end
 133
 134   def test_invalid_observer
 135     assert_raise(ArgumentError) { Topic.observers = Object.new; Topic.instantiate_observers }
 136   end
 137 end