rakedotnet 1.1.51

data/lib/tools.rb

@@ -0,0 +1,3 @@
+def with(value)
+    yield(value)
+end

data/lib/windowspaths.rb

@@ -0,0 +1,28 @@
+require 'registry_accessor'
+
+module BradyW
+	module WindowsPaths
+        private
+          # Fetches the path for tools like bcp.exe and sqlcmd.exe from the registry
+          def sql_tool version
+              regvalue "SOFTWARE\\Microsoft\\Microsoft SQL Server\\#{version}\\Tools\\ClientSetup", 'Path'
+          end
+
+          # Fetches the path for Visual Studio tools like MSTest.exe from the registry
+          def visual_studio version
+              regvalue "SOFTWARE\\Microsoft\\VisualStudio\\#{version}", 'InstallDir'
+          end
+
+          # Fetches the .NET Framework path from the registry
+          def dotnet subpath
+              regvalue "SOFTWARE\\Microsoft\\NET Framework Setup\\NDP\\#{subpath}", "InstallPath"
+          end
+
+          def regvalue(key, value)
+              keyAndVal = "#{key}\\#{value}"
+              log "Retrieving registry key #{keyAndVal}"
+              regacc = BradyW::RegistryAccessor.new
+              regacc.regvalue(key,value)
+          end
+	end
+end

data/spec/base.rb

@@ -0,0 +1,32 @@
+$: << File.expand_path(File.dirname(__FILE__) +"/../lib")
+require "rspec"
+# Needed to mock out our config/props
+require 'config'
+require "singleton"
+
+include FileUtils
+
+RSpec.configure do |config|
+
+  # File dependent tests, so we always baseline ourselves in the test directory
+  config.before(:all) do
+    @current = pwd
+    cd File.expand_path(File.dirname(__FILE__))
+  end
+
+  config.after(:all) do
+    cd @current
+  end
+
+  config.before(:each) do
+    @config = BradyW::BaseConfig.new
+    class MockConfig
+      include Singleton
+      attr_accessor :values
+    end
+
+    # Force only our base class to be returned
+    BradyW::Config.stub!(:instance).and_return(MockConfig.instance)
+    MockConfig.instance.values = @config
+  end
+end

data/spec/basetask_spec.rb

@@ -0,0 +1,66 @@
+require "base"
+require "basetask"
+require "basetaskmocking"
+
+module BradyW
+  class BaseTask < Rake::TaskLib
+    attr_accessor :dependencies
+  end
+end
+
+describe BradyW::BaseTask do
+  before(:each) do
+    Rake::Task.clear
+  end
+
+  it "Task with no dependencies/default name" do
+    task = BradyW::BaseTask.new
+    task.should_receive(:log).with("Running task: task")
+    task.name.should == :task
+    task.dependencies.should == nil
+    task.unless.should == nil
+    task.should_receive(:exectask)
+    Rake::Task[:task].invoke
+  end
+
+  it "Task with no dependencies/custom name" do
+    task = BradyW::BaseTask.new "mytask"
+    task.should_receive(:log).with("Running task: mytask")
+    task.name.should == "mytask"
+    task.dependencies.should == nil
+    task.unless.should == nil
+    task.should_receive(:exectask)
+    Rake::Task[:mytask].invoke
+  end
+
+  it "Task with dependencies/custom name" do
+    task = BradyW::BaseTask.new "mytask" => :dependenttask
+    task.name.should == "mytask"
+    task.dependencies.should == :dependenttask
+    task.unless.should == nil
+
+    dtask = BradyW::BaseTask.new "dependenttask"
+    task.should_receive(:exectask)
+    dtask.should_receive(:exectask)
+
+    task.should_receive(:log).with("Running task: mytask")
+    dtask.should_receive(:log).with("Running task: dependenttask")
+    Rake::Task[:mytask].invoke
+  end
+
+  it "Task with dependencies/custom name + block" do
+    task = BradyW::BaseTask.new "mytask" => [:dependenttask, :test] do |t|
+      t.unless = "yes"
+    end
+    task.name.should == "mytask"
+    task.dependencies.should == [:dependenttask, :test]
+    task.unless.should == "yes"
+
+    dtask = BradyW::BaseTask.new "dependenttask"
+    task.should_not_receive(:exectask)
+    dtask.should_not_receive(:exectask)
+
+    task.should_receive(:log).with("Skipping task: mytask due to unless condition specified in rakefile")
+    Rake::Task[:mytask].invoke
+  end
+end

