fixed-vagrant 0.7.4.dev

data/test/vagrant/systems/base_test.rb

@@ -0,0 +1,18 @@
+require "test_helper"
+
+class BaseSystemTest < Test::Unit::TestCase
+  setup do
+    @klass = Vagrant::Systems::Base
+
+    @vm = mock("vm")
+    @instance = @klass.new(@vm)
+  end
+
+  should "provide access to the VM" do
+    assert_equal @vm, @instance.vm
+  end
+
+  should "error on preparing host only network" do
+    assert_raises(@klass::BaseError) { @instance.prepare_host_only_network }
+  end
+end

data/test/vagrant/systems/linux_test.rb

@@ -0,0 +1,114 @@
+require "test_helper"
+
+class LinuxSystemTest < Test::Unit::TestCase
+  setup do
+    @klass = Vagrant::Systems::Linux
+    @ssh = mock("ssh")
+    @mock_env = vagrant_env
+    @vm = mock("vm")
+    @vm.stubs(:env).returns(@mock_env)
+    @instance = @klass.new(@vm)
+  end
+
+  context "halting" do
+    setup do
+      @ssh_session = mock("ssh_session")
+      @ssh.stubs(:execute).yields(@ssh_session)
+      @vm.stubs(:ssh).returns(@ssh)
+
+      @real_vm = mock("real_vm")
+      @real_vm.stubs(:state).returns(:powered_off)
+      @vm.stubs(:vm).returns(@real_vm)
+    end
+
+    should "execute halt via SSH" do
+      @ssh_session.expects(:exec!).with("sudo halt").once
+      @instance.halt
+    end
+  end
+
+  context "mounting shared folders" do
+    setup do
+      @name = "foo"
+      @guestpath = "/bar"
+    end
+
+    should "create the dir, mount the folder, then set permissions" do
+      mount_seq = sequence("mount_seq")
+      @ssh.expects(:exec!).with("sudo mkdir -p #{@guestpath}").in_sequence(mount_seq)
+      @instance.expects(:mount_folder).with(@ssh, @name, @guestpath).in_sequence(mount_seq)
+      @ssh.expects(:exec!).with("sudo chown #{@vm.env.config.ssh.username} #{@guestpath}").in_sequence(mount_seq)
+
+      @instance.mount_shared_folder(@ssh, @name, @guestpath)
+    end
+  end
+
+  #-------------------------------------------------------------------
+  # "Private" methods tests
+  #-------------------------------------------------------------------
+  context "mounting the main folder" do
+    setup do
+      @name = "foo"
+      @guestpath = "bar"
+      @sleeptime = 0
+      @limit = 10
+
+      @success_return = false
+    end
+
+    def mount_folder
+      @instance.mount_folder(@ssh, @name, @guestpath, @sleeptime)
+    end
+
+    should "execute the proper mount command" do
+      @ssh.expects(:exec!).with("sudo mount -t vboxsf -o uid=`id -u #{@vm.env.config.ssh.username}`,gid=`id -g #{@vm.env.config.ssh.username}` #{@name} #{@guestpath}").returns(@success_return)
+      mount_folder
+    end
+
+    should "test type of text and text string to detect error" do
+      data = mock("data")
+      data.expects(:[]=).with(:result, !@success_return)
+
+      @ssh.expects(:exec!).yields(data, :stderr, "No such device").returns(@success_return)
+      mount_folder
+    end
+
+    should "test type of text and test string to detect success" do
+      data = mock("data")
+      data.expects(:[]=).with(:result, @success_return)
+
+      @ssh.expects(:exec!).yields(data, :stdout, "Nothing such device").returns(@success_return)
+      mount_folder
+    end
+
+    should "raise an ActionException if the command fails constantly" do
+      @ssh.expects(:exec!).times(@limit).returns(!@success_return)
+
+      assert_raises(Vagrant::Systems::Linux::LinuxError) {
+        mount_folder
+      }
+    end
+
+    should "not raise any exception if the command succeeded" do
+      @ssh.expects(:exec!).once.returns(@success_return)
+
+      assert_nothing_raised {
+        mount_folder
+      }
+    end
+
+    should "add uid AND gid to mount" do
+      uid = "foo"
+      gid = "bar"
+      env = vagrant_env(vagrantfile(<<-vf))
+        config.vm.shared_folder_uid = "#{uid}"
+        config.vm.shared_folder_gid = "#{gid}"
+      vf
+
+      @vm.stubs(:env).returns(env)
+
+      @ssh.expects(:exec!).with("sudo mount -t vboxsf -o uid=`id -u #{uid}`,gid=`id -g #{gid}` #{@name} #{@guestpath}").returns(@success_return)
+      mount_folder
+    end
+  end
+end

data/test/vagrant/ui_test.rb

