vagrantup 0.5.2 → 0.5.3

data/test/vagrant/commands/reload_test.rb

@@ -0,0 +1,47 @@
+require "test_helper"
+
+class CommandsReloadTest < Test::Unit::TestCase
+  setup do
+    @klass = Vagrant::Commands::Reload
+
+    @env = mock_environment
+    @env.stubs(:require_root_path)
+    @instance = @klass.new(@env)
+  end
+
+  context "executing" do
+    should "call all or single for the method" do
+      seq = sequence("seq")
+      @env.expects(:require_root_path).in_sequence(seq)
+      @instance.expects(:all_or_single).with([], :reload).in_sequence(seq)
+      @instance.execute
+    end
+  end
+
+  context "reloading a single VM" do
+    setup do
+      @foo_vm = mock("vm")
+      @foo_vm.stubs(:env).returns(@env)
+      vms = { :foo => @foo_vm }
+      @env.stubs(:vms).returns(vms)
+    end
+
+    should "error and exit if the VM doesn't exist" do
+      @env.stubs(:vms).returns({})
+      @instance.expects(:error_and_exit).with(:unknown_vm, :vm => :foo).once
+      @instance.reload_single(:foo)
+    end
+
+    should "reload if its created" do
+      @foo_vm.stubs(:created?).returns(true)
+      @foo_vm.expects(:reload).once
+      @instance.execute(["foo"])
+    end
+
+    should "do nothing if its not created" do
+      @foo_vm.stubs(:created?).returns(false)
+      @foo_vm.expects(:reload).never
+      @instance.reload_single(:foo)
+    end
+  end
+end

data/test/vagrant/commands/resume_test.rb

@@ -0,0 +1,44 @@
+require "test_helper"
+
+class CommandsResumeTest < Test::Unit::TestCase
+  setup do
+    @klass = Vagrant::Commands::Resume
+
+    @env = mock_environment
+    @instance = @klass.new(@env)
+  end
+
+  context "executing" do
+    should "call all or single for the method" do
+      @instance.expects(:all_or_single).with([], :resume)
+      @instance.execute
+    end
+  end
+
+  context "resuming a single VM" do
+    setup do
+      @foo_vm = mock("vm")
+      @foo_vm.stubs(:env).returns(@env)
+      vms = { :foo => @foo_vm }
+      @env.stubs(:vms).returns(vms)
+    end
+
+    should "error and exit if the VM doesn't exist" do
+      @env.stubs(:vms).returns({})
+      @instance.expects(:error_and_exit).with(:unknown_vm, :vm => :foo).once
+      @instance.resume_single(:foo)
+    end
+
+    should "resume if its created" do
+      @foo_vm.stubs(:created?).returns(true)
+      @foo_vm.expects(:resume).once
+      @instance.execute(["foo"])
+    end
+
+    should "do nothing if its not created" do
+      @foo_vm.stubs(:created?).returns(false)
+      @foo_vm.expects(:resume).never
+      @instance.resume_single(:foo)
+    end
+  end
+end

data/test/vagrant/commands/ssh_config_test.rb

@@ -0,0 +1,77 @@
+require "test_helper"
+
+class CommandsSSHConfigTest < Test::Unit::TestCase
+  setup do
+    @klass = Vagrant::Commands::SSHConfig
+
+    @env = mock_environment
+    @env.stubs(:require_root_path)
+    @instance = @klass.new(@env)
+  end
+
+  context "executing" do
+    setup do
+      @instance.stubs(:show_single)
+    end
+
+    should "require root path" do
+      @env.expects(:require_root_path).once
+      @instance.execute
+    end
+
+    should "call show_single with given argument" do
+      @instance.expects(:show_single).with("foo").once
+      @instance.execute(["foo"])
+    end
+  end
+
+  context "showing a single entry" do
+    setup do
+      @ssh = mock("ssh")
+      @ssh.stubs(:port).returns(2197)
+
+      @bar = mock("vm")
+      @bar.stubs(:env).returns(mock_environment)
+      @bar.stubs(:ssh).returns(@ssh)
+
+      @vms = {:bar => @bar}
+      @env.stubs(:multivm?).returns(true)
+      @env.stubs(:vms).returns(@vms)
+
+      @data = {
+        :host_key => "vagrant",
+        :ssh_user => @bar.env.config.ssh.username,
+        :ssh_port => @bar.ssh.port,
+        :private_key_path => @bar.env.config.ssh.private_key_path
+      }
+
+      @instance.stubs(:puts)
+    end
+
+    should "error if name is nil and multivm" do
+      @env.stubs(:multivm?).returns(true)
+      @instance.expects(:error_and_exit).with(:ssh_config_multivm).once
+      @instance.show_single(nil)
+    end
+
+    should "error if the VM is not found" do
+      @instance.expects(:error_and_exit).with(:unknown_vm, :vm => "foo").once
+      @instance.show_single("foo")
+    end
+
+    should "output rendered template" do
+      result = mock("result")
+      Vagrant::Util::TemplateRenderer.expects(:render).with("ssh_config", @data).returns(result)
+
+      @instance.expects(:puts).with(result).once
+      @instance.show_single(:bar)
+    end
+
+    should "render with the given host name if given" do
+      host = "foo"
+      @data[:host_key] = host
+      Vagrant::Util::TemplateRenderer.expects(:render).with("ssh_config", @data)
+      @instance.execute(["bar", "--host", host])
+    end
+  end
+end

data/test/vagrant/commands/ssh_test.rb

