switchtower 0.9.0

data/test/command_test.rb

@@ -0,0 +1,43 @@
+$:.unshift File.dirname(__FILE__) + "/../lib"
+
+require 'stringio'
+require 'test/unit'
+require 'switchtower/command'
+
+class CommandTest < Test::Unit::TestCase
+  class MockSession
+    def open_channel
+      { :closed => true, :status => 0 }
+    end
+  end
+
+  class MockActor
+    attr_reader :sessions
+
+    def initialize
+      @sessions = Hash.new { |h,k| h[k] = MockSession.new }
+    end
+  end
+
+  def setup
+    @actor = MockActor.new
+  end
+
+  def test_command_executes_on_all_servers
+    command = SwitchTower::Command.new(%w(server1 server2 server3),
+      "hello", nil, {}, @actor)
+    assert_equal %w(server1 server2 server3), @actor.sessions.keys.sort
+  end
+
+  def test_command_with_newlines
+    command = SwitchTower::Command.new(%w(server1), "hello\nworld", nil, {},
+      @actor)
+    assert_equal "hello\\\nworld", command.command
+  end
+
+  def test_command_with_windows_newlines
+    command = SwitchTower::Command.new(%w(server1), "hello\r\nworld", nil, {},
+      @actor)
+    assert_equal "hello\\\nworld", command.command
+  end
+end

data/test/configuration_test.rb

