From ce53cbe2393611bb7106f0303b9f7734bad7901d Mon Sep 17 00:00:00 2001
From: Michael Deering <mdeering@mdeering.com>
Date: Thu, 12 Nov 2009 00:13:53 -0700
Subject: [PATCH] Matched up how it searches for files to annotate so that it
 would catch my factory girl files.

---
 lib/annotate_models.rb | 28 +++++++++++++++-------------
 1 file changed, 15 insertions(+), 13 deletions(-)

diff --git a/lib/annotate_models.rb b/lib/annotate_models.rb
index b734dda..94b8e33 100755
--- a/lib/annotate_models.rb
+++ b/lib/annotate_models.rb
@@ -1,11 +1,12 @@
 require "config/environment"
 
-MODEL_DIR         = File.join(RAILS_ROOT, "app/models" )
-UNIT_TEST_DIR     = File.join(RAILS_ROOT, "test/unit"  )
-SPEC_MODEL_DIR    = File.join(RAILS_ROOT, "spec/models")
-FIXTURES_DIR      = File.join(RAILS_ROOT, "test/fixtures")
-SPEC_FIXTURES_DIR = File.join(RAILS_ROOT, "spec/fixtures")
-SORT_COLUMNS      = ENV['SORT'] != 'no'
+MODEL_DIR          = File.join(RAILS_ROOT, "app/models" )
+UNIT_TEST_DIR      = File.join(RAILS_ROOT, "test/unit"  )
+SPEC_MODEL_DIR     = File.join(RAILS_ROOT, "spec/models")
+FIXTURES_DIR       = File.join(RAILS_ROOT, "test/fixtures")
+SPEC_FIXTURES_DIR  = File.join(RAILS_ROOT, "spec/fixtures")
+SPEC_FACTORIES_DIR = File.join(RAILS_ROOT, "spec/factories")
+SORT_COLUMNS       = ENV['SORT'] != 'no'
 
 module AnnotateModels
 
@@ -99,13 +100,14 @@ def self.annotate(klass, header)
     fixtures_name = "#{klass.table_name}.yml"
 
     [
-      File.join(MODEL_DIR,          "#{model_name}.rb"),      # model
-      File.join(UNIT_TEST_DIR,      "#{model_name}_test.rb"), # test
-      File.join(FIXTURES_DIR,       fixtures_name),           # fixture
-      File.join(SPEC_MODEL_DIR,     "#{model_name}_spec.rb"), # spec
-      File.join(SPEC_FIXTURES_DIR,  fixtures_name),           # spec fixture
-      File.join(RAILS_ROOT,         'test', 'factories.rb'),  # factories file
-      File.join(RAILS_ROOT,         'spec', 'factories.rb'),  # factories file
+      File.join(MODEL_DIR,          "#{model_name}.rb"),          # model
+      File.join(UNIT_TEST_DIR,      "#{model_name}_test.rb"),     # test
+      File.join(FIXTURES_DIR,       fixtures_name),               # fixture
+      File.join(SPEC_MODEL_DIR,     "#{model_name}_spec.rb"),     # spec
+      File.join(SPEC_FIXTURES_DIR,  fixtures_name),               # spec fixture
+      File.join(RAILS_ROOT,         'test', 'factories.rb'),      # factories file
+      File.join(RAILS_ROOT,         'spec', 'factories.rb'),      # factories file
+      File.join(SPEC_FACTORIES_DIR, "#{model_name.pluralize}.rb") # factories files
     ].each { |file| annotate_one_file(file, info) }
   end