Merge pull request #415 from pschambacher/rails5.0

Remove deprecation warning for Rails 5.0
DatabaseCleaner · Mar 11, 2016 · f318781 · f318781
2 parents 2840197 + 430a957
commit f318781
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/lib/database_cleaner/active_record/truncation.rb b/lib/database_cleaner/active_record/truncation.rb
@@ -27,7 +27,7 @@ def database_cleaner_view_cache
 
       def database_cleaner_table_cache
         # the adapters don't do caching (#130) but we make the assumption that the list stays the same in tests
-        @database_cleaner_tables ||= tables
+        @database_cleaner_tables ||= ::ActiveRecord::VERSION::MAJOR >= 5 ? data_sources : tables
       end
 
       def truncate_table(table_name)