do_sqlserver-tinytds 0.10.17.alpha

data/lib/do_sqlserver_tinytds/addressable_extension.rb

@@ -0,0 +1,69 @@
+#require "addressable/uri"
+#
+#module Addressable
+#  class URI
+#    def self.parse(uri)
+#
+#      # If we were given nil, return nil.
+#      return nil unless uri
+#      # If a URI object is passed, just return itself.
+#      return uri.dup if uri.kind_of?(self)
+#
+#      # If a URI object of the Ruby standard library variety is passed,
+#      # convert it to a string, then parse the string.
+#      # We do the check this way because we don't want to accidentally
+#      # cause a missing constant exception to be thrown.
+#      if uri.class.name =~ /^URI\b/
+#        uri = uri.to_s
+#      end
+#
+#      # Otherwise, convert to a String
+#      begin
+#        uri = uri.to_str
+#      rescue TypeError, NoMethodError
+#        raise TypeError, "Can't convert #{uri.class} into String."
+#      end if not uri.is_a? String
+#
+#      # This Regexp supplied as an example in RFC 3986, and it works great.
+#      scan = uri.scan(URIREGEX)
+#      fragments = scan[0]
+#      scheme = fragments[1]
+#      authority = fragments[3]
+#      path = fragments[4]
+#      query = fragments[6]
+#      fragment = fragments[8]
+#      user = nil
+#      password = nil
+#      host = nil
+#      port = nil
+#      database = nil
+#      if authority != nil
+#        # The Regexp above doesn't split apart the authority.
+#        userinfo = authority[/^([^\[\]]*)@/, 1]
+#        if userinfo != nil
+#          user = userinfo.strip[/^([^:]*):?/, 1]
+#          password = userinfo.strip[/:(.*)$/, 1]
+#        end
+#        host = authority.gsub(/^([^\[\]]*)@/, EMPTYSTR).gsub(/:([^:@\[\]]*?)$/, EMPTYSTR)
+#        port = authority[/:(\d+).*$/ , 1]#authority[/:([^:@\[\]]*?)$/, 1]
+#        database = authority[/:\d+(.*)$/ , 1]
+#      end
+#      if port == EMPTYSTR
+#        port = nil
+#      end
+#
+#      return Addressable::URI.new(
+#        :scheme => scheme,
+#        :user => user,
+#        :password => password,
+#        :host => host,
+#        :port => port,
+#        :path => path,
+#        :query => query,
+#        :fragment => fragment,
+#        :database => database
+#      )
+#
+#    end
+#  end
+#end

data/lib/do_sqlserver_tinytds/tiny_tds_extension.rb

