database_cleaner 1.8.5 → 2.0.2

data/lib/database_cleaner/generic/base.rb

@@ -1,29 +0,0 @@
-module ::DatabaseCleaner
-  module Generic
-    module Base
-
-      def self.included(base)
-        base.extend(ClassMethods)
-      end
-
-      def db
-        :default
-      end
-
-      def cleaning(&block)
-        begin
-          start
-          yield
-        ensure
-          clean
-        end
-      end
-
-      module ClassMethods
-        def available_strategies
-          %W[]
-        end
-      end
-    end
-  end
-end

data/lib/database_cleaner/generic/transaction.rb

@@ -1,11 +0,0 @@
-module DatabaseCleaner
-  module Generic
-    module Transaction
-      def initialize(opts = {})
-        if !opts.empty?
-          raise ArgumentError, "Options are not available for transaction strategies."
-        end
-      end
-    end
-  end
-end

data/lib/database_cleaner/generic/truncation.rb

@@ -1,40 +0,0 @@
-module DatabaseCleaner
-  module Generic
-    module Truncation
-      def initialize(opts={})
-        if !opts.empty? && !(opts.keys - [:only, :except, :pre_count, :reset_ids, :cache_tables]).empty?
-          raise ArgumentError, "The only valid options are :only, :except, :pre_count, :reset_ids or :cache_tables. You specified #{opts.keys.join(',')}."
-        end
-        if opts.has_key?(:only) && opts.has_key?(:except)
-          raise ArgumentError, "You may only specify either :only or :except.  Doing both doesn't really make sense does it?"
-        end
-
-        @only = opts[:only]
-        @tables_to_exclude = Array( (opts[:except] || []).dup ).flatten
-        @tables_to_exclude += migration_storage_names
-        @pre_count = opts[:pre_count]
-        @reset_ids = opts[:reset_ids]
-        @cache_tables = opts.has_key?(:cache_tables) ? !!opts[:cache_tables] : true
-      end
-
-      def start
-        #included for compatability reasons, do nothing if you don't need to
-      end
-
-      def clean
-        raise NotImplementedError
-      end
-
-      private
-      def tables_to_truncate
-        raise NotImplementedError
-      end
-
-      # overwrite in subclasses
-      # default implementation given because migration storage need not be present
-      def migration_storage_names
-        %w[]
-      end
-    end
-  end
-end

data/lib/database_cleaner/null_strategy.rb

@@ -1,20 +0,0 @@
-module DatabaseCleaner
-  class NullStrategy
-    def start
-      # no-op
-    end
-    
-    def db=(connection)
-      # no-op
-    end
-     
-    def clean
-      # no-op
-    end
-
-    def cleaning(&block)
-      # no-op
-      yield
-    end
-  end
-end

data/lib/database_cleaner/orm_autodetector.rb

@@ -1,41 +0,0 @@
-module DatabaseCleaner
-  class ORMAutodetector
-    ORMS = {
-      active_record: "ActiveRecord",
-      data_mapper: "DataMapper",
-      mongo_mapper: "MongoMapper",
-      mongoid: "Mongoid",
-      mongo: "Mongo",
-      couch_potato: "CouchPotato",
-      sequel: "Sequel",
-      moped: "Moped",
-      ohm: "Ohm",
-      redis: "Redis",
-      neo4j: "Neo4j",
-    }
-
-    def orm
-      @autodetected = true
-      autodetected_orm or raise no_orm_detected_error
-      ORMS.key(autodetected_orm.to_s)
-    end
-
-    def autodetected?
-      !!@autodetected
-    end
-
-    private
-
-    def autodetected_orm
-      ORMS.values.find do |orm|
-        Kernel.const_get(orm) rescue next
-      end
-    end
-
-    def no_orm_detected_error
-      orm_list = ORMS.values.join(", ").sub(ORMS.values.last, "or #{ORMS.values.last}")
-      NoORMDetected.new("No known ORM was detected!  Is #{orm_list} loaded?")
-    end
-  end
-  private_constant :ORMAutodetector
-end

data/lib/database_cleaner/safeguard.rb