@@ -0,0 +1,210 @@
+$:.unshift File.dirname(__FILE__) + "/../lib"
+
+require 'test/unit'
+require 'switchtower/configuration'
+require 'flexmock'
+
+class ConfigurationTest < Test::Unit::TestCase
+  class MockActor
+    attr_reader :tasks
+
+    def initialize(config)
+    end
+
+    def define_task(*args, &block)
+      (@tasks ||= []).push [args, block].flatten
+    end
+  end
+
+  class MockSCM
+    attr_reader   :configuration
+
+    def initialize(config)
+      @configuration = config
+    end
+  end
+
+  def setup
+    @config = SwitchTower::Configuration.new(MockActor)
+    @config.set :scm, MockSCM
+  end
+
+  def test_version_dir_default
+    assert "releases", @config.version_dir
+  end
+
+  def test_current_dir_default
+    assert "current", @config.current_dir
+  end
+
+  def test_shared_dir_default
+    assert "shared", @config.shared_dir
+  end
+
+  def test_set_repository
+    @config.set :repository, "/foo/bar/baz"
+    assert_equal "/foo/bar/baz", @config.repository
+  end
+
+  def test_set_user
+    @config.set :user, "flippy"
+    assert_equal "flippy", @config.user
+  end
+
+  def test_define_single_role
+    @config.role :app, "somewhere.example.com"
+    assert_equal 1, @config.roles[:app].length
+    assert_equal "somewhere.example.com", @config.roles[:app].first.host
+    assert_equal Hash.new, @config.roles[:app].first.options
+  end
+
+  def test_define_single_role_with_options
+    @config.role :app, "somewhere.example.com", :primary => true
+    assert_equal 1, @config.roles[:app].length
+    assert_equal "somewhere.example.com", @config.roles[:app].first.host
+    assert_equal({:primary => true}, @config.roles[:app].first.options)
+  end
+
+  def test_define_multi_role
+    @config.role :app, "somewhere.example.com", "else.example.com"
+    assert_equal 2, @config.roles[:app].length
+    assert_equal "somewhere.example.com", @config.roles[:app].first.host
+    assert_equal "else.example.com", @config.roles[:app].last.host
+    assert_equal({}, @config.roles[:app].first.options)
+    assert_equal({}, @config.roles[:app].last.options)
+  end
+
+  def test_define_multi_role_with_options
+    @config.role :app, "somewhere.example.com", "else.example.com", :primary => true
+    assert_equal 2, @config.roles[:app].length
+    assert_equal "somewhere.example.com", @config.roles[:app].first.host
+    assert_equal "else.example.com", @config.roles[:app].last.host
+    assert_equal({:primary => true}, @config.roles[:app].first.options)
+    assert_equal({:primary => true}, @config.roles[:app].last.options)
+  end
+
+  def test_load_string_unnamed
+    @config.load :string => "set :repository, __FILE__"
+    assert_equal "<eval>", @config.repository
+  end
+
+  def test_load_string_named
+    @config.load :string => "set :repository, __FILE__", :name => "test.rb"
+    assert_equal "test.rb", @config.repository
+  end
+
+  def test_load
+    file = File.dirname(__FILE__) + "/fixtures/config.rb"
+    @config.load file
+    assert_equal "1/2/foo", @config.repository
+    assert_equal "./#{file}.example.com", @config.gateway
+    assert_equal 1, @config.roles[:web].length
+  end
+
+  def test_load_explicit_name
+    file = File.dirname(__FILE__) + "/fixtures/config.rb"
+    @config.load file, :name => "config"
+    assert_equal "1/2/foo", @config.repository
+    assert_equal "config.example.com", @config.gateway
+    assert_equal 1, @config.roles[:web].length
+  end
+
+  def test_load_file_implied_name
+    file = File.dirname(__FILE__) + "/fixtures/config.rb"
+    @config.load :file => file
+    assert_equal "1/2/foo", @config.repository
+    assert_equal "./#{file}.example.com", @config.gateway
+    assert_equal 1, @config.roles[:web].length
+  end
+
+  def test_load_file_explicit_name
+    file = File.dirname(__FILE__) + "/fixtures/config.rb"
+    @config.load :file => file, :name => "config"
+    assert_equal "1/2/foo", @config.repository
+    assert_equal "config.example.com", @config.gateway
+    assert_equal 1, @config.roles[:web].length
+  end
+
+  def test_task_without_options
+    block = Proc.new { }
+    @config.task :hello, &block
+    assert_equal 1, @config.actor.tasks.length
+    assert_equal :hello, @config.actor.tasks[0][0]
+    assert_equal({}, @config.actor.tasks[0][1])
+    assert_equal block, @config.actor.tasks[0][2]
+  end
+
+  def test_task_with_options
+    block = Proc.new { }
+    @config.task :hello, :roles => :app, &block
+    assert_equal 1, @config.actor.tasks.length
+    assert_equal :hello, @config.actor.tasks[0][0]
+    assert_equal({:roles => :app}, @config.actor.tasks[0][1])
+    assert_equal block, @config.actor.tasks[0][2]
+  end
+
+  def test_source
+    @config.set :repository, "/foo/bar/baz"
+    assert_equal "/foo/bar/baz", @config.source.configuration.repository
+  end
+
+  def test_releases_path_default
+    @config.set :deploy_to, "/start/of/path"
+    assert_equal "/start/of/path/releases", @config.releases_path
+  end
+
+  def test_releases_path_custom
+    @config.set :deploy_to, "/start/of/path"
+    @config.set :version_dir, "right/here"
+    assert_equal "/start/of/path/right/here", @config.releases_path
+  end
+
+  def test_current_path_default
+    @config.set :deploy_to, "/start/of/path"
+    assert_equal "/start/of/path/current", @config.current_path
+  end
+
+  def test_current_path_custom
+    @config.set :deploy_to, "/start/of/path"
+    @config.set :current_dir, "right/here"
+    assert_equal "/start/of/path/right/here", @config.current_path
+  end
+
+  def test_shared_path_default
+    @config.set :deploy_to, "/start/of/path"
+    assert_equal "/start/of/path/shared", @config.shared_path
+  end
+
+  def test_shared_path_custom
+    @config.set :deploy_to, "/start/of/path"
+    @config.set :shared_dir, "right/here"
+    assert_equal "/start/of/path/right/here", @config.shared_path
+  end
+
+  def test_release_path_implicit
+    @config.set :deploy_to, "/start/of/path"
+    assert_equal "/start/of/path/releases/#{@config.now.strftime("%Y%m%d%H%M%S")}", @config.release_path
+  end
+
+  def test_release_path_explicit
+    @config.set :deploy_to, "/start/of/path"
+    assert_equal "/start/of/path/releases/silly", @config.release_path("silly")
+  end
+
+  def test_task_description
+    block = Proc.new { }
+    @config.desc "A sample task"
+    @config.task :hello, &block
+    assert_equal "A sample task", @config.actor.tasks[0][1][:desc]
+  end
+
+  def test_set_scm_to_darcs
+    @config.set :scm, :darcs
+    assert_equal "SwitchTower::SCM::Darcs", @config.source.class.name
+  end
+
+  def test_set_scm_to_subversion
+    @config.set :scm, :subversion
+    assert_equal "SwitchTower::SCM::Subversion", @config.source.class.name
+  end
+end

