Merge pull request #126 from davejachimiak/dj-attempt_only_if_table_exists

Attempt model annotation only if table exists

Author: ctran

lib/annotate/annotate_models.rb

@@ -411,7 +411,7 @@ def do_annotations(options={})
     def annotate_model_file(annotated, file, header, options)
       begin
         klass = get_model_class(file)
-        if klass && klass < ActiveRecord::Base && !klass.abstract_class?
+        if klass && klass < ActiveRecord::Base && !klass.abstract_class? && klass.table_exists?
           if annotate(klass, file, header, options)
             annotated << klass
           end

spec/annotate/annotate_models_spec.rb

@@ -461,4 +461,20 @@ class User < ActiveRecord::Base
        end
     end
   end
+
+  describe '.annotate_model_file' do
+    before do
+      class Foo < ActiveRecord::Base; end;
+      AnnotateModels.stub(:get_model_class).with('foo.rb') { Foo }
+      Foo.stub(:table_exists?) { false }
+    end
+
+    after { Object.send :remove_const, 'Foo' }
+
+    it 'skips attempt to annotate if no table exists for model' do
+      annotate_model_file = AnnotateModels.annotate_model_file([], 'foo.rb', nil, nil)
+
+      annotate_model_file.should eq nil
+    end
+  end
 end