data/spec/basetaskmocking.rb

@@ -0,0 +1,44 @@
+require 'base'
+require 'rake'
+require 'rake/tasklib'
+
+module BradyW
+	class BaseTask < Rake::TaskLib
+        def exectaskpublic
+          exectask
+        end       
+
+        def shell(*cmd, &block)
+          command = cmd.first
+          puts command
+          # We aren't testing concurrent tasks here, so no thread safety worries
+          if !@sh
+            @sh = []
+          end
+
+          @sh << command
+          
+          # Make it look like it went OK
+          yield true, nil if block_given?
+        end
+
+        def excecutedPop
+          return nil unless @sh
+          @sh.pop()
+        end
+    end
+end
+
+class SimulateProcessFailure
+  def exitstatus
+    return "BW Rake Task Problem"
+  end
+end
+
+def rm_safe directory
+  # Before we delete the files, copy them to a place where we can verify their correctness
+  if File.exist? directory
+    FileUtils::cp_r directory, "data/output"
+  end
+  FileUtils::rm_rf directory
+end

data/spec/bcp_spec.rb

@@ -0,0 +1,137 @@
+require "base"
+require "bcp"
+require "basetaskmocking"
+
+describe BradyW::BCP do
+  before(:each) do
+    def @config.db_name
+      "regulardb"
+    end
+
+    def @config.db_hostname
+      "myhostname"
+    end
+
+    def @config.db_general_authmode
+      :sqlauth
+    end
+
+    def @config.db_general_user
+      "theuser"
+    end
+
+    def @config.db_general_password
+      "thepassword"
+    end
+
+    def @config.project_prefix
+      "PRE"
+    end
+  end
+
+  after(:each) do
+    # Remove our generated test data
+    FileUtils::rm_rf "data/output/bcp"
+  end
+
+  it "Works OK with standard delimiters and SQL Auth" do
+    task = BradyW::BCP.new do |bcp|
+      bcp.files = FileList["data/bcp/01-firsttable.csv",
+                           "data/bcp/02-nexttable.csv"]
+    end
+
+    # Don't want to depend on specific registry setting
+    task.should_receive(:sql_tool).any_number_of_times.with("100").and_return("z:\\")
+
+    task.exectaskpublic
+    task.excecutedPop.should == "\"z:\\bcp.exe\" \"nexttable\" in 02-nexttable.csv -U theuser -P thepassword /Smyhostname -t \"|d3l1m1t3r|\" /c -m 1 -F 2"
+    task.excecutedPop.should == "\"z:\\bcp.exe\" \"firsttable\" in 01-firsttable.csv -U theuser -P thepassword /Smyhostname -t \"|d3l1m1t3r|\" /c -m 1 -F 2"
+
+    expected = IO.readlines("data/bcp/01-firsttable-expectedout.csv")
+    actual = IO.readlines("data/output/bcp/01-firsttable.csv")
+
+    actual.should == expected
+
+    expected = IO.readlines("data/bcp/02-nexttable-expectedout.csv")
+    actual = IO.readlines("data/output/bcp/02-nexttable.csv")
+
+    actual.should == expected
+
+  end
+
+  it "Works OK with custom delimiters, Custom Version, and Windows Auth" do
+    def @config.db_general_authmode
+      :winauth
+    end
+
+    task = BradyW::BCP.new do |bcp|
+      bcp.files = FileList["data/bcp/01-firsttable.csv",
+                           "data/bcp/02-nexttable.csv"]
+      bcp.delimiter = "foobar"
+      bcp.version = "852"
+    end
+
+    # Don't want to depend on specific registry setting
+    task.should_receive(:sql_tool).any_number_of_times.with("852").and_return("z:\\")
+
+    task.exectaskpublic
+    task.excecutedPop.should == "\"z:\\bcp.exe\" \"regulardb.dbo.nexttable\" in 02-nexttable.csv -T -S myhostname -t \"foobar\" /c -m 1 -F 2"
+    task.excecutedPop.should == "\"z:\\bcp.exe\" \"regulardb.dbo.firsttable\" in 01-firsttable.csv -T -S myhostname -t \"foobar\" /c -m 1 -F 2"
+
+    expected = IO.readlines("data/bcp/01-firsttable-expectedout2.csv")
+    actual = IO.readlines("data/output/bcp/01-firsttable.csv")
+
+    actual.should == expected
+
+    expected = IO.readlines("data/bcp/02-nexttable-expectedout2.csv")
+    actual = IO.readlines("data/output/bcp/02-nexttable.csv")
+
+    actual.should == expected
+  end
+
+  it "Handles delimiter interference Properly" do
+    def @config.db_general_authmode
+          :winauth
+        end
+
+    task = BradyW::BCP.new do |bcp|
+      bcp.files = FileList["data/bcp/delimInData.csv"]
+    end
+
+    lambda { task.exectaskpublic }.should raise_exception()
+  end
+
+  it "Handles failure gracefully" do
+    task = BradyW::BCP.new do |bcp|
+      bcp.files = FileList["data/bcp/01-firsttable.csv",
+                           "data/bcp/02-nexttable.csv"]
+    end
+
+    # Don't want to depend on specific registry setting
+    task.should_receive(:sql_tool).any_number_of_times.with("100").and_return("z:\\")
+
+    task.stub!(:shell).and_yield(nil, SimulateProcessFailure.new)
+
+    lambda { task.exectaskpublic }.should raise_exception("Command failed with status (BW Rake Task Problem):")
+
+    # This means our temporary file was correctly cleaned up
+    File.exist?("#{ENV['tmp']}/bcp").should_not == true
+    # Our test code should have done this
+    File.exist?("data/output/bcp").should == true
+  end
+
+  it "Properly disables identity inserts when set" do
+    task = BradyW::BCP.new do |bcp|
+      bcp.identity_inserts = true
+      bcp.files = FileList["data/bcp/01-firsttable.csv",
+                           "data/bcp/02-nexttable.csv"]
+    end
+    # Don't want to depend on specific registry setting
+    task.should_receive(:sql_tool).any_number_of_times.with("100").and_return("z:\\")
+
+    task.exectaskpublic
+    task.excecutedPop.should == "\"z:\\bcp.exe\" \"nexttable\" in 02-nexttable.csv -U theuser -P thepassword /Smyhostname -t \"|d3l1m1t3r|\" /c -E -m 1 -F 2"
+    task.excecutedPop.should == "\"z:\\bcp.exe\" \"firsttable\" in 01-firsttable.csv -U theuser -P thepassword /Smyhostname -t \"|d3l1m1t3r|\" /c -E -m 1 -F 2"
+
+  end
+end

