require 'mongo_mapper_ext'

# 
# disabling :use_database, all tests will use the same :test database.
# 
module MongoMapper::Plugins::DbConfig::ClassMethods
  alias_method :_use_database, :use_database
  def use_database database_alias; end
end


# 
# Disabling indexes in testing environment
# 
if Object.const_defined?(:Rad)
  MongoMapper::Plugins::Indexes::ClassMethods.class_eval do        
    alias_method :original_ensure_index, :ensure_index
    def ensure_index_stub *args; end
  end
  
  rad.after :mode, bang: false do
    if rad.production?
      MongoMapper::Plugins::Indexes::ClassMethods.send :alias_method, :ensure_index, :original_ensure_index
    else
      MongoMapper::Plugins::Indexes::ClassMethods.send :alias_method, :ensure_index, :ensure_index_stub
    end
  end  
end


# 
# Clear db before each test
# 
rspec do
  def self.with_mongo_mapper
    before :all do
      MongoMapper.db_config = {
        'default' => {'name' => "test"}
      }
      
      MongoMapper.database = 'test'      
    end
    
    before do
      MongoMapper.db_config.each do |db_alias, opt|      
        db = MongoMapper.databases[db_alias]    
        db.collection_names.each do |name|
          next if name =~ /^system\./
          db.collection(name).drop
        end
      end
    end
  end
end


# 
# Silence logger
# 
MongoMapper.logger = Logger.new(nil)
if defined? Paperclip
  Paperclip.class_eval do
    def self.logger
      @logger ||= Logger.new(nil)
    end
  end
end


# 
# Files
# 
rspec do
  class << self
    def with_files
      path, cache_path = '/tmp/spec_fs', '/tmp/spec_fs_cache'
      
      before do
        rad.config.merge!({fs: {path: path, cache_path: cache_path}}, override: true)
        
        Models::FileUploader.storage :file
        
        CarrierWave.configure do |config|          
          config.storage = :file
          config.enable_processing = false
          
          config.cache_dir = rad.config.fs.cache_path!
          config.root = rad.config.fs.path!
        end
      end
      
      before do
        path.to_dir.destroy
        cache_path.to_dir.destroy
      end
      
      after do
        path.to_dir.destroy
        cache_path.to_dir.destroy
      end
    end    
  end
end