felix-vlad 1.2.0.3

data/test/test_vlad.rb

@@ -0,0 +1,211 @@
+require 'test/vlad_test_case'
+require 'vlad'
+
+$TESTING = true
+
+class TestVlad < VladTestCase
+  def test_all_hosts
+    util_set_hosts
+    assert_equal %w[app.example.com db.example.com], @vlad.all_hosts
+  end
+
+  def test_fetch
+    set :foo, 5
+    assert_equal 5, @vlad.fetch(:foo)
+  end
+
+  def test_fetch_with_default
+    assert_equal 5, @vlad.fetch(:not_here, 5)
+  end
+
+  def test_host
+    @vlad.host "test.example.com", :app, :db
+    expected = {"test.example.com" => {}}
+    assert_equal expected, @vlad.roles[:app]
+    assert_equal expected, @vlad.roles[:db]
+  end
+
+  def test_host_invalid
+    assert_raise(ArgumentError) { @vlad.host "", :app, :db }
+    assert_raise(ArgumentError) { @vlad.host nil, :web }
+  end
+
+  def test_host_multiple_hosts
+    @vlad.host "test.example.com", :app, :db
+    @vlad.host "yarr.example.com", :app, :db, :no_release => true
+
+    expected = {
+      "test.example.com" => {},
+      "yarr.example.com" => {:no_release => true}
+    }
+
+    assert_equal expected, @vlad.roles[:app]
+    assert_equal expected, @vlad.roles[:db]
+    assert_not_equal(@vlad.roles[:db]["test.example.com"].object_id,
+                     @vlad.roles[:app]["test.example.com"].object_id)
+  end
+
+  def test_hosts_for_array_of_roles
+    util_set_hosts
+    assert_equal %w[app.example.com db.example.com], @vlad.hosts_for([:app, :db])
+  end
+
+  def test_hosts_for_one_role
+    util_set_hosts
+    @vlad.host "app2.example.com", :app
+    assert_equal %w[app.example.com app2.example.com], @vlad.hosts_for(:app)
+  end
+
+  def test_hosts_for_multiple_roles
+    util_set_hosts
+    assert_equal %w[app.example.com db.example.com], @vlad.hosts_for(:app, :db)
+  end
+
+  def test_hosts_for_unique
+    util_set_hosts
+    @vlad.host "app.example.com", :web
+    assert_equal %w[app.example.com db.example.com], @vlad.hosts_for(:app, :db, :web)
+  end
+
+  def test_initialize
+    @vlad.set_defaults # ensure these three are virginal
+    assert_raise(Vlad::ConfigurationError) { @vlad.code_repo }
+    assert_raise(Vlad::ConfigurationError) { @vlad.deploy_to }
+    assert_raise(Vlad::ConfigurationError) { @vlad.domain }
+  end
+
+  def test_role
+    @vlad.role :app, "test.example.com"
+    expected = {"test.example.com" => {}}
+    assert_equal expected, @vlad.roles[:app]
+  end
+
+  def test_role_multiple_hosts
+    @vlad.role :app, "test.example.com"
+    @vlad.role :app, "yarr.example.com", :no_release => true
+    expected = {
+      "test.example.com" => {},
+      "yarr.example.com" => {:no_release => true}
+    }
+    assert_equal expected, @vlad.roles[:app]
+  end
+
+  def test_role_multiple_roles
+    @vlad.role :app, "test.example.com", :primary => true
+    @vlad.role :db, "yarr.example.com", :no_release => true
+    expected_db = { "yarr.example.com" => {:no_release => true} }
+    assert_equal expected_db, @vlad.roles[:db]
+    expected_app = { "test.example.com" => {:primary => true} }
+    assert_equal expected_app, @vlad.roles[:app]
+  end
+
+  def test_remote_task
+    t = @vlad.remote_task(:test_task) { 5 }
+    assert_equal @task_count + 1, Rake.application.tasks.size
+    assert_equal Hash.new, t.options
+  end
+
+  def test_remote_task_all_hosts_by_default
+    util_set_hosts
+    t = @vlad.remote_task(:test_task) { 5 }
+    assert_equal %w[app.example.com db.example.com], t.target_hosts
+  end
+
+  def test_remote_task_environment_override
+    old_env_hosts = ENV["HOSTS"]
+    ENV["HOSTS"] = 'other1.example.com,   other2.example.com'
+    util_set_hosts
+    t = @vlad.remote_task(:test_task) { 5 }
+    assert_equal %w[other1.example.com other2.example.com], t.target_hosts
+  ensure 
+    ENV["HOSTS"] = old_env_hosts
+  end
+
+  def test_remote_task_body_set
+    set(:some_variable, 5)
+    @vlad.host 'www.example.com', :app
+    @vlad.remote_task(:some_task) do $some_task_result = some_variable end
+
+    Rake::Task['some_task'].execute nil
+    assert_equal @vlad.fetch(:some_variable), $some_task_result
+  end
+
+  def test_remote_task_with_options
+    t = @vlad.remote_task :test_task, :roles => [:app, :db] do
+      fail "should not run"
+    end
+    assert_equal({:roles => [:app, :db]}, t.options)
+  end
+
+  def test_remote_task_before_host_declaration
+    t = @vlad.remote_task :test_task, :roles => :web do 5 end
+    @vlad.host 'www.example.com', :web
+    assert_equal %w[www.example.com], t.target_hosts
+  end
+
+  def test_remote_task_role_override
+    host "db1", :db
+    host "db2", :db
+    host "db3", :db
+    host "master", :master_db
+
+    remote_task(:migrate_the_db, :roles => [:db]) { flunk "bad!" }
+    task = Rake::Task["migrate_the_db"]
+    assert_equal %w[db1 db2 db3], task.target_hosts
+
+    task.options[:roles] = :master_db
+    assert_equal %w[master], task.target_hosts
+
+    task.options[:roles] = [:master_db]
+    assert_equal %w[master], task.target_hosts
+  end
+
+  def test_set
+    set :test, 5
+    assert_equal 5, @vlad.test
+  end
+
+  def test_set_lazy_block_evaluation
+    set(:test) { fail "lose" }
+    assert_raise(RuntimeError) { @vlad.test }
+  end
+
+  def test_set_with_block
+    x = 1
+    set(:test) { x += 2 }
+
+    assert_equal 3, @vlad.test
+    assert_equal 3, @vlad.test
+  end
+
+  def test_set_with_reference
+    @vlad.instance_eval do
+      set(:var_one) { var_two }
+      set(:var_two) { var_three }
+      set(:var_three) { 5 }
+    end
+
+    assert_equal 5, @vlad.var_one
+  end
+
+  def test_set_with_block_and_value
+    e = assert_raise(ArgumentError) do
+      set(:test, 5) { 6 }
+    end
+    assert_equal "cannot provide both a value and a block", e.message
+  end
+
+  def test_set_with_nil
+    set(:test, nil)
+    assert_equal nil, @vlad.test
+  end
+
+  def test_set_with_reserved_name
+    $TESTING = false
+    e = assert_raise(ArgumentError) { set(:all_hosts, []) }
+    assert_equal "cannot set reserved name: 'all_hosts'", e.message
+  ensure
+    $TESTING = true
+  end
+end
+