@@ -0,0 +1,29 @@
+require "test_helper"
+
+class ShellUITest < Test::Unit::TestCase
+  setup do
+    @klass = Vagrant::UI::Shell
+    @shell = mock("shell")
+    @instance = @klass.new(vagrant_env, @shell)
+  end
+
+  context "prefixing with resource" do
+    should "prefix message with environment resource" do
+      @shell.expects(:say).with() do |message, color|
+        assert message =~ /\[#{@instance.env.resource}\]/
+        true
+      end
+
+      @instance.info("vagrant.errors.test_key")
+    end
+
+    should "not prefix the message if given false" do
+      @shell.expects(:say).with() do |message, color|
+        assert message !~ /\[#{@instance.env.resource}\]/
+        true
+      end
+
+      @instance.info("vagrant.errors.test_key", :prefix => false)
+    end
+  end
+end

data/test/vagrant/util/busy_test.rb

@@ -0,0 +1,106 @@
+require "test_helper"
+
+class BusyUtilTest < Test::Unit::TestCase
+  setup do
+    @klass = Vagrant::Util::Busy
+  end
+
+  context "registering" do
+    setup do
+      @callback = lambda { puts "FOO" }
+      Signal.stubs(:trap)
+    end
+
+    teardown do
+      @klass.registered.clear
+    end
+
+    should "trap the signal on the first registration" do
+      Signal.expects(:trap).with("INT").once
+      @klass.register(@callback)
+      @klass.register(lambda { puts "BAR" })
+    end
+
+    should "not register the same callback multiple times" do
+      @klass.register(@callback)
+      @klass.register(@callback)
+      @klass.register(@callback)
+      assert_equal 1, @klass.registered.length
+      assert_equal @callback, @klass.registered.first
+    end
+  end
+
+  context "unregistering" do
+    setup do
+      Signal.stubs(:trap)
+
+      @callback = lambda { puts "FOO" }
+    end
+
+    teardown do
+      @klass.registered.clear
+    end
+
+    should "remove the callback and set the trap to DEFAULT when removing final" do
+      @klass.register(@callback)
+      Signal.expects(:trap).with("INT", "DEFAULT").once
+      @klass.unregister(@callback)
+      assert @klass.registered.empty?
+    end
+
+    should "not reset signal trap if not final callback" do
+      @klass.register(@callback)
+      @klass.register(lambda { puts "BAR" })
+      Signal.expects(:trap).never
+      @klass.unregister(@callback)
+    end
+  end
+
+  context "marking for busy" do
+    setup do
+      @callback = lambda { "foo" }
+    end
+
+    should "register, call the block, then unregister" do
+      waiter = mock("waiting")
+      proc = lambda { waiter.ping! }
+
+      seq = sequence('seq')
+      @klass.expects(:register).with(@callback).in_sequence(seq)
+      waiter.expects(:ping!).in_sequence(seq)
+      @klass.expects(:unregister).with(@callback).in_sequence(seq)
+
+      @klass.busy(@callback, &proc)
+    end
+
+    should "unregister callback even if block raises exception" do
+      waiter = mock("waiting")
+      proc = lambda { waiter.ping! }
+
+      seq = sequence('seq')
+      @klass.expects(:register).with(@callback).in_sequence(seq)
+      waiter.expects(:ping!).raises(Exception.new("uh oh!")).in_sequence(seq)
+      @klass.expects(:unregister).with(@callback).in_sequence(seq)
+
+      assert_raises(Exception) { @klass.busy(@callback, &proc) }
+    end
+  end
+
+  context "firing callbacks" do
+    setup do
+      Signal.stubs(:trap)
+    end
+
+    teardown do
+      @klass.registered.clear
+    end
+
+    should "just call the registered callbacks" do
+      waiting = mock("waiting")
+      waiting.expects(:ping!).once
+
+      @klass.register(lambda { waiting.ping! })
+      @klass.fire_callbacks
+    end
+  end
+end

data/test/vagrant/util/hash_with_indifferent_access_test.rb

@@ -0,0 +1,39 @@
+require "test_helper"
+
+class HashWithIndifferentAccessUtilTest < Test::Unit::TestCase
+  setup do
+    @klass = Vagrant::Util::HashWithIndifferentAccess
+    @instance = @klass.new
+  end
+
+  should "be a hash" do
+    assert @instance.is_a?(Hash)
+  end
+
+  should "allow indifferent access when setting with a string" do
+    @instance["foo"] = "bar"
+    assert_equal "bar", @instance[:foo]
+  end
+
+  should "allow indifferent access when setting with a symbol" do
+    @instance[:foo] = "bar"
+    assert_equal "bar", @instance["foo"]
+  end
+
+  should "allow indifferent key lookup" do
+    @instance["foo"] = "bar"
+    assert @instance.key?(:foo)
+    assert @instance.has_key?(:foo)
+    assert @instance.include?(:foo)
+    assert @instance.member?(:foo)
+  end
+
+  should "forward up block to Hash if given to initializer" do
+    instance = @klass.new do |h,k|
+      h[k] = "foo"
+    end
+
+    assert_equal "foo", instance[:foo]
+    assert_equal "foo", instance["foo"]
+  end
+end

data/test/vagrant/util/plain_logger_test.rb

@@ -0,0 +1,17 @@
+require "test_helper"
+
+class PlainLoggerUtilTest < Test::Unit::TestCase
+  setup do
+    @klass = Vagrant::Util::PlainLogger
+    @instance = @klass.new(nil)
+  end
+
+  should "inherit from the standard logger" do
+    assert @instance.is_a?(::Logger)
+  end
+
+  should "just add a newline to the message" do
+    msg = "foo bar baz"
+    assert_equal "#{msg}\n", @instance.format_message("1", "2", "3", msg)
+  end
+end

data/test/vagrant/util/platform_test.rb

@@ -0,0 +1,18 @@
+require "test_helper"
+
+class PlatformTest < Test::Unit::TestCase
+  context "file options" do
+    should "include add binary bit to options on windows platform" do
+      # This constant is not defined on non-windows platforms, so define it here
+      File::BINARY = 4096 unless defined?(File::BINARY)
+
+      Mario::Platform.expects(:windows?).returns(true)
+      assert_equal Vagrant::Util::Platform.tar_file_options, File::CREAT|File::EXCL|File::WRONLY|File::BINARY
+    end
+
+    should "not include binary bit on other platforms" do
+      Mario::Platform.expects(:windows?).returns(false)
+      assert_equal Vagrant::Util::Platform.tar_file_options, File::CREAT|File::EXCL|File::WRONLY
+    end
+  end
+end

data/test/vagrant/util/resource_logger_test.rb

@@ -0,0 +1,78 @@
+require "test_helper"
+
+class ResourceLoggerUtilTest < Test::Unit::TestCase
+  setup do
+    @klass = Vagrant::Util::ResourceLogger
+  end
+
+  context "singleton logger" do
+    setup do
+      @klass.reset_singleton_logger!
+
+      @result = mock("result")
+    end
+
+    should "return a nil plain logger if the environment is not loaded" do
+      env = vagrant_env
+      env.stubs(:loaded?).returns(false)
+
+      Vagrant::Util::PlainLogger.expects(:new).with(nil).returns(@result)
+      assert_equal @result, @klass.singleton_logger(env)
+    end
+
+    should "return a logger with the output file set if environment is ready" do
+      env = vagrant_env
+
+      Vagrant::Util::PlainLogger.expects(:new).returns(@result).with() do |path|
+        assert path.to_s =~ /logs/
+        true
+      end
+
+      assert_equal @result, @klass.singleton_logger(env)
+    end
+
+    should "only load the logger once" do
+      env = vagrant_env
+
+      Vagrant::Util::PlainLogger.expects(:new).with(anything).returns(@result)
+      assert_equal @result, @klass.singleton_logger(env)
+      assert_equal @result, @klass.singleton_logger(env)
+      assert_equal @result, @klass.singleton_logger(env)
+    end
+  end
+
+  context "initialization" do
+    should "setup the logger and attributes" do
+      env = vagrant_env
+      resource = mock("resource")
+      result = mock("result")
+
+      @klass.expects(:singleton_logger).with(env).returns(result)
+      instance = @klass.new(resource, env)
+      assert_equal resource, instance.resource
+      assert_equal env, instance.env
+      assert_equal result, instance.logger
+    end
+  end
+
+  context "with an instance" do
+    setup do
+      @resource = "foo"
+      @env = vagrant_env
+      @logger = mock("logger")
+
+      @klass.stubs(:singleton_logger).returns(@logger)
+      @instance = @klass.new(@resource, @env)
+    end
+
+    context "logging methods" do
+      [:debug, :info, :error, :fatal].each do |method|
+        should "log with the proper format on #{method}" do
+          message = "bar"
+          @logger.expects(method).with("[#{@resource}] #{message}").once
+          @instance.send(method, message)
+        end
+      end
+    end
+  end
+end

data/test/vagrant/util/retryable_test.rb

@@ -0,0 +1,50 @@
+require "test_helper"
+
+class RetryableUtilTest < Test::Unit::TestCase
+  setup do
+    @klass = Class.new do
+      extend Vagrant::Util::Retryable
+    end
+  end
+
+  should "retry specified number of times if exception is raised" do
+    proc = mock("proc")
+    proc.expects(:call).twice
+
+    assert_raises(RuntimeError) {
+      @klass.retryable(:tries => 2, :on => RuntimeError) do
+        proc.call
+        raise "An error"
+      end
+    }
+  end
+
+  should "only retry on specified exception" do
+    proc = mock("proc")
+    proc.expects(:call).once
+
+    assert_raises(StandardError) {
+      @klass.retryable(:tries => 5, :on => RuntimeError) do
+        proc.call
+        raise StandardError.new
+      end
+    }
+  end
+
+  should "retry on multiple exceptions given" do
+    proc = mock("proc")
+    proc.expects(:call).twice
+
+    assert_raises(StandardError) {
+      @klass.retryable(:tries => 2, :on => [StandardError, RuntimeError]) do
+        proc.call
+        raise StandardError
+      end
+    }
+  end
+
+  should "return the value of the block" do
+    result = @klass.retryable { 7 }
+    assert_equal 7, result
+  end
+end