Contents

module DatabaseRewinder
  class Cleaner
    attr_accessor :config, :connection_name, :only, :except, :inserted_tables, :pool

    def initialize(config: nil, connection_name: nil, only: nil, except: nil)
      @config, @connection_name, @only, @except = config, connection_name, Array(only), Array(except)
      reset
    end

    def db
      config['database']
    end

    def clean
      return if !pool || inserted_tables.empty?

      # When the application uses multiple database connections, a connection
      # pool used in test could be already removed (i.e., pool.connected? = false).
      # In this case, we have to reconnect to the database to clean inserted
      # tables.
      with_automatic_reconnect(pool) do
        delete_all (ar_conn = pool.connection), DatabaseRewinder.all_table_names(ar_conn) & inserted_tables
      end
      reset
    end

    def clean_all
      if pool
        ar_conn = pool.connection
        delete_all ar_conn, DatabaseRewinder.all_table_names(ar_conn)
      else
        require 'database_rewinder/dummy_model'
        DummyModel.with_temporary_connection(config) do |temporary_connection|
          delete_all temporary_connection, DatabaseRewinder.all_table_names(temporary_connection)
        end
      end

      reset
    end

    private
    def delete_all(ar_conn, tables)
      tables = tables & @only if @only.any?
      tables -= @except if @except.any?
      return if tables.empty?

      ar_conn.disable_referential_integrity do
        tables.each do |table_name|
          ar_conn.execute "DELETE FROM #{ar_conn.quote_table_name(table_name)};"
        end
      end
    end

    def reset
      @inserted_tables = []
    end

    def with_automatic_reconnect(pool)
      reconnect = pool.automatic_reconnect
      pool.automatic_reconnect = true
      yield
    ensure
      pool.automatic_reconnect = reconnect
    end
  end
end

require_relative 'compatibility'

Version data entries

7 entries across 7 versions & 1 rubygems

Version	Path
database_rewinder-0.6.5	lib/database_rewinder/cleaner.rb
database_rewinder-0.6.4	lib/database_rewinder/cleaner.rb
database_rewinder-0.6.3	lib/database_rewinder/cleaner.rb
database_rewinder-0.6.2	lib/database_rewinder/cleaner.rb
database_rewinder-0.6.1	lib/database_rewinder/cleaner.rb
database_rewinder-0.6.0	lib/database_rewinder/cleaner.rb
database_rewinder-0.5.3	lib/database_rewinder/cleaner.rb