kotovalexarian-likes-github/database_cleaner
1
0
Fork 0
mirror of https://github.com/DatabaseCleaner/database_cleaner synced 2023-03-27 23:22:03 -04:00
Code Releases Activity

Get schema_migrations table name from ActiveRecord

Browse source 
This makes ActiveRecord responsible for determining the table name to
handle subtle nuisances like table_name_prefix and table_name_suffix.
This commit is contained in:
Kyle Stevens 2013-05-24 15:19:39 -04:00
parent a44abba53d
commit 30db965a99
2 changed files with 11 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
lib/database_cleaner/active_record/truncation.rb
View file

@ -244,7 +244,7 @@ module DatabaseCleaner::ActiveRecord
# overwritten
def migration_storage_names
%w[schema_migrations]
[::ActiveRecord::Migrator.schema_migrations_table_name]
end
def pre_count?

10
spec/database_cleaner/active_record/truncation_spec.rb
View file

@ -39,6 +39,16 @@ module DatabaseCleaner
Truncation.new.clean
end
it "should use ActiveRecord's schema_migrations_table_name" do
connection.stub!(:database_cleaner_table_cache).and_return(%w[pre_schema_migrations_suf widgets dogs])
::ActiveRecord::Base.stub!(:table_name_prefix).and_return('pre_')
::ActiveRecord::Base.stub!(:table_name_suffix).and_return('_suf')
connection.should_receive(:truncate_tables).with(['widgets', 'dogs'])
Truncation.new.clean
end
it "should only truncate the tables specified in the :only option when provided" do
connection.stub!(:database_cleaner_table_cache).and_return(%w[schema_migrations widgets dogs])