Add support for multi-statement query

Since Rails 5.2, fixtures use multi-statement query when inserting.
rails/rails#31422

However, since `database_rewinder` does not support multi-statement
query, tables inserted by fixture is not cleaned.
This patch also adds support to allow to retrieve the table where insert
was done even if a query is multi-statement.

lib/database_rewinder.rb

@@ -48,13 +48,15 @@ def record_inserted_table(connection, sql)
         end
       end or return
 
-      match = sql.match(/\A\s*INSERT(?:\s+IGNORE)?(?:\s+INTO)?\s+(?:\.*[`"]?([^.\s`"]+)[`"]?)*/i)
-      return unless match
-
-      table = match[1]
-      if table
-        cleaner.inserted_tables << table unless cleaner.inserted_tables.include? table
-        cleaner.pool ||= connection.pool
+      sql.split(';').each do |statement|
+        match = statement.match(/\A\s*INSERT(?:\s+IGNORE)?(?:\s+INTO)?\s+(?:\.*[`"]?([^.\s`"]+)[`"]?)*/i)
+        next unless match
+
+        table = match[1]
+        if table
+          cleaner.inserted_tables << table unless cleaner.inserted_tables.include? table
+          cleaner.pool ||= connection.pool
+        end
       end
     end
 

test/database_rewinder_test.rb

@@ -94,6 +94,14 @@ def perform_insert(sql)
         SQL
         assert_equal ['foos'], @cleaner.inserted_tables
       end
+
+      test 'with multi statement query' do
+        perform_insert <<-SQL
+          INSERT INTO "foos" ("name") VALUES (?);
+          INSERT INTO "bars" ("name") VALUES (?)
+        SQL
+        assert_equal ['foos', 'bars'], @cleaner.inserted_tables
+      end
     end
 
     sub_test_case 'Database accepts more than one dots in an object notation (e.g. SQLServer)' do