@@ -0,0 +1,123 @@
+require 'tiny_tds'
+
+module TinyTds
+  class Client
+    #raw_execute - freetds does not support dynamic sql so
+    #we're going to translate dynamic sql statements into a plain
+    #sql statement
+    def raw_execute(text , *var_list)
+
+      dynamic_sql = text.clone
+      flat_sql = var_list.empty? ? dynamic_sql : process_questioned_sql(dynamic_sql , *var_list)
+      execute(flat_sql)
+    end
+
+    :private
+
+    def process_questioned_sql(sql , *vars)
+      sql_type = classify_sql(sql)
+
+      result = substitute_quoted_questions(sql)
+      _sql = result[:sql]
+
+      vars.each do |var|
+        _sql.sub!("?" , convert_type_to_s(var , sql_type))
+      end
+
+      #place back strings in the sql that contained "?"
+      result[:container].each do |k,v|
+        _sql.gsub!(k , v)
+      end
+
+      _sql
+    end
+
+    def substitute_quoted_questions(sql)
+      container = {}
+      #collect strings in the sql that contains a question_mark
+      qstrings = sql.scan(/"[^"]*"/).select{|s| s.include?("?")}
+
+      #temporarily replace quoted values
+      counter = 0
+      qstrings.each do |qstring|
+        key = "(((####{counter}###)))"
+        sql.gsub!(qstring, key)
+
+        container[key] = qstring
+        counter += 1
+      end
+
+      {:sql => sql, :container => container}
+    end
+
+    #convert_type_to_s - convert Ruby collection objects fed into the query into
+    #sql strings
+    def convert_type_to_s(arg , sql_type)
+
+      case sql_type
+        when :between
+          case arg
+            when Range , Array
+              "#{sql_stringify_value(arg.first)} AND #{sql_stringify_value(arg.last)}"
+            else
+              raise "Type not found..."
+          end
+        when :in
+          case arg
+            when Range , Array
+              " (#{arg.collect{|e| "#{sql_stringify_value(e)}"}.join(" , ")}) "
+            else
+              raise "Type not found..."
+          end
+        else
+          case arg
+            when Range , Array
+              arg.collect{|e| "#{sql_stringify_value(e)}"}.join(" , ")
+            else
+              sql_stringify_value(arg)
+          end
+      end
+    end
+
+    #convert_type_to_s - convert Ruby object fed into the query into
+    #sql string
+    def sql_stringify_value(value)
+      case
+        when value.is_a?(String)
+          "N'#{value}'"
+        when value.is_a?(Numeric)
+          value.to_s
+        when value.is_a?(NilClass)
+          "NULL"
+        when value.is_a?(DateTime)
+          value.strftime("'%m/%d/%Y %I:%M:%S %p'")
+        when value.is_a?(Time)
+          DateTime.new(value.year, value.month, value.day,
+                     value.hour, value.min, value.sec,
+                     Rational(value.gmt_offset / 3600, 24)).strftime("'%m/%d/%Y %I:%M:%S %p'")
+        else
+          "N'#{value.to_s}'"
+      end
+    end
+
+    #classify_sql - determine if this sql has a between or in
+    def classify_sql(sql)
+      case
+        when sql[/between.+\?/i] #var should be an array
+          :between
+        when sql[/\sin\s+\?/i]
+          :in
+      end
+    end
+
+  end
+
+  class Error
+    alias :to_str :to_s
+
+    def errstr
+      @errstr ||= []
+      @errstr
+    end
+  end
+end

data/lib/do_sqlserver_tinytds/transaction.rb

@@ -0,0 +1,36 @@
+module DataObjects
+
+  module SqlServer
+
+    class Transaction < DataObjects::Transaction
+
+      def begin
+        connection.instance_variable_get("@connection").autocommit = false
+      end
+
+      def commit
+        connection.instance_variable_get("@connection").commit
+      ensure
+        connection.instance_variable_get("@connection").autocommit = true
+      end
+
+      def rollback
+        connection.instance_variable_get("@connection").rollback
+      ensure
+        connection.instance_variable_get("@connection").autocommit = true
+      end
+
+      def rollback_prepared
+        # TODO: what should be done differently?
+        rollback
+      end
+
+      def prepare
+        # TODO: what should be done here?
+      end
+
+    end
+
+  end
+
+end

data/lib/do_sqlserver_tinytds/version.rb

@@ -0,0 +1,5 @@
+module DataObjects
+  module SqlServer
+    VERSION = '0.10.17.alpha'
+  end
+end

data/spec/command_spec.rb

@@ -0,0 +1,9 @@
+# encoding: utf-8
+
+require File.expand_path(File.join(File.dirname(__FILE__), 'spec_helper'))
+require 'data_objects/spec/shared/command_spec'
+
+describe DataObjects::SqlServer::Command do
+  it_should_behave_like 'a Command'
+  it_should_behave_like 'a Command with async'
+end

data/spec/connection_spec.rb

@@ -0,0 +1,21 @@
+# encoding: utf-8
+
+require File.expand_path(File.join(File.dirname(__FILE__), 'spec_helper'))
+require 'data_objects/spec/shared/connection_spec'
+
+describe DataObjects::SqlServer::Connection do
+
+  before :all do
+    @driver = CONFIG.scheme
+    @user   = CONFIG.user
+    @password = CONFIG.pass
+    @host   = CONFIG.host
+    @port   = CONFIG.port
+    @database = CONFIG.database
+  end
+
+  it_should_behave_like 'a Connection'
+  it_should_behave_like 'a Connection with authentication support'
+  it_should_behave_like 'a Connection with JDBC URL support' if JRUBY
+  it_should_behave_like 'a Connection via JDNI' if JRUBY
+end

