105 lines
3.0 KiB
Ruby
105 lines
3.0 KiB
Ruby
require File.join(File.dirname(__FILE__), '..', '..', 'test_helper')
|
|
|
|
class CommandsPackageTest < Test::Unit::TestCase
|
|
setup do
|
|
@klass = Vagrant::Commands::Package
|
|
|
|
@env = mock_environment
|
|
@instance = @klass.new(@env)
|
|
end
|
|
|
|
context "executing" do
|
|
should "package base if a base is given" do
|
|
@instance.expects(:package_base).once
|
|
@instance.execute(["--base","foo"])
|
|
end
|
|
|
|
should "package single if no name is given" do
|
|
@instance.expects(:package_single).with(nil).once
|
|
@instance.execute
|
|
end
|
|
|
|
should "package single if a name is given" do
|
|
@instance.expects(:package_single).with("foo").once
|
|
@instance.execute(["foo"])
|
|
end
|
|
end
|
|
|
|
context "packaging base" do
|
|
should "error and exit if no VM is found" do
|
|
Vagrant::VM.expects(:find).with("foo", @instance.env).returns(nil)
|
|
@instance.expects(:error_and_exit).with(:vm_base_not_found, :name => "foo").once
|
|
@instance.execute(["--base", "foo"])
|
|
end
|
|
|
|
should "package the VM like any other VM" do
|
|
vm = mock("vm")
|
|
Vagrant::VM.expects(:find).with("foo", @instance.env).returns(vm)
|
|
@instance.expects(:package_vm).with(vm).once
|
|
@instance.execute(["--base", "foo"])
|
|
end
|
|
end
|
|
|
|
context "packaging a single VM" 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 name is given in a multi-vm env" do
|
|
@env.stubs(:multivm?).returns(true)
|
|
@instance.expects(:error_and_exit).with(:package_multivm).once
|
|
@instance.package_single(nil)
|
|
end
|
|
|
|
should "error and exit if the VM doesn't exist" do
|
|
@instance.expects(:error_and_exit).with(:unknown_vm, :vm => :foo).once
|
|
@instance.package_single(:foo)
|
|
end
|
|
|
|
should "error and exit if the VM is not created" do
|
|
@vm.stubs(:created?).returns(false)
|
|
@instance.expects(:error_and_exit).with(:environment_not_created).once
|
|
@instance.package_single(:bar)
|
|
end
|
|
|
|
should "use the first VM is no name is given in a single VM environment" do
|
|
@instance.expects(:package_vm).with(@vm).once
|
|
@instance.package_single(nil)
|
|
end
|
|
|
|
should "package the VM" do
|
|
@instance.expects(:package_vm).with(@vm).once
|
|
@instance.package_single(:bar)
|
|
end
|
|
end
|
|
|
|
context "packaging a VM" do
|
|
setup do
|
|
@vm = mock("vm")
|
|
@vm.stubs(:powered_off?).returns(true)
|
|
|
|
@options = {}
|
|
@instance.stubs(:options).returns(@options)
|
|
end
|
|
|
|
should "error and exit if VM is not powered off" do
|
|
@vm.stubs(:powered_off?).returns(false)
|
|
@instance.expects(:error_and_exit).with(:vm_power_off_to_package).once
|
|
@instance.package_vm(@vm)
|
|
end
|
|
|
|
should "package the VM with the proper arguments" do
|
|
@options[:output] = "foo.box"
|
|
@options[:include] = :bar
|
|
|
|
@vm.expects(:package).with(@options[:output], @options[:include]).once
|
|
@instance.package_vm(@vm)
|
|
end
|
|
end
|
|
end
|