require 'spec_helper'
require 'albacore/sqlcmd'

describe SQLCmd, "when running a script the easy way" do
  before :all do
    @cmd = SQLCmd.new
    @cmd.log_level = :verbose
    @cmd.extend(SystemPatch)
    @cmd.disable_system = true
    
    @cmd.scripts "somescript.sql"
    
    @cmd.execute
  end

  it "should use localhost as server default" do
	@cmd.system_command.should include("-S \".\"")
  end
  
  it "should have no database designation (because it could be embedded in the sql file)" do
	@cmd.system_command.should_not include("-d")
  end
  
  it "should use integrated security instead of username/password" do
	@cmd.system_command.should include("-E")
  end

  it "should not include username" do
	@cmd.system_command.should_not include("-U")
  end

  it "should not include password" do
	@cmd.system_command.should_not include("-P")
  end

  it "should find the location of the sqlcmd exe for the user" do
    @cmd.system_command.downcase.should include("sqlcmd.exe")
  end
  
  it "should specify the script file" do
    @cmd.system_command.should include("-i \"somescript.sql\"")
  end
  
  it "should not contain the -b option" do
    @cmd.system_command.should_not include("-b")
  end
end

describe SQLCmd, "when turning off trusted_connection" do
  before :all do
    @cmd = SQLCmd.new
    @cmd.log_level = :verbose
    @cmd.extend(SystemPatch)
    @cmd.disable_system = true
    
    @cmd.scripts "somescript.sql"
    @cmd.trusted_connection = false
    
    @cmd.execute
  end

  it "should not specify the -E option" do
    @cmd.system_command.should_not include("-E")
  end
end

describe SQLCmd, "when using a trusted connection with a username and password" do
  before :all do
    @cmd = SQLCmd.new
    @cmd.log_level = :verbose
    @cmd.extend(SystemPatch)
    @cmd.disable_system = true
    
    @cmd.scripts "somescript.sql"
    @cmd.trusted_connection = true
    @cmd.username = "user"
    @cmd.password = "password"
    
    @cmd.execute
  end

  it "should override the trusted connection and not specify the -E option" do
    @cmd.system_command.should_not include("-E")
  end

  it "should specify the username" do
    @cmd.system_command.should include("-U \"user\"")
  end
  
  it "should specify the password" do
    @cmd.system_command.should include("-P \"password\"")
  end
end

describe SQLCmd, "when running a script file against a database with authentication information" do
  before :all do
    @cmd = SQLCmd.new
    @cmd.command = "sqlcmd.exe"
    @cmd.log_level = :verbose
    @cmd.extend(SystemPatch)
    @cmd.disable_system = true
    
    @cmd.server="a server"
    @cmd.database="a database"
    @cmd.username="some user"
    @cmd.password="shh! it's a secret!"
    @cmd.scripts "somescript.sql"
    
    @cmd.execute
  end
  
  it "should specify the location of the sqlcmd exe" do
    @cmd.system_command.downcase.should include("sqlcmd.exe")
  end
  
  it "should specify the script file" do
    @cmd.system_command.should include("-i \"somescript.sql\"")
  end
  
  it "should specify the server" do
    @cmd.system_command.should include("-S \"a server\"")
  end
  
  it "should specify the database" do
    @cmd.system_command.should include("-d \"a database\"")
  end

  it "should specify the username" do
    @cmd.system_command.should include("-U \"some user\"")
  end
  
  it "should specify the password" do
    @cmd.system_command.should include("-P \"shh! it's a secret!\"")
  end
  
  it "should not contain the -b option" do
    @cmd.system_command.should_not include("-b")
  end
end

describe SQLCmd, "when running with no command path specified" do
  before :all do
    strio = StringIO.new
    @cmd = SQLCmd.new
    @cmd.log_level = :verbose
    @cmd.log_device = strio
    @cmd.extend(SystemPatch)
    @cmd.extend(FailPatch)
    @cmd.disable_system = true

    @cmd.execute
    @log_data = strio.string

    @all_possible_sqlcmd_paths = []

    ["program files", "program files (x86)"].each do |program_files|
      ["90", "100"].each do |sql_version|
        sqlcmd_path = File.join(ENV['SystemDrive'], program_files,'microsoft sql server', sql_version, 'tools','binn', 'sqlcmd.exe')
        @all_possible_sqlcmd_paths << sqlcmd_path
      end
    end
  end

  it "should find sqlcmd in the standard places or bail" do
    any_sqlcmd_exists = false

    @all_possible_sqlcmd_paths.each do |sqlcmd_path|
      sqlcmd_exists = File.exists?(sqlcmd_path)
      any_sqlcmd_exists = true if sqlcmd_exists
      $task_failed.should be_false if sqlcmd_exists
    end
    $task_failed.should be_true if any_sqlcmd_exists == false
  end

  it "should log a failure message if it cannot find sqlcmd in the standard places" do
    any_sqlcmd_exists = false

    @all_possible_sqlcmd_paths.each do |sqlcmd_path|
      sqlcmd_exists = File.exists?(sqlcmd_path)
      any_sqlcmd_exists = true if File.exists?(sqlcmd_path)
    end
    @log_data.should include('SQLCmd.command cannot be nil.') if any_sqlcmd_exists == false
  end
