diff --git a/lib/database_cleaner/active_record/truncation.rb b/lib/database_cleaner/active_record/truncation.rb index 229ddd7..6e043d1 100644 --- a/lib/database_cleaner/active_record/truncation.rb +++ b/lib/database_cleaner/active_record/truncation.rb @@ -192,7 +192,7 @@ module DatabaseCleaner FROM pg_tables WHERE tablename !~ '_prt_' AND - tablename <> '#{::ActiveRecord::SchemaMigration.table_name}' AND + tablename <> '#{migration_table_name}' AND schemaname = ANY (current_schemas(false)) _SQL rows.collect { |result| result.first } @@ -257,7 +257,15 @@ module DatabaseCleaner::ActiveRecord # overwritten def migration_storage_names - [::ActiveRecord::SchemaMigration.table_name] + [migration_table_name] + end + + def migration_table_name + if ActiveRecord::VERSION::MAJOR < 5 + ::ActiveRecord::Migrator.schema_migrations_table_name + else + ::ActiveRecord::SchemaMigration.table_name + end end def cache_tables?