rails-sqlserver-2000-2005-adapter 1.0.0

data/test/cases/column_test_sqlserver.rb

@@ -0,0 +1,66 @@
+require 'cases/sqlserver_helper'
+require 'models/binary'
+
+class ColumnTestSqlserver < ActiveRecord::TestCase
+  
+  def setup
+    @column_klass = ActiveRecord::ConnectionAdapters::SQLServerColumn
+  end
+  
+  should 'return real_number as float' do
+    assert_equal :float, TableWithRealColumn.columns_hash["real_number"].type
+  end
+  
+  should 'know its #table_name and #table_klass' do
+    Topic.columns.each do |column|
+      assert_equal 'topics', column.table_name, "This column #{column.inspect} did not know it's #table_name"
+      assert_equal Topic, column.table_klass, "This column #{column.inspect} did not know it's #table_klass"
+    end
+  end
+  
+  context 'For :binary columns' do
+
+    setup do
+      @binary_string = "GIF89a\001\000\001\000\200\000\000\377\377\377\000\000\000!\371\004\000\000\000\000\000,\000\000\000\000\001\000\001\000\000\002\002D\001\000;"
+      @saved_bdata = Binary.create!(:data => @binary_string)
+    end
+    
+    should 'read and write binary data equally' do
+      assert_equal @binary_string, Binary.find(@saved_bdata).data
+    end
+    
+    should 'quote data for sqlserver with literal 0x prefix' do
+      # See the output of the stored procedure: 'exec sp_datatype_info'
+      sqlserver_encoded_bdata = "0x47494638396101000100800000ffffff00000021f90400000000002c00000000010001000002024401003b"
+      assert_equal sqlserver_encoded_bdata, @column_klass.string_to_binary(@binary_string) 
+    end
+
+  end
+  
+  context 'For .columns method' do
+
+    should 'return correct scales and precisions for NumericData' do
+      bank_balance = NumericData.columns_hash['bank_balance']
+      big_bank_balance = NumericData.columns_hash['big_bank_balance']
+      world_population = NumericData.columns_hash['world_population']
+      my_house_population = NumericData.columns_hash['my_house_population']
+      assert_equal [2,10], [bank_balance.scale, bank_balance.precision]
+      assert_equal [2,15], [big_bank_balance.scale, big_bank_balance.precision]
+      assert_equal [0,10], [world_population.scale, world_population.precision]
+      assert_equal [0,2], [my_house_population.scale, my_house_population.precision]
+    end
+    
+    should 'return correct null, limit, and default for Topic' do
+      tch = Topic.columns_hash
+      assert_equal false, tch['id'].null
+      assert_equal true,  tch['title'].null
+      assert_equal 255,   tch['author_name'].limit
+      assert_equal true,  tch['approved'].default
+      assert_equal 0,     tch['replies_count'].default
+    end
+
+  end
+  
+  
+  
+end

data/test/cases/connection_test_sqlserver.rb

