X-Git-Url: https://git.openstreetmap.org./rails.git/blobdiff_plain/2f9291ba5764fe104264ae7e3b6a361e11212e8b..aa573ce1bb82f778ed7f7a84aeceb66a74ff352f:/lib/tasks/auto_annotate_models.rake

diff --git a/lib/tasks/auto_annotate_models.rake b/lib/tasks/auto_annotate_models.rake
index 954acbf22..103e7fcc9 100644
--- a/lib/tasks/auto_annotate_models.rake
+++ b/lib/tasks/auto_annotate_models.rake
@@ -1,8 +1,8 @@
 # NOTE: only doing this in development as some production environments (Heroku)
 # NOTE: are sensitive to local FS writes, and besides -- it's just not proper
 # NOTE: to have a dev-mode tool do its thing in production.
-if Rails.env.development?
-  task :set_annotation_options do
+if Rails.env.development? || Rails.env.test?
+  task :set_annotation_options => :environment do
     # You can override any of these by setting an environment variable of the
     # same name.
     Annotate.set_defaults(