Sha256: 932e085da7710d4e4cea197070ca4c53fe793e667f50e467d7bc6ee7978006e2
Contents?: true
Size: 1.3 KB
Versions: 10
Compression:
Stored size: 1.3 KB
Contents
require "database_cleaner/generic/truncation"
require 'database_cleaner/sequel/base'
module DatabaseCleaner
module Sequel
class Truncation
include ::DatabaseCleaner::Sequel::Base
include ::DatabaseCleaner::Generic::Truncation
def clean
case db.database_type
when :postgres
# PostgreSQL requires all tables with FKs to be truncates in the same command, or have the CASCADE keyword
# appended. Bulk truncation without CASCADE is:
# * Safer. Tables outside of tables_to_truncate won't be affected.
# * Faster. Less roundtrips to the db.
unless (tables= tables_to_truncate(db)).empty?
all_tables= tables.map{|t| %["#{t}"]}.join ','
db.run "TRUNCATE TABLE #{all_tables};"
end
else
# Truncate each table normally
each_table do |db, table|
db[table].truncate
end
end
end
def each_table
tables_to_truncate(db).each do |table|
yield db, table.to_sym
end
end
private
def tables_to_truncate(db)
(@only || db.tables.map(&:to_s)) - @tables_to_exclude
end
# overwritten
def migration_storage_names
%w[schema_info schema_migrations]
end
end
end
end
Version data entries
10 entries across 5 versions & 2 rubygems