Move lib/active_record/sqlserver_test_case.rb to test/support/sql_counter_sqlserver.rb

Author: metaskills

lib/active_record/sqlserver_test_case.rb

@@ -4,7 +4,7 @@
 require 'cases/helper'
 require 'support/load_schema_sqlserver'
 require 'support/coerced_test_sqlserver'
-require 'cases/sqlserver_test_case'
+require 'support/sql_counter_sqlserver'
 
 module ActiveRecord
   class TestCase < ActiveSupport::TestCase

test/cases/helper_sqlserver.rb

@@ -0,0 +1,31 @@
+module ARTest
+  module Sqlserver
+
+    extend self
+
+    attr_accessor :sql_counter_listenter
+
+    def ignored_sql
+      [ /SELECT SCOPE_IDENTITY/,
+        /INFORMATION_SCHEMA\.(TABLES|VIEWS|COLUMNS)/,
+        /SELECT @@version/,
+        /SELECT @@TRANCOUNT/,
+        /(BEGIN|COMMIT|ROLLBACK|SAVE) TRANSACTION/ ]
+    end
+
+    def sql_counter_listenters
+      ActiveSupport::Notifications.notifier.listeners_for('sql.active_record').select do |listener|
+        listener.inspect =~ /ActiveRecord::SQLCounter/
+      end
+    end
+
+    def sql_counter_listenters_unsubscribe
+      sql_counter_listenters.each { |listener| ActiveSupport::Notifications.unsubscribe(listener) }
+    end
+
+  end
+end
+
+ActiveRecord::SQLCounter.ignored_sql.concat ARTest::Sqlserver.ignored_sql
+ARTest::Sqlserver.sql_counter_listenters_unsubscribe
+ARTest::Sqlserver.sql_counter_listenter = ActiveSupport::Notifications.subscribe 'sql.active_record', ActiveRecord::SQLCounter.new