capistrano-file-transfer-ext 0.0.4 → 0.1.0

data/README.md

@@ -26,6 +26,9 @@ Now you can use them in your tasks.
 
 * `safe_upload` - Transfers a file from the local host to multiple remote hosts. ([more...](#safe_upload))
 * `safe_put` - Store the contents of multiple servers. ([more...](#safe_put))
+* `transfer_if_modified` - Transfer files from/to remote servers only if they are different. ([more...](#transfer_if_modified))
+* `install` - Install files on remote servers. ([more...](#install))
+* `install_if_modified` - Install files on remote servers only if they are different. ([more...](#install_if_modified))
 
 ## Reference
 
@@ -59,9 +62,9 @@ This must be a string indicating the path on the remote server that should be up
 All of the options of `upload` are sensible. In addition, there are some extra options.
 
 * `:install` It must be either `:always` (the default), or `:if_modified`. If `:if_modified` is given, install the file only if the checksums are different.
-* `:digest` Thi is a symbol indicating which algorithm should be used to calculate the checksum of files. `:md5` is default.
+* `:digest` This is a symbol indicating which algorithm should be used to calculate the checksum of files. `:md5` is default.
 * `:digest_cmd` The command to calculate checksum of files. `md5sum` is default.
-* `:sudo` It must be a boolean. If set true, use `sudo` on placing files. `false` by default.
+* `:via` specify the method to run commands. `:run` by default. you may need to set this value as `:sudo` if you want to overwrite system files.
 
 
 ### `safe_put`
@@ -91,6 +94,104 @@ This is a string naming the file on the remote server(s) that should be created
 All of the options of `safe_upload` are sensible.
 
 
+### `transfer_if_modified`
+
+**Definition**
+
+    transfer_if_modified(direction, from, to, options={}, &block)
+
+**Module**
+
+    Capistrano::Configuration::Actions::FileTransferExt
+
+The `transfer_if_modified` action is `transfer`'s senior brother.
+Transfer files from/to remote servers only if they are different.
+
+#### Arguments
+
+**direction**
+
+The direction of transfer. `:up` or `:down` is sensible.
+
+**from**
+
+Source file.
+
+**to**
+
+Destination file.
+
+**options**
+
+All of the options of `transfer` are sensible. In addition, there are some extra options.
+
+* `:digest` This is a symbol indicating which algorithm should be used to calculate the checksum of files. `:md5` is default.
+* `:digest_cmd` The command to calculate checksum of files. `md5sum` is default.
+
+
+### `install`
+
+**Definition**
+
+    install(from, to, options={}, &block)
+
+**Module**
+
+    Capistrano::Configuration::Actions::FileTransferExt
+
+Install files on remote servers. This method acts just like `mv`.
+
+#### Arguments
+
+**from**
+
+Source file on remote server.
+
+**to**
+
+Destination file on remote server.
+
+**options**
+
+* `:via` specify the method to run commands. `:run` by default. you may need to set this value as `:sudo` if you want to overwrite system files.
+* `:mode` The mode of destination file. If not given, preserve original mode of `to`.
+* `:owner` The owner of destination file. If not given and `:via` is `:sudo`, preserve original mode of `to`.
+* `:group` The group of destination file. If not given and `:via` is `:sudo`, preserve original mode of `to`.
+
+
+### `install_if_modified`
+
+**Definition**
+
+    install_if_modified(from, to, options={}, &block)
+
+**Module**
+
+    Capistrano::Configuration::Actions::FileTransferExt
+
+The `install_if_modified` action is `install`'s senior brother.
+Install files on remote servers only if they are different.
+
+#### Arguments
+
+**from**
+
+Source file on remote server.
+
+**to**
+
+Destination file on remote server.
+
+**options**
+
+* `:via` specify the method to run commands. `:run` by default. you may need to set this value as `:sudo` if you want to overwrite system files.
+* `:mode` The mode of destination file. If not given, preserve original mode of `to`.
+* `:owner` The owner of destination file. If not given and `:via` is `:sudo`, preserve original mode of `to`.
+* `:group` The group of destination file. If not given and `:via` is `:sudo`, preserve original mode of `to`.
+* `:digest` This is a symbol indicating which algorithm should be used to calculate the checksum of files. `:md5` is default.
+* `:digest_cmd` The command to calculate checksum of files. `md5sum` is default.
+
+
 ## Contributing
 
 1. Fork it

data/capistrano-file-transfer-ext.gemspec

@@ -16,4 +16,7 @@ Gem::Specification.new do |gem|
   gem.version       = Capistrano::Configuration::Actions::FileTransferExt::VERSION
 
   gem.add_dependency("capistrano")
+  gem.add_development_dependency("net-scp", "~> 1.0.4")
+  gem.add_development_dependency("net-ssh", "~> 2.2.2")
+  gem.add_development_dependency("vagrant", "~> 1.0.6")
 end

data/lib/capistrano/configuration/actions/file_transfer_ext.rb

@@ -64,7 +64,7 @@ module Capistrano
         # * :digest_cmd - the digest command. the default is "#{digest}sum".
         #
         def transfer_if_modified(direction, from, to, options={}, &block)
-          digest_method = options.fetch(:digest, "md5")
+          digest_method = options.fetch(:digest, :md5).to_s
           digest_cmd = options.fetch(:digest_cmd, "#{digest_method.downcase}sum")
           require "digest/#{digest_method.downcase}"
           target = direction == :up ? from : to
@@ -75,7 +75,7 @@ module Capistrano
             target.pos = pos
           else
             begin
-              digest = Digest::const_get(digest_method.upcase).hexdigest(File.read(target))
+              digest = Digest.const_get(digest_method.upcase).hexdigest(File.read(target))
             rescue SystemCallError
               digest = nil
             end
@@ -102,18 +102,63 @@ module Capistrano
         # The +options+ hash may include any of the following keys:
         #
         # * :mode - permission of the file.
-        # * :sudo - use sudo if set true. the default is false.
+        # * :via - :run by default.
         #
         def install(from, to, options={}, &block)
-          mode = options.delete(:mode)
-          try_sudo = options.delete(:sudo) ? sudo : ""
+          via = options.delete(:via)
+          if via == :sudo or options.delete(:sudo) # check :sudo for backward compatibility
+            # ignore {:via => :sudo} since `sudo()` cannot handle multiple commands properly.
+            try_sudo = sudo
+          else
+            try_sudo = ""
+            options[:via] = via
+          end
+          if options.key?(:mode)
+            mode = options.delete(:mode)
+          elsif fetch(:install_preserve_mode, true)
+            begin
+              # respect mode of original file
+              # `stat -c` for GNU, `stat -f` for BSD
+              s = capture("test -f #{to.dump} && ( stat -c '%a' #{to.dump} || stat -f '%p' #{to.dump} )", options)
+              mode = s.to_i(8) & 0777 if /^[0-7]+$/ =~ s
+              logger.debug("preserve original file mode #{mode.to_s(8)}.")
+            rescue
+              # nop
+            end
+          end
+          if options.key?(:owner)
+            owner = options.delete(:owner)
+          elsif fetch(:install_preserve_owner, true) and via == :sudo
+            begin
+              owner = capture("test -f #{to.dump} && ( stat -c '%u' #{to.dump} || stat -f '%u' #{to.dump} )", options).strip
+              logger.debug("preserve original file owner #{owner.dump}.")
+            rescue
+              # nop
+            end
+          end
+          if options.key?(:group)
+            group = options.delete(:group)
+          elsif fetch(:install_preserve_group, true) and via == :sudo
+            begin
+              group = capture("test -f #{to.dump} && ( stat -c '%g' #{to.dump} || stat -f '%g' #{to.dump} )", options).strip
+              logger.debug("preserve original file grop #{group.dump}.")
+            rescue
+              # nop
+            end
+          end
           execute = []
-          execute << "( test -d #{File.dirname(to).dump} || #{try_sudo} mkdir -p #{File.dirname(to).dump} )"
-          execute << "#{try_sudo} mv -f #{from.dump} #{to.dump}"
+          if block_given?
+            execute << yield(from, to)
+          else
+            execute << "( test -d #{File.dirname(to).dump} || #{try_sudo} mkdir -p #{File.dirname(to).dump} )"
+            execute << "#{try_sudo} mv -f #{from.dump} #{to.dump}"
+          end
           if mode
             mode = mode.is_a?(Numeric) ? mode.to_s(8) : mode.to_s
-            execute << "#{try_sudo} chmod #{mode} #{to.dump}"
+            execute << "#{try_sudo} chmod #{mode.dump} #{to.dump}"
           end
+          execute << "#{try_sudo} chown #{owner.to_s.dump} #{to.dump}" if owner
+          execute << "#{try_sudo} chgrp #{group.to_s.dump} #{to.dump}" if group
           invoke_command(execute.join(" && "), options)
         end
         alias place install
@@ -123,37 +168,32 @@ module Capistrano
         # The +options+ hash may include any of the following keys:
         #
         # * :mode - permission of the file.
-        # * :sudo - use sudo if set true. the default is false.
+        # * :via - :run by default.
         # * :digest - digest algorithm. the default is "md5".
         # * :digest_cmd - the digest command. the default is "#{digest}sum".
         #
         def install_if_modified(from, to, options={}, &block)
-          mode = options.delete(:mode)
-          try_sudo = options.delete(:sudo) ? sudo : ""
-          digest_method = options.fetch(:digest, "md5")
+          digest_method = options.fetch(:digest, :md5).to_s
           digest_cmd = options.fetch(:digest_cmd, "#{digest_method.downcase}sum")
-          execute = []
-          execute << %{( test -d #{File.dirname(to).dump} || #{try_sudo} mkdir -p #{File.dirname(to).dump} )}
-          # calculate hexdigest of `from'
-          execute << %{from=$(#{digest_cmd} #{from.dump} 2>/dev/null | #{DIGEST_FILTER_CMD} || true)}
-          execute << %{echo %s} % ["#{digest_method.upcase}(#{from}) = ${from}".dump]
-          # calculate hexdigest of `to'
-          execute << %{to=$(#{digest_cmd} #{to.dump} 2>/dev/null | #{DIGEST_FILTER_CMD} || true)}
-          execute << %{echo %s} % ["#{digest_method.upcase}(#{to}) = ${to}".dump]
-          # check the hexdigests
-          execute << (<<-EOS).gsub(/\s+/, " ").strip
-            if [ -n "${from}" -a "${to}" ] && [ "${from}" = "${to}" ]; then
-              echo "skip placing since no changes.";
-            else
-              #{try_sudo} mv -f #{from.dump} #{to.dump};
-            fi
-          EOS
-
-          if mode
-            mode = mode.is_a?(Numeric) ? mode.to_s(8) : mode.to_s
-            execute << "#{try_sudo} chmod #{mode} #{to.dump}"
+          install(from, to, options) do |from, to|
+            execute = []
+            execute << %{( test -d #{File.dirname(to).dump} || #{try_sudo} mkdir -p #{File.dirname(to).dump} )}
+            # calculate hexdigest of `from'
+            execute << %{from=$(#{digest_cmd} #{from.dump} 2>/dev/null | #{DIGEST_FILTER_CMD} || true)}
+            execute << %{echo %s} % ["#{digest_method.upcase}(#{from}) = ${from}".dump]
+            # calculate hexdigest of `to'
+            execute << %{to=$(#{digest_cmd} #{to.dump} 2>/dev/null | #{DIGEST_FILTER_CMD} || true)}
+            execute << %{echo %s} % ["#{digest_method.upcase}(#{to}) = ${to}".dump]
+            # check the hexdigests
+            execute << (<<-EOS).gsub(/\s+/, " ").strip
+              if [ -n "${from}" -a "${to}" ] && [ "${from}" = "${to}" ]; then
+                echo "skip installing since no changes.";
+              else
+                #{try_sudo} mv -f #{from.dump} #{to.dump};
+              fi
+            EOS
+            execute.join(" && ")
           end
-          invoke_command(execute.join(" && "), options)
         end
         alias place_if_modified install_if_modified
       end

data/lib/capistrano/configuration/actions/file_transfer_ext/version.rb

@@ -2,7 +2,7 @@ module Capistrano
   class Configuration
     module Actions
       module FileTransferExt
-        VERSION = "0.0.4"
+        VERSION = "0.1.0"
       end
     end
   end

data/test/centos6-64/.gitignore

@@ -0,0 +1,5 @@
+/.bundle
+/.vagrant
+/known_hosts
+/tmp
+/vendor

data/test/centos6-64/Capfile

@@ -0,0 +1,2 @@
+load "deploy"
+load "../config/deploy"

data/test/centos6-64/Gemfile

@@ -0,0 +1,2 @@
+source "https://rubygems.org"
+gemspec(:path => "../..")

data/test/centos6-64/Vagrantfile

@@ -0,0 +1,99 @@
+# -*- mode: ruby -*-
+# vi: set ft=ruby :
+
+Vagrant::Config.run do |config|
+  # All Vagrant configuration is done here. The most common configuration
+  # options are documented and commented below. For a complete reference,
+  # please see the online documentation at vagrantup.com.
+
+  # Every Vagrant virtual environment requires a box to build off of.
+  config.vm.box = "centos6-64"
+
+  # The url from where the 'config.vm.box' box will be fetched if it
+  # doesn't already exist on the user's system.
+  config.vm.box_url = "http://developer.nrel.gov/downloads/vagrant-boxes/CentOS-6.3-x86_64-v20130101.box"
+
+  # Boot with a GUI so you can see the screen. (Default is headless)
+  # config.vm.boot_mode = :gui
+
+  # Assign this VM to a host-only network IP, allowing you to access it
+  # via the IP. Host-only networks can talk to the host machine as well as
+  # any other machines on the same network, but cannot be accessed (through this
+  # network interface) by any external networks.
+  config.vm.network :hostonly, "192.168.33.10"
+
+  # Assign this VM to a bridged network, allowing you to connect directly to a
+  # network using the host's network device. This makes the VM appear as another
+  # physical device on your network.
+  # config.vm.network :bridged
+
+  # Forward a port from the guest to the host, which allows for outside
+  # computers to access the VM, whereas host only networking does not.
+  # config.vm.forward_port 80, 8080
+
+  # Share an additional folder to the guest VM. The first argument is
+  # an identifier, the second is the path on the guest to mount the
+  # folder, and the third is the path on the host to the actual folder.
+  # config.vm.share_folder "v-data", "/vagrant_data", "../data"
+
+  # Enable provisioning with Puppet stand alone.  Puppet manifests
+  # are contained in a directory path relative to this Vagrantfile.
+  # You will need to create the manifests directory and a manifest in
+  # the file precise-amd64.pp in the manifests_path directory.
+  #
+  # An example Puppet manifest to provision the message of the day:
+  #
+  # # group { "puppet":
+  # #   ensure => "present",
+  # # }
+  # #
+  # # File { owner => 0, group => 0, mode => 0644 }
+  # #
+  # # file { '/etc/motd':
+  # #   content => "Welcome to your Vagrant-built virtual machine!
+  # #               Managed by Puppet.\n"
+  # # }
+  #
+  # config.vm.provision :puppet do |puppet|
+  #   puppet.manifests_path = "manifests"
+  #   puppet.manifest_file  = "precise-amd64.pp"
+  # end
+
+  # Enable provisioning with chef solo, specifying a cookbooks path, roles
+  # path, and data_bags path (all relative to this Vagrantfile), and adding 
+  # some recipes and/or roles.
+  #
+  # config.vm.provision :chef_solo do |chef|
+  #   chef.cookbooks_path = "../my-recipes/cookbooks"
+  #   chef.roles_path = "../my-recipes/roles"
+  #   chef.data_bags_path = "../my-recipes/data_bags"
+  #   chef.add_recipe "mysql"
+  #   chef.add_role "web"
+  #
+  #   # You may also specify custom JSON attributes:
+  #   chef.json = { :mysql_password => "foo" }
+  # end
+
+  # Enable provisioning with chef server, specifying the chef server URL,
+  # and the path to the validation key (relative to this Vagrantfile).
+  #
+  # The Opscode Platform uses HTTPS. Substitute your organization for
+  # ORGNAME in the URL and validation key.
+  #
+  # If you have your own Chef Server, use the appropriate URL, which may be
+  # HTTP instead of HTTPS depending on your configuration. Also change the
+  # validation key to validation.pem.
+  #
+  # config.vm.provision :chef_client do |chef|
+  #   chef.chef_server_url = "https://api.opscode.com/organizations/ORGNAME"
+  #   chef.validation_key_path = "ORGNAME-validator.pem"
+  # end
+  #
+  # If you're using the Opscode platform, your validator client is
+  # ORGNAME-validator, replacing ORGNAME with your organization name.
+  #
+  # IF you have your own Chef Server, the default validation client name is
+  # chef-validator, unless you changed the configuration.
+  #
+  #   chef.validation_client_name = "ORGNAME-validator"
+end

data/test/centos6-64/run.sh

@@ -0,0 +1,7 @@
+#!/bin/sh -e
+
+bundle exec vagrant up
+bundle exec cap test_all
+bundle exec vagrant destroy -f
+
+# vim:set ft=sh :

data/test/config/deploy.rb

@@ -0,0 +1,476 @@
+set :application, "capistrano-file-transfer-ext"
+set :repository,  "."
+set :deploy_to do
+  File.join("/home", user, application)
+end
+set :deploy_via, :copy
+set :scm, :none
+set :use_sudo, false
+set :user, "vagrant"
+set :password, "vagrant"
+set :ssh_options, {:user_known_hosts_file => "/dev/null"}
+
+role :web, "192.168.33.10"
+role :app, "192.168.33.10"
+role :db,  "192.168.33.10", :primary => true
+
+$LOAD_PATH.push(File.expand_path("../../lib", File.dirname(__FILE__)))
+require "capistrano/configuration/actions/file_transfer_ext"
+require "stringio"
+
+task(:test_all) {
+  find_and_execute_task("test_default")
+  find_and_execute_task("test_transfer_if_modified")
+  find_and_execute_task("test_install")
+  find_and_execute_task("test_install_if_modified")
+}
+
+def _invoke_command(cmdline, options={})
+  via = options.delete(:via)
+  if via == :run_locally
+    run_locally(cmdline)
+  else
+    invoke_command(cmdline, options.merge(:via => via))
+  end
+end
+
+def assert_timestamp_equals(x, y, options={})
+  begin
+    _invoke_command("test \! #{x.dump} -nt #{y.dump} -a \! #{x.dump} -ot #{y.dump}", options)
+  rescue
+    logger.debug("assert_timestamp_equals(#{x}, #{y}) failed.")
+    _invoke_command("ls -l #{x.dump} #{y.dump}", options)
+    raise
+  end
+end
+
+def assert_timestamp_not_equals(x, y, options={})
+  begin
+    _invoke_command("test #{x.dump} -nt #{y.dump} -o #{x.dump} -ot #{y.dump}", options)
+  rescue
+    logger.debug("assert_timestamp_not_equals(#{x}, #{y}) failed.")
+    _invoke_command("ls -l #{x.dump} #{y.dump}", options)
+    raise
+  end
+end
+
+def assert_file_equals(x, y, options={})
+  begin
+    _invoke_command("cmp #{x.dump} #{y.dump}", options)
+  rescue
+    logger.debug("assert_file_equals(#{x}, #{y}) failed.")
+    _invoke_command("ls -l #{x.dump} #{y.dump}", options)
+    raise
+  end
+end
+
+def assert_file_mode_equals(x, y, options={})
+  begin
+    _invoke_command("test $(ls -l #{x.dump} | cut -d ' ' -f 1) = $(ls -l #{y.dump} | cut -d ' ' -f 1)", options)
+  rescue
+    logger.debug("assert_file_mode_equals(#{x}, #{y}) failed.")
+    _invoke_command("ls -l #{x.dump} #{y.dump}", options)
+    raise
+  end
+end
+
+def assert_file_mode(mode, file, options={})
+  mode = mode.to_i(8) if mode.is_a?(String)
+  smode = "%c%c%c%c%c%c%c%c%c%c" % [
+    ?-,
+    mode & 0400 == 0 ? ?- : ?r, mode & 0200 == 0 ? ?- : ?w, mode & 0100 == 0 ? ?- : ?x,
+    mode & 0040 == 0 ? ?- : ?r, mode & 0020 == 0 ? ?- : ?w, mode & 0010 == 0 ? ?- : ?x,
+    mode & 0004 == 0 ? ?- : ?r, mode & 0002 == 0 ? ?- : ?w, mode & 0001 == 0 ? ?- : ?x,
+  ]
+  begin
+    _invoke_command("test #{smode.dump} = $(ls -l #{file.dump} | cut -d ' ' -f 1)", options)
+  rescue
+    logger.debug("assert_file_mode(#{mode.to_s(8)}, #{file}) failed.")
+    _invoke_command("ls -l #{file.dump}", options)
+    raise
+  end
+end
+
+def assert_file_owner(uid, file, options={})
+  uid = uid.to_i
+  # `stat -c` => GNU, `stat -f` => BSD
+  begin
+    _invoke_command("test #{uid} -eq $( stat -c '%u' #{file.dump} || stat -f '%u' #{file.dump} )", options)
+  rescue
+    logger.debug("assert_file_owner(#{uid}, #{file}) failed.")
+    _invoke_command("ls -l #{file.dump}", options)
+    raise
+  end
+end
+
+def assert_file_group(gid, file, options={})
+  gid = gid.to_i
+  # `stat -c` => GNU, `stat -f` => BSD
+  begin
+    _invoke_command("test #{gid} -eq $( stat -c '%g' #{file.dump} || stat -f '%g' #{file.dump} )", options)
+  rescue
+    logger.debug("assert_file_group(#{gid}, #{file}) failed.")
+    _invoke_command("ls -l #{file.dump}", options)
+    raise
+  end
+end
+
+namespace(:test_default) {
+  task(:default) {
+    methods.grep(/^test_/).each do |m|
+      send(m)
+    end
+  }
+  before "test_default", "test_default:setup"
+  after "test_default", "test_default:teardown"
+
+  task(:setup) {
+    run_locally("mkdir -p tmp")
+    run("mkdir -p tmp")
+  }
+
+  task(:teardown) {
+    run_locally("rm -rf tmp")
+    sudo("rm -rf tmp")
+  }
+
+  task(:test_safe_upload) {
+    body = "foo"
+    from = "tmp/foo"
+    to = "tmp/rfoo"
+    run_locally("rm -f #{from.dump}; echo #{body.dump} > #{from.dump}")
+    run("rm -f #{to.dump}")
+    safe_upload(from, to)
+    run("test -f #{to.dump}")
+    run("test #{body.dump} = $(cat #{to.dump})")
+  }
+
+  task(:test_safe_put) {
+    body = "bar"
+    to = "tmp/rbar"
+    run("rm -f #{to.dump}")
+    safe_put(body, to)
+    run("test -f #{to.dump}")
+    run("test #{body.dump} = $(cat #{to.dump})")
+  }
+}
+
+namespace(:test_transfer_if_modified) {
+  task(:default) {
+    methods.grep(/^test_/).each do |m|
+      send(m)
+    end
+  }
+  before "test_transfer_if_modified", "test_transfer_if_modified:setup"
+  after "test_transfer_if_modified", "test_transfer_if_modified:teardown"
+
+  task(:setup) {
+    run_locally("mkdir -p tmp")
+    run("mkdir -p tmp")
+  }
+
+  task(:teardown) {
+    run_locally("rm -rf tmp")
+    sudo("rm -rf tmp")
+  }
+
+  def _test_transfer_if_modified_up(from, to, options={})
+    run("rm -f #{to.dump}")
+    transfer_if_modified(:up, from, to, options)
+    if from.respond_to?(:read)
+      pos = from.pos
+      body = from.dup.read
+      from.pos = pos
+    else
+      body = File.read(from)
+    end
+    tempbody = capture("mktemp tmp/body.XXXXXXXXXX").strip
+    run("rm -f #{tempbody.dump}")
+    put(body, tempbody)
+    assert_file_equals(tempbody, to)
+
+## should not transfer without changes
+    2.times do
+      sleep(1)
+      tempto = capture("mktemp tmp/to.XXXXXXXXXX").strip
+      run("rm -f #{tempto.dump}; cp -p #{to.dump} #{tempto.dump}")
+      transfer_if_modified(:up, from, to, options) # re-transfer
+      assert_file_equals(tempto, to)
+      assert_timestamp_equals(tempto, to)
+    end
+
+## should transfer if `from' is changed
+    tempfrom = run_locally("mktemp tmp/from.XXXXXXXXXX").strip
+    tempto = capture("mktemp tmp/to.XXXXXXXXXX").strip
+    File.write(tempfrom, body * 4)
+    run("rm -f #{tempto.dump}; cp -p #{to.dump} #{tempto.dump}")
+    transfer_if_modified(:up, tempfrom, to, options)
+    run("test #{tempto.dump} -ot #{to.dump}") # check if `to' is overwritten
+  end
+
+  def _test_transfer_if_modified_down(from, to, options={})
+    to = to.gsub(/\$CAPISTRANO:HOST\$/, "192.168.33.10")
+    run_locally("rm -f #{to.dump}")
+    transfer_if_modified(:down, from, to, options)
+    tempbody = capture("mktemp tmp/body.XXXXXXXXXX").strip
+    run_locally("rm -f #{tempbody.dump}")
+    download(from, tempbody)
+    assert_file_equals(tempbody, to, :via => :run_locally)
+
+## should not transfer without changes
+    2.times do
+      sleep(1)
+      tempto = capture("mktemp tmp/to.XXXXXXXXXX").strip
+      run_locally("rm -f #{tempto.dump}; cp -p #{to.dump} #{tempto.dump}")
+      transfer_if_modified(:down, from, to, options) # re-transfer
+      assert_file_equals(tempto, to, :via => :run_locally)
+      assert_timestamp_equals(tempto, to, :via => :run_locally)
+    end
+
+## should transfer if `from' is changed
+    tempfrom = capture("mktemp tmp/from.XXXXXXXXXX").strip
+    tempto = run_locally("mktemp tmp/to.XXXXXXXXXX").strip
+    run("for i in 0 1 2 3; do cat #{from.dump} >> #{tempfrom.dump}; done")
+    run_locally("rm -f #{tempto.dump}; cp -p #{to.dump} #{tempto.dump}")
+    transfer_if_modified(:down, tempfrom, to, options)
+    run_locally("test #{tempto.dump} -ot #{to.dump}") # check if `to' is overwritten
+  end
+
+  task(:test_transfer_if_modified_up) {
+    run_locally("rm -f tmp/foo; echo foo > tmp/foo")
+    _test_transfer_if_modified_up("tmp/foo", "tmp/rfoo")
+  }
+
+  task(:test_transfer_if_modified_up_with_stringio) {
+    _test_transfer_if_modified_up(StringIO.new("bar"), "tmp/rbar", :digest => :sha1)
+  }
+
+  task(:test_transfer_if_modified_down) {
+    run("rm -f tmp/foo; echo foo > tmp/foo")
+    _test_transfer_if_modified_down("tmp/foo", "tmp/lfoo", :digest => :sha1)
+  }
+
+  task(:test_transfer_if_modified_down_with_capistrano_host) {
+    run("rm -f tmp/bar; echo bar > tmp/bar")
+    _test_transfer_if_modified_down("tmp/bar", "tmp/lbar_$CAPISTRANO:HOST$")
+  }
+}
+
+def _test_install(frombody, tobody, options={})
+  from = capture("mktemp tmp/from.XXXXXXXXXX").strip
+  run("rm -f #{from.dump}")
+  put(frombody, from)
+
+  sleep(1)
+  to = capture("mktemp tmp/to.XXXXXXXXXX").strip
+  run("rm -f #{to.dump}")
+  if tobody
+    put(tobody, to)
+    orig_mode = options.delete(:orig_mode)
+    run("chmod #{orig_mode.to_s(8)} #{to.dump}") if orig_mode
+    orig_owner = options.delete(:orig_owner)
+    sudo("chown #{orig_owner} #{to.dump}") if orig_owner
+    orig_group = options.delete(:orig_group)
+    sudo("chgrp #{orig_group} #{to.dump}") if orig_group
+  end
+
+  tempfrom = capture("mktemp tmp/from2.XXXXXXXXXX").strip
+  run("rm -f #{tempfrom.dump}; #{sudo} cp -p #{from.dump} #{tempfrom.dump}")
+
+  tempto = capture("mktemp tmp/to2.XXXXXXXXXX").strip
+  if tobody
+    run("rm -f #{tempto.dump}; #{sudo} cp -p #{to.dump} #{tempto.dump}")
+  else
+    run("rm -f #{tempto.dump}")
+  end
+
+  method = ( options.delete(:install) || :always )
+  send(method == :if_modified ? :install_if_modified : :install, from, to, options)
+  yield(from, to, tempfrom, tempto)
+end
+
+namespace(:test_install) {
+  task(:default) {
+    methods.grep(/^test_/).each do |m|
+      send(m)
+    end
+  }
+  before "test_install", "test_install:setup"
+  after "test_install", "test_install:teardown"
+
+  task(:setup) {
+    run_locally("mkdir -p tmp")
+    run("mkdir -p tmp")
+  }
+
+  task(:teardown) {
+    run_locally("rm -rf tmp")
+    sudo("rm -rf tmp")
+  }
+
+  task(:test_if_not_modified) {
+    _test_install("foo", "foo") do |from, to, tempfrom, tempto|
+      assert_file_equals(tempto, to)
+      assert_timestamp_equals(tempfrom, to)
+      assert_timestamp_not_equals(tempto, to)
+      assert_file_mode_equals(tempto, to)
+    end
+  }
+
+  task(:test_if_modified) {
+    _test_install("foo", "bar", :digest => :sha1) do |from, to, tempfrom, tempto|
+      assert_file_equals(tempfrom, to)
+      assert_timestamp_equals(tempfrom, to)
+      assert_timestamp_not_equals(tempto, to)
+      assert_file_mode_equals(tempto, to)
+    end
+  }
+
+  task(:test_if_missing) {
+    _test_install("baz", nil) do |from, to, tempfrom, tempto|
+      assert_file_equals(tempfrom, to)
+      assert_timestamp_equals(tempfrom, to)
+    end
+  }
+
+  task(:test_with_mode) {
+    _test_install("foo", "bar", :mode => 0755) do |from, to, tempfrom, tempto|
+      assert_file_mode(0755, to)
+    end
+  }
+
+  task(:test_via_sudo) {
+    _test_install("bar", "baz", :orig_owner => 0, :via => :sudo) do |from, to, tempfrom, tempto|
+      assert_file_owner(0, to)
+    end
+  }
+
+  task(:test_with_mode_via_sudo) {
+    _test_install("bar", "baz", :mode => 0644, :orig_owner => 0, :via => :sudo) do |from, to, tempfrom, tempto|
+      assert_file_mode(0644, to)
+      assert_file_owner(0, to)
+    end
+  }
+
+  task(:test_preserve_original_mode) {
+    _test_install("foo", "bar", :orig_mode => 0600) do |from, to, tempfrom, tempto|
+      assert_file_mode(0600, to)
+    end
+  }
+
+  task(:test_ignore_original_mode) {
+    _test_install("foo", "bar", :orig_mode => 0400, :mode => 0700) do |from, to, tempfrom, tempto|
+      assert_file_mode(0700, to)
+    end
+  }
+
+  task(:test_preserve_original_owner_group_mode) {
+    _test_install("foo", "bar", :orig_owner => 0, :orig_group => 0, :orig_mode => 0640, :via => :sudo) do |from, to, tempfrom, tempto|
+      assert_file_mode(0640, to)
+      assert_file_owner(0, to)
+      assert_file_group(0, to)
+    end
+  }
+
+  task(:test_ignore_original_owner_group_mode) {
+    _test_install("bar", "baz", :owner => 0, :group => 0, :mode => 0640, :via => :sudo) do |from, to, tempfrom, tempto|
+      assert_file_mode(0640, to)
+      assert_file_owner(0, to)
+      assert_file_group(0, to)
+    end
+  }
+}
+
+namespace(:test_install_if_modified) {
+  task(:default) {
+    methods.grep(/^test_/).each do |m|
+      send(m)
+    end
+  }
+  before "test_install_if_modified", "test_install_if_modified:setup"
+  after "test_install_if_modified", "test_install_if_modified:teardown"
+
+  task(:setup) {
+    run_locally("mkdir -p tmp")
+    run("mkdir -p tmp")
+  }
+
+  task(:teardown) {
+    run_locally("rm -rf tmp")
+    sudo("rm -rf tmp")
+  }
+
+  task(:test_if_not_modified) {
+    _test_install("foo", "foo", :install => :if_modified) do |from, to, tempfrom, tempto|
+      assert_file_equals(tempto, to)
+      assert_timestamp_equals(tempto, to)
+      assert_timestamp_not_equals(tempfrom, to)
+      assert_file_mode_equals(tempto, to)
+    end
+  }
+
+  task(:test_if_modified) {
+    _test_install("foo", "bar", :install => :if_modified, :digest => :sha1) do |from, to, tempfrom, tempto|
+      assert_file_equals(tempfrom, to)
+      assert_timestamp_equals(tempfrom, to)
+      assert_timestamp_not_equals(tempto, to)
+      assert_file_mode_equals(tempto, to)
+    end
+  }
+
+  task(:test_if_missing) {
+    _test_install("baz", nil, :install => :if_modified) do |from, to, tempfrom, tempto|
+      assert_file_equals(tempfrom, to)
+      assert_timestamp_equals(tempfrom, to)
+    end
+  }
+
+  task(:test_with_mode) {
+    _test_install("foo", "bar", :install => :if_modified, :mode => 0755) do |from, to, tempfrom, tempto|
+      assert_file_mode(0755, to)
+    end
+  }
+
+  task(:test_via_sudo) {
+    _test_install("bar", "baz", :install => :if_modified, :orig_owner => 0, :via => :sudo) do |from, to, tempfrom, tempto|
+      assert_file_owner(0, to)
+    end
+  }
+
+  task(:test_with_mode_via_sudo) {
+    _test_install("bar", "baz", :install => :if_modified, :orig_owner => 0, :mode => 0644, :via => :sudo) do |from, to, tempfrom, tempto|
+      assert_file_mode(0644, to)
+      assert_file_owner(0, to)
+    end
+  }
+
+  task(:test_preserve_original_mode) {
+    _test_install("foo", "bar", :install => :if_modified, :orig_mode => 0600) do |from, to, tempfrom, tempto|
+      assert_file_mode(0600, to)
+    end
+  }
+
+  task(:test_ignore_original_mode) {
+    _test_install("foo", "bar", :install => :if_modified, :orig_mode => 0400, :mode => 0700) do |from, to, tempfrom, tempto|
+      assert_file_mode(0700, to)
+    end
+  }
+
+  task(:test_preserve_original_owner_group_mode) {
+    _test_install("foo", "bar", :install => :if_modified, :orig_owner => 0, :orig_group => 0, :orig_mode => 0640, :via => :sudo) do |from, to, tempfrom, tempto|
+      assert_file_mode(0640, to)
+      assert_file_owner(0, to)
+      assert_file_group(0, to)
+    end
+  }
+
+  task(:test_ignore_original_owner_group_mode) {
+    _test_install("bar", "baz", :install => :if_modified, :owner => 0, :group => 0, :mode => 0640, :via => :sudo) do |from, to, tempfrom, tempto|
+      assert_file_mode(0640, to)
+      assert_file_owner(0, to)
+      assert_file_group(0, to)
+    end
+  }
+}
+
+# vim:set ft=ruby sw=2 ts=2 :

data/test/precise64/.gitignore

@@ -0,0 +1,5 @@
+/.bundle
+/.vagrant
+/known_hosts
+/tmp
+/vendor

data/test/precise64/Capfile

@@ -0,0 +1,2 @@
+load "deploy"
+load "../config/deploy"

data/test/precise64/Gemfile

@@ -0,0 +1,2 @@
+source "https://rubygems.org"
+gemspec(:path => "../..")

data/test/precise64/Vagrantfile

@@ -0,0 +1,99 @@
+# -*- mode: ruby -*-
+# vi: set ft=ruby :
+
+Vagrant::Config.run do |config|
+  # All Vagrant configuration is done here. The most common configuration
+  # options are documented and commented below. For a complete reference,
+  # please see the online documentation at vagrantup.com.
+
+  # Every Vagrant virtual environment requires a box to build off of.
+  config.vm.box = "precise64"
+
+  # The url from where the 'config.vm.box' box will be fetched if it
+  # doesn't already exist on the user's system.
+  config.vm.box_url = "http://files.vagrantup.com/precise64.box"
+
+  # Boot with a GUI so you can see the screen. (Default is headless)
+  # config.vm.boot_mode = :gui
+
+  # Assign this VM to a host-only network IP, allowing you to access it
+  # via the IP. Host-only networks can talk to the host machine as well as
+  # any other machines on the same network, but cannot be accessed (through this
+  # network interface) by any external networks.
+  config.vm.network :hostonly, "192.168.33.10"
+
+  # Assign this VM to a bridged network, allowing you to connect directly to a
+  # network using the host's network device. This makes the VM appear as another
+  # physical device on your network.
+  # config.vm.network :bridged
+
+  # Forward a port from the guest to the host, which allows for outside
+  # computers to access the VM, whereas host only networking does not.
+  # config.vm.forward_port 80, 8080
+
+  # Share an additional folder to the guest VM. The first argument is
+  # an identifier, the second is the path on the guest to mount the
+  # folder, and the third is the path on the host to the actual folder.
+  # config.vm.share_folder "v-data", "/vagrant_data", "../data"
+
+  # Enable provisioning with Puppet stand alone.  Puppet manifests
+  # are contained in a directory path relative to this Vagrantfile.
+  # You will need to create the manifests directory and a manifest in
+  # the file precise-amd64.pp in the manifests_path directory.
+  #
+  # An example Puppet manifest to provision the message of the day:
+  #
+  # # group { "puppet":
+  # #   ensure => "present",
+  # # }
+  # #
+  # # File { owner => 0, group => 0, mode => 0644 }
+  # #
+  # # file { '/etc/motd':
+  # #   content => "Welcome to your Vagrant-built virtual machine!
+  # #               Managed by Puppet.\n"
+  # # }
+  #
+  # config.vm.provision :puppet do |puppet|
+  #   puppet.manifests_path = "manifests"
+  #   puppet.manifest_file  = "precise-amd64.pp"
+  # end
+
+  # Enable provisioning with chef solo, specifying a cookbooks path, roles
+  # path, and data_bags path (all relative to this Vagrantfile), and adding 
+  # some recipes and/or roles.
+  #
+  # config.vm.provision :chef_solo do |chef|
+  #   chef.cookbooks_path = "../my-recipes/cookbooks"
+  #   chef.roles_path = "../my-recipes/roles"
+  #   chef.data_bags_path = "../my-recipes/data_bags"
+  #   chef.add_recipe "mysql"
+  #   chef.add_role "web"
+  #
+  #   # You may also specify custom JSON attributes:
+  #   chef.json = { :mysql_password => "foo" }
+  # end
+
+  # Enable provisioning with chef server, specifying the chef server URL,
+  # and the path to the validation key (relative to this Vagrantfile).
+  #
+  # The Opscode Platform uses HTTPS. Substitute your organization for
+  # ORGNAME in the URL and validation key.
+  #
+  # If you have your own Chef Server, use the appropriate URL, which may be
+  # HTTP instead of HTTPS depending on your configuration. Also change the
+  # validation key to validation.pem.
+  #
+  # config.vm.provision :chef_client do |chef|
+  #   chef.chef_server_url = "https://api.opscode.com/organizations/ORGNAME"
+  #   chef.validation_key_path = "ORGNAME-validator.pem"
+  # end
+  #
+  # If you're using the Opscode platform, your validator client is
+  # ORGNAME-validator, replacing ORGNAME with your organization name.
+  #
+  # IF you have your own Chef Server, the default validation client name is
+  # chef-validator, unless you changed the configuration.
+  #
+  #   chef.validation_client_name = "ORGNAME-validator"
+end

data/test/precise64/run.sh

@@ -0,0 +1,7 @@
+#!/bin/sh -e
+
+bundle exec vagrant up
+bundle exec cap test_all
+bundle exec vagrant destroy -f
+
+# vim:set ft=sh :

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: capistrano-file-transfer-ext
 version: !ruby/object:Gem::Version
-  version: 0.0.4
+  version: 0.1.0
   prerelease: 
 platform: ruby
 authors:
@@ -9,7 +9,7 @@ authors:
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2013-03-07 00:00:00.000000000 Z
+date: 2013-03-14 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: capistrano
@@ -27,6 +27,54 @@ dependencies:
     - - ! '>='
       - !ruby/object:Gem::Version
         version: '0'
+- !ruby/object:Gem::Dependency
+  name: net-scp
+  requirement: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: 1.0.4
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: 1.0.4
+- !ruby/object:Gem::Dependency
+  name: net-ssh
+  requirement: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: 2.2.2
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: 2.2.2
+- !ruby/object:Gem::Dependency
+  name: vagrant
+  requirement: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: 1.0.6
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: 1.0.6
 description: A sort of utilities which helps you transferring files with Capistrano.
 email:
 - yamashita@geishatokyo.com
@@ -42,6 +90,17 @@ files:
 - capistrano-file-transfer-ext.gemspec
 - lib/capistrano/configuration/actions/file_transfer_ext.rb
 - lib/capistrano/configuration/actions/file_transfer_ext/version.rb
+- test/centos6-64/.gitignore
+- test/centos6-64/Capfile
+- test/centos6-64/Gemfile
+- test/centos6-64/Vagrantfile
+- test/centos6-64/run.sh
+- test/config/deploy.rb
+- test/precise64/.gitignore
+- test/precise64/Capfile
+- test/precise64/Gemfile
+- test/precise64/Vagrantfile
+- test/precise64/run.sh
 homepage: https://github.com/yyuu/capistrano-file-transfer-ext
 licenses: []
 post_install_message: 
@@ -66,4 +125,15 @@ rubygems_version: 1.8.23
 signing_key: 
 specification_version: 3
 summary: A sort of utilities which helps you transferring files with Capistrano.
-test_files: []
+test_files:
+- test/centos6-64/.gitignore
+- test/centos6-64/Capfile
+- test/centos6-64/Gemfile
+- test/centos6-64/Vagrantfile
+- test/centos6-64/run.sh
+- test/config/deploy.rb
+- test/precise64/.gitignore
+- test/precise64/Capfile
+- test/precise64/Gemfile
+- test/precise64/Vagrantfile
+- test/precise64/run.sh