@@ -0,0 +1,103 @@
+require 'cases/sqlserver_helper'
+require 'models/reply' 
+
+class ConnectionTestSqlserver < ActiveRecord::TestCase
+  
+  self.use_transactional_fixtures = false
+  
+  fixtures :topics, :accounts
+  
+  def setup
+    @connection = ActiveRecord::Base.connection
+  end
+  
+  
+  should 'return finished DBI statment handle from #execute without block' do
+    handle = @connection.execute('SELECT * FROM [topics]')
+    assert_instance_of DBI::StatementHandle, handle
+    assert handle.finished?
+  end
+  
+  should 'finish DBI statment handle from #execute with block' do
+    assert_all_statements_used_are_closed do
+      @connection.execute('SELECT * FROM [topics]') { }
+    end
+  end
+  
+  should 'return an unfinished DBI statement handler from #raw_execute' do
+    handle = @connection.send(:raw_execute,'SELECT * FROM [topics]')
+    assert_instance_of DBI::StatementHandle, handle
+    assert !handle.finished?
+  end
+  
+  should 'finish connection from #raw_select' do
+    assert_all_statements_used_are_closed do
+      @connection.send(:raw_select,'SELECT * FROM [topics]')
+    end
+  end
+  
+  should 'affect rows' do
+    assert Topic.connection.instance_variable_get("@connection")["AutoCommit"]
+    topic_data = { 1 => { "content" => "1 updated" }, 2 => { "content" => "2 updated" } }
+    updated = Topic.update(topic_data.keys, topic_data.values)
+    assert_equal 2, updated.size
+    assert_equal "1 updated", Topic.find(1).content
+    assert_equal "2 updated", Topic.find(2).content
+    assert_equal 2, Topic.delete([1, 2])        
+  end
+  
+  should 'execute without block closes statement' do
+    assert_all_statements_used_are_closed do
+      @connection.execute("SELECT 1")
+    end
+  end
+  
+  should 'execute with block closes statement' do
+    assert_all_statements_used_are_closed do
+      @connection.execute("SELECT 1") do |sth|
+        assert !sth.finished?, "Statement should still be alive within block"
+      end
+    end
+  end
+  
+  should 'insert with identity closes statement' do
+    assert_all_statements_used_are_closed do
+      @connection.insert("INSERT INTO accounts ([id], [firm_id],[credit_limit]) values (999, 1, 50)")
+    end
+  end
+  
+  should 'insert without identity closes statement' do
+    assert_all_statements_used_are_closed do
+      @connection.insert("INSERT INTO accounts ([firm_id],[credit_limit]) values (1, 50)")
+    end
+  end
+
+  should 'active closes statement' do
+    assert_all_statements_used_are_closed do
+      @connection.active?
+    end
+  end
+  
+  
+  private
+  
+  def assert_all_statements_used_are_closed(&block)
+    existing_handles = []
+    ObjectSpace.each_object(DBI::StatementHandle) {|handle| existing_handles << handle}
+    GC.disable
+    yield
+    used_handles = []
+    ObjectSpace.each_object(DBI::StatementHandle) {|handle| used_handles << handle unless existing_handles.include? handle}
+    assert_block "No statements were used within given block" do
+      used_handles.size > 0
+    end
+    ObjectSpace.each_object(DBI::StatementHandle) do |handle|
+      assert_block "Statement should have been closed within given block" do
+        handle.finished?
+      end
+    end
+  ensure
+    GC.enable
+  end
+
+end

data/test/cases/eager_association_test_sqlserver.rb

@@ -0,0 +1,22 @@
+require 'cases/sqlserver_helper'
+require 'models/author'
+require 'models/post'
+require 'models/comment'
+
+class EagerAssociationTestSqlserver < ActiveRecord::TestCase
+end
+
+class EagerAssociationTest < ActiveRecord::TestCase
+  
+  COERCED_TESTS = [:test_count_with_include]
+  
+  include SqlserverCoercedTest
+  
+  fixtures :authors, :posts, :comments
+  
+  def test_coerced_test_count_with_include
+    assert_equal 3, authors(:david).posts_with_comments.count(:conditions => "len(comments.body) > 15")
+  end
+  
+  
+end

data/test/cases/inheritance_test_sqlserver.rb

@@ -0,0 +1,28 @@
+require 'cases/sqlserver_helper'
+require 'models/company'
+
+class InheritanceTestSqlserver < ActiveRecord::TestCase
+end
+
+class InheritanceTest < ActiveRecord::TestCase
+  
+  COERCED_TESTS = [
+    :test_eager_load_belongs_to_primary_key_quoting,
+    :test_a_bad_type_column
+  ]
+  
+  include SqlserverCoercedTest
+  
+  def test_coerced_test_eager_load_belongs_to_primary_key_quoting
+    assert_sql(/\(\[companies\].\[id\] = 1\)/) do
+      Account.find(1, :include => :firm)
+    end
+  end
+  
+  def test_coerced_test_a_bad_type_column
+    Company.connection.insert "INSERT INTO [companies] ([id], #{QUOTED_TYPE}, [name]) VALUES(100, 'bad_class!', 'Not happening')"
+    assert_raises(ActiveRecord::SubclassNotFound) { Company.find(100) }
+  end
+  
+  
+end

data/test/cases/migration_test_sqlserver.rb