data/test/fixtures/config.rb

@@ -0,0 +1,5 @@
+set :application, "foo"
+set :repository, "1/2/#{application}"
+set :gateway, "#{__FILE__}.example.com"
+
+role :web, "www.example.com", :primary => true

data/test/scm/cvs_test.rb

@@ -0,0 +1,164 @@
+$:.unshift File.dirname(__FILE__) + "/../../lib"
+
+require File.dirname(__FILE__) + "/../utils"
+require 'test/unit'
+require 'switchtower/scm/cvs'
+
+class ScmCvsTest < Test::Unit::TestCase
+  class CvsTest < SwitchTower::SCM::Cvs
+    attr_accessor :story
+    attr_reader   :last_path
+
+    def cvs_log(path)
+      @last_path = path
+      story.shift
+    end
+  end
+
+  class MockChannel
+    attr_reader :sent_data
+
+    def send_data(data)
+      @sent_data ||= []
+      @sent_data << data
+    end
+
+    def [](name)
+      "value"
+    end
+  end
+
+  class MockActor
+    attr_reader :command
+    attr_reader :channels
+    attr_accessor :story
+
+    def initialize(config)
+      @config = config
+    end
+
+    def run(command)
+      @command = command
+      @channels ||= []
+      @channels << MockChannel.new
+      story.each { |stream, line| yield @channels.last, stream, line }
+    end
+
+    def release_path
+      (@config[:now] || Time.now.utc).strftime("%Y%m%d%H%M%S")
+    end
+
+    def method_missing(sym, *args)
+      @config.send(sym, *args)
+    end
+  end
+
+  def setup
+    @config = MockConfiguration.new
+    @config[:repository] = ":ext:joetester@rubyforge.org:/hello/world"
+    @config[:local] = "/hello/world"
+    @config[:cvs] = "/path/to/cvs"
+    @config[:password] = "chocolatebrownies"
+    @config[:now] = Time.utc(2005,8,24,12,0,0)
+    @scm = CvsTest.new(@config)
+    @actor = MockActor.new(@config)
+    @log_msg = <<MSG.strip
+RCS file: /var/cvs/copland/copland/LICENSE,v
+Working file: LICENSE
+head: 1.1
+branch:
+locks: strict
+access list:
+keyword substitution: kv
+total revisions: 1;     selected revisions: 1
+description:
+----------------------------
+revision 1.1
+date: 2004/08/29 04:23:36;  author: minam;  state: Exp;
+New implementation.
+=============================================================================
+
+RCS file: /var/cvs/copland/copland/Rakefile,v
+Working file: Rakefile
+head: 1.7
+branch:
+locks: strict
+access list:
+keyword substitution: kv
+total revisions: 7;     selected revisions: 1
+description:
+----------------------------
+revision 1.7
+date: 2004/09/15 16:35:01;  author: minam;  state: Exp;  lines: +2 -1
+Rakefile now publishes package documentation from doc/packages instead of
+doc/packrat. Updated "latest updates" in manual.
+=============================================================================
+
+RCS file: /var/cvs/copland/copland/TODO,v
+Working file: TODO
+head: 1.18
+branch:
+locks: strict
+access list:
+keyword substitution: kv
+total revisions: 18;    selected revisions: 1
+description:
+----------------------------
+revision 1.18
+date: 2004/10/12 02:21:02;  author: minam;  state: Exp;  lines: +4 -1
+Added RubyConf 2004 presentation.
+=============================================================================
+
+RCS file: /var/cvs/copland/copland/Attic/build-gemspec.rb,v
+Working file: build-gemspec.rb
+head: 1.5
+branch:
+locks: strict
+access list:
+keyword substitution: kv
+total revisions: 5;     selected revisions: 1
+description:
+----------------------------
+revision 1.5
+date: 2004/08/29 04:10:17;  author: minam;  state: dead;  lines: +0 -0
+Here we go -- point of no return. Deleting existing implementation to make
+way for new implementation.
+=============================================================================
+
+RCS file: /var/cvs/copland/copland/copland.gemspec,v
+Working file: copland.gemspec
+head: 1.12
+branch:
+locks: strict
+access list:
+keyword substitution: kv
+total revisions: 13;    selected revisions: 1
+description:
+----------------------------
+revision 1.12
+date: 2004/09/11 21:45:58;  author: minam;  state: Exp;  lines: +4 -4
+Minor change in how version is communicated to gemspec.
+=============================================================================
+MSG
+    @scm.story = [ @log_msg ]
+  end
+
+  def test_latest_revision
+    @scm.story = [ @log_msg ]
+    assert_equal "2004-10-12 02:21:02", @scm.latest_revision
+    assert_equal "/hello/world", @scm.last_path
+  end
+
+  def test_checkout
+    @actor.story = []
+    assert_nothing_raised { @scm.checkout(@actor) }
+    assert_nil @actor.channels.last.sent_data
+    assert_match %r{/path/to/cvs}, @actor.command
+  end
+
+  def test_checkout_needs_ssh_password
+    @actor.story = [[:out, "joetester@rubyforge.org's password: "]]
+    assert_nothing_raised { @scm.checkout(@actor) }
+    assert_equal ["chocolatebrownies\n"], @actor.channels.last.sent_data
+  end
+end

