lib/combustion.rb in combustion-0.1.1 vs lib/combustion.rb in combustion-0.2.0

@@ -1,18 +1,28 @@
 require 'rails'
 require 'active_support/dependencies'
-require 'active_record/railtie'
-require 'action_controller/railtie'
-require 'action_view/railtie'
-require 'sprockets/railtie'
 
 module Combustion
-  def self.initialize!
+  Modules = %w( active_record action_controller action_view action_mailer
+    sprockets )
+
+  def self.initialize!(*modules)
+    modules = Modules if modules.empty? || modules == [:all]
+    modules.each { |mod| require "#{mod}/railtie" }
+
+    Combustion::Application.configure_for_combustion
     Combustion::Application.initialize!
 
     silence_stream(STDOUT) do
       load "#{Rails.root}/db/schema.rb"
     end
+
+    RSpec.configure do |config|
+      config.include(Capybara) if defined?(Capybara)
+
+      config.include(Combustion::Application.routes.url_helpers)
+      config.include(Combustion::Application.routes.mounted_helpers)
+    end if defined?(RSpec) && RSpec.respond_to?(:configure)
   end
 end
 
 require 'combustion/application'
 require 'combustion/version'