From 2f58b509ffc24442be9722aced7dfdf44ebddb49 Mon Sep 17 00:00:00 2001
From: Brendon Murphy <xternal1+github@gmail.com>
Date: Sun, 10 Mar 2013 12:13:07 -0700
Subject: [PATCH] Add very basic initializer to enforce system wide

I've added the config initializer referenced in the README on every
Rails 3.x project I've deployed StrongParameters in, so figured
this is useful.
---
 .../strong_parameters_initializer_generator.rb | 15 +++++++++++++++
 test/initializer_generator_test.rb             | 18 ++++++++++++++++++
 2 files changed, 33 insertions(+)
 create mode 100644 lib/generators/rails/strong_parameters_initializer_generator.rb
 create mode 100644 test/initializer_generator_test.rb

diff --git a/lib/generators/rails/strong_parameters_initializer_generator.rb b/lib/generators/rails/strong_parameters_initializer_generator.rb
new file mode 100644
index 0000000..50ee631
--- /dev/null
+++ b/lib/generators/rails/strong_parameters_initializer_generator.rb
@@ -0,0 +1,15 @@
+module Rails
+  module Generators
+    class StrongParametersInitializerGenerator < Rails::Generators::Base
+      desc "Creates an initializer file to enforce strong parameters on all ActiveRecord models"
+
+      def create_initializer_file
+        create_file "config/initializers/strong_parameters.rb", initializer_content
+      end
+
+      def initializer_content
+        "ActiveRecord::Base.send(:include, ActiveModel::ForbiddenAttributesProtection)\n"
+      end
+    end
+  end
+end
diff --git a/test/initializer_generator_test.rb b/test/initializer_generator_test.rb
new file mode 100644
index 0000000..385def0
--- /dev/null
+++ b/test/initializer_generator_test.rb
@@ -0,0 +1,18 @@
+require 'rails/generators/test_case'
+require 'generators/rails/strong_parameters_initializer_generator'
+
+class StrongParametersInitializerGeneratorTest < Rails::Generators::TestCase
+  tests Rails::Generators::StrongParametersInitializerGenerator
+  destination File.expand_path("../tmp", File.dirname(__FILE__))
+  setup :prepare_destination
+
+  def test_initializer_content
+    Rails.stubs(:application).returns(nil)
+    run_generator
+
+    assert_file "config/initializers/strong_parameters.rb" do |content|
+      init_code = 'ActiveRecord::Base.send(:include, ActiveModel::ForbiddenAttributesProtection)'
+      assert_match init_code, content
+    end
+  end
+end