X-Git-Url: https://git.openstreetmap.org./rails.git/blobdiff_plain/16bef0c8ecad24ac0ca93963196bc844adbb57de..4c4150e0d0dc43c0468388956728d8ffc43e6c9c:/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(