From a2f3faf1327d217f5f39ace8609cc4d201549506 Mon Sep 17 00:00:00 2001
From: Matt Solt <mattsolt@gmail.com>
Date: Sun, 5 Mar 2017 13:27:04 -0500
Subject: [PATCH] Rails 5.1 removes
 ActiveRecord::Migrator.schema_migrations_table_name and replaces it with
 ActiveRecord::SchemaMigration.table_name (available since Rails 4.0)

---
 lib/database_cleaner/active_record/truncation.rb | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/lib/database_cleaner/active_record/truncation.rb b/lib/database_cleaner/active_record/truncation.rb
index 82a61678..229ddd7d 100644
--- a/lib/database_cleaner/active_record/truncation.rb
+++ b/lib/database_cleaner/active_record/truncation.rb
@@ -192,7 +192,7 @@ def tables_with_schema
           FROM pg_tables
           WHERE
             tablename !~ '_prt_' AND
-            tablename <> '#{::ActiveRecord::Migrator.schema_migrations_table_name}' AND
+            tablename <> '#{::ActiveRecord::SchemaMigration.table_name}' AND
             schemaname = ANY (current_schemas(false))
         _SQL
         rows.collect { |result| result.first }
@@ -257,7 +257,7 @@ def tables_to_truncate(connection)
 
     # overwritten
     def migration_storage_names
-      [::ActiveRecord::Migrator.schema_migrations_table_name]
+      [::ActiveRecord::SchemaMigration.table_name]
     end
 
     def cache_tables?