@@ -1,107 +0,0 @@
-module DatabaseCleaner
-  class Safeguard
-    class Error < Exception
-      class RemoteDatabaseUrl < Error
-        def initialize
-          super("ENV['DATABASE_URL'] is set to a remote URL. Please refer to https://github.com/DatabaseCleaner/database_cleaner#safeguards")
-        end
-      end
-
-      class ProductionEnv < Error
-        def initialize(env)
-          super("ENV['#{env}'] is set to production. Please refer to https://github.com/DatabaseCleaner/database_cleaner#safeguards")
-        end
-      end
-
-      class NotWhitelistedUrl < Error
-        def initialize
-          super("ENV['DATABASE_URL'] is set to a URL that is not on the whitelist. Please refer to https://github.com/DatabaseCleaner/database_cleaner#safeguards")
-        end
-      end
-    end
-
-    class WhitelistedUrl
-      def run
-        return if skip?
-        raise Error::NotWhitelistedUrl if database_url_not_whitelisted?
-      end
-
-      private
-
-        def database_url_not_whitelisted?
-          !DatabaseCleaner.url_whitelist.include?(ENV['DATABASE_URL'])
-        end
-
-        def skip?
-          !DatabaseCleaner.url_whitelist
-        end
-    end
-
-
-    class RemoteDatabaseUrl
-      LOCAL = %w(localhost 127.0.0.1)
-
-      def run
-        raise Error::RemoteDatabaseUrl if !skip? && given?
-      end
-
-      private
-
-        def given?
-          remote?(ENV['DATABASE_URL'])
-        end
-
-        def remote?(url)
-          return false unless url
-
-          parsed = URI.parse(url)
-          return false if parsed.scheme == 'sqlite3:'
-
-          host = parsed.host
-          return false unless host
-          return false if LOCAL.include?(host)
-          return false if host.end_with? '.local'
-          true
-        end
-
-        def skip?
-          ENV['DATABASE_CLEANER_ALLOW_REMOTE_DATABASE_URL'] ||
-            DatabaseCleaner.allow_remote_database_url ||
-            DatabaseCleaner.url_whitelist
-        end
-    end
-
-    class Production
-      KEYS = %w(ENV RACK_ENV RAILS_ENV)
-
-      def run
-        raise Error::ProductionEnv.new(key) if !skip? && given?
-      end
-
-      private
-
-        def given?
-          !!key
-        end
-
-        def key
-          @key ||= KEYS.detect { |key| ENV[key] == 'production' }
-        end
-
-        def skip?
-          ENV['DATABASE_CLEANER_ALLOW_PRODUCTION'] ||
-            DatabaseCleaner.allow_production
-        end
-    end
-
-    CHECKS = [
-      RemoteDatabaseUrl,
-      Production,
-      WhitelistedUrl
-    ]
-
-    def run
-      CHECKS.each { |const| const.new.run }
-    end
-  end
-end

data/lib/database_cleaner/spec/database_helper.rb

@@ -1,82 +0,0 @@
-require 'yaml'
-
-module DatabaseCleaner
-  module Spec
-    class DatabaseHelper < Struct.new(:db)
-      def self.with_all_dbs &block
-        %w[mysql mysql2 sqlite3 postgres].map(&:to_sym).each do |db|
-          yield new(db)
-        end
-      end
-
-      def setup
-        create_db
-        establish_connection
-        load_schema
-      end
-
-      attr_reader :connection
-
-      def teardown
-        drop_db
-      end
-
-      private
-
-      def establish_connection(config = default_config)
-        raise NotImplementedError
-      end
-
-      def create_db
-        if db == :sqlite3
-          # NO-OP
-        elsif db == :postgres
-          establish_connection default_config.merge('database' => 'postgres')
-          connection.execute "CREATE DATABASE #{default_config['database']}" rescue nil
-        else
-          establish_connection default_config.merge("database" => nil)
-          connection.execute "CREATE DATABASE IF NOT EXISTS #{default_config['database']}"
-        end
-      end
-
-      def load_schema
-        connection.execute <<-SQL
-          CREATE TABLE IF NOT EXISTS users (
-            id SERIAL PRIMARY KEY,
-            name INTEGER
-          );
-        SQL
-
-        connection.execute <<-SQL
-          CREATE TABLE IF NOT EXISTS agents (
-            name INTEGER
-          );
-        SQL
-      end
-
-      def drop_db
-        if db == :sqlite3
-          begin
-            File.unlink(db_config['sqlite3']['database'])
-          rescue Errno::ENOENT
-          end
-        elsif db == :postgres
-          # FIXME
-          connection.execute "DROP TABLE IF EXISTS users"
-          connection.execute "DROP TABLE IF EXISTS agents"
-        else
-          connection.execute "DROP DATABASE IF EXISTS #{default_config['database']}"
-        end
-      end
-
-      def db_config
-        config_path = 'spec/support/config.yml'
-        @db_config ||= YAML.load(IO.read(config_path))
-      end
-
-      def default_config
-        db_config[db.to_s]
-      end
-    end
-  end
-end

data/lib/database_cleaner/spec/shared_examples.rb

@@ -1,15 +0,0 @@
-RSpec.shared_examples_for "a generic strategy" do
-  it { is_expected.to respond_to(:db)  }
-end
-
-RSpec.shared_examples_for "a generic truncation strategy" do
-  it { is_expected.to respond_to(:start) }
-  it { is_expected.to respond_to(:clean) }
-  it { is_expected.to respond_to(:cleaning) }
-end
-
-RSpec.shared_examples_for "a generic transaction strategy" do
-  it { is_expected.to respond_to(:start) }
-  it { is_expected.to respond_to(:clean) }
-  it { is_expected.to respond_to(:cleaning) }
-end

data/lib/database_cleaner/spec.rb

@@ -1,2 +0,0 @@
-require "database_cleaner/spec/database_helper"
-require "database_cleaner/spec/shared_examples"

data/lib/database_cleaner/version.rb

@@ -1,3 +0,0 @@
-module DatabaseCleaner
-  VERSION = "1.8.5"
-end