data/spec/encoding_spec.rb

@@ -0,0 +1,8 @@
+# encoding: utf-8
+
+require File.expand_path(File.join(File.dirname(__FILE__), 'spec_helper'))
+require 'data_objects/spec/shared/encoding_spec'
+
+describe DataObjects::SqlServer::Connection do
+  it_should_behave_like 'a driver supporting different encodings'
+end

data/spec/reader_spec.rb

@@ -0,0 +1,8 @@
+# encoding: utf-8
+
+require File.expand_path(File.join(File.dirname(__FILE__), 'spec_helper'))
+require 'data_objects/spec/shared/reader_spec'
+
+describe DataObjects::SqlServer::Reader do
+  it_should_behave_like 'a Reader'
+end

data/spec/result_spec.rb

@@ -0,0 +1,16 @@
+# encoding: utf-8
+
+require File.expand_path(File.join(File.dirname(__FILE__), 'spec_helper'))
+require 'data_objects/spec/shared/result_spec'
+
+# splitting the descibe into two separate declaration avoids
+# concurrent execution of the "it_should_behave_like ....."
+# needed by some databases (sqlite3)
+
+describe DataObjects::SqlServer::Result do
+  it_should_behave_like 'a Result'
+end
+
+describe DataObjects::SqlServer::Result do
+  it_should_behave_like 'a Result which returns inserted key with sequences'
+end

data/spec/spec_helper.rb