@@ -0,0 +1,57 @@
+require 'cases/sqlserver_helper'
+require 'models/person'
+
+class MigrationTestSqlserver < ActiveRecord::TestCase
+  
+  context 'For transactions' do
+    
+    setup do
+      @connection = ActiveRecord::Base.connection
+      @trans_test_table1 = 'sqlserver_trans_table1'
+      @trans_test_table2 = 'sqlserver_trans_table2'
+      @trans_tables = [@trans_test_table1,@trans_test_table2]
+    end
+    
+    teardown do
+      @trans_tables.each do |table_name|
+        ActiveRecord::Migration.drop_table(table_name) if @connection.tables.include?(table_name)
+      end
+    end
+    
+    should 'not create a tables if error in migrations' do
+      begin
+        ActiveRecord::Migrator.up(SQLSERVER_MIGRATIONS_ROOT+'/transaction_table')
+      rescue Exception => e
+        assert_match %r|this and all later migrations canceled|, e.message
+      end
+      assert_does_not_contain @trans_test_table1, @connection.tables
+      assert_does_not_contain @trans_test_table2, @connection.tables
+    end
+    
+  end
+  
+  
+end
+
+
+class MigrationTest < ActiveRecord::TestCase
+  
+  COERCED_TESTS = [:test_add_column_not_null_without_default]
+  
+  include SqlserverCoercedTest
+  
+  
+  def test_coerced_test_add_column_not_null_without_default
+    Person.connection.create_table :testings do |t| 
+      t.column :foo, :string
+      t.column :bar, :string, :null => false
+    end
+    assert_raises(ActiveRecord::StatementInvalid) do
+      Person.connection.execute "INSERT INTO [testings] ([foo], [bar]) VALUES ('hello', NULL)"
+    end
+  ensure
+    Person.connection.drop_table :testings rescue nil
+  end
+  
+  
+end

data/test/cases/offset_and_limit_test_sqlserver.rb

@@ -0,0 +1,82 @@
+require 'cases/sqlserver_helper'
+require 'models/book'
+
+class OffsetAndLimitTestSqlserver < ActiveRecord::TestCase
+  
+  class Account < ActiveRecord::Base; end
+  
+  def setup
+    @connection = ActiveRecord::Base.connection
+  end
+  
+  context 'When selecting with limit' do
+
+    setup do
+      @select_sql = 'SELECT * FROM schema'
+    end
+
+    should 'alter SQL to limit number of records returned' do
+      options = { :limit => 10 }
+      assert_equal('SELECT TOP 10 * FROM schema', @connection.add_limit_offset!(@select_sql, options))
+    end
+
+    should 'only allow integers for limit' do
+      options = { :limit => 'ten' }
+      assert_raise(ArgumentError) {@connection.add_limit_offset!(@select_sql, options) }
+    end
+
+    should 'convert strings which look like integers to integers' do
+      options = { :limit => '42' }
+      assert_nothing_raised(ArgumentError) {@connection.add_limit_offset!(@select_sql, options)}
+    end
+
+    should 'not allow sql injection via limit' do
+      options = { :limit => '1 * FROM schema; DELETE * FROM table; SELECT TOP 10 *'}
+      assert_raise(ArgumentError) { @connection.add_limit_offset!(@select_sql, options) }
+    end
+
+  end
+  
+  context 'When selecting with limit and offset' do
+
+    setup do
+      @select_sql = 'SELECT * FROM books'
+      @books = (1..10).map {|i| Book.create!}
+    end
+    
+    teardown do
+      @books.each {|b| b.destroy}
+    end
+
+    should 'have limit if offset is passed' do
+      options = { :offset => 1 }
+      assert_raise(ArgumentError) { @connection.add_limit_offset!(@select_sql, options) }
+    end
+
+    should 'only allow integers for offset' do
+      options = { :limit => 10, :offset => 'five' }
+      assert_raise(ArgumentError) { @connection.add_limit_offset!(@select_sql, options)}
+    end
+
+    should 'convert strings which look like integers to integers' do
+      options = { :limit => 10, :offset => '5' }
+      assert_nothing_raised(ArgumentError) {@connection.add_limit_offset!(@select_sql, options)}
+    end
+
+    should 'alter SQL to limit number of records returned offset by specified amount' do
+      options = { :limit => 3, :offset => 5 }
+      expected_sql = %&SELECT * FROM (SELECT TOP 3 * FROM (SELECT TOP 8 * FROM books) AS tmp1) AS tmp2&
+      assert_equal(expected_sql, @connection.add_limit_offset!(@select_sql, options))
+    end
+
+    # Not really sure what an offset sql injection might look like
+    should 'not allow sql injection via offset' do
+      options = { :limit => 10, :offset => '1 * FROM schema; DELETE * FROM table; SELECT TOP 10 *'}
+      assert_raise(ArgumentError) { @connection.add_limit_offset!(@select_sql, options) }
+    end
+
+  end
+  
+  
+end
+