data/test/test_vlad_git.rb

@@ -0,0 +1,39 @@
+require 'test/vlad_test_case'
+require 'vlad'
+require 'vlad/git'
+
+class TestVladGit < VladTestCase
+  def setup
+    super
+    @scm = Vlad::Git.new
+    set :code_repo, "git@myhost:/home/john/project1"
+  end
+
+  def test_checkout
+    # Checkout to the current directory (which is what the :update task passes)
+    cmd = @scm.checkout 'master', '.'
+    assert_equal 'rm -rf repo && git clone git@myhost:/home/john/project1 repo && cd repo && git checkout -f -b deployed-master master', cmd
+
+    # Mimic :update task
+    # 'head' should become HEAD
+    cmd = @scm.checkout 'head', '.'
+    assert_equal 'rm -rf repo && git clone git@myhost:/home/john/project1 repo && cd repo && git checkout -f -b deployed-HEAD HEAD', cmd
+
+    # Checkout to a relative path
+    cmd = @scm.checkout 'master', 'some/relative/path'
+    assert_equal 'rm -rf some/relative/path && git clone git@myhost:/home/john/project1 some/relative/path && cd some/relative/path && git checkout -f -b deployed-master master', cmd
+  end
+
+  def test_export
+    cmd = @scm.export 'master', 'the/release/path'
+    assert_equal 'mkdir -p the/release/path && git archive --format=tar master | (cd the/release/path && tar xf -)', cmd
+  end
+
+  def test_revision
+    ['head', 'HEAD'].each do |head|
+      cmd = @scm.revision(head)
+      expected = "`git rev-parse HEAD`"
+      assert_equal expected, cmd
+    end
+  end
+end

data/test/test_vlad_mercurial.rb

@@ -0,0 +1,26 @@
+require 'test/vlad_test_case'
+require 'vlad'
+require 'vlad/mercurial'
+
+class TestVladMercurial < Test::Unit::TestCase
+  def setup
+    @scm = Vlad::Mercurial.new
+    set :code_repo, "http://repo/project"
+  end
+
+  def test_checkout
+    cmd = @scm.checkout 'head', '/the/place'
+    assert_equal 'hg pull -r tip -R /the/place http://repo/project', cmd
+  end
+
+  def test_export
+    cmd = @scm.export 'head', '/the/place'
+    assert_equal 'hg archive -r tip -R http://repo/project /the/place', cmd
+  end
+
+  def test_revision
+    cmd = @scm.revision('tip')
+    expected = "`hg identify -R http://repo/project | cut -f1 -d\\ `"
+    assert_equal expected, cmd
+  end
+end

data/test/test_vlad_perforce.rb