@@ -0,0 +1,155 @@
+$TESTING=true
+JRUBY = nil
+
+require 'rubygems'
+require 'rspec'
+require 'date'
+require 'ostruct'
+require 'fileutils'
+
+driver_lib = File.expand_path('../../lib', __FILE__)
+$LOAD_PATH.unshift(driver_lib) unless $LOAD_PATH.include?(driver_lib)
+
+
+repo_root = File.expand_path('../../..', __FILE__)
+
+
+require 'data_objects'
+require 'data_objects/spec/setup'
+require 'data_objects/spec/lib/pending_helpers'
+require 'do_sqlserver_tinytds'
+require 'ruby-debug'
+
+DataObjects::SqlServer.logger = DataObjects::Logger.new(STDOUT, :off)
+at_exit { DataObjects.logger.flush }
+
+
+CONFIG = OpenStruct.new
+CONFIG.scheme   = 'sqlserver'
+CONFIG.user           = ENV['DO_SQLSERVER_USER'] || 'sa'
+CONFIG.pass           = ENV['DO_SQLSERVER_PASS'] || 'p@$$word'
+CONFIG.host           = ENV['DO_SQLSERVER_HOST'] || 'MSSQLLOCAL'
+CONFIG.port     = ENV['DO_SQLSERVER_PORT'] || '1433'
+CONFIG.instance = ENV['DO_SQLSERVER_INSTANCE'] || 'SQLEXPRESS'
+CONFIG.database = ENV['DO_SQLSERVER_DATABASE'] || "/datamapper_test"
+
+CONFIG.driver       = 'sqlserver'
+CONFIG.uri          = ENV["DO_SQLSERVER_SPEC_URI"] ||"#{CONFIG.scheme}://#{CONFIG.user}:#{CONFIG.pass}@#{CONFIG.host}:#{CONFIG.port}#{CONFIG.database}"
+CONFIG.sleep        = "WAITFOR DELAY '00:00:00:10'"
+
+module DataObjectsSpecHelpers
+
+  def setup_test_environment
+    conn = DataObjects::Connection.new(CONFIG.uri)
+
+    conn.create_command(<<-EOF).execute_non_query
+      IF OBJECT_ID('invoices') IS NOT NULL DROP TABLE invoices
+    EOF
+
+    conn.create_command(<<-EOF).execute_non_query
+      IF OBJECT_ID('users') IS NOT NULL DROP TABLE users
+    EOF
+
+    conn.create_command(<<-EOF).execute_non_query
+      IF OBJECT_ID('widgets') IS NOT NULL DROP TABLE widgets
+    EOF
+
+    conn.create_command(<<-EOF).execute_non_query
+      CREATE TABLE [users] (
+        [id] int NOT NULL IDENTITY,
+        [name] nvarchar(200) default 'Billy' NULL,
+        [fired_at] timestamp,
+        PRIMARY KEY ([id])
+      );
+    EOF
+
+    conn.create_command(<<-EOF).execute_non_query
+      CREATE TABLE [invoices] (
+        [id] int NOT NULL IDENTITY,
+        [invoice_number] varchar(50) NOT NULL,
+        PRIMARY KEY ([id])
+      );
+    EOF
+
+    conn.create_command(<<-EOF).execute_non_query
+      CREATE TABLE [widgets] (
+        [id] int NOT NULL IDENTITY,
+        [code] char(8) default 'A14' NULL,
+        [name] varchar(200) default 'Super Widget' NULL,
+        [shelf_location] nvarchar(max) NULL,
+        [description] nvarchar(max) NULL,
+        [image_data] image NULL,
+        [ad_description] varchar(8000) NULL,
+        [ad_image] image NULL,
+        [whitepaper_text] nvarchar(max) NULL,
+        [cad_drawing] image NULL,
+        [flags] bit default 0 NULL,
+        [number_in_stock] smallint default 500,
+        [number_sold] int default 0,
+        [super_number] bigint default 9223372036854775807,
+        [weight] float default 1.23,
+        [cost1] real default 10.23 NULL,
+        [cost2] decimal(8,2) default 50.23 NULL,
+        [release_date] smalldatetime default '2008-02-14' NULL,   -- date type is SQL Server 2008 only
+        [release_datetime] datetime default '2008-02-14 00:31:12' NULL,
+        [release_timestamp] smalldatetime /* default '2008-02-14 00:31:31' */ NULL,
+        -- [status] enum('active','out of stock') NOT NULL default 'active',
+        PRIMARY KEY ([id])
+      );
+    EOF
+
+    1.upto(16) do |n|
+      conn.create_command(<<-EOF).execute_non_query #(::Extlib::ByteArray.new("CAD \001 \000 DRAWING"))
+        insert into widgets(code, name, shelf_location, description, image_data, ad_description, ad_image, whitepaper_text, cad_drawing, super_number, weight)
+        VALUES ('W#{n.to_s.rjust(7,"0")}', 'Widget #{n}', 'A14', 'This is a description', 'IMAGE DATA', 'Buy this product now!', 'AD IMAGE DATA', 'String', 0x2145475754578785 , 1234, 13.4);
+      EOF
+    end
+
+    conn.create_command(<<-EOF).execute_non_query
+      update widgets set flags = 1 where id = 2
+    EOF
+
+    conn.create_command(<<-EOF).execute_non_query
+      update widgets set ad_description = NULL where id = 3
+    EOF
+
+    conn.create_command(<<-EOF).execute_non_query
+      update widgets set flags = NULL where id = 4
+    EOF
+
+    conn.create_command(<<-EOF).execute_non_query
+      update widgets set cost1 = NULL where id = 5
+    EOF
+
+    conn.create_command(<<-EOF).execute_non_query
+      update widgets set cost2 = NULL where id = 6
+    EOF
+
+    conn.create_command(<<-EOF).execute_non_query
+      update widgets set release_date = NULL where id = 7
+    EOF
+
+    conn.create_command(<<-EOF).execute_non_query
+      update widgets set release_datetime = NULL where id = 8
+    EOF
+
+    # (cannot update a Timestamp column w/MSSQL)
+    # so we use a smalldatetime
+    conn.create_command(<<-EOF).execute_non_query
+      update widgets set release_timestamp = NULL where id = 9
+    EOF
+
+    conn.create_command(<<-EOF).execute_non_query
+      update widgets set release_datetime = '2008-07-14 00:31:12' where id = 10
+    EOF
+
+    conn.close
+
+  end
+
+end
+
+RSpec.configure do |config|
+  config.include(DataObjectsSpecHelpers)
+  config.include(DataObjects::Spec::PendingHelpers)
+end