Class: DatabaseCleaner::Sequel::Deletion

Inherits:
Truncation show all
Defined in:
lib/database_cleaner/sequel/deletion.rb

Instance Method Summary collapse

Methods inherited from Truncation

#initialize, #start

Methods inherited from Base

#db

Constructor Details

This class inherits a constructor from DatabaseCleaner::Sequel::Truncation

Instance Method Details

#clean ⇒ Object 



35
36
37
38
39
40
41
42
43
44
 
# File 'lib/database_cleaner/sequel/deletion.rb', line 35

def clean
  return unless dirty?

  tables = tables_to_truncate(db)
  db.transaction do
    disable_referential_integrity(tables) do
      delete_tables(db, tables)
    end
  end
end

#delete_tables(db, tables) ⇒ Object 



29
30
31
32
33
 
# File 'lib/database_cleaner/sequel/deletion.rb', line 29

def delete_tables(db, tables)
  tables.each do |table|
    db[table.to_sym].delete
  end
end

#disable_referential_integrity(tables) ⇒ Object 



6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
 
# File 'lib/database_cleaner/sequel/deletion.rb', line 6

def disable_referential_integrity(tables)
  case db.database_type
  when :postgres
    db.run('SET CONSTRAINTS ALL DEFERRED')
    tables_to_truncate(db).each do |table|
      db.run("ALTER TABLE \"#{table}\" DISABLE TRIGGER ALL")
    end
  when :mysql
    old = db.fetch('SELECT @@FOREIGN_KEY_CHECKS').first[:@@FOREIGN_KEY_CHECKS]
    db.run('SET FOREIGN_KEY_CHECKS = 0')
  end
  yield
ensure
  case db.database_type
  when :postgres
    tables.each do |table|
      db.run("ALTER TABLE \"#{table}\" ENABLE TRIGGER ALL")
    end
  when :mysql
    db.run("SET FOREIGN_KEY_CHECKS = #{old}")
  end
end