issue 36 fix: warn about macros, to mitigate when we're included during a…

Gemfile

@@ -5,4 +5,5 @@ group :development do
   gem "rdoc"
   gem "mg"
   gem 'activesupport', '>= 2.1.0'
+  gem "wrong"
 end

History.txt

 == 2.5.0 2011-06-27 ?
 
+* issue 36 fix: warn about macros, to mitigate when we're included during a production run, not just a rakefile run -- possibly at the expense of too much noise
+
 *   85f6e53 (HEAD, master) Merge branch 'master' of github.com:ctran/annotate_models
 |\  
 

lib/annotate/annotate_models.rb

@@ -330,7 +330,10 @@ end
 module ::ActiveRecord
   class Base
     def self.method_missing(name, *args)
+      super
+    rescue NoMethodError => e
       # ignore this, so unknown/unloaded macros won't cause parsing to fail
+      warn "Annotate Models ignoring #{e.class}: #{e.message}"
     end
   end
 end

spec/annotate/annotate_models_spec.rb

@@ -46,6 +46,8 @@ EOS
   describe "#get_model_class" do
     module ::ActiveRecord
       class Base
+        def self.has_many name
+        end
       end
     end
 
@@ -69,15 +71,32 @@ EOS
           acts_as_awesome :yah
         end
       EOS
+      create('foo_with_known_macro.rb', <<-EOS)
+        class FooWithKnownMacro < ActiveRecord::Base
+          has_many :yah
+        end
+      EOS
     end
+
     it "should work" do
       klass = AnnotateModels.get_model_class("foo.rb")
       klass.name.should == "Foo"
     end
+
     it "should not care about unknown macros" do
-      klass = AnnotateModels.get_model_class("foo_with_macro.rb")
-      klass.name.should == "FooWithMacro"
+      capturing(:stderr) do
+        klass = AnnotateModels.get_model_class("foo_with_macro.rb")
+        klass.name.should == "FooWithMacro"
+      end.should include("undefined method `acts_as_awesome'")
+    end
+
+    it "should allow known macros" do
+      capturing(:stderr) do
+        klass = AnnotateModels.get_model_class("foo_with_known_macro.rb")
+        klass.name.should == "FooWithKnownMacro"
+      end.should == ""
     end
+
   end
 
 end

spec/spec_helper.rb

@@ -6,5 +6,7 @@ rescue LoadError
   require 'rspec'
 end
 
+require "wrong/adapters/rspec"
+
 $:.unshift(File.dirname(__FILE__) + '/../lib')
 require 'annotate'