tiny_tds 2.1.6-x64-mingw-ucrt

data/test/benchmark/query.rb

@@ -0,0 +1,77 @@
+$:.unshift File.expand_path('../../../lib',__FILE__)
+require 'rubygems'
+require 'bench_press'
+require 'tiny_tds'
+require 'odbc'
+require 'odbc_utf8'
+
+extend BenchPress
+
+author 'Ken Collins'
+summary 'Query everything.'
+
+reps 1_000
+
+@odbc = ODBC.connect ENV['TINYTDS_UNIT_DATASERVER'], 'tinytds', ''
+@odbc.use_time = true
+
+@odbc_utf8 = ODBC_UTF8.connect ENV['TINYTDS_UNIT_DATASERVER'], 'tinytds', ''
+@odbc_utf8.use_time = true
+
+@tinytds = TinyTds::Client.new(
+  :dataserver    => ENV['TINYTDS_UNIT_DATASERVER'],
+  :username      => 'tinytds',
+  :password      => '',
+  :database      => 'tinytdstest',
+  :appname       => 'TinyTds Dev',
+  :login_timeout => 5,
+  :timeout       => 5 )
+
+@query_all = "SELECT * FROM [datatypes]"
+
+
+measure "ODBC (ascii-8bit)" do
+  h = @odbc.run(@query_all)
+  h.fetch_all
+  h.drop
+end
+
+# measure "ODBC (utf8)" do
+#   h = @odbc_utf8.run(@query_all)
+#   h.fetch_all
+#   h.drop
+# end
+
+measure "TinyTDS (row caching)" do
+  @tinytds.execute(@query_all).each
+end
+
+measure "TinyTDS (no caching)" do
+  @tinytds.execute(@query_all).each(:cache_rows => false)
+end
+
+
+
+=begin
+
+Author: Ken Collins  
+Date: January 22, 2011  
+Summary: Query everything.  
+
+System Information
+------------------
+    Operating System:    Mac OS X 10.6.6 (10J567)
+    CPU:                 Intel Core 2 Duo 1.6 GHz
+    Processor Count:     2
+    Memory:              4 GB
+    ruby 1.8.7 (2010-04-19 patchlevel 253) [i686-darwin10.4.3], MBARI 0x6770, Ruby Enterprise Edition 2010.02
+
+"TinyTDS (row caching)" is up to 79% faster over 1,000 repetitions
+------------------------------------------------------------------
+
+    TinyTDS (row caching)    4.90862512588501 secs    Fastest
+    TinyTDS (no caching)     4.91626906394958 secs    0% Slower
+    ODBC (ascii-8bit)        23.959536075592  secs    79% Slower
+
+=end
+

data/test/benchmark/query_odbc.rb

