vaguerent/test/vagrant/actions/vm/shared_folders_test.rb

require File.join(File.dirname(__FILE__), '..', '..', '..', 'test_helper')

class SharedFoldersActionTest < Test::Unit::TestCase
  setup do
    @runner, @vm, @action = mock_action(Vagrant::Actions::VM::SharedFolders)
    @runner.stubs(:system).returns(linux_system(@vm))
  end

  def stub_shared_folders
    env = mock_environment do |config|
      config.vm.shared_folders.clear

      if block_given?
        yield config
      else
        folders = [%w{foo fooguest foohost}, %w{bar barguest barhost}]
        folders.each do |data|
          config.vm.share_folder(*data)
        end
      end
    end

    @runner.stubs(:env).returns(env)
    env.config.vm.shared_folders
  end

  context "before boot" do
    should "clear folders and create metadata, in order" do
      before_seq = sequence("before")
      @action.expects(:clear_shared_folders).once.in_sequence(before_seq)
      @action.expects(:create_metadata).once.in_sequence(before_seq)
      @action.before_boot
    end
  end

  context "after boot" do
    should "mount folders then setup unison" do
      seq = sequence("after")
      @action.expects(:mount_shared_folders).once.in_sequence(seq)
      @action.expects(:setup_unison).once.in_sequence(seq)
      @action.after_boot
    end
  end

  context "collecting shared folders" do
    setup do
      File.stubs(:expand_path).returns("baz")
    end

    should "return a hash of the shared folders" do
      data = {
        "foo" => %W[bar baz],
        "bar" => %W[foo baz]
      }

      stub_shared_folders do |config|
        data.each do |name, value|
          config.vm.share_folder(name, *value)
        end
      end

      result = @action.shared_folders
      assert_equal data.length, result.length
      data.each do |name, value|
        guest, host = value
        assert_equal guest, result[name][:guestpath]
        assert_equal host, result[name][:hostpath]
      end
    end

    should "append sync suffix if sync enabled to a folder" do
      name = "foo"
      guest = "bar"
      host = "baz"

      stub_shared_folders do |config|
        config.vm.share_folder(name, guest, host, :sync => true)
      end

      result = @action.shared_folders
      assert_equal "#{guest}#{@runner.env.config.unison.folder_suffix}", result[name][:guestpath]
      assert_equal guest, result[name][:original][:guestpath]
    end

    should "not destroy original hash" do
      @folders = stub_shared_folders do |config|
        config.vm.share_folder("foo", "bar", "baz", :sync => true)
      end

      folder = @folders["foo"].dup

      @action.shared_folders
      assert_equal folder, @runner.env.config.vm.shared_folders["foo"]
    end
  end

  context "unison shared folders" do
    setup do
      @folders = stub_shared_folders do |config|
        config.vm.share_folder("foo", "bar", "baz", :sync => true)
        config.vm.share_folder("bar", "foo", "baz")
      end
    end

    should "only return the folders marked for syncing" do
      result = @action.unison_folders
      assert_equal 1, result.length
      assert result.has_key?("foo")
      assert !result.has_key?("bar")
    end
  end

  context "clearing shared folders" do
    setup do
      @shared_folder = mock("shared_folder")
      @shared_folders = [@shared_folder]
      @vm.stubs(:shared_folders).returns(@shared_folders)
    end

    should "call destroy on each shared folder then reload" do
      destroy_seq = sequence("destroy")
      @shared_folders.each do |sf|
        sf.expects(:destroy).once.in_sequence(destroy_seq)
      end

      @runner.expects(:reload!).once.in_sequence(destroy_seq)
      @action.clear_shared_folders
    end

    should "do nothing if no shared folders existed" do
      @shared_folders.clear
      @runner.expects(:reload!).never
      @action.clear_shared_folders
    end
  end

  context "setting up shared folder metadata" do
    setup do
      stub_shared_folders
    end

    should "add all shared folders to the VM" do
      shared_folders = []
      data = %W[foo bar]
      shared_folders.expects(:<<).times(data.length).with() do |sf|
        hostpath = File.expand_path("#{sf.name}host", @runner.env.root_path)
        assert data.include?(sf.name)
        assert_equal hostpath, sf.host_path
        true
      end

      @vm.stubs(:shared_folders).returns(shared_folders)
      @vm.expects(:save).once

      @action.create_metadata
    end
  end

  context "mounting the shared folders" do
    setup do
      @folders = stub_shared_folders
      @ssh = mock("ssh")
      @runner.ssh.stubs(:execute).yields(@ssh)
      @runner.system.stubs(:mount_shared_folder)
    end

    should "mount all shared folders to the VM" do
      mount_seq = sequence("mount_seq")
      @folders.each do |name, data|
        @runner.system.expects(:mount_shared_folder).with(@ssh, name, data[:guestpath]).in_sequence(mount_seq)
      end

      @action.mount_shared_folders
    end
  end

  context "setting up unison" do
    setup do
      @ssh = mock("ssh")
      @runner.ssh.stubs(:execute).yields(@ssh)

      @folders = stub_shared_folders do |config|
        config.vm.share_folder("foo", "bar", "baz", :sync => true)
        config.vm.share_folder("bar", "foo", "baz")
      end
    end

    should "do nothing if unison folders is empty" do
      @action.stubs(:unison_folders).returns({})
      @runner.ssh.expects(:execute).never
      @action.setup_unison
    end

    should "prepare unison then create for each folder" do
      seq = sequence("unison seq")
      @runner.system.expects(:prepare_unison).with(@ssh).once.in_sequence(seq)
      @action.unison_folders.each do |name, data|
        if data[:sync]
          @runner.system.expects(:create_unison).with do |ssh, opts|
            assert_equal @ssh, ssh
            assert_equal data, opts

            true
          end
        end
      end

      @action.setup_unison
    end
  end
end