Debug

Author: aidanharan

docker-compose.ci.yml

@@ -9,6 +9,6 @@ services:
     build:
       context: .
       dockerfile: Dockerfile.ci
-    command: wait-for sqlserver:1433 -- bundle exec rake test TESTOPTS="-v"
+    command: wait-for sqlserver:1433 -- bundle exec rake test TEST_FILES="test/cases/temp_test_sqlserver.rb" TESTOPTS="-v"
     depends_on:
       - "sqlserver"

lib/active_record/connection_adapters/sqlserver/database_statements.rb

@@ -14,6 +14,8 @@ def write_query?(sql) # :nodoc:
         end
 
         def raw_execute(sql, name, async: false, allow_retry: false, materialize_transactions: true)
+          # puts "raw_execute: #{sql}" if $AIDO
+
           log(sql, name, async: async) do |notification_payload|
             with_raw_connection(allow_retry: allow_retry, materialize_transactions: materialize_transactions) do |conn|
               result = if id_insert_table_name = query_requires_identity_insert?(sql)
@@ -312,8 +314,11 @@ def sql_for_insert(sql, pk, binds, returning)
         # === SQLServer Specific ======================================== #
 
         def set_identity_insert(table_name, conn, enable)
+          puts "set_identity_insert: #{table_name} #{enable}"
+
           internal_raw_execute("SET IDENTITY_INSERT #{table_name} #{enable ? 'ON' : 'OFF'}", conn , perform_do: true)
         rescue Exception
+          puts "IDENTITY_INSERT could not be turned #{enable ? 'ON' : 'OFF'} for table #{table_name}"
           raise ActiveRecordError, "IDENTITY_INSERT could not be turned #{enable ? 'ON' : 'OFF'} for table #{table_name}"
         end
 
@@ -404,12 +409,37 @@ def exclude_output_inserted_id_sql_type(pk, exclude_output_inserted)
         end
 
         def query_requires_identity_insert?(sql)
+          # puts "query_requires_identity_insert?: #{sql}" if $AIDO
+
           return false unless insert_sql?(sql)
 
           raw_table_name = get_raw_table_name(sql)
+
+          # binding.pry
+          # if $AIDO
+          #
+          #   puts "xxxx"
+          #   return true
+          # end
+
+
+
+
           id_column = identity_columns(raw_table_name).first
 
-          id_column && sql =~ /^\s*(INSERT|EXEC sp_executesql N'INSERT)[^(]+\([^)]*\b(#{id_column.name})\b,?[^)]*\)/i ? SQLServer::Utils.extract_identifiers(raw_table_name).quoted : false
+
+          if id_column && sql =~ /^\s*(INSERT|EXEC sp_executesql N'INSERT)[^(]+\([^)]*\b(#{id_column.name})\b,?[^)]*\)/i
+
+            # puts "xxx: raw_table_name=#{raw_table_name}, id_column=#{id_column.name}" if $AIDO
+            # puts "xxx: quoted=#{SQLServer::Utils.extract_identifiers(raw_table_name).quoted}" if $AIDO
+
+            SQLServer::Utils.extract_identifiers(raw_table_name).quoted
+          else
+            false
+
+          end
+
+          # id_column && sql =~ /^\s*(INSERT|EXEC sp_executesql N'INSERT)[^(]+\([^)]*\b(#{id_column.name})\b,?[^)]*\)/i ? SQLServer::Utils.extract_identifiers(raw_table_name).quoted : false
         end
 
         def insert_sql?(sql)
@@ -453,6 +483,8 @@ def finish_statement_handle(handle)
         # Getting around this by raising an exception ourselves while PR
         # https://github.com/rails-sqlserver/tiny_tds/pull/469 is not released.
         def internal_raw_execute(sql, conn, perform_do: false)
+          # puts "internal_raw_execute: #{sql}" if $AIDO
+
           result = conn.execute(sql).tap do |_result|
             raise TinyTds::Error, "failed to execute statement" if _result.is_a?(FalseClass)
           end

test/cases/temp_test_sqlserver.rb

@@ -20,17 +20,21 @@ class TempTestSQLServer < ActiveRecord::TestCase
       INSERT INTO #{arunit2_database}.dbo.dogs(id) SELECT id FROM #{arunit_database}.dbo.dogs (NOLOCK)
     SQL
 
+    ActiveSupport::Notifications.subscribe('sql.active_record') do |_name, _start, _finish, _id, payload|
+      puts payload[:sql]
+    end
+
     OtherDog.destroy_all
 
     assert Dog.count, 1
     assert OtherDog.count, 0
 
     # Dog.transaction do
-    arunit_connection.transaction do
+    # arunit_connection.transaction do
         assert_nothing_raised do
           arunit_connection.execute(sql)
         end
-      end
+      # end
     # end
 
     assert Dog.count, 1