@@ -0,0 +1,106 @@
+require 'rubygems'
+require 'bench_press'
+begin gem 'odbc', '0.99992' ; rescue Gem::LoadError ; end
+require 'odbc'
+
+extend BenchPress
+
+author 'Ken Collins'
+summary 'Benchmarking ODBC Querys'
+
+reps 1_000
+
+@client = ODBC.connect ENV['TINYTDS_UNIT_DATASERVER'], 'tinytds', ''
+@client.use_time = true
+
+@query_nothing  = "SELECT NULL AS [null]"
+@query_ints     = "SELECT [int], [bigint], [smallint], [tinyint] FROM [datatypes]"
+@query_binaries = "SELECT [binary_50], [image], [varbinary_50] FROM [datatypes]"
+@query_bits     = "SELECT [bit] FROM [datatypes]"
+@query_chars    = "SELECT [char_10], [nchar_10], [ntext], [nvarchar_50], [text], [varchar_50] FROM [datatypes]"
+@query_dates    = "SELECT [datetime], [smalldatetime] FROM [datatypes]"
+@query_decimals = "SELECT [decimal_9_2], [decimal_16_4], [numeric_18_0], [numeric_36_2] FROM [datatypes]"
+@query_floats   = "SELECT [float], [real] FROM [datatypes]"
+@query_moneys   = "SELECT [money], [smallmoney] FROM [datatypes]"
+@query_guids    = "SELECT [uniqueidentifier] FROM [datatypes]"
+@query_all      = "SELECT * FROM [datatypes]"
+
+def select_all(query)
+  h = @client.run(query)
+  h.fetch_all
+  h.drop
+end
+
+
+measure "Nothing" do
+  select_all @query_nothing
+end
+
+measure "Integers" do
+  select_all @query_ints
+end
+
+measure "Binaries" do
+  select_all @query_binaries
+end
+
+measure "Bits" do
+  select_all @query_bits
+end
+
+measure "Chars" do
+  select_all @query_chars
+end
+
+measure "Dates" do
+  select_all @query_dates
+end
+
+measure "Decimals" do
+  select_all @query_decimals
+end
+
+measure "Floats" do
+  select_all @query_floats
+end
+
+measure "Moneys" do
+  select_all @query_moneys
+end
+
+measure "Guids" do
+  select_all @query_guids
+end
+
+measure "All" do
+  select_all @query_all
+end
+
+
+=begin
+
+System Information
+------------------
+    Operating System:    Mac OS X 10.6.4 (10F569)
+    CPU:                 Intel Core 2 Duo 2.4 GHz
+    Processor Count:     2
+    Memory:              4 GB
+    ruby 1.8.7 (2010-08-16 patchlevel 302) [i686-darwin10.4.0]
+
+"Nothing" is up to 98% faster over 1,000 repetitions
+----------------------------------------------------
+
+    Nothing     0.297961950302124 secs    Fastest
+    Bits        0.377611875534058 secs    21% Slower
+    Guids       0.381000995635986 secs    21% Slower
+    Moneys      0.405518054962158 secs    26% Slower
+    Floats      0.409428119659424 secs    27% Slower
+    Integers    0.448167085647583 secs    33% Slower
+    Decimals    0.471596956253052 secs    36% Slower
+    Dates       0.52501106262207  secs    43% Slower
+    Binaries    3.66349482536316  secs    91% Slower
+    Chars       6.82928085327148  secs    95% Slower
+    All         28.4982612133026  secs    98% Slower
+
+=end
+

data/test/benchmark/query_tinytds.rb

@@ -0,0 +1,126 @@
+require 'rubygems'
+require 'bench_press'
+$:.unshift File.expand_path('../../../lib',__FILE__)
+require 'tiny_tds'
+
+extend BenchPress
+
+author 'Ken Collins'
+summary 'Benchmark TinyTds Querys'
+
+reps 1_000
+
+@client = TinyTds::Client.new({ 
+  :dataserver    => ENV['TINYTDS_UNIT_DATASERVER'],
+  :username      => 'tinytds',
+  :password      => '',
+  :database      => 'tinytdstest',
+  :appname       => 'TinyTds Dev',
+  :login_timeout => 5,
+  :timeout       => 5
+})
+
+@query_nothing  = "SELECT NULL AS [null]"
+@query_ints     = "SELECT [int], [bigint], [smallint], [tinyint] FROM [datatypes]"
+@query_binaries = "SELECT [binary_50], [image], [varbinary_50] FROM [datatypes]"
+@query_bits     = "SELECT [bit] FROM [datatypes]"
+@query_chars    = "SELECT [char_10], [nchar_10], [ntext], [nvarchar_50], [text], [varchar_50] FROM [datatypes]"
+@query_dates    = "SELECT [datetime], [smalldatetime] FROM [datatypes]"
+@query_decimals = "SELECT [decimal_9_2], [decimal_16_4], [numeric_18_0], [numeric_36_2] FROM [datatypes]"
+@query_floats   = "SELECT [float], [real] FROM [datatypes]"
+@query_moneys   = "SELECT [money], [smallmoney] FROM [datatypes]"
+@query_guids    = "SELECT [uniqueidentifier] FROM [datatypes]"
+@query_all      = "SELECT * FROM [datatypes]"
+
+def select_all(query)
+  @client.execute(query).each
+end
+
+
+measure "Nothing" do
+  select_all @query_nothing
+end
+
+measure "Integers" do
+  select_all @query_ints
+end
+
+measure "Binaries" do
+  select_all @query_binaries
+end
+
+measure "Bits" do
+  select_all @query_bits
+end
+
+measure "Chars" do
+  select_all @query_chars
+end
+
+measure "Dates" do
+  select_all @query_dates
+end
+
+measure "Decimals" do
+  select_all @query_decimals
+end
+
+measure "Floats" do
+  select_all @query_floats
+end
+
+measure "Moneys" do
+  select_all @query_moneys
+end
+
+measure "Guids" do
+  select_all @query_guids
+end
+
+measure "All" do
+  select_all @query_all
+end
+
+
+=begin
+
+Query Tinytds
+=============
+Author: Ken Collins  
+Date: September 11, 2011  
+Summary: Benchmark TinyTds Querys  
+
+System Information
+------------------
+    Operating System:    Mac OS X 10.7.1 (11B26)
+    CPU:                 Quad-Core Intel Xeon 2.66 GHz
+    Processor Count:     4
+    Memory:              24 GB
+    ruby 1.8.7 (2011-02-18 patchlevel 334) [i686-darwin11.1.0], MBARI 0x6770, Ruby Enterprise Edition 2011.03
+
+----------------------------------------------------
+    (before 64bit times)                    (after 64bit times)
+    Nothing     0.287657022476196 secs      Nothing     0.289273977279663 secs
+    Bits        0.406533002853394 secs      Bits        0.424988031387329 secs
+    Guids       0.419962882995605 secs      Guids       0.427381992340088 secs
+    Floats      0.452103137969971 secs      Floats      0.455377101898193 secs
+    Moneys      0.481696844100952 secs      Moneys      0.485175132751465 secs
+    Integers    0.496185064315796 secs      Integers    0.525003910064697 secs
+    Binaries    0.538873195648193 secs      Decimals    0.541536808013916 secs
+    Decimals    0.540570974349976 secs      Binaries    0.542865991592407 secs
+    Dates       0.761389970779419 secs      Dates       1.51440119743347  secs
+    Chars       0.793163061141968 secs      Chars       0.666505098342896 secs
+    All         4.4630811214447   secs      All         5.17242312431335  secs
+
+=end
+
+
+
+
+
+
+
+
+
+
+

