Module: DatabaseRewinder
- Extended by:
- Compatibility
- Defined in:
- lib/database_rewinder.rb,
lib/database_rewinder/cleaner.rb,
lib/database_rewinder/railtie.rb,
lib/database_rewinder/dummy_model.rb,
lib/database_rewinder/compatibility.rb,
lib/database_rewinder/active_record_monkey.rb
Defined Under Namespace
Modules: Compatibility, InsertRecorder
Classes: Cleaner, DummyModel, Railtie
Constant Summary
collapse
- VERSION =
Gem.loaded_specs['database_rewinder'].version.to_s
Class Attribute Summary collapse
Class Method Summary
collapse
[], clean_with, cleaning, start, strategy=
Class Attribute Details
.database_configuration ⇒ Object
14
15
16
|
# File 'lib/database_rewinder.rb', line 14
def database_configuration
@database_configuration || ActiveRecord::Base.configurations
end
|
Class Method Details
.[](connection) ⇒ Object
24
25
26
|
# File 'lib/database_rewinder.rb', line 24
def [](connection)
@cleaners.detect {|c| c.connection_name == connection} || create_cleaner(connection)
end
|
.all=(v) ⇒ Object
28
29
30
|
# File 'lib/database_rewinder.rb', line 28
def all=(v)
@clean_all = v
end
|
.all_table_names(connection) ⇒ Object
cache AR connection.tables
73
74
75
76
77
78
79
80
81
|
# File 'lib/database_rewinder.rb', line 73
def all_table_names(connection)
db = connection.pool.spec.config[:database]
tables = ActiveSupport::Deprecation.silence { connection.tables }
@table_names_cache[db] ||= tables.reject do |t|
(t == ActiveRecord::Migrator.schema_migrations_table_name) ||
(ActiveRecord::Base.respond_to?(:internal_metadata_table_name) && (t == ActiveRecord::Base.internal_metadata_table_name))
end
end
|
.clean ⇒ Object
60
61
62
63
64
65
66
|
# File 'lib/database_rewinder.rb', line 60
def clean
if @clean_all
clean_all
else
cleaners.each(&:clean)
end
end
|
.clean_all ⇒ Object
68
69
70
|
# File 'lib/database_rewinder.rb', line 68
def clean_all
cleaners.each(&:clean_all)
end
|
.cleaners ⇒ Object
32
33
34
35
|
# File 'lib/database_rewinder.rb', line 32
def cleaners
create_cleaner 'test' if @cleaners.empty?
@cleaners
end
|
.create_cleaner(connection_name) ⇒ Object
18
19
20
21
22
|
# File 'lib/database_rewinder.rb', line 18
def create_cleaner(connection_name)
config = database_configuration[connection_name] or raise %Q[Database configuration named "#{connection_name}" is not configured.]
Cleaner.new(config: config, connection_name: connection_name, only: @only, except: @except).tap {|c| @cleaners << c}
end
|
.init ⇒ Object
10
11
12
|
# File 'lib/database_rewinder.rb', line 10
def init
@cleaners, @table_names_cache, @clean_all, @only, @except, @database_configuration = [], {}, false
end
|
.record_inserted_table(connection, sql) ⇒ Object
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
|
# File 'lib/database_rewinder.rb', line 37
def record_inserted_table(connection, sql)
config = connection.instance_variable_get(:'@config')
database = config[:database]
root_dir = defined?(Rails) ? Rails.root : Dir.pwd
cleaner = cleaners.detect do |c|
if (config[:adapter] == 'sqlite3') && (config[:database] != ':memory:')
File.expand_path(c.db, root_dir) == File.expand_path(database, root_dir)
else
c.db == database
end
end or return
match = sql.match(/\AINSERT(?:\s+IGNORE)?(?:\s+INTO)?\s+(?:\.*[`"]?([^.\s`"]+)[`"]?)*/i)
return unless match
table = match[1]
if table
cleaner.inserted_tables << table unless cleaner.inserted_tables.include? table
cleaner.pool ||= connection.pool
end
end
|