rvagrant 0.8.7.dev

data/test/unit/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/unit/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/unit/vagrant/util/counter_test.rb

@@ -0,0 +1,29 @@
+require "test_helper"
+
+class CounterUtilTest < Test::Unit::TestCase
+  setup do
+    @klass = Class.new do
+      extend Vagrant::Util::Counter
+    end
+  end
+
+  context "basic counter" do
+    should "get and update the counter" do
+      assert_equal 1, @klass.get_and_update_counter
+      assert_equal 2, @klass.get_and_update_counter
+    end
+  end
+
+  context "multiple classes with a counter" do
+    setup do
+      @klass2 = Class.new do
+        extend Vagrant::Util::Counter
+      end
+    end
+
+    should "not affect other classes" do
+      assert_equal 1, @klass.get_and_update_counter
+      assert_equal 1, @klass2.get_and_update_counter
+    end
+  end
+end

data/test/unit/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/unit/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)
+
+      Vagrant::Util::Platform.stubs(: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
+      Vagrant::Util::Platform.stubs(:windows?).returns(false)
+      assert_equal Vagrant::Util::Platform.tar_file_options, File::CREAT|File::EXCL|File::WRONLY
+    end
+  end
+end

data/test/unit/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

data/test/unit/vagrant/util/stacked_proc_runner_test.rb

@@ -0,0 +1,43 @@
+require "test_helper"
+
+class StackedProcRunnerUtilTest < Test::Unit::TestCase
+  class TestClass
+    include Vagrant::Util::StackedProcRunner
+  end
+
+  setup do
+    @instance = TestClass.new
+    @instance.proc_stack.clear
+  end
+
+  should "not run the procs right away" do
+    obj = mock("obj")
+    obj.expects(:foo).never
+    @instance.push_proc { |config| obj.foo }
+    @instance.push_proc { |config| obj.foo }
+    @instance.push_proc { |config| obj.foo }
+  end
+
+  should "run the blocks when run_procs! is ran" do
+    obj = mock("obj")
+    obj.expects(:foo).times(2)
+    @instance.push_proc { obj.foo }
+    @instance.push_proc { obj.foo }
+    @instance.run_procs!
+  end
+
+  should "run the blocks with the same arguments" do
+    passed_config = mock("config")
+    @instance.push_proc { |config| assert passed_config.equal?(config) }
+    @instance.push_proc { |config| assert passed_config.equal?(config) }
+    @instance.run_procs!(passed_config)
+  end
+
+  should "not clear the blocks after running" do
+    obj = mock("obj")
+    obj.expects(:foo).times(2)
+    @instance.push_proc { obj.foo }
+    @instance.run_procs!
+    @instance.run_procs!
+  end
+end

data/test/unit/vagrant/util/template_renderer_test.rb

@@ -0,0 +1,145 @@
+require "test_helper"
+
+class TemplateRendererUtilTest < Test::Unit::TestCase
+  context "initializing" do
+    should "set the template to the given argument" do
+      r = Vagrant::Util::TemplateRenderer.new("foo")
+      assert_equal "foo", r.template
+    end
+
+    should "set any additional variables" do
+      r = Vagrant::Util::TemplateRenderer.new("foo", {:bar => :baz})
+      assert_equal :baz, r.bar
+    end
+  end
+
+  context "rendering" do
+    setup do
+      @template = "foo"
+      @r = Vagrant::Util::TemplateRenderer.new(@template)
+      @r.stubs(:full_template_path).returns(@template + "!")
+
+      @contents = "bar"
+
+      @file = mock("file")
+      @file.stubs(:read).returns(@contents)
+      File.stubs(:open).yields(@file)
+    end
+
+    should "open the template file for reading" do
+      File.expects(:open).with(@r.full_template_path, 'r').once
+      @r.render
+    end
+
+    should "set the template to the file contents, render, then set it back" do
+      result = "bar"
+
+      template_seq = sequence("template_seq")
+      @r.expects(:template=).with(@file.read).in_sequence(template_seq)
+      @r.expects(:render_string).returns(result).in_sequence(template_seq)
+      @r.expects(:template=).with(@template).in_sequence(template_seq)
+      assert_equal result, @r.render
+    end
+
+    should "render the ERB file in the context of the renderer" do
+      result = "bar"
+      template = "<%= foo %>"
+      @r.foo = result
+      @file.expects(:read).returns(template)
+      assert_equal result, @r.render
+    end
+  end
+
+  context "rendering as string" do
+    setup do
+      @result = "foo"
+      @erb = mock("erb")
+      @erb.stubs(:result).returns(@result)
+
+      @r = Vagrant::Util::TemplateRenderer.new("foo")
+    end
+
+    should "simply render the template as a string" do
+      Erubis::Eruby.expects(:new).with(@r.template, :trim => true).returns(@erb)
+      @erb.expects(:result).returns(@result)
+      assert_equal @result, @r.render_string
+    end
+  end
+
+  context "the full template path" do
+    setup do
+      @template = "foo"
+      @r = Vagrant::Util::TemplateRenderer.new(@template)
+    end
+
+    should "be the ERB file in the templates directory" do
+      result = Vagrant.source_root.join("templates", "#{@template}.erb")
+      assert_equal result.to_s, @r.full_template_path
+    end
+
+    should "remove duplicate path separators" do
+      @r.template = "foo///bar"
+      result = Vagrant.source_root.join("templates", "foo", "bar.erb")
+      assert_equal result.to_s, @r.full_template_path
+    end
+  end
+
+  context "class methods" do
+    context "render_with method" do
+      setup do
+        @template = "foo"
+        @r = Vagrant::Util::TemplateRenderer.new(@template)
+        @r.stubs(:render)
+
+        @method = :rawr
+
+        Vagrant::Util::TemplateRenderer.stubs(:new).with(@template, {}).returns(@r)
+      end
+
+      should "use the second argument as the template" do
+        Vagrant::Util::TemplateRenderer.expects(:new).with(@template, {}).returns(@r)
+        Vagrant::Util::TemplateRenderer.render_with(@method, @template)
+      end
+
+      should "send in additional argument to the renderer" do
+        data = {:hey => :foo}
+        Vagrant::Util::TemplateRenderer.expects(:new).with(@template, data).returns(@r)
+        Vagrant::Util::TemplateRenderer.render_with(@method, @template, data)
+      end
+
+      should "yield a block if given with the renderer as the argument" do
+        @r.expects(:yielded=).with(true).once
+        Vagrant::Util::TemplateRenderer.render_with(@method, @template) do |r|
+          r.yielded = true
+        end
+      end
+
+      should "render the result using the given method" do
+        result = mock('result')
+        @r.expects(@method).returns(result)
+        assert_equal result, Vagrant::Util::TemplateRenderer.render_with(@method, @template)
+      end
+
+      should "convert the given method to a sym prior to calling" do
+        @r.expects(@method.to_sym).returns(nil)
+        Vagrant::Util::TemplateRenderer.render_with(@method.to_s, @template)
+      end
+    end
+
+    context "render method" do
+      should "call render_with the render! method" do
+        args = ["foo", "bar", "baz"]
+        Vagrant::Util::TemplateRenderer.expects(:render_with).with(:render, *args)
+        Vagrant::Util::TemplateRenderer.render(*args)
+      end
+    end
+
+    context "render_string method" do
+      should "call render_with the render! method" do
+        args = ["foo", "bar", "baz"]
+        Vagrant::Util::TemplateRenderer.expects(:render_with).with(:render_string, *args)
+        Vagrant::Util::TemplateRenderer.render_string(*args)
+      end
+    end
+  end
+end