aboutsummaryrefslogtreecommitdiffstats
path: root/railties/lib/generators/active_record
diff options
context:
space:
mode:
Diffstat (limited to 'railties/lib/generators/active_record')
-rw-r--r--railties/lib/generators/active_record/model/model_generator.rb2
-rw-r--r--railties/lib/generators/active_record/observer/observer_generator.rb2
2 files changed, 4 insertions, 0 deletions
diff --git a/railties/lib/generators/active_record/model/model_generator.rb b/railties/lib/generators/active_record/model/model_generator.rb
index 59310f6c61..42ff48c75c 100644
--- a/railties/lib/generators/active_record/model/model_generator.rb
+++ b/railties/lib/generators/active_record/model/model_generator.rb
@@ -23,6 +23,8 @@ module ActiveRecord
end
end
+ hook_for :test_framework
+
protected
def parent_class_name
diff --git a/railties/lib/generators/active_record/observer/observer_generator.rb b/railties/lib/generators/active_record/observer/observer_generator.rb
index fa51c23336..a6b57423b8 100644
--- a/railties/lib/generators/active_record/observer/observer_generator.rb
+++ b/railties/lib/generators/active_record/observer/observer_generator.rb
@@ -8,6 +8,8 @@ module ActiveRecord
def create_observer_file
template 'observer.rb', File.join('app/models', class_path, "#{file_name}_observer.rb")
end
+
+ hook_for :test_framework
end
end
end