data/test/cases/pessimistic_locking_test_sqlserver.rb

@@ -0,0 +1,100 @@
+require 'cases/sqlserver_helper'
+require 'models/person'
+require 'models/reader'
+
+class PessimisticLockingTestSqlserver < ActiveRecord::TestCase
+  
+  self.use_transactional_fixtures = false
+  fixtures :people, :readers
+  
+  def setup
+    Person.columns; Reader.columns # Avoid introspection queries during tests.
+  end
+  
+  context 'For simple finds with default lock option' do
+
+    should 'lock with simple find' do
+      assert_nothing_raised do
+        Person.transaction do
+          Person.find 1, :lock => true
+        end
+      end
+    end
+
+    should 'lock with scoped find' do
+      assert_nothing_raised do
+        Person.transaction do
+          Person.with_scope(:find => { :lock => true }) do
+            Person.find 1
+          end
+        end
+      end
+    end
+
+    should 'lock with eager find' do
+      assert_nothing_raised do
+        Person.transaction do
+          Person.find 1, :include => :readers, :lock => true
+        end
+      end
+    end
+
+    should 'reload with lock when #lock! called' do
+      assert_nothing_raised do
+        Person.transaction do
+          person = Person.find 1
+          old, person.first_name = person.first_name, 'fooman'
+          person.lock!
+          assert_equal old, person.first_name
+        end
+      end
+    end
+
+  end
+  
+  context 'For dueling concurrent connections' do
+    
+    use_concurrent_connections
+
+    should 'no locks does not wait' do
+      first, second = duel { Person.find 1 }
+      assert first.end > second.end
+    end
+    
+    should 'that second lock waits' do
+      assert [0.2, 1, 5].any? { |zzz|
+        first, second = duel(zzz) { Person.find 1, :lock => true }
+        second.end > first.end
+      }
+    end
+    
+  end
+  
+    
+  protected
+  
+  def duel(zzz = 5)
+    t0, t1, t2, t3 = nil, nil, nil, nil
+    a = Thread.new do
+      t0 = Time.now
+      Person.transaction do
+        yield
+        sleep zzz       # block thread 2 for zzz seconds
+      end
+      t1 = Time.now
+    end
+    b = Thread.new do
+      sleep zzz / 2.0   # ensure thread 1 tx starts first
+      t2 = Time.now
+      Person.transaction { yield }
+      t3 = Time.now
+    end
+    a.join
+    b.join
+    assert t1 > t0 + zzz
+    assert t2 > t0
+    assert t3 > t2
+    [t0.to_f..t1.to_f, t2.to_f..t3.to_f]
+  end
+  
+end

data/test/cases/query_cache_test_sqlserver.rb

@@ -0,0 +1,24 @@
+require 'cases/sqlserver_helper'
+require 'models/task'
+
+class QueryCacheTestSqlserver < ActiveRecord::TestCase
+end
+
+class QueryCacheTest < ActiveRecord::TestCase
+  
+  COERCED_TESTS = [:test_cache_does_not_wrap_string_results_in_arrays]
+  
+  include SqlserverCoercedTest
+  
+  fixtures :tasks
+  
+
+  def test_coerced_test_cache_does_not_wrap_string_results_in_arrays
+    Task.cache do
+      assert_instance_of Fixnum, Task.connection.select_value("SELECT count(*) AS count_all FROM tasks")
+    end
+  end
+  
+end
+
+

