vaguerent/test/vagrant/commands/package_test.rb
Mitchell Hashimoto b3277e015f Package subcommand
2010-04-13 16:46:51 -07:00

85 lines
2.4 KiB
Ruby

require File.join(File.dirname(__FILE__), '..', '..', 'test_helper')
class CommandsPackageTest < Test::Unit::TestCase
setup do
@klass = Vagrant::Commands::Package
@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
setup do
@persisted_vm.stubs(:package)
@persisted_vm.stubs(:powered_off?).returns(true)
end
context "with no base specified" do
should "require a persisted vm" do
@env.expects(:require_persisted_vm).once
@instance.execute
end
end
context "with base specified" do
setup do
@vm = mock("vm")
Vagrant::VM.stubs(:find).with(@name).returns(@vm)
@vm.stubs(:env=).with(@env)
@env.stubs(:vm=)
@name = "bar"
end
should "find the given base and set it on the env" do
Vagrant::VM.expects(:find).with(@name).returns(@vm)
@vm.expects(:env=).with(@env)
@env.expects(:vm=).with(@vm)
@instance.execute(["foo", "--base", @name])
end
should "error if the VM is not found" do
Vagrant::VM.expects(:find).with(@name).returns(nil)
@instance.expects(:error_and_exit).with(:vm_base_not_found, :name => @name).once
@instance.execute(["foo", "--base", @name])
end
end
context "shared (with and without base specified)" do
should "error and exit if the VM is not powered off" do
@persisted_vm.stubs(:powered_off?).returns(false)
@instance.expects(:error_and_exit).with(:vm_power_off_to_package).once
@persisted_vm.expects(:package).never
@instance.execute
end
should "call package on the persisted VM" do
@persisted_vm.expects(:package).once
@instance.execute
end
should "pass the out path and include_files to the package method" do
out_path = mock("out_path")
include_files = "foo"
@persisted_vm.expects(:package).with(out_path, [include_files]).once
@instance.execute([out_path, "--include", include_files])
end
should "default to an empty array when not include_files are specified" do
out_path = mock("out_path")
@persisted_vm.expects(:package).with(out_path, []).once
@instance.execute([out_path])
end
end
end
end