data/test/bin/install-freetds.sh

@@ -0,0 +1,20 @@
+#!/usr/bin/env bash
+
+set -x
+set -e
+
+if [ -z "$FREETDS_VERSION" ]; then
+  FREETDS_VERSION=$(ruby -r "./ext/tiny_tds/extconsts.rb" -e "puts FREETDS_VERSION")
+fi
+
+wget http://www.freetds.org/files/stable/freetds-$FREETDS_VERSION.tar.gz
+tar -xzf freetds-$FREETDS_VERSION.tar.gz
+cd freetds-$FREETDS_VERSION
+./configure --prefix=/opt/local \
+            --with-openssl=/opt/local \
+            --with-tdsver=7.3
+make
+make install
+cd ..
+rm -rf freetds-$FREETDS_VERSION
+rm freetds-$FREETDS_VERSION.tar.gz

data/test/bin/install-mssql.ps1

@@ -0,0 +1,31 @@
+$ProgressPreference = 'SilentlyContinue'
+
+if (-not(Test-path "C:\Downloads"))
+{
+    mkdir "C:\Downloads"
+}
+
+$sqlInstallationFile = "C:\Downloads\sqlexpress.exe"
+if (-not(Test-path $sqlInstallationFile -PathType leaf))
+{
+    Write-Host "Downloading SQL Express ..."
+    Invoke-WebRequest -Uri "https://go.microsoft.com/fwlink/?linkid=829176" -OutFile "C:\Downloads\sqlexpress.exe"
+}
+
+Write-Host "Installing SQL Express ..."
+Start-Process -Wait -FilePath "C:\Downloads\sqlexpress.exe" -ArgumentList /qs, /x:"C:\Downloads\setup"
+C:\Downloads\setup\setup.exe /q /ACTION=Install /INSTANCENAME=SQLEXPRESS /FEATURES=SQLEngine /UPDATEENABLED=0 /SQLSVCACCOUNT='NT AUTHORITY\System' /SQLSYSADMINACCOUNTS='BUILTIN\ADMINISTRATORS' /TCPENABLED=1 /NPENABLED=0 /IACCEPTSQLSERVERLICENSETERMS
+
+Write-Host "Configuring SQL Express ..."
+stop-service MSSQL`$SQLEXPRESS
+set-itemproperty -path 'HKLM:\software\microsoft\microsoft sql server\mssql14.SQLEXPRESS\mssqlserver\supersocketnetlib\tcp\ipall' -name tcpdynamicports -value ''
+set-itemproperty -path 'HKLM:\software\microsoft\microsoft sql server\mssql14.SQLEXPRESS\mssqlserver\supersocketnetlib\tcp\ipall' -name tcpport -value 1433
+set-itemproperty -path 'HKLM:\software\microsoft\microsoft sql server\mssql14.SQLEXPRESS\mssqlserver\' -name LoginMode -value 2
+
+Write-Host "Starting SQL Express ..."
+start-service MSSQL`$SQLEXPRESS
+
+Write-Host "Configuring MSSQL for TinyTDS ..."
+& sqlcmd -Q "CREATE DATABASE [tinytdstest];"
+& sqlcmd -Q "CREATE LOGIN [tinytds] WITH PASSWORD = '', CHECK_POLICY = OFF, DEFAULT_DATABASE = [tinytdstest];"
+& sqlcmd -Q "USE [tinytdstest]; CREATE USER [tinytds] FOR LOGIN [tinytds]; EXEC sp_addrolemember N'db_owner', N'tinytds';"