data/spec/config_spec.rb

@@ -0,0 +1,43 @@
+require "base"
+require "config"
+
+describe "Configuration Works Properly" do
+  before(:all) do
+    $: << File.expand_path(File.dirname(__FILE__))+"/config_testdata"
+  end
+
+  after(:each) do
+    File.delete "newuserfile.rb" if File.exist? "newuserfile.rb"
+  end
+
+  def fetchprops(defaultfile,userfile)
+    # get around the private method
+    BradyW::Config.send(:new,defaultfile,userfile).values
+  end
+
+  it "Should work fine with only default properties" do
+    props = fetchprops("onlydefault.rb",
+                               "newuserfile.rb")
+
+    props.setting.should == "yep"
+    props.setting2.should == "nope"
+    props.setting3.should == "yep"
+  end
+
+  it "Should work OK with default + partially filled out user properties" do
+    props = fetchprops("defaultpartialuser_default.rb",
+                                    "defaultpartialuser_user.rb")
+
+    props.setting.should == "overrodethis"
+    props.setting2.should == "nope"
+    props.setting3.should == "yep"
+  end
+
+  it "Should work OK with default + completely filled out user properties" do
+    props = BradyW::Config.send(:new).values
+
+    props.setting.should == "yep2"
+    props.setting2.should == "nope2"
+    props.setting3.should == "yep2"
+  end
+end