end

describe SQLCmd, "when execution of sqlcmd fails" do
  before :all do
    strio = StringIO.new
    @cmd = SQLCmd.new
    @cmd.command="sqlcmd.exe"
    @cmd.log_level = :verbose
    @cmd.log_device = strio
    @cmd.extend(SystemPatch)
    @cmd.extend(FailPatch)
    @cmd.disable_system = true
    @cmd.force_system_failure = true
    
    @cmd.execute
    @log_data = strio.string
  end
  
  
  it "should fail" do
    $task_failed.should be_true
  end
  
  it "should log a failure message" do
    @log_data.should include('SQLCmd Failed. See Build Log For Detail.')
  end
end

describe SQLCmd, "when running multiple script files" do
  before :all do
    @cmd = SQLCmd.new
    @cmd.command = "sqlcmd.exe"
    @cmd.log_level = :verbose
    @cmd.extend(SystemPatch)
    @cmd.disable_system = true
    
    scriptnames = Array.new
    scriptnames << "did you get.sql\r\n"
    scriptnames << "that thing.sql"
    scriptnames << "i sent you.sql\r\n"
    @cmd.scripts = scriptnames
    
    @cmd.execute
  end
  
  it "should specify the first script file" do
    @cmd.system_command.should include("-i \"did you get.sql\"")
  end

  it "should specify the second script file" do
    @cmd.system_command.should include("-i \"that thing.sql\"")
  end
  
  it "should specify the third script file" do
    @cmd.system_command.should include("-i \"i sent you.sql\"")
  end
  
  it "should include the -b option" do
    @cmd.system_command.should include("-b")
  end
  
end

describe SQLCmd, "when running multiple script files with batch_abort set to false" do
  before :all do
    @cmd = SQLCmd.new
    @cmd.command = "sqlcmd.exe"
    @cmd.log_level = :verbose
    @cmd.extend(SystemPatch)
    @cmd.disable_system = true
    @cmd.batch_abort = false
    
    scriptnames = Array.new
    scriptnames << "did you get.sql\r\n"
    scriptnames << "that thing.sql"
    scriptnames << "i sent you.sql\r\n"
    @cmd.scripts = scriptnames
    
    @cmd.execute
  end
  
  it "should specify the first script file" do
    @cmd.system_command.should include("-i \"did you get.sql\"")
  end

  it "should specify the second script file" do
    @cmd.system_command.should include("-i \"that thing.sql\"")
  end
  
  it "should specify the third script file" do
    @cmd.system_command.should include("-i \"i sent you.sql\"")
  end
  
  it "should not include the -b option" do
    @cmd.system_command.should_not include("-b")
  end
  
end

describe SQLCmd, "when running with variables specified" do
  before :all do
    @cmd = SQLCmd.new
    @cmd.command = "sqlcmd.exe"
    @cmd.log_level = :verbose
    @cmd.extend(SystemPatch)
    @cmd.disable_system = true
    @cmd.scripts "somescript.sql"
    
    @cmd.variables :myvar => "my value", :another_var => :another_value
    
    @cmd.execute
  end
  
  it "should supply the variables to sqlcmd" do
    @cmd.system_command.should include("-v myvar=my value")
    @cmd.system_command.should include("-v another_var=another_value")
  end
end

describe SQLCmd, "when providing configuration" do
  let :sqlcmd do
    Albacore.configure do |config|
      config.sqlcmd.command = "configured"
    end
    sqlcmd = SQLCmd.new
  end

  it "should use the configured values" do
    sqlcmd.command.should == "configured"
  end
end

describe SQLCmd, "when severity it set" do
  before :all do
    @cmd = SQLCmd.new
    @cmd.extend(SystemPatch)
    @cmd.disable_system = true
    @cmd.scripts "somescript.sql"

    @cmd.severity = 1
    
    @cmd.execute
  end

  it "should have severity option set" do
    @cmd.system_command.should include("-V")
  end
  it "should have severity set to correct value" do
    @cmd.system_command.should include("-V \"1\"")
  end
end