data/test/bin/install-mssqltools.sh

@@ -0,0 +1,9 @@
+#!/usr/bin/env bash
+
+set -x
+set -e
+
+curl https://packages.microsoft.com/keys/microsoft.asc | sudo apt-key add -
+curl https://packages.microsoft.com/config/ubuntu/20.04/prod.list | sudo tee /etc/apt/sources.list.d/msprod.list
+sudo apt-get update
+sudo ACCEPT_EULA=Y apt-get -y install mssql-tools unixodbc-dev

data/test/bin/install-openssl.sh

@@ -0,0 +1,18 @@
+#!/usr/bin/env bash
+
+set -x
+set -e
+
+if [ -z "$OPENSSL_VERSION" ]; then
+  OPENSSL_VERSION=$(ruby -r "./ext/tiny_tds/extconsts.rb" -e "puts OPENSSL_VERSION")
+fi
+
+wget https://www.openssl.org/source/openssl-$OPENSSL_VERSION.tar.gz
+tar -xzf openssl-$OPENSSL_VERSION.tar.gz
+cd openssl-$OPENSSL_VERSION
+./config --prefix=/opt/local --openssldir=/opt/local
+make
+make install_sw install_ssldirs
+cd ..
+rm -rf openssl-$OPENSSL_VERSION
+rm openssl-$OPENSSL_VERSION.tar.gz

data/test/bin/setup_tinytds_db.sh

@@ -0,0 +1,7 @@
+#!/usr/bin/env bash
+
+set -x
+set -e
+
+/opt/mssql-tools/bin/sqlcmd -S localhost -U sa -P $SA_PASSWORD -i ./test/sql/db-create.sql
+/opt/mssql-tools/bin/sqlcmd -S localhost -U sa -P $SA_PASSWORD -i ./test/sql/db-login.sql

data/test/bin/setup_volume_permissions.sh

@@ -0,0 +1,10 @@
+#!/usr/bin/env bash
+
+set -x
+
+sudo groupadd -g 3434 circleci_tinytds
+sudo usermod -a -G circleci_tinytds $USER
+sudo useradd circleci_tinytds -u 3434 -g 3434
+sudo usermod -a -G circleci_tinytds circleci_tinytds
+sudo chgrp -R circleci_tinytds .
+sudo chmod -R g+rwx .

data/test/client_test.rb