data/spec/config_testdata/defaultpartialuser_default.rb

@@ -0,0 +1,17 @@
+require "base_config"
+
+module TestCase_2
+  class ADefaultConfig < BradyW::BaseConfig
+    def setting
+      "yep"
+    end
+
+    def setting2
+      "nope"
+    end
+
+    def setting3
+      "yep"
+    end
+  end
+end

data/spec/config_testdata/defaultpartialuser_user.rb

@@ -0,0 +1,9 @@
+require "defaultpartialuser_default"
+
+module TestCase_2
+  class UserConfig < TestCase_2::ADefaultConfig
+    def setting
+      "overrodethis"
+    end
+  end
+end

data/spec/config_testdata/local_properties.rb

@@ -0,0 +1,13 @@
+module TestCase_3
+  class AUserConfig < ADefaultConfig
+    def setting
+      "yep2"
+    end
+    def setting2
+      "nope2"
+    end
+    def setting3
+      "yep2"
+    end
+  end
+end

data/spec/config_testdata/local_properties_default.rb

@@ -0,0 +1,15 @@
+require "base_config"
+
+module TestCase_3
+  class ADefaultConfig < BradyW::BaseConfig
+    def setting
+      "yep"
+    end
+    def setting2
+      "nope"
+    end
+    def setting3
+      "yep"
+    end
+  end
+end

data/spec/config_testdata/onlydefault.rb

@@ -0,0 +1,17 @@
+require "config"
+
+module TestCase_1
+  class Config < BradyW::BaseConfig
+    def setting
+      "yep"
+    end
+
+    def setting2
+      "nope"
+    end
+
+    def setting3
+      "yep"
+    end
+  end
+end

data/spec/db_spec.rb

@@ -0,0 +1,88 @@
+require "base"
+require "database"
+
+describe BradyW::Database do
+  before(:each) do
+    @db = BradyW::Database.new
+    def @config.db_name
+          "regulardb"
+    end
+    def @config.project_prefix
+      "PRE"
+    end
+  end
+
+  it "DB Name Plain" do
+    @db.name.should == "regulardb"
+  end
+
+  it "DB Name with hostname/prefix" do
+    def @config.db_name
+      "@prefix@-@thismachinehostname@"
+    end
+    @db.name.should == "PRE-"+Socket.gethostname
+  end
+
+  it "User plain" do
+    def @config.db_general_user
+      "username2"
+    end
+
+    @db.user.should == "username2"
+  end
+
+  it "Password" do
+    def @config.db_general_password
+        "thepassword"
+    end
+
+    @db.password.should == "thepassword"
+  end
+
+  it "User hostname/prefix" do
+    def @config.db_general_user
+          "@prefix@-@thismachinehostname@"
+    end
+    @db.user.should == "PRE-"+Socket.gethostname
+  end
+
+  it "Connect String for .NET Code/SQL Auth" do
+    def @config.db_hostname
+      "myhostname"
+    end
+
+    def @config.db_general_authmode
+      :sqlauth
+    end
+
+    def @config.db_general_user
+      "theuser"
+    end
+
+    def @config.db_general_password
+      "thepassword"
+    end
+
+    def @config.db_connect_string_sqlauth
+      "user @user@ pass @password@ host @host@ db @initialcatalog@"
+    end
+
+    @db.connect_code.should == "user theuser pass thepassword host myhostname db regulardb"
+  end
+
+  it "Connect String for .NET Code/Windows Auth" do
+    def @config.db_hostname
+          "myhostname"
+    end
+
+    def @config.db_general_authmode
+          :winauth
+    end
+
+    def @config.db_connect_string_winauth
+      "host @host@ db @initialcatalog@"
+    end
+
+    @db.connect_code.should == "host myhostname db regulardb"
+  end
+end