@@ -0,0 +1,129 @@
+require "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
+
+    should "execute if commands are given" do
+      @env.stubs(:vms).returns(:foo => mock("foo"))
+      @instance.expects(:ssh_execute).with("foo", @env.vms[:foo]).once
+      @instance.execute(["foo","-e","bar"])
+    end
+
+    should "execute over every VM if none is specified with a command" do
+      vms = {
+        :foo => mock("foo"),
+        :bar => mock("bar")
+      }
+
+      @env.stubs(:vms).returns(vms)
+      vms.each do |key, vm|
+        @instance.expects(:ssh_execute).with(key, vm).once
+      end
+
+      @instance.execute(["--execute", "bar"])
+    end
+  end
+
+  context "ssh executing" do
+    setup do
+      @name = :foo
+
+      @ssh_conn = mock("connection")
+      @ssh_conn.stubs(:exec!)
+
+      @ssh = mock("ssh")
+      @ssh.stubs(:execute).yields(@ssh_conn)
+
+      @vm = mock("vm")
+      @vm.stubs(:created?).returns(true)
+      @vm.stubs(:ssh).returns(@ssh)
+      @vm.stubs(:env).returns(@env)
+    end
+
+    should "error and exit if invalid VM given" do
+      @instance.expects(:error_and_exit).with(:unknown_vm, :vm => @name).once
+      @instance.ssh_execute(@name, nil)
+    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_execute(@name, @vm)
+    end
+
+    should "execute each command" do
+      commands = [:a,:b,:c]
+      @instance.stubs(:options).returns(:execute => commands)
+      commands.each do |cmd|
+        @ssh_conn.expects(:exec!).with(cmd).once
+      end
+
+      @instance.ssh_execute(@name, @vm)
+    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

data/test/vagrant/commands/status_test.rb

@@ -0,0 +1,40 @@
+require "test_helper"
+
+class CommandsStatusTest < Test::Unit::TestCase
+  setup do
+    @klass = Vagrant::Commands::Status
+
+    @persisted_vm = mock("persisted_vm")
+    @persisted_vm.stubs(:execute!)
+
+    @env = mock_environment
+    @env.stubs(:require_persisted_vm)
+    @env.stubs(:vm).returns(@persisted_vm)
+
+    @instance = @klass.new(@env)
+  end
+
+  context "executing" do
+    should "show local status by default" do
+      @instance.expects(:show_local_status).once
+      @instance.expects(:show_global_status).never
+      @instance.execute
+    end
+
+    should "show global status if flagged" do
+      @instance.expects(:show_local_status).never
+      @instance.expects(:show_global_status).once
+      @instance.execute(["--global"])
+    end
+
+    should "show help if too many args are given" do
+      @instance.expects(:show_help).once
+      @instance.execute(["1","2","3"])
+    end
+
+    should "pass the VM name to local status if given" do
+      @instance.expects(:show_local_status).with("foo").once
+      @instance.execute(["foo"])
+    end
+  end
+end

data/test/vagrant/commands/suspend_test.rb

@@ -0,0 +1,44 @@
+require "test_helper"
+
+class CommandsSuspendTest < Test::Unit::TestCase
+  setup do
+    @klass = Vagrant::Commands::Suspend
+
+    @env = mock_environment
+    @instance = @klass.new(@env)
+  end
+
+  context "executing" do
+    should "call all or single for the method" do
+      @instance.expects(:all_or_single).with([], :suspend)
+      @instance.execute
+    end
+  end
+
+  context "suspending a single VM" do
+    setup do
+      @foo_vm = mock("vm")
+      @foo_vm.stubs(:env).returns(@env)
+      vms = { :foo => @foo_vm }
+      @env.stubs(:vms).returns(vms)
+    end
+
+    should "error and exit if the VM doesn't exist" do
+      @env.stubs(:vms).returns({})
+      @instance.expects(:error_and_exit).with(:unknown_vm, :vm => :foo).once
+      @instance.suspend_single(:foo)
+    end
+
+    should "suspend if its created" do
+      @foo_vm.stubs(:created?).returns(true)
+      @foo_vm.expects(:suspend).once
+      @instance.execute(["foo"])
+    end
+
+    should "do nothing if its not created" do
+      @foo_vm.stubs(:created?).returns(false)
+      @foo_vm.expects(:suspend).never
+      @instance.suspend_single(:foo)
+    end
+  end
+end

data/test/vagrant/commands/up_test.rb

@@ -0,0 +1,49 @@
+require "test_helper"
+
+class CommandsUpTest < Test::Unit::TestCase
+  setup do
+    @klass = Vagrant::Commands::Up
+
+    @env = mock_environment
+    @env.stubs(:require_root_path)
+    @instance = @klass.new(@env)
+  end
+
+  context "executing" do
+    should "call all or single for the method" do
+      seq = sequence("seq")
+      @env.expects(:require_root_path).in_sequence(seq)
+      @instance.expects(:all_or_single).with([], :up).in_sequence(seq)
+      @instance.execute
+    end
+  end
+
+  context "upping a single VM" do
+    setup do
+      @vm = mock("vm")
+      @vm.stubs(:env).returns(@env)
+
+      @vms = {:foo => @vm}
+      @env.stubs(:vms).returns(@vms)
+    end
+
+    should "error and exit if the VM doesn't exist" do
+      @env.stubs(:vms).returns({})
+      @instance.expects(:error_and_exit).with(:unknown_vm, :vm => :foo).once
+      @instance.up_single(:foo)
+    end
+
+    should "start created VMs" do
+      @vm.stubs(:created?).returns(true)
+      @vm.expects(:start).once
+      @instance.up_single(:foo)
+    end
+
+    should "up non-created VMs" do
+      @vm.stubs(:created?).returns(false)
+      @vm.expects(:up).once
+      @vm.expects(:start).never
+      @instance.up_single(:foo)
+    end
+  end
+end