lib/stationed/generators/plugins/factory_girl.rb in stationed-0.0.1 vs lib/stationed/generators/plugins/factory_girl.rb in stationed-0.1.0

@@ -1,9 +1,17 @@
 module Stationed
   module Generators
     module Plugins
       module FactoryGirl
+        def self.prepended(base)
+          base.class_option :factory_girl,
+            type: :boolean,
+            default: true,
+            desc: 'Include FactoryGirl and default tests for valid factories'
+        end
+
         def finish_template
+          return super unless options[:factory_girl]
           gem 'factory_girl_rails', group: [:development, :test]
           copy_file 'factory_girl.rb', 'spec/support/factory_girl.rb'
           copy_file 'factories_spec.rb', 'spec/models/factories_spec.rb'
           super
         end