@@ -0,0 +1,275 @@
+# encoding: utf-8
+require 'test_helper'
+
+class ClientTest < TinyTds::TestCase
+  describe 'with valid credentials' do
+    before do
+      @client = new_connection
+    end
+
+    it 'must not be closed' do
+      assert !@client.closed?
+      assert @client.active?
+    end
+
+    it 'allows client connection to be closed' do
+      assert @client.close
+      assert @client.closed?
+      assert !@client.active?
+      assert @client.dead?
+      action = lambda { @client.execute('SELECT 1 as [one]').each }
+      assert_raise_tinytds_error(action) do |e|
+        assert_match %r{closed connection}i, e.message, 'ignore if non-english test run'
+      end
+    end
+
+    it 'has getters for the tds version information (brittle since conf takes precedence)' do
+      if sybase_ase?
+        assert_equal 7, @client.tds_version
+        assert_equal 'DBTDS_5_0 - 5.0 SQL Server', @client.tds_version_info
+      elsif @client.tds_73?
+        assert_equal 11, @client.tds_version
+        assert_equal 'DBTDS_7_3 - Microsoft SQL Server 2008', @client.tds_version_info
+      else
+        assert_equal 9, @client.tds_version
+        assert_equal 'DBTDS_7_1/DBTDS_8_0 - Microsoft SQL Server 2000', @client.tds_version_info
+      end
+    end
+
+    it 'uses UTF-8 client charset/encoding by default' do
+      assert_equal 'UTF-8', @client.charset
+      assert_equal Encoding.find('UTF-8'), @client.encoding
+    end
+
+    it 'has a #escape method used for quote strings' do
+      assert_equal "''hello''", @client.escape("'hello'")
+    end
+
+    ['CP850', 'CP1252', 'ISO-8859-1'].each do |encoding|
+      it "allows valid iconv character set - #{encoding}" do
+        begin
+          client = new_connection(:encoding => encoding)
+          assert_equal encoding, client.charset
+          assert_equal Encoding.find(encoding), client.encoding
+        ensure
+          client.close if client
+        end
+      end
+    end
+
+    it 'must be able to use :host/:port connection' do
+      host = ENV['TINYTDS_UNIT_HOST_TEST'] || ENV['TINYTDS_UNIT_HOST'] || 'localhost'
+      port = ENV['TINYTDS_UNIT_PORT_TEST'] || ENV['TINYTDS_UNIT_PORT'] || 1433
+      begin
+        client = new_connection dataserver: nil, host: host, port: port
+      ensure
+        client.close if client
+      end
+    end unless sqlserver_azure?
+  end
+
+  describe 'With in-valid options' do
+    before(:all) do
+      init_toxiproxy
+    end
+
+    it 'raises an argument error when no :host given and :dataserver is blank' do
+      assert_raises(ArgumentError) { new_connection :dataserver => nil, :host => nil }
+    end
+
+    it 'raises an argument error when no :username is supplied' do
+      assert_raises(ArgumentError) { TinyTds::Client.new :username => nil }
+    end
+
+    it 'raises TinyTds exception with undefined :dataserver' do
+      options = connection_options :login_timeout => 1, :dataserver => 'DOESNOTEXIST'
+      action = lambda { new_connection(options) }
+      assert_raise_tinytds_error(action) do |e|
+        # Not sure why tese are different.
+        if ruby_darwin?
+          assert_equal 20009, e.db_error_number
+          assert_equal 9, e.severity
+          assert_match %r{is unavailable or does not exist}i, e.message, 'ignore if non-english test run'
+        else
+          assert_equal 20012, e.db_error_number
+          assert_equal 2, e.severity
+          assert_match %r{server name not found in configuration files}i, e.message, 'ignore if non-english test run'
+        end
+      end
+      assert_new_connections_work
+    end
+
+    it 'raises TinyTds exception with long query past :timeout option' do
+      client = new_connection :timeout => 1
+      action = lambda { client.execute("WaitFor Delay '00:00:02'").do }
+      assert_raise_tinytds_error(action) do |e|
+        assert_equal 20003, e.db_error_number
+        assert_equal 6, e.severity
+        assert_match %r{timed out}i, e.message, 'ignore if non-english test run'
+      end
+      assert_client_works(client)
+      close_client(client)
+      assert_new_connections_work
+    end
+
+    it 'must not timeout per sql batch when not under transaction' do
+      client = new_connection :timeout => 2
+      client.execute("WaitFor Delay '00:00:01'").do
+      client.execute("WaitFor Delay '00:00:01'").do
+      client.execute("WaitFor Delay '00:00:01'").do
+      close_client(client)
+    end
+
+    it 'must not timeout per sql batch when under transaction' do
+      client = new_connection :timeout => 2
+      begin
+        client.execute("BEGIN TRANSACTION").do
+        client.execute("WaitFor Delay '00:00:01'").do
+        client.execute("WaitFor Delay '00:00:01'").do
+        client.execute("WaitFor Delay '00:00:01'").do
+      ensure
+        client.execute("COMMIT TRANSACTION").do
+        close_client(client)
+      end
+    end
+
+    it 'raises TinyTds exception with tcp socket network failure' do
+      begin
+        client = new_connection timeout: 2, port: 1234, host: ENV['TOXIPROXY_HOST']
+        assert_client_works(client)
+        action = lambda { client.execute("waitfor delay '00:00:05'").do }
+
+        # Use toxiproxy to close the TCP socket after 1 second.
+        # We want TinyTds to execute the statement, hit the timeout configured above, and then not be able to use the network to cancel
+        # the network connection needs to close after the sql batch is sent and before the timeout above is hit
+        Toxiproxy[:sqlserver_test].toxic(:slow_close, delay: 1000).apply do
+          assert_raise_tinytds_error(action) do |e|
+            assert_equal 20003, e.db_error_number
+            assert_equal 6, e.severity
+            assert_match %r{timed out}i, e.message, 'ignore if non-english test run'
+          end
+        end
+      ensure
+        assert_new_connections_work
+      end
+    end
+
+    it 'raises TinyTds exception with dead connection network failure' do
+      skip if ruby_windows?
+
+      begin
+        client = new_connection timeout: 2, port: 1234, host: ENV['TOXIPROXY_HOST']
+        assert_client_works(client)
+        action = lambda { client.execute("waitfor delay '00:00:05'").do }
+
+        # Use toxiproxy to close the network connection after 1 second.
+        # We want TinyTds to execute the statement, hit the timeout configured above, and then not be able to use the network to cancel
+        # the network connection needs to close after the sql batch is sent and before the timeout above is hit
+        Toxiproxy[:sqlserver_test].toxic(:timeout, timeout: 1000).apply do
+          assert_raise_tinytds_error(action) do |e|
+            assert_equal 20047, e.db_error_number
+            assert_includes [1,9], e.severity
+            assert_match %r{dead or not enabled}i, e.message, 'ignore if non-english test run'
+          end
+        end
+      ensure
+        assert_new_connections_work
+      end
+    end
+
+    it 'raises TinyTds exception with login timeout' do
+      begin
+        action = lambda do
+          Toxiproxy[:sqlserver_test].toxic(:timeout, timeout: 0).apply do
+            new_connection login_timeout: 1, port: 1234, host: ENV['TOXIPROXY_HOST']
+          end
+        end
+        assert_raise_tinytds_error(action) do |e|
+          assert_equal 20003, e.db_error_number
+          assert_equal 6, e.severity
+          assert_match %r{timed out}i, e.message, 'ignore if non-english test run'
+        end
+      ensure
+        assert_new_connections_work
+      end
+    end
+
+    it 'raises TinyTds exception with wrong :username' do
+      skip if ENV['CI'] && sqlserver_azure? # Some issue with db_error_number.
+      options = connection_options :username => 'willnotwork'
+      action = lambda { new_connection(options) }
+      assert_raise_tinytds_error(action) do |e|
+        assert_equal sybase_ase? ? 4002 : 18456, e.db_error_number
+        assert_equal 14, e.severity
+        assert_match %r{login failed}i, e.message, 'ignore if non-english test run'
+      end
+      assert_new_connections_work
+    end
+
+  end
+
+  describe '#parse_username' do
+    let(:client) { @client = new_connection }
+
+    it 'returns username if azure is not true' do
+      _(
+        client.send(:parse_username, username: 'user@abc123.database.windows.net')
+      ).must_equal 'user@abc123.database.windows.net'
+    end
+
+    it 'returns short username if azure is true' do
+      _(
+        client.send(
+          :parse_username,
+          username: 'user@abc123.database.windows.net',
+          host:     'abc123.database.windows.net',
+          azure:    true
+        )
+      ).must_equal 'user@abc123'
+    end
+
+    it 'returns full username if azure is false' do
+      _(
+        client.send(
+          :parse_username,
+          username: 'user@abc123.database.windows.net',
+          host:     'abc123.database.windows.net',
+          azure:    false
+        )
+      ).must_equal 'user@abc123.database.windows.net'
+    end
+
+    it 'returns short username if passed and azure is true' do
+      _(
+        client.send(
+          :parse_username,
+          username: 'user@abc123',
+          host:     'abc123.database.windows.net',
+          azure:    true
+        )
+      ).must_equal 'user@abc123'
+    end
+
+    it 'returns username with servername if passed and azure is true' do
+      _(
+        client.send(
+          :parse_username,
+          username: 'user',
+          host:     'abc123.database.windows.net',
+          azure:    true
+        )
+      ).must_equal 'user@abc123'
+    end
+
+    it 'returns username with servername if passed and azure is false' do
+      _(
+        client.send(
+          :parse_username,
+          username: 'user',
+          host:     'abc123.database.windows.net',
+          azure:    false
+        )
+      ).must_equal 'user'
+    end
+  end
+end