require File.expand_path("../../spec_helper", File.dirname(__FILE__))
require 'tempfile'

describe Sprinkle::Installers::Transfer do
  include Sprinkle::Deployment

  before do
    @package = mock(Sprinkle::Package, :name => 'package')
    @empty = Proc.new { }
    @delivery = mock(Sprinkle::Deployment, :process => true)
		@source = 'source'
		@destination = 'destination'
    @installer = create_transfer(@source, @destination)
    @roles = []
    @deployment = deployment do
      delivery :capistrano
      installer do; prefix '/usr/bin'; end
    end
  end

  def create_transfer(source, dest, options={}, &block)
    i = Sprinkle::Installers::Transfer.new(@package, source, dest, options, &block)
    i.delivery = @delivery
		i
  end

  describe 'when created' do
    it 'should accept a source and destination to install' do
      @installer.source.should == @source
      @installer.destination.should == @destination
    end
  end

  describe 'during installation' do

    context 'single pre/post commands' do
      before do
        @installer = create_transfer @source, @destination do
          pre :install, 'op1'
          post :install, 'op2'
        end

        @delivery = @installer.delivery
      end

      it "should call the pre and post install commands around the file transfer" do
        @delivery.should_receive(:process).with(@package.name, ['op1'], @roles).and_return
        @delivery.should_receive(:transfer).and_return
        @delivery.should_receive(:process).with(@package.name, ['op2'], @roles).and_return
      end

      it "should call transfer with recursive defaulted to nil" do
        @delivery.should_receive(:process).and_return
        @delivery.should_receive(:transfer).with(@package.name, @source, @destination, @roles, nil)
      end

    end

    context 'multiple pre/post commands' do
      before do
        @installer = create_transfer @source, @destination do
          pre :install, 'op1', 'op1-1'
          post :install, 'op2', 'op2-1'
        end

        @delivery = @installer.delivery
      end

      it "should call the pre and post install commands around the file transfer" do
        @delivery.should_receive(:process).with(@package.name, ['op1', 'op1-1'], @roles).and_return
        @delivery.should_receive(:transfer).and_return
        @delivery.should_receive(:process).with(@package.name, ['op2', 'op2-1'], @roles).and_return
      end

    end

		after do
      @installer.process @roles
    end
  end

	describe "if the :render flag is true" do
		before do
      @installer = create_transfer @source, @destination, :render => true
			@delivery = @installer.delivery
			@delivery.stub!(:render_template_file)
    end

		it "should render the source file as a template to a tempfile" do
			@tempfile = Tempfile.new("foo")
			@installer.should_receive(:render_template_file).with(@source, anything, @package.name).and_return(@tempfile)
			@delivery.stub!(:transfer)
		end

		it "should call transfer with recursive set to false" do
			@tempfile = Tempfile.new("foo")
			@installer.should_receive(:render_template_file).with(@source, anything, @package.name).and_return(@tempfile)
			@delivery.should_receive(:transfer).with(@package.name, @tempfile.path, @destination, @roles, false).ordered.and_return
		end

		after do
      @installer.process @roles
    end
	end

	describe "if the :recursive flag is explicitly set to false" do
		before do
      @installer = create_transfer @source, @destination, :recursive => false
    end

		it "should call transfer with recursive set to false" do
			delivery = @installer.delivery
			delivery.should_receive(:transfer).with(@package.name, @source, @destination, @roles, false).ordered.and_return
		end

		after do
      @installer.process @roles
    end
	end
end