Sha256: 605c03b0e6e895ea2520651b81012bc37826a7541742c8936370a6710c227308
Contents?: true
Size: 1.32 KB
Versions: 13
Compression:
Stored size: 1.32 KB
Contents
# encoding: utf-8
module Specjour
module DbScrub
begin
require 'rake'
extend Rake::DSL if defined?(Rake::DSL)
if defined?(Rails)
Rake::Task.define_task(:environment) { }
load 'rails/tasks/misc.rake'
load 'active_record/railties/databases.rake'
end
rescue LoadError
Specjour.logger.debug "Failed to load Rails rake tasks"
end
extend self
def drop
Rake::Task['db:drop'].invoke
end
def scrub
connect_to_database
puts "Resetting database #{ENV['TEST_ENV_NUMBER']}"
schema_load_task.invoke
end
protected
def connect_to_database
ActiveRecord::Base.remove_connection
ActiveRecord::Base.configurations = Rails.application.config.database_configuration
ActiveRecord::Base.establish_connection
connection
rescue # assume the database doesn't exist
Rake::Task['db:create'].invoke
end
def connection
ActiveRecord::Base.connection
end
def pending_migrations?
ActiveRecord::Migrator.new(:up, 'db/migrate').pending_migrations.any?
end
def schema_load_task
Rake::Task[{ :sql => "db:test:load_structure", :ruby => "db:test:load" }[ActiveRecord::Base.schema_format]]
end
def tables_to_purge
connection.tables - ['schema_migrations']
end
end
end
Version data entries
13 entries across 13 versions & 1 rubygems