activerecord-sqlserver-adapter 5.2.1 → 6.0.2

data/lib/arel_sqlserver.rb

@@ -1,2 +1,4 @@
-require 'arel'
-require 'arel/visitors/sqlserver'
+# frozen_string_literal: true
+
+require "arel"
+require "arel/visitors/sqlserver"

data/test/appveyor/dbsetup.ps1

@@ -5,15 +5,15 @@ Write-Output "Setting up..."
 
 Write-Output "Setting variables..."
 $serverName = $env:COMPUTERNAME
-$instances = @('SQL2012SP1', 'SQL2014')
+$instanceNames = @('SQL2014')
 $smo = 'Microsoft.SqlServer.Management.Smo.'
 $wmi = new-object ($smo + 'Wmi.ManagedComputer')
 
 Write-Output "Configure Instances..."
-foreach ($instance in $instances) {
-  Write-Output "Instance $instance ..."
+foreach ($instanceName in $instanceNames) {
+  Write-Output "Instance $instanceName ..."
   Write-Output "Enable TCP/IP and port 1433..."
-  $uri = "ManagedComputer[@Name='$serverName']/ServerInstance[@Name='$instance']/ServerProtocol[@Name='Tcp']"
+  $uri = "ManagedComputer[@Name='$serverName']/ServerInstance[@Name='$instanceName']/ServerProtocol[@Name='Tcp']"
   $tcp = $wmi.GetSmoObject($uri)
   $tcp.IsEnabled = $true
   foreach ($ipAddress in $Tcp.IPAddresses) {

data/test/cases/adapter_test_sqlserver.rb

@@ -1,19 +1,21 @@
-require 'cases/helper_sqlserver'
-require 'models/topic'
-require 'models/task'
-require 'models/post'
-require 'models/subscriber'
-require 'models/minimalistic'
+# frozen_string_literal: true
 
-class AdapterTestSQLServer < ActiveRecord::TestCase
+require "cases/helper_sqlserver"
+require "models/topic"
+require "models/task"
+require "models/post"
+require "models/subscriber"
+require "models/minimalistic"
+require "models/college"
 
+class AdapterTestSQLServer < ActiveRecord::TestCase
   fixtures :tasks
 
   let(:basic_insert_sql) { "INSERT INTO [funny_jokes] ([name]) VALUES('Knock knock')" }
   let(:basic_update_sql) { "UPDATE [customers] SET [address_street] = NULL WHERE [id] = 2" }
   let(:basic_select_sql) { "SELECT * FROM [customers] WHERE ([customers].[id] = 1)" }
 
-  it 'has basic and non-senstive information in the adpaters inspect method' do
+  it "has basic and non-sensitive information in the adapters inspect method" do
     string = connection.inspect
     _(string).must_match %r{ActiveRecord::ConnectionAdapters::SQLServerAdapter}
     _(string).must_match %r{version\: \d.\d}
@@ -25,48 +27,97 @@ class AdapterTestSQLServer < ActiveRecord::TestCase
     _(string).wont_match %r{port}
   end
 
-  it 'has a 128 max #table_alias_length' do
+  it "has a 128 max #table_alias_length" do
     assert connection.table_alias_length <= 128
   end
 
-  it 'raises invalid statement error for bad SQL' do
+  it "raises invalid statement error for bad SQL" do
     assert_raise(ActiveRecord::StatementInvalid) { Topic.connection.update("UPDATE XXX") }
   end
 
-  it 'is has our adapter_name' do
-    assert_equal 'SQLServer', connection.adapter_name
+  it "is has our adapter_name" do
+    assert_equal "SQLServer", connection.adapter_name
   end
 
-  it 'support DDL in transactions' do
+  it "support DDL in transactions" do
     assert connection.supports_ddl_transactions?
   end
 
-  it 'allow owner table name prefixs like dbo to still allow table exists to return true' do
+  it "table exists works if table name prefixed by schema and owner" do
     begin
-      assert_equal 'topics', Topic.table_name
+      assert_equal "topics", Topic.table_name
       assert Topic.table_exists?
-      Topic.table_name = 'dbo.topics'
-      assert Topic.table_exists?, 'Tasks table name of dbo.topics should return true for exists.'
+
+      # Test when owner included in table name.
+      Topic.table_name = "dbo.topics"
+      assert Topic.table_exists?, "Topics table name of 'dbo.topics' should return true for exists."
+
+      # Test when database and owner included in table name.
+      Topic.table_name = "#{ActiveRecord::Base.configurations["arunit"]['database']}.dbo.topics"
+      assert Topic.table_exists?, "Topics table name of '[DATABASE].dbo.topics' should return true for exists."
     ensure
-      Topic.table_name = 'topics'
+      Topic.table_name = "topics"
     end
   end
 
-  it 'return true to insert sql query for inserts only' do
-    assert connection.send(:insert_sql?,'INSERT...')
+  it "test table existence across database schemas" do
+    arunit_connection = Topic.connection
+    arunit2_connection = College.connection
+
+    arunit_database = arunit_connection.pool.spec.config[:database]
+    arunit2_database = arunit2_connection.pool.spec.config[:database]
+
+    # Assert that connections use different default databases schemas.
+    assert_not_equal arunit_database, arunit2_database
+
+    # Assert that the Topics table exists when using the Topics connection.
+    assert arunit_connection.table_exists?('topics'), 'Topics table exists using table name'
+    assert arunit_connection.table_exists?('dbo.topics'), 'Topics table exists using owner and table name'
+    assert arunit_connection.table_exists?("#{arunit_database}.dbo.topics"), 'Topics table exists using database, owner and table name'
+
+    # Assert that the Colleges table exists when using the Colleges connection.
+    assert arunit2_connection.table_exists?('colleges'), 'College table exists using table name'
+    assert arunit2_connection.table_exists?('dbo.colleges'), 'College table exists using owner and table name'
+    assert arunit2_connection.table_exists?("#{arunit2_database}.dbo.colleges"), 'College table exists using database, owner and table name'
+
+    # Assert that the tables exist when using each others connection.
+    assert arunit_connection.table_exists?("#{arunit2_database}.dbo.colleges"), 'Colleges table exists using Topics connection'
+    assert arunit2_connection.table_exists?("#{arunit_database}.dbo.topics"), 'Topics table exists using Colleges connection'
+  end
+
+  it "return true to insert sql query for inserts only" do
+    assert connection.send(:insert_sql?, "INSERT...")
     assert connection.send(:insert_sql?, "EXEC sp_executesql N'INSERT INTO [fk_test_has_fks] ([fk_id]) VALUES (@0); SELECT CAST(SCOPE_IDENTITY() AS bigint) AS Ident', N'@0 int', @0 = 0")
-    assert !connection.send(:insert_sql?,'UPDATE...')
-    assert !connection.send(:insert_sql?,'SELECT...')
+    assert !connection.send(:insert_sql?, "UPDATE...")
+    assert !connection.send(:insert_sql?, "SELECT...")
+  end
+
+  it "return unquoted table name object from basic INSERT UPDATE and SELECT statements" do
+    assert_equal "funny_jokes", connection.send(:get_table_name, basic_insert_sql)
+    assert_equal "customers", connection.send(:get_table_name, basic_update_sql)
+    assert_equal "customers", connection.send(:get_table_name, basic_select_sql)
   end
 
-  it 'return unquoted table name object from basic INSERT UPDATE and SELECT statements' do
-    assert_equal 'funny_jokes', connection.send(:get_table_name, basic_insert_sql)
-    assert_equal 'customers', connection.send(:get_table_name, basic_update_sql)
-    assert_equal 'customers', connection.send(:get_table_name, basic_select_sql)
+  it "test bad connection" do
+    assert_raise ActiveRecord::NoDatabaseError do
+      config = ActiveRecord::Base.configurations["arunit"].merge(database: "inexistent_activerecord_unittest")
+      ActiveRecord::Base.sqlserver_connection config
+    end
+  end
+
+  it "test database exists returns false if database does not exist" do
+    config = ActiveRecord::Base.configurations["arunit"].merge(database: "inexistent_activerecord_unittest")
+    assert_not ActiveRecord::ConnectionAdapters::SQLServerAdapter.database_exists?(config),
+               "expected database to not exist"
   end
 
-  describe 'with different language' do
+  it "test database exists returns true when the database exists" do
+    config = ActiveRecord::Base.configurations["arunit"]
+    assert ActiveRecord::ConnectionAdapters::SQLServerAdapter.database_exists?(config),
+           "expected database #{config[:database]} to exist"
+  end
 
+  describe "with different language" do
     before do
       @default_language = connection.user_options_language
     end
@@ -76,18 +127,18 @@ class AdapterTestSQLServer < ActiveRecord::TestCase
       connection.send :initialize_dateformatter
     end
 
-    it 'memos users dateformat' do
+    it "memos users dateformat" do
       connection.execute("SET LANGUAGE us_english") rescue nil
       dateformat = connection.instance_variable_get(:@database_dateformat)
-      assert_equal 'mdy', dateformat
+      assert_equal "mdy", dateformat
     end
 
-    it 'has a dateformatter' do
+    it "has a dateformatter" do
       assert Date::DATE_FORMATS[:_sqlserver_dateformat]
       assert Time::DATE_FORMATS[:_sqlserver_dateformat]
     end
 
-    it 'does a datetime insertion when language is german' do
+    it "does a datetime insertion when language is german" do
       connection.execute("SET LANGUAGE deutsch")
       connection.send :initialize_dateformatter
       assert_nothing_raised do
@@ -96,40 +147,36 @@ class AdapterTestSQLServer < ActiveRecord::TestCase
         Task.create! starting: starting, ending: ending
       end
     end
-
   end
 
-  describe 'testing #lowercase_schema_reflection' do
-
+  describe "testing #lowercase_schema_reflection" do
     before do
       SSTestUpper.delete_all
-      SSTestUpper.create COLUMN1: 'Got a minute?', COLUMN2: 419
-      SSTestUpper.create COLUMN1: 'Favorite number?', COLUMN2: 69
+      SSTestUpper.create COLUMN1: "Got a minute?", COLUMN2: 419
+      SSTestUpper.create COLUMN1: "Favorite number?", COLUMN2: 69
     end
 
     after do
       connection.lowercase_schema_reflection = false
     end
 
-    it 'not lowercase schema reflection by default' do
-      assert SSTestUpper.columns_hash['COLUMN1']
-      assert_equal 'Got a minute?', SSTestUpper.first.COLUMN1
-      assert_equal 'Favorite number?', SSTestUpper.last.COLUMN1
-      assert SSTestUpper.columns_hash['COLUMN2']
+    it "not lowercase schema reflection by default" do
+      assert SSTestUpper.columns_hash["COLUMN1"]
+      assert_equal "Got a minute?", SSTestUpper.first.COLUMN1
+      assert_equal "Favorite number?", SSTestUpper.last.COLUMN1
+      assert SSTestUpper.columns_hash["COLUMN2"]
     end
 
-    it 'lowercase schema reflection when set' do
+    it "lowercase schema reflection when set" do
       connection.lowercase_schema_reflection = true
-      assert SSTestUppered.columns_hash['column1']
-      assert_equal 'Got a minute?', SSTestUppered.first.column1
-      assert_equal 'Favorite number?', SSTestUppered.last.column1
-      assert SSTestUppered.columns_hash['column2']
+      assert SSTestUppered.columns_hash["column1"]
+      assert_equal "Got a minute?", SSTestUppered.first.column1
+      assert_equal "Favorite number?", SSTestUppered.last.column1
+      assert SSTestUppered.columns_hash["column2"]
     end
-
   end
 
-  describe 'identity inserts' do
-
+  describe "identity inserts" do
     before do
       @identity_insert_sql = "INSERT INTO [funny_jokes] ([id],[name]) VALUES(420,'Knock knock')"
       @identity_insert_sql_unquoted = "INSERT INTO funny_jokes (id, name) VALUES(420, 'Knock knock')"
@@ -139,62 +186,60 @@ class AdapterTestSQLServer < ActiveRecord::TestCase
       @identity_insert_sql_unordered_sp = "EXEC sp_executesql N'INSERT INTO [funny_jokes] ([name],[id]) VALUES (@0, @1)', N'@0 nvarchar(255), @1  int', @0 = N'Knock knock', @1 = 420"
     end
 
-    it 'return quoted table_name to #query_requires_identity_insert? when INSERT sql contains id column' do
-      assert_equal 'funny_jokes', connection.send(:query_requires_identity_insert?,@identity_insert_sql)
-      assert_equal 'funny_jokes', connection.send(:query_requires_identity_insert?,@identity_insert_sql_unquoted)
-      assert_equal 'funny_jokes', connection.send(:query_requires_identity_insert?,@identity_insert_sql_unordered)
-      assert_equal 'funny_jokes', connection.send(:query_requires_identity_insert?,@identity_insert_sql_sp)
-      assert_equal 'funny_jokes', connection.send(:query_requires_identity_insert?,@identity_insert_sql_unquoted_sp)
-      assert_equal 'funny_jokes', connection.send(:query_requires_identity_insert?,@identity_insert_sql_unordered_sp)
+    it "return quoted table_name to #query_requires_identity_insert? when INSERT sql contains id column" do
+      assert_equal "funny_jokes", connection.send(:query_requires_identity_insert?, @identity_insert_sql)
+      assert_equal "funny_jokes", connection.send(:query_requires_identity_insert?, @identity_insert_sql_unquoted)
+      assert_equal "funny_jokes", connection.send(:query_requires_identity_insert?, @identity_insert_sql_unordered)
+      assert_equal "funny_jokes", connection.send(:query_requires_identity_insert?, @identity_insert_sql_sp)
+      assert_equal "funny_jokes", connection.send(:query_requires_identity_insert?, @identity_insert_sql_unquoted_sp)
+      assert_equal "funny_jokes", connection.send(:query_requires_identity_insert?, @identity_insert_sql_unordered_sp)
     end
 
-    it 'return false to #query_requires_identity_insert? for normal SQL' do
+    it "return false to #query_requires_identity_insert? for normal SQL" do
       [basic_insert_sql, basic_update_sql, basic_select_sql].each do |sql|
-        assert !connection.send(:query_requires_identity_insert?,sql), "SQL was #{sql}"
+        assert !connection.send(:query_requires_identity_insert?, sql), "SQL was #{sql}"
       end
     end
 
-    it 'find identity column using #identity_columns' do
-      task_id_column = Task.columns_hash['id']
+    it "find identity column using #identity_columns" do
+      task_id_column = Task.columns_hash["id"]
       assert_equal task_id_column.name, connection.send(:identity_columns, Task.table_name).first.name
       assert_equal task_id_column.sql_type, connection.send(:identity_columns, Task.table_name).first.sql_type
     end
 
-    it 'return an empty array when calling #identity_columns for a table_name with no identity' do
+    it "return an empty array when calling #identity_columns for a table_name with no identity" do
       _(connection.send(:identity_columns, Subscriber.table_name)).must_equal []
     end
-
   end
 
-  describe 'quoting' do
-
-    it 'return 1 for #quoted_true' do
-      assert_equal '1', connection.quoted_true
+  describe "quoting" do
+    it "return 1 for #quoted_true" do
+      assert_equal "1", connection.quoted_true
     end
 
-    it 'return 0 for #quoted_false' do
-      assert_equal '0', connection.quoted_false
+    it "return 0 for #quoted_false" do
+      assert_equal "0", connection.quoted_false
     end
 
-    it 'not escape backslash characters like abstract adapter' do
+    it "not escape backslash characters like abstract adapter" do
       string_with_backslashs = "\\n"
       assert_equal string_with_backslashs, connection.quote_string(string_with_backslashs)
     end
 
-    it 'quote column names with brackets' do
-      assert_equal '[foo]', connection.quote_column_name(:foo)
-      assert_equal '[foo]', connection.quote_column_name('foo')
-      assert_equal '[foo].[bar]', connection.quote_column_name('foo.bar')
+    it "quote column names with brackets" do
+      assert_equal "[foo]", connection.quote_column_name(:foo)
+      assert_equal "[foo]", connection.quote_column_name("foo")
+      assert_equal "[foo].[bar]", connection.quote_column_name("foo.bar")
     end
 
-    it 'not quote already quoted column names with brackets' do
-      assert_equal '[foo]', connection.quote_column_name('[foo]')
-      assert_equal '[foo].[bar]', connection.quote_column_name('[foo].[bar]')
+    it "not quote already quoted column names with brackets" do
+      assert_equal "[foo]", connection.quote_column_name("[foo]")
+      assert_equal "[foo].[bar]", connection.quote_column_name("[foo].[bar]")
     end
 
-    it 'quote table names like columns' do
-      assert_equal '[foo].[bar]', connection.quote_column_name('foo.bar')
-      assert_equal '[foo].[bar].[baz]', connection.quote_column_name('foo.bar.baz')
+    it "quote table names like columns" do
+      assert_equal "[foo].[bar]", connection.quote_column_name("foo.bar")
+      assert_equal "[foo].[bar].[baz]", connection.quote_column_name("foo.bar.baz")
     end
 
     it "surround string with national prefix" do
@@ -204,44 +249,40 @@ class AdapterTestSQLServer < ActiveRecord::TestCase
     it "escape all single quotes by repeating them" do
       assert_equal "N'''quotation''s'''", connection.quote("'quotation's'")
     end
-
   end
 
-  describe 'disabling referential integrity' do
-
+  describe "disabling referential integrity" do
     before do
       connection.disable_referential_integrity { SSTestHasPk.delete_all; SSTestHasFk.delete_all }
       @parent = SSTestHasPk.create!
       @member = SSTestHasFk.create!(fk_id: @parent.id)
     end
 
-    it 'NOT ALLOW by default the deletion of a referenced parent' do
-      SSTestHasPk.connection.disable_referential_integrity { }
+    it "NOT ALLOW by default the deletion of a referenced parent" do
+      SSTestHasPk.connection.disable_referential_integrity {}
       assert_raise(ActiveRecord::StatementInvalid) { @parent.destroy }
     end
 
-    it 'ALLOW deletion of referenced parent using #disable_referential_integrity block' do
+    it "ALLOW deletion of referenced parent using #disable_referential_integrity block" do
       SSTestHasPk.connection.disable_referential_integrity { @parent.destroy }
     end
 
-    it 'again NOT ALLOW deletion of referenced parent after #disable_referential_integrity block' do
+    it "again NOT ALLOW deletion of referenced parent after #disable_referential_integrity block" do
       assert_raise(ActiveRecord::StatementInvalid) do
-        SSTestHasPk.connection.disable_referential_integrity { }
+        SSTestHasPk.connection.disable_referential_integrity {}
         @parent.destroy
       end
     end
 
-    it 'not disable referential integrity for the same table twice' do
+    it "not disable referential integrity for the same table twice" do
       tables = SSTestHasPk.connection.tables_with_referential_integrity
       assert_equal tables.size, tables.uniq.size
     end
-
   end
 
-  describe 'database statements' do
-
+  describe "database statements" do
     it "run the database consistency checker useroptions command" do
-      skip 'on azure' if connection_sqlserver_azure?
+      skip "on azure" if connection_sqlserver_azure?
       keys = [:textsize, :language, :isolation_level, :dateformat]
       user_options = connection.user_options
       keys.each do |key|
@@ -251,183 +292,217 @@ class AdapterTestSQLServer < ActiveRecord::TestCase
     end
 
     it "return a underscored key hash with indifferent access of the results" do
-      skip 'on azure' if connection_sqlserver_azure?
+      skip "on azure" if connection_sqlserver_azure?
       user_options = connection.user_options
-      assert_equal 'read committed', user_options['isolation_level']
-      assert_equal 'read committed', user_options[:isolation_level]
+      assert_equal "read committed", user_options["isolation_level"]
+      assert_equal "read committed", user_options[:isolation_level]
     end
-
   end
 
-  describe 'schema statements' do
-
-    it 'create integers when no limit supplied' do
-      assert_equal 'integer', connection.type_to_sql(:integer)
+  describe "schema statements" do
+    it "create integers when no limit supplied" do
+      assert_equal "integer", connection.type_to_sql(:integer)
     end
 
-    it 'create integers when limit is 4' do
-      assert_equal 'integer', connection.type_to_sql(:integer, limit: 4)
+    it "create integers when limit is 4" do
+      assert_equal "integer", connection.type_to_sql(:integer, limit: 4)
     end
 
-    it 'create integers when limit is 3' do
-      assert_equal 'integer', connection.type_to_sql(:integer, limit: 3)
+    it "create integers when limit is 3" do
+      assert_equal "integer", connection.type_to_sql(:integer, limit: 3)
     end
 
-    it 'create smallints when limit is 2' do
-      assert_equal 'smallint', connection.type_to_sql(:integer, limit: 2)
+    it "create smallints when limit is 2" do
+      assert_equal "smallint", connection.type_to_sql(:integer, limit: 2)
     end
 
-    it 'create tinyints when limit is 1' do
-      assert_equal 'tinyint', connection.type_to_sql(:integer, limit: 1)
+    it "create tinyints when limit is 1" do
+      assert_equal "tinyint", connection.type_to_sql(:integer, limit: 1)
     end
 
-    it 'create bigints when limit is greateer than 4' do
-      assert_equal 'bigint', connection.type_to_sql(:integer, limit: 5)
-      assert_equal 'bigint', connection.type_to_sql(:integer, limit: 6)
-      assert_equal 'bigint', connection.type_to_sql(:integer, limit: 7)
-      assert_equal 'bigint', connection.type_to_sql(:integer, limit: 8)
+    it "create bigints when limit is greateer than 4" do
+      assert_equal "bigint", connection.type_to_sql(:integer, limit: 5)
+      assert_equal "bigint", connection.type_to_sql(:integer, limit: 6)
+      assert_equal "bigint", connection.type_to_sql(:integer, limit: 7)
+      assert_equal "bigint", connection.type_to_sql(:integer, limit: 8)
     end
 
-    it 'create floats when no limit supplied' do
-      assert_equal 'float', connection.type_to_sql(:float)
+    it "create floats when no limit supplied" do
+      assert_equal "float", connection.type_to_sql(:float)
     end
-
   end
 
-  describe 'views' do
-
+  describe "views" do
     # Using connection.views
 
-    it 'return an array' do
+    it "return an array" do
       assert_instance_of Array, connection.views
     end
 
-    it 'find SSTestCustomersView table name' do
-      _(connection.views).must_include 'sst_customers_view'
+    it "find SSTestCustomersView table name" do
+      _(connection.views).must_include "sst_customers_view"
     end
 
-    it 'work with dynamic finders' do
-      name = 'MetaSkills'
+    it "work with dynamic finders" do
+      name = "MetaSkills"
       customer = SSTestCustomersView.create! name: name
       assert_equal customer, SSTestCustomersView.find_by_name(name)
     end
 
-    it 'not contain system views' do
-      systables = ['sysconstraints','syssegments']
+    it "not contain system views" do
+      systables = ["sysconstraints", "syssegments"]
       systables.each do |systable|
         assert !connection.views.include?(systable), "This systable #{systable} should not be in the views array."
       end
     end
 
-    it 'allow the connection#view_information method to return meta data on the view' do
-      view_info = connection.send(:view_information,'sst_customers_view')
-      assert_equal('sst_customers_view', view_info['TABLE_NAME'])
-      assert_match(/CREATE VIEW sst_customers_view/, view_info['VIEW_DEFINITION'])
+    it "allow the connection#view_information method to return meta data on the view" do
+      view_info = connection.send(:view_information, "sst_customers_view")
+      assert_equal("sst_customers_view", view_info["TABLE_NAME"])
+      assert_match(/CREATE VIEW sst_customers_view/, view_info["VIEW_DEFINITION"])
     end
 
-    it 'allow the connection#view_table_name method to return true table_name for the view' do
-      assert_equal 'customers', connection.send(:view_table_name,'sst_customers_view')
-      assert_equal 'topics', connection.send(:view_table_name,'topics'), 'No view here, the same table name should come back.'
+    it "allow the connection#view_table_name method to return true table_name for the view" do
+      assert_equal "customers", connection.send(:view_table_name, "sst_customers_view")
+      assert_equal "topics", connection.send(:view_table_name, "topics"), "No view here, the same table name should come back."
     end
 
     # With same column names
 
-    it 'have matching column objects' do
-      columns = ['id','name','balance']
+    it "have matching column objects" do
+      columns = ["id", "name", "balance"]
       assert !SSTestCustomersView.columns.blank?
       assert_equal columns.size, SSTestCustomersView.columns.size
       columns.each do |colname|
         assert_instance_of ActiveRecord::ConnectionAdapters::SQLServerColumn,
-          SSTestCustomersView.columns_hash[colname],
-          "Column name #{colname.inspect} was not found in these columns #{SSTestCustomersView.columns.map(&:name).inspect}"
+                           SSTestCustomersView.columns_hash[colname],
+                           "Column name #{colname.inspect} was not found in these columns #{SSTestCustomersView.columns.map(&:name).inspect}"
       end
     end
 
-    it 'find identity column' do
-      _(SSTestCustomersView.primary_key).must_equal 'id'
-      _(connection.primary_key(SSTestCustomersView.table_name)).must_equal 'id'
-      _(SSTestCustomersView.columns_hash['id']).must_be :is_identity?
+    it "find identity column" do
+      _(SSTestCustomersView.primary_key).must_equal "id"
+      _(connection.primary_key(SSTestCustomersView.table_name)).must_equal "id"
+      _(SSTestCustomersView.columns_hash["id"]).must_be :is_identity?
     end
 
-    it 'find default values' do
+    it "find default values" do
       assert_equal 0, SSTestCustomersView.new.balance
     end
 
-    it 'respond true to data_source_exists?' do
+    it "respond true to data_source_exists?" do
       assert SSTestCustomersView.connection.data_source_exists?(SSTestCustomersView.table_name)
     end
 
     # With aliased column names
 
-    it 'have matching column objects' do
-      columns = ['id','pretend_null']
+    it "have matching column objects" do
+      columns = ["id", "pretend_null"]
       assert !SSTestStringDefaultsView.columns.blank?
       assert_equal columns.size, SSTestStringDefaultsView.columns.size
       columns.each do |colname|
         assert_instance_of ActiveRecord::ConnectionAdapters::SQLServerColumn,
-          SSTestStringDefaultsView.columns_hash[colname],
-          "Column name #{colname.inspect} was not found in these columns #{SSTestStringDefaultsView.columns.map(&:name).inspect}"
+                           SSTestStringDefaultsView.columns_hash[colname],
+                           "Column name #{colname.inspect} was not found in these columns #{SSTestStringDefaultsView.columns.map(&:name).inspect}"
       end
     end
 
-    it 'find identity column' do
-      _(SSTestStringDefaultsView.primary_key).must_equal 'id'
-      _(connection.primary_key(SSTestStringDefaultsView.table_name)).must_equal 'id'
-      _(SSTestStringDefaultsView.columns_hash['id']).must_be :is_identity?
+    it "find identity column" do
+      _(SSTestStringDefaultsView.primary_key).must_equal "id"
+      _(connection.primary_key(SSTestStringDefaultsView.table_name)).must_equal "id"
+      _(SSTestStringDefaultsView.columns_hash["id"]).must_be :is_identity?
     end
 
-    it 'find default values' do
-      assert_equal 'null', SSTestStringDefaultsView.new.pretend_null,
-        SSTestStringDefaultsView.columns_hash['pretend_null'].inspect
+    it "find default values" do
+      assert_equal "null", SSTestStringDefaultsView.new.pretend_null,
+                   SSTestStringDefaultsView.columns_hash["pretend_null"].inspect
     end
 
-    it 'respond true to data_source_exists?' do
+    it "respond true to data_source_exists?" do
       assert SSTestStringDefaultsView.connection.data_source_exists?(SSTestStringDefaultsView.table_name)
     end
 
     # That have more than 4000 chars for their defintion
 
-    it 'cope with null returned for the defintion' do
+    it "cope with null returned for the defintion" do
       assert_nothing_raised() { SSTestStringDefaultsBigView.columns }
     end
 
-    it 'using alternate view defintion still be able to find real default' do
-      assert_equal 'null', SSTestStringDefaultsBigView.new.pretend_null,
-        SSTestStringDefaultsBigView.columns_hash['pretend_null'].inspect
+    it "using alternate view defintion still be able to find real default" do
+      assert_equal "null", SSTestStringDefaultsBigView.new.pretend_null,
+                   SSTestStringDefaultsBigView.columns_hash["pretend_null"].inspect
     end
-
   end
 
-  describe 'database_prefix_remote_server?' do
-
+  describe "database_prefix_remote_server?" do
     after do
       connection_options.delete(:database_prefix)
     end
 
-    it 'returns false if database_prefix is not configured' do
+    it "returns false if database_prefix is not configured" do
       assert_equal false, connection.database_prefix_remote_server?
     end
 
-    it 'returns true if database_prefix has been set' do
+    it "returns true if database_prefix has been set" do
       connection_options[:database_prefix] = "server.database.schema."
       assert_equal true, connection.database_prefix_remote_server?
     end
 
-    it 'returns false if database_prefix has been set incorrectly' do
+    it "returns false if database_prefix has been set incorrectly" do
       connection_options[:database_prefix] = "server.database.schema"
       assert_equal false, connection.database_prefix_remote_server?
     end
-
   end
 
-  it 'in_memory_oltp' do
-    if ENV['IN_MEMORY_OLTP'] && connection.supports_in_memory_oltp?
-      _(SSTMemory.primary_key).must_equal 'id'
-      _(SSTMemory.columns_hash['id']).must_be :is_identity?
+  it "in_memory_oltp" do
+    if ENV["IN_MEMORY_OLTP"] && connection.supports_in_memory_oltp?
+      _(SSTMemory.primary_key).must_equal "id"
+      _(SSTMemory.columns_hash["id"]).must_be :is_identity?
     else
-      skip 'supports_in_memory_oltp? => false'
+      skip "supports_in_memory_oltp? => false"
     end
   end
 
-end
+  describe "block writes to a database" do
+    def setup
+      @conn = ActiveRecord::Base.connection
+      @connection_handler = ActiveRecord::Base.connection_handler
+    end
+
+    def test_errors_when_an_insert_query_is_called_while_preventing_writes
+      assert_raises(ActiveRecord::ReadOnlyError) do
+        @connection_handler.while_preventing_writes do
+          @conn.insert("INSERT INTO [subscribers] ([nick]) VALUES ('aido')")
+        end
+      end
+    end
 
+    def test_errors_when_an_update_query_is_called_while_preventing_writes
+      @conn.insert("INSERT INTO [subscribers] ([nick]) VALUES ('aido')")
+
+      assert_raises(ActiveRecord::ReadOnlyError) do
+        @connection_handler.while_preventing_writes do
+          @conn.update("UPDATE [subscribers] SET [subscribers].[name] = 'Aidan' WHERE [subscribers].[nick] = 'aido'")
+        end
+      end
+    end
+
+    def test_errors_when_a_delete_query_is_called_while_preventing_writes
+      @conn.execute("INSERT INTO [subscribers] ([nick]) VALUES ('aido')")
+
+      assert_raises(ActiveRecord::ReadOnlyError) do
+        @connection_handler.while_preventing_writes do
+          @conn.execute("DELETE FROM [subscribers] WHERE [subscribers].[nick] = 'aido'")
+        end
+      end
+    end
+
+    def test_doesnt_error_when_a_select_query_is_called_while_preventing_writes
+      @conn.execute("INSERT INTO [subscribers] ([nick]) VALUES ('aido')")
+
+      @connection_handler.while_preventing_writes do
+        assert_equal 1, @conn.execute("SELECT * FROM [subscribers] WHERE [subscribers].[nick] = 'aido'")
+      end
+    end
+  end
+end