Add savepoint placeholder for each savepoint created (#1179)

Author: aidanharan

test/support/query_assertions.rb

@@ -10,10 +10,7 @@ def assert_queries_count(count = nil, include_schema: false, &block)
           queries = include_schema ? counter.log_all : counter.log
 
           # Start of monkey-patch
-          # Rails tests expect a save-point to be released at the end of the test. SQL Server does not release
-          # save-points and so the number of queries will be off by one. This monkey patch adds a placeholder query
-          # to the end of the queries array to account for the missing save-point release.
-          queries.append "/* release savepoint placeholder for testing */" if queries.first =~ /SAVE TRANSACTION \S+/
+          queries = include_release_savepoint_placeholder_queries(queries)
           # End of monkey-patch
 
           if count
@@ -24,6 +21,29 @@ def assert_queries_count(count = nil, include_schema: false, &block)
           result
         end
       end
+
+      private
+
+      # Rails tests expect a save-point to be created and released. SQL Server does not release
+      # save-points and so the number of queries will be off. This monkey patch adds a placeholder queries
+      # to replace the missing save-point releases.
+      def include_release_savepoint_placeholder_queries(queries)
+        grouped_queries = [[]]
+
+        queries.each do |query|
+          if query =~ /SAVE TRANSACTION \S+/
+            grouped_queries << [query]
+          else
+            grouped_queries.last << query
+          end
+        end
+
+        grouped_queries.each do |group|
+          group.append "/* release savepoint placeholder for testing */" if group.first =~ /SAVE TRANSACTION \S+/
+        end
+
+        grouped_queries.flatten
+      end
     end
   end
 end