Allow models with non standard capitalization

Author: malmckay

lib/annotate/annotate_models.rb

@@ -239,7 +239,11 @@ def get_model_class(file)
       begin
         parts.inject(Object) {|klass, part| klass.const_get(part) }
       rescue LoadError, NameError
-        Object.const_get(parts.last)
+        begin
+          Object.const_get(parts.last)
+        rescue LoadError, NameError
+          Object.const_get(Module.constants.detect{|c|parts.last.downcase == c.downcase})
+        end
       end
     end
 

spec/annotate/annotate_models_spec.rb

@@ -69,6 +69,11 @@ class FooWithMacro < ActiveRecord::Base
           acts_as_awesome :yah
         end
       EOS
+      create('foo_with_capitals.rb', <<-EOS)
+        class FooWithCAPITALS < ActiveRecord::Base
+          acts_as_awesome :yah
+        end
+      EOS
     end
     it "should work" do
       klass = AnnotateModels.get_model_class("foo.rb")
@@ -78,6 +83,10 @@ class FooWithMacro < ActiveRecord::Base
       klass = AnnotateModels.get_model_class("foo_with_macro.rb")
       klass.name.should == "FooWithMacro"
     end
+    it "should find models with non standard capitalization" do
+      klass = AnnotateModels.get_model_class("foo_with_capitals.rb")
+      klass.name.should == "FooWithCAPITALS"
+    end
   end
 
 end