From 18fd049832ed6000677560a96af05eb4772eff10 Mon Sep 17 00:00:00 2001
From: Jon Rowe <jon@mischievousmonkey.co.uk>
Date: Thu, 1 Apr 2010 07:40:15 +0100
Subject: [PATCH] Split off instance functionality into own file

---
 lib/database_cleaner/base.rb | 116 +++++++++++++++++++++++++++++++++++
 1 file changed, 116 insertions(+)
 create mode 100644 lib/database_cleaner/base.rb

diff --git a/lib/database_cleaner/base.rb b/lib/database_cleaner/base.rb
new file mode 100644
index 0000000..ece0439
--- /dev/null
+++ b/lib/database_cleaner/base.rb
@@ -0,0 +1,116 @@
+module DatabaseCleaner
+  class Base
+    
+    def initialize(desired_orm = nil,opts = {})
+      if desired_orm == :autodetect || desired_orm.nil?
+        autodetect 
+      else
+        self.orm = desired_orm
+      end
+      @db = opts[:connection] if opts.has_key? :connection
+    end
+    
+    def db
+      @db || :default
+    end
+      
+    def create_strategy(*args)
+      strategy, *strategy_args = args
+      orm_strategy(strategy).new(*strategy_args)
+    end
+
+    def clean_with(*args)
+      strategy = create_strategy(*args)
+      strategy.clean
+      strategy
+    end
+
+    alias clean_with! clean_with
+
+    def strategy=(args)
+      strategy, *strategy_args = args
+       if strategy.is_a?(Symbol)
+          @strategy = create_strategy(*args)
+       elsif strategy_args.empty?
+         @strategy = strategy
+       else
+         raise ArgumentError, "You must provide a strategy object, or a symbol for a know strategy along with initialization params."
+       end
+    end
+    
+    def strategy
+      return @strategy if @strategy
+      raise NoStrategySetError, "Please set a strategy with DatabaseCleaner.strategy=."
+    end
+
+    def orm=(desired_orm)
+      @orm = desired_orm
+    end
+    
+    def orm
+      @orm || autodetect
+    end
+    
+    def start
+      strategy.start
+    end
+
+    def clean
+      strategy.clean
+    end
+
+    alias clean! clean
+       
+    def auto_detected
+      return true unless @autodetected.nil?
+    end
+    
+    #TODO make strategies directly comparable
+    def ==(other)
+      self.orm == other.orm && self.db == other.db && self.strategy.class == other.strategy.class
+    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
+      raise UnknownStrategySpecified, "The '#{strategy}' strategy does not exist for the #{orm} ORM!  Available strategies: #{orm_module.available_strategies.join(', ')}"
+    end
+    
+    def autodetect
+      @orm ||= begin
+        @autodetected = true
+        if defined? ::ActiveRecord
+          :active_record
+        elsif defined? ::DataMapper
+          :data_mapper
+        elsif defined? ::MongoMapper
+          :mongo_mapper
+        elsif defined? ::CouchPotato
+          :couch_potato
+        else
+          raise NoORMDetected, "No known ORM was detected!  Is ActiveRecord, DataMapper, MongoMapper or CouchPotato loaded?"
+        end
+      end
+    end
+    
+    def orm_module
+      case orm
+        when :active_record
+          DatabaseCleaner::ActiveRecord
+        when :data_mapper
+          DatabaseCleaner::DataMapper
+        when :mongo_mapper
+          DatabaseCleaner::MongoMapper
+        when :couch_potato
+          DatabaseCleaner::CouchPotato
+      end
+    end    
+  end
+end
\ No newline at end of file