Fixes #tables deprecation in Rails 5.0 and keep compatibility with Ra…

…ils 4.x
thoughtbot · Jul 21, 2016 · cedb538 · cedb538
1 parent bb76d55
commit cedb538
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 1 deletion.
diff --git a/lib/shoulda/matchers/active_record/association_matchers/join_table_matcher.rb b/lib/shoulda/matchers/active_record/association_matchers/join_table_matcher.rb
@@ -38,7 +38,7 @@ def join_table_option_correct?
           end
 
           def join_table_exists?
-            if connection.tables.include?(join_table_name)
+            if RailsShim.tables_and_views(connection).include?(join_table_name)
               true
             else
               @failure_message = missing_table_message

diff --git a/lib/shoulda/matchers/rails_shim.rb b/lib/shoulda/matchers/rails_shim.rb
@@ -55,6 +55,14 @@ def self.simply_generate_validation_message(attribute, type, model_name, options
         I18n.translate(primary_translation_key, translate_options)
       end
 
+      def self.tables_and_views(connection)
+        if active_record_major_version >= 5
+          connection.data_sources
+        else
+          connection.tables
+        end
+      end
+
       def self.active_record_major_version
         ::ActiveRecord::VERSION::MAJOR
       end