Expose ORM autodetection publicly

lib/database_cleaner.rb

@@ -1,3 +1,8 @@
 $LOAD_PATH.unshift(File.expand_path(File.dirname(__FILE__))) unless $LOAD_PATH.include?(File.expand_path(File.dirname(__FILE__)))
 require 'database_cleaner/configuration'
 
+module DatabaseCleaner
+  def self.can_detect_orm?
+    DatabaseCleaner::Base.autodetect_orm
+  end
+end

lib/database_cleaner/base.rb

@@ -1,7 +1,6 @@
 require 'database_cleaner/null_strategy'
 module DatabaseCleaner
   class Base
-
     def initialize(desired_orm = nil,opts = {})
       if [:autodetect, nil, "autodetect"].include?(desired_orm)
         autodetect
@@ -98,26 +97,7 @@ module DatabaseCleaner
       self.orm == other.orm && self.db == other.db
     end
 
-    private
+    def autodetect_orm
-
-    def orm_module
-      ::DatabaseCleaner.orm_module(orm)
-    end
-
-    def orm_strategy(strategy)
-      require "database_cleaner/#{orm.to_s}/#{strategy.to_s}"
-      orm_module.const_get(strategy.to_s.capitalize)
-    rescue LoadError => e
-      if orm_module.respond_to? :available_strategies
-        raise UnknownStrategySpecified, "The '#{strategy}' strategy does not exist for the #{orm} ORM!  Available strategies: #{orm_module.available_strategies.join(', ')}"
-      else
-        raise UnknownStrategySpecified, "The '#{strategy}' strategy does not exist for the #{orm} ORM!"
-      end
-    end
-
-    def autodetect
-      @orm ||= begin
-        @autodetected = true
       if defined? ::ActiveRecord
         :active_record
       elsif defined? ::DataMapper
@@ -136,10 +116,31 @@ module DatabaseCleaner
         :ohm
       elsif defined? ::Redis
         :redis
+      end
+    end
+
+    private
+
+    def orm_module
+      ::DatabaseCleaner.orm_module(orm)
+    end
+
+    def orm_strategy(strategy)
+      require "database_cleaner/#{orm.to_s}/#{strategy.to_s}"
+      orm_module.const_get(strategy.to_s.capitalize)
+    rescue LoadError => e
+      if orm_module.respond_to? :available_strategies
+        raise UnknownStrategySpecified, "The '#{strategy}' strategy does not exist for the #{orm} ORM!  Available strategies: #{orm_module.available_strategies.join(', ')}"
       else
-          raise NoORMDetected, "No known ORM was detected!  Is ActiveRecord, DataMapper, Sequel, MongoMapper, Mongoid, Moped, or CouchPotato, Redis or Ohm loaded?"
+        raise UnknownStrategySpecified, "The '#{strategy}' strategy does not exist for the #{orm} ORM!"
       end
     end
+
+    def autodetect
+      @autodetected = true
+
+      @orm ||= autodetect_orm ||
+               raise(NoORMDetected, "No known ORM was detected!  Is ActiveRecord, DataMapper, Sequel, MongoMapper, Mongoid, Moped, or CouchPotato, Redis or Ohm loaded?")
     end
 
     def set_default_orm_strategy