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

class CommandsSSHTest < Test::Unit::TestCase
  setup do
    @klass = Vagrant::Commands::SSH

    @env = mock_environment
    @instance = @klass.new(@env)
  end

  context "executing" do
    should "connect to the given argument" do
      @instance.expects(:ssh_connect).with("foo").once
      @instance.execute(["foo"])
    end

    should "connect with nil name if none is given" do
      @instance.expects(:ssh_connect).with(nil).once
      @instance.execute
    end
  end

  context "ssh connecting" do
    setup do
      @vm = mock("vm")
      @vm.stubs(:created?).returns(true)

      @vms = {:bar => @vm}
      @env.stubs(:vms).returns(@vms)
      @env.stubs(:multivm?).returns(false)
    end

    should "error and exit if no VM is specified and multivm and no primary VM" do
      @env.stubs(:multivm?).returns(true)
      @env.stubs(:primary_vm).returns(nil)
      @instance.expects(:error_and_exit).with(:ssh_multivm).once
      @instance.ssh_connect(nil)
    end

    should "use the primary VM if it exists and no name is specified" do
      vm = mock("vm")
      ssh = mock("ssh")
      vm.stubs(:created?).returns(true)
      vm.stubs(:ssh).returns(ssh)

      @env.stubs(:multivm?).returns(true)
      @env.stubs(:primary_vm).returns(vm)
      ssh.expects(:connect).once
      @instance.ssh_connect(nil)
    end

    should "error and exit if VM is nil" do
      @instance.expects(:error_and_exit).with(:unknown_vm, :vm => :foo).once
      @instance.ssh_connect(:foo)
    end

    should "error and exit if VM isn't created" do
      @vm.stubs(:created?).returns(false)
      @instance.expects(:error_and_exit).with(:environment_not_created).once
      @instance.ssh_connect(:bar)
    end

    should "ssh connect" do
      ssh = mock("ssh")
      @vm.stubs(:ssh).returns(ssh)
      ssh.expects(:connect)

      @instance.ssh_connect(:bar)
    end
  end
end