@@ -0,0 +1,37 @@
+require 'test/vlad_test_case'
+require 'vlad'
+require 'vlad/perforce'
+
+class TestVladPerforce < VladTestCase
+  def setup
+    super
+    @scm = Vlad::Perforce.new
+  end
+
+  def test_checkout
+    cmd = @scm.checkout 'head', '/the/place'
+    assert_equal 'p4 sync ...#head', cmd
+  end
+
+  def test_checkout_revision
+    cmd = @scm.checkout 555, '/the/place'
+    assert_equal 'p4 sync ...@555', cmd
+  end
+
+  def test_export
+    cmd = @scm.export 'head', '/the/place'
+    assert_equal '(cd /the/place && p4 sync ...#head)', cmd
+  end
+
+  def test_revision
+    cmd = @scm.revision('head')
+    assert_equal '`p4 changes -s submitted -m 1 ...#head | cut -f 2 -d\\ `', cmd
+  end
+
+  def test_rev_no
+    assert_equal "@555", @scm.rev_no(555)
+    assert_equal "#head", @scm.rev_no('head')
+    assert_equal "#head", @scm.rev_no('HEAD')
+    assert_equal "@666", @scm.rev_no("@666")
+  end
+end

data/test/test_vlad_subversion.rb

@@ -0,0 +1,27 @@
+require 'test/vlad_test_case'
+require 'vlad'
+require 'vlad/subversion'
+
+class TestVladSubversion < VladTestCase
+  def setup
+    super
+    @scm = Vlad::Subversion.new
+    set :code_repo, "svn+ssh://repo/myproject"
+  end
+
+  def test_checkout
+    cmd = @scm.checkout 'HEAD', '/the/place'
+    assert_equal 'svn co -r HEAD svn+ssh://repo/myproject /the/place', cmd
+  end
+
+  def test_export
+    cmd = @scm.export 'HEAD', '/the/place'
+    assert_equal 'svn export -r HEAD svn+ssh://repo/myproject /the/place', cmd
+  end
+
+  def test_revision
+    cmd = @scm.revision('HEAD')
+    expected = "`svn info svn+ssh://repo/myproject | grep 'Revision:' | cut -f2 -d\\ `"
+    assert_equal expected, cmd
+  end
+end

data/test/vlad_test_case.rb

@@ -0,0 +1,71 @@
+require 'test/unit'
+require 'stringio'
+require 'vlad'
+
+class StringIO
+  def readpartial(size) read end # suck!
+end
+
+module Process
+  def self.expected status
+    @@expected ||= []
+    @@expected << status
+  end
+
+  def self.waitpid2(pid)
+    [ @@expected.shift ]
+  end
+end
+
+
+class Rake::RemoteTask
+  attr_accessor :commands, :action, :input, :output, :error
+
+  Status = Struct.new :exitstatus
+
+  class Status
+    def success?() exitstatus == 0 end
+  end
+
+  def system *command
+    @commands << command
+    self.action ? self.action[command.join(' ')] : true
+  end
+
+  def popen4 *command
+    @commands << command
+
+    @input = StringIO.new
+    out = StringIO.new @output.shift.to_s
+    err = StringIO.new @error.shift.to_s
+
+    raise if block_given?
+
+    status = self.action ? self.action[command.join(' ')] : 0
+    Process.expected Status.new(status)
+
+    return 42, @input, out, err
+  end
+
+  def select reads, writes, errs, timeout
+    [reads, writes, errs]
+  end
+
+end
+
+class VladTestCase < Test::Unit::TestCase
+  undef_method :default_test
+
+  def setup
+    @vlad = Rake::RemoteTask
+    @vlad.reset
+    Rake.application.clear
+    @task_count = Rake.application.tasks.size
+    @vlad.set :domain, "example.com"
+  end
+
+  def util_set_hosts
+    @vlad.host "app.example.com", :app
+    @vlad.host "db.example.com", :db
+  end
+end

data/vladdemo.sh

@@ -0,0 +1,64 @@
+#!/bin/bash
+
+killall mongrel svnserve 2> /dev/null
+
+rm -rf ~/demo
+mkdir ~/demo
+cd ~/demo
+
+pause() {
+  echo -n "waiting... hit return... "
+  read
+  echo
+}
+
+echo "Starting demo from scratch"
+echo "  This step creates a subversion code repo and imports a new rails app"
+echo "  It modifies the Rakefile to load Vlad and creates config/deploy.rb"
+echo
+pause
+
+svnadmin create svnrepo
+echo "anon-access = write" >> svnrepo/conf/svnserve.conf
+
+svnserve -d --foreground -r svnrepo --listen-host localhost &
+
+rails mydemoapp
+
+cd mydemoapp
+
+echo "require 'rubygems'
+require 'vlad'
+Vlad.load" >> Rakefile
+
+echo "set :code_repo, 'svn://localhost/blah'
+set :domain, 'localhost'
+set :deploy_to, '/Users/ryan/demo/website'
+set :web_command, 'sudo apachectl'" > config/deploy.rb
+
+svn import -m Added . svn://localhost/blah
+
+echo
+echo "Here are the tasks available:"
+echo
+
+rake -T vlad
+
+echo
+echo "The next step deploys and fires up the application"
+echo
+pause
+
+rake -t vlad:setup vlad:update vlad:start
+
+open http://localhost:8000/
+
+echo
+echo "done! check it out"
+echo
+pause
+
+rake vlad:stop
+
+kill %1
+