data/test/cases/schema_dumper_test_sqlserver.rb

@@ -0,0 +1,40 @@
+require 'cases/sqlserver_helper'
+
+class SchemaDumperTestSqlserver < ActiveRecord::TestCase
+  
+  context 'In schema dump' do
+    
+    should 'include limit constraint for integer columns' do
+      output = standard_dump(/^(?!integer_limits)/)
+      assert_match %r{c_int_1.*:limit => 2}, output
+      assert_match %r{c_int_2.*:limit => 2}, output
+      assert_match %r{c_int_3.*}, output
+      assert_match %r{c_int_4.*}, output
+      assert_no_match %r{c_int_3.*:limit}, output
+      assert_no_match %r{c_int_4.*:limit}, output
+      assert_match %r{c_int_5.*:limit => 8}, output
+      assert_match %r{c_int_6.*:limit => 8}, output
+      assert_match %r{c_int_7.*:limit => 8}, output
+      assert_match %r{c_int_8.*:limit => 8}, output
+    end
+    
+    should 'honor nonstandard primary keys' do
+      output = standard_dump
+      match = output.match(%r{create_table "movies"(.*)do})
+      assert_not_nil(match, "nonstandardpk table not found")
+      assert_match %r(:primary_key => "movieid"), match[1], "non-standard primary key not preserved"
+    end
+    
+  end
+  
+  
+  private
+  
+  def standard_dump(ignore_tables = [])
+    stream = StringIO.new
+    ActiveRecord::SchemaDumper.ignore_tables = [*ignore_tables]
+    ActiveRecord::SchemaDumper.dump(ActiveRecord::Base.connection, stream)
+    stream.string
+  end
+  
+end

data/test/cases/specific_schema_test_sqlserver.rb

@@ -0,0 +1,25 @@
+require 'cases/sqlserver_helper'
+
+class StringDefault < ActiveRecord::Base; end;
+
+class SpecificSchemaTestSqlserver < ActiveRecord::TestCase
+  
+  should 'default strings before save' do
+    default = StringDefault.new
+    assert_equal nil, default.string_with_null_default
+    assert_equal 'null', default.string_with_pretend_null_one
+    assert_equal '(null)', default.string_with_pretend_null_two
+    assert_equal 'NULL', default.string_with_pretend_null_three
+    assert_equal '(NULL)', default.string_with_pretend_null_four
+  end
+
+  should 'default strings after save' do
+    default = StringDefault.create
+    assert_equal nil, default.string_with_null_default
+    assert_equal 'null', default.string_with_pretend_null_one
+    assert_equal '(null)', default.string_with_pretend_null_two
+    assert_equal 'NULL', default.string_with_pretend_null_three
+    assert_equal '(NULL)', default.string_with_pretend_null_four
+  end
+  
+end

data/test/cases/sqlserver_helper.rb

