Ticket #9558: records_affected_test.rb

require 'rubygems'
require 'active_record'
require 'test/unit'

class RecordsAffectedTest < Test::Unit::TestCase
  def setup
    ActiveRecord::Base.establish_connection(  
                                            :adapter => 'sqlserver',
                                            :host => 'HOST',
                                            :database => 'DATABASE',
                                            :username => 'USERNAME',
                                            :password => 'PASSWORD') if RUBY_PLATFORM =~ /mswin32/i

    ActiveRecord::Base.establish_connection(  
                                            :adapter => 'sqlserver',
                                            :mode => 'odbc',
                                            :dsn => 'DSN',
                                            :username => 'USERNAME',
                                            :password => 'PASSWORD') if RUBY_PLATFORM =~ /linux/i
    @conn = ActiveRecord::Base.connection

    @conn.execute "CREATE TABLE #RecordsAffectedTest(foo int)"
  end

  def teardown
    @conn.execute "DROP TABLE #RecordsAffectedTest"
  end

  def test_update
    assert ActiveRecord::Base.connection.instance_variable_get("@connection")["AutoCommit"], "This only seems to happen when auto-commit is on"
    #add 10 records
    (1..10).each do |i|
      @conn.execute "INSERT INTO #RecordsAffectedTest(foo) VALUES (#{i})"
    end
    assert_equal 10, @conn.update("UPDATE #RecordsAffectedTest SET foo = -1"), "@@ROWCOUNT should be 10"
  end
end