lib/generator_spec/main.rb in generator-spec-0.4.7 vs lib/generator_spec/main.rb in generator-spec-0.4.8

@@ -40,9 +40,31 @@
       RSpec::Generator.setup_generator test_method_name, &block
     end
   end
 end
 
+class Class
+  def use_orm orm
+    class_eval do
+      raise ArgumentError, "Unknown orm #{orm}" if ![:active_record, :mongoid, :mongo_mapper, :data_mapper].include?(orm)
+      include "RSpec::Rails::Orm::#{orm.to_s.camelize}".constantize
+    end
+  end
+  
+  def helpers *types
+    types.each{|type| include_helper type}
+  end    
+
+  protected
+  
+  def include_helper type
+    raise ArgumentError, "Can not specify for #{type}" if ![:migration, :model, :helper, :controller, :view, :observer, :mailer].include?(type)
+    class_eval do
+      include "RSpec::Rails::#{type.to_s.camelize}".constantize
+    end
+  end      
+end
+
 require 'generator_spec/require_generator'
 
 require 'generator_spec/matchers/content/content_matcher'
 require_all File.dirname(__FILE__) + '/matchers/content'