@@ -0,0 +1,88 @@
+require 'rubygems'
+require 'shoulda'
+require 'mocha'
+require 'cases/helper'
+require 'models/topic'
+
+SQLSERVER_TEST_ROOT       = File.expand_path(File.join(File.dirname(__FILE__),'..'))
+SQLSERVER_ASSETS_ROOT     = SQLSERVER_TEST_ROOT + "/assets"
+SQLSERVER_FIXTURES_ROOT   = SQLSERVER_TEST_ROOT + "/fixtures"
+SQLSERVER_MIGRATIONS_ROOT = SQLSERVER_TEST_ROOT + "/migrations"
+SQLSERVER_SCHEMA_ROOT     = SQLSERVER_TEST_ROOT + "/schema"
+ACTIVERECORD_TEST_ROOT    = File.expand_path(SQLSERVER_TEST_ROOT + "/../../../../rails/activerecord/test/")
+
+ActiveRecord::Migration.verbose = false
+
+# Defining our classes in one place as well as soem core tests that need coercing date/time types.
+
+class TableWithRealColumn < ActiveRecord::Base; end
+class FkTestHasFk < ActiveRecord::Base ; end
+class FkTestHasPk < ActiveRecord::Base ; end
+class NumericData < ActiveRecord::Base ; self.table_name = 'numeric_data' ; end
+class SqlServerChronic < ActiveRecord::Base
+  coerce_sqlserver_date :date
+  coerce_sqlserver_time :time
+  default_timezone = :utc
+end
+class Topic < ActiveRecord::Base
+  coerce_sqlserver_date :last_read
+  coerce_sqlserver_time :bonus_time
+end
+class Person < ActiveRecord::Base
+  coerce_sqlserver_date :favorite_day
+end
+
+# A module that we can include in classes where we want to override an active record test.
+
+module SqlserverCoercedTest
+  def self.included(base)
+    base.extend ClassMethods
+  end
+  module ClassMethods
+    def coerced_tests
+      self.const_get(:COERCED_TESTS) rescue nil
+    end
+    def method_added(method)
+      undef_method(method) if coerced_tests && coerced_tests.include?(method)
+    end
+  end
+end
+
+# Change the text database type to support ActiveRecord's tests for = on text columns which 
+# is not supported in SQL Server text columns, so use varchar(8000) instead.
+
+if ActiveRecord::Base.connection.sqlserver_2000?
+  ActiveRecord::ConnectionAdapters::SQLServerAdapter.native_text_database_type = 'varchar(8000)'
+end
+
+# Our changes/additions to ActiveRecord test helpers specific for SQL Server.
+
+ActiveRecord::Base.connection.class.class_eval do
+  IGNORED_SQL << /SELECT SCOPE_IDENTITY/ << /INFORMATION_SCHEMA.TABLES/ << /INFORMATION_SCHEMA.COLUMNS/
+end
+
+ActiveRecord::ConnectionAdapters::SQLServerAdapter.class_eval do
+  def raw_select_with_query_record(sql, name = nil)
+    $queries_executed ||= []
+    $queries_executed << sql unless IGNORED_SQL.any? { |r| sql =~ r }
+    raw_select_without_query_record(sql,name)
+  end
+  alias_method_chain :raw_select, :query_record
+end
+
+module ActiveRecord 
+  class TestCase < ActiveSupport::TestCase
+    def assert_sql(*patterns_to_match)
+      $queries_executed = []
+      yield
+    ensure
+      failed_patterns = []
+      patterns_to_match.each do |pattern|
+        failed_patterns << pattern unless $queries_executed.any?{ |sql| pattern === sql }
+      end
+      assert failed_patterns.empty?, "Query pattern(s) #{failed_patterns.map(&:inspect).join(', ')} not found in:\n#{$queries_executed.inspect}"
+    end
+  end
+end
+
+

data/test/connections/native_sqlserver/connection.rb

@@ -0,0 +1,23 @@
+print "Using native SQLServer\n"
+require_dependency 'models/course'
+require 'logger'
+
+ActiveRecord::Base.logger = Logger.new("debug.log")
+
+ActiveRecord::Base.configurations = {
+  'arunit' => {
+    :adapter  => 'sqlserver',
+    :host     => 'localhost',
+    :username => 'rails',
+    :database => 'activerecord_unittest'
+  },
+  'arunit2' => {
+    :adapter  => 'sqlserver',
+    :host     => 'localhost',
+    :username => 'rails',
+    :database => 'activerecord_unittest2'
+  }
+}
+
+ActiveRecord::Base.establish_connection 'arunit'
+Course.establish_connection 'arunit2'

data/test/connections/native_sqlserver_odbc/connection.rb

@@ -0,0 +1,25 @@
+print "Using native SQLServer via ODBC\n"
+require_dependency 'models/course'
+require 'logger'
+
+ActiveRecord::Base.logger = Logger.new("debug.log")
+
+ActiveRecord::Base.configurations = {
+  'arunit' => {
+    :adapter  => 'sqlserver',
+    :mode     => 'ODBC',
+    :host     => 'localhost',
+    :username => 'rails',
+    :dsn => 'activerecord_unittest'
+  },
+  'arunit2' => {
+    :adapter  => 'sqlserver',
+    :mode     => 'ODBC',
+    :host     => 'localhost',
+    :username => 'rails',
+    :dsn => 'activerecord_unittest2'
+  }
+}
+
+ActiveRecord::Base.establish_connection 'arunit'
+Course.establish_connection 'arunit2'