data/test/scm/subversion_test.rb

@@ -0,0 +1,100 @@
+$:.unshift File.dirname(__FILE__) + "/../../lib"
+
+require File.dirname(__FILE__) + "/../utils"
+require 'test/unit'
+require 'switchtower/scm/subversion'
+
+class ScmSubversionTest < Test::Unit::TestCase
+  class SubversionTest < SwitchTower::SCM::Subversion
+    attr_accessor :story
+    attr_reader   :last_path
+
+    def svn_log(path)
+      @last_path = path
+      story.shift
+    end
+  end
+
+  class MockChannel
+    attr_reader :sent_data
+
+    def send_data(data)
+      @sent_data ||= []
+      @sent_data << data
+    end
+
+    def [](name)
+      "value"
+    end
+  end
+
+  class MockActor
+    attr_reader :command
+    attr_reader :channels
+    attr_accessor :story
+
+    def initialize(config)
+      @config = config
+    end
+
+    def run(command)
+      @command = command
+      @channels ||= []
+      @channels << MockChannel.new
+      story.each { |stream, line| yield @channels.last, stream, line }
+    end
+
+    def method_missing(sym, *args)
+      @config.send(sym, *args)
+    end
+  end
+
+  def setup
+    @config = MockConfiguration.new
+    @config[:repository] = "/hello/world"
+    @config[:svn] = "/path/to/svn"
+    @config[:password] = "chocolatebrownies"
+    @scm = SubversionTest.new(@config)
+    @actor = MockActor.new(@config)
+    @log_msg = <<MSG.strip
+------------------------------------------------------------------------
+r1967 | minam | 2005-08-03 06:59:03 -0600 (Wed, 03 Aug 2005) | 2 lines
+
+Initial commit of the new switchtower utility
+
+------------------------------------------------------------------------
+MSG
+    @scm.story = [ @log_msg ]
+  end
+
+  def test_latest_revision
+    @scm.story = [ @log_msg ]
+    assert_equal "1967", @scm.latest_revision
+    assert_equal "/hello/world", @scm.last_path
+  end
+
+  def test_latest_revision_searching_upwards
+    @scm.story = [ "-----------------------------\n", @log_msg ]
+    assert_equal "1967", @scm.latest_revision
+    assert_equal "/hello", @scm.last_path
+  end
+
+  def test_checkout
+    @actor.story = []
+    assert_nothing_raised { @scm.checkout(@actor) }
+    assert_nil @actor.channels.last.sent_data
+    assert_match %r{/path/to/svn}, @actor.command
+  end
+
+  def test_checkout_needs_ssh_password
+    @actor.story = [[:out, "Password: "]]
+    assert_nothing_raised { @scm.checkout(@actor) }
+    assert_equal ["chocolatebrownies\n"], @actor.channels.last.sent_data
+  end
+
+  def test_checkout_needs_http_password
+    @actor.story = [[:out, "Password for (something): "]]
+    assert_nothing_raised { @scm.checkout(@actor) }
+    assert_equal ["chocolatebrownies\n"], @actor.channels.last.sent_data
+  end
+end