Merge pull request #68 from sinsoku/fix_an_issue_that_finding_wrong_c…

…leaner

Fix an issue recording table names to wrong cleaner

lib/database_rewinder.rb

@@ -39,13 +39,14 @@ def cleaners
     def record_inserted_table(connection, sql)
       config = connection.instance_variable_get(:'@config')
       database = config[:database]
+      host = config[:host]
       #NOTE What's the best way to get the app dir besides Rails.root? I know Dir.pwd here might not be the right solution, but it should work in most cases...
       root_dir = defined?(Rails) && Rails.respond_to?(:root) ? Rails.root : Dir.pwd
       cleaner = cleaners.detect do |c|
         if (config[:adapter] == 'sqlite3') && (config[:database] != ':memory:')
           File.expand_path(c.db, root_dir) == File.expand_path(database, root_dir)
         else
-          c.db == database
+          c.db == database && c.host == host
         end
       end or return
 

lib/database_rewinder/cleaner.rb

@@ -19,6 +19,10 @@ def db
       config['database']
     end
 
+    def host
+      config['host']
+    end
+
     def clean(multiple: true)
       return if !pool || inserted_tables.empty?