Sha256: a8549b393d50d81f4da244d9d90e66d054b97ec821d1ea796d1863fc55110843
Contents?: true
Size: 1.28 KB
Versions: 16
Compression:
Stored size: 1.28 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_type= db.url.sub(/:.+/,'').to_sym
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
end
end
private
def tables_to_truncate(db)
(@only || db.tables) - @tables_to_exclude
end
# overwritten
def migration_storage_name
:schema_info
end
end
end
end
Version data entries
16 entries across 10 versions & 4 rubygems