vagrant 0.5.2 → 0.5.3

data/test/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/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 = File.join(PROJECT_ROOT, "templates", "#{@template}.erb")
+      assert_equal result, @r.full_template_path
+    end
+
+    should "remove duplicate path separators" do
+      @r.template = "foo///bar"
+      result = File.join(PROJECT_ROOT, "templates", "foo", "bar.erb")
+      assert_equal result, @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

data/test/vagrant/util/translator_test.rb

@@ -0,0 +1,61 @@
+require "test_helper"
+
+class TranslatorUtilTest < Test::Unit::TestCase
+  include Vagrant::Util
+
+  setup do
+    @klass = Translator
+  end
+
+  context "loading the errors from the YML" do
+    setup do
+      YAML.stubs(:load_file)
+      @klass.reset!
+    end
+
+    should "load the file initially, then never again unless reset" do
+      YAML.expects(:load_file).with(File.join(PROJECT_ROOT, "templates", "strings.yml")).once
+      @klass.strings
+      @klass.strings
+      @klass.strings
+      @klass.strings
+    end
+
+    should "reload if reset! is called" do
+      YAML.expects(:load_file).with(File.join(PROJECT_ROOT, "templates", "strings.yml")).twice
+      @klass.strings
+      @klass.reset!
+      @klass.strings
+    end
+  end
+
+  context "getting the string translated" do
+    setup do
+      @strings = {}
+      @strings[:foo] = "foo bar baz"
+      @klass.stubs(:strings).returns(@strings)
+    end
+
+    should "render the error string" do
+      TemplateRenderer.expects(:render_string).with(@strings[:foo], anything).once
+      @klass.t(:foo)
+    end
+
+    should "pass in any data entries" do
+      data = mock("data")
+      TemplateRenderer.expects(:render_string).with(@strings[:foo], data).once
+      @klass.t(:foo, data)
+    end
+
+    should "return the result of the render" do
+      result = mock("result")
+      TemplateRenderer.expects(:render_string).returns(result)
+      assert_equal result, @klass.t(:foo)
+    end
+
+    should "return an unknown if the key doesn't exist" do
+      result = @klass.t(:unknown)
+      assert result =~ /Unknown/i
+    end
+  end
+end

data/test/vagrant/util_test.rb

@@ -0,0 +1,27 @@
+require "test_helper"
+
+class UtilTest < Test::Unit::TestCase
+  class UtilIncludeTest
+    include Vagrant::Util
+  end
+
+  setup do
+    @klass = UtilIncludeTest
+  end
+
+  context "with a class" do
+    should "have the util methods" do
+      assert @klass.respond_to?(:error_and_exit)
+    end
+  end
+
+  context "with an instance" do
+    setup do
+      @instance = @klass.new
+    end
+
+    should "have the util methods" do
+      assert @instance.respond_to?(:error_and_exit)
+    end
+  end
+end

data/test/vagrant/vm_test.rb

@@ -0,0 +1,228 @@
+require "test_helper"
+
+class VMTest < Test::Unit::TestCase
+  setup do
+    @mock_vm = mock("vm")
+
+    @persisted_vm = mock("persisted_vm")
+
+    @env = mock_environment
+    @env.stubs(:vm).returns(@persisted_vm)
+
+    Net::SSH.stubs(:start)
+  end
+
+  context "finding a VM" do
+    should "return return an uncreated VM object if the VM is not found" do
+      VirtualBox::VM.expects(:find).returns(nil)
+      result = Vagrant::VM.find("foo")
+      assert result.is_a?(Vagrant::VM)
+      assert !result.created?
+    end
+
+    should "return a Vagrant::VM object for that VM if found" do
+      VirtualBox::VM.expects(:find).with("foo").returns("bar")
+      result = Vagrant::VM.find("foo", mock_environment)
+      assert result.is_a?(Vagrant::VM)
+      assert_equal "bar", result.vm
+    end
+  end
+
+  context "vagrant VM instance" do
+    setup do
+      @vm_name = "foo"
+      @vm = Vagrant::VM.new(:env => @env, :vm => @mock_vm, :vm_name => @vm_name)
+      @mock_vm.stubs(:uuid).returns("foo")
+    end
+
+    context "checking if created" do
+      should "return true if the VM object is not nil" do
+        @vm.stubs(:vm).returns(:foo)
+        assert @vm.created?
+      end
+
+      should "return false if the VM object is nil" do
+        @vm.stubs(:vm).returns(nil)
+        assert !@vm.created?
+      end
+    end
+
+    context "accessing the SSH object" do
+      setup do
+        # Reset this to nil to force the reload
+        @vm.instance_variable_set(:@ssh, nil)
+
+        @ssh = mock("ssh")
+        Vagrant::SSH.stubs(:new).returns(@ssh)
+      end
+
+      should "load it the first time" do
+        Vagrant::SSH.expects(:new).with(@vm.env).once.returns(@ssh)
+        @vm.ssh
+        @vm.ssh
+        @vm.ssh
+      end
+
+      should "use the same value once its loaded" do
+        result = @vm.ssh
+        assert_equal result, @vm.ssh
+      end
+    end
+
+    context "loading associated system" do
+      should "error and exit if system is not specified" do
+        @vm.env.config.vm.system = nil
+
+        @vm.expects(:error_and_exit).with(:system_unspecified).once
+        @vm.load_system!
+      end
+
+      context "with a class" do
+        class FakeSystemClass
+          def initialize(vm); end
+        end
+
+        should "initialize class if given" do
+          @vm.env.config.vm.system = Vagrant::Systems::Linux
+
+          @vm.expects(:error_and_exit).never
+          @vm.load_system!
+
+          assert @vm.system.is_a?(Vagrant::Systems::Linux)
+        end
+
+        should "error and exit if class has invalid parent" do
+          @vm.env.config.vm.system = FakeSystemClass
+          @vm.expects(:error_and_exit).with(:system_invalid_class, :system => @vm.env.config.vm.system.to_s).once
+          @vm.load_system!
+        end
+      end
+
+      context "with a symbol" do
+        should "initialize proper symbols" do
+          valid = {
+            :linux => Vagrant::Systems::Linux
+          }
+
+          valid.each do |symbol, klass|
+            @vm.env.config.vm.system = symbol
+            @vm.expects(:error_and_exit).never
+            @vm.load_system!
+
+            assert @vm.system.is_a?(klass)
+            assert_equal @vm, @vm.system.vm
+          end
+        end
+
+        should "error and exit with invalid symbol" do
+          @vm.env.config.vm.system = :shall_never_exist
+          @vm.expects(:error_and_exit).with(:system_unknown_type, :system => @vm.env.config.vm.system.to_s).once
+          @vm.load_system!
+        end
+      end
+    end
+
+    context "uuid" do
+      should "call UUID on VM object" do
+        uuid = mock("uuid")
+        @mock_vm.expects(:uuid).once.returns(uuid)
+        assert_equal uuid, @vm.uuid
+      end
+
+      should "return nil if vm is nil" do
+        @vm.expects(:vm).returns(nil)
+        assert @vm.uuid.nil?
+      end
+    end
+
+    context "reloading" do
+      should "load the same VM and set it" do
+        new_vm = mock("vm")
+        VirtualBox::VM.expects(:find).with(@mock_vm.uuid).returns(new_vm)
+        @vm.reload!
+        assert_equal new_vm, @vm.vm
+      end
+    end
+
+    context "packaging" do
+      should "execute the package action" do
+        @vm.env.actions.expects(:run).with(:package, :foo => :bar).once
+        @vm.package(:foo => :bar)
+      end
+    end
+
+    context "upping" do
+      should "execute the up action" do
+        @vm.env.actions.expects(:run).with(:up, nil).once
+        @vm.up
+      end
+    end
+
+    context "halting" do
+      should "execute the halt action" do
+        @vm.env.actions.expects(:run).with(:halt, :foo => :bar).once
+        @vm.halt({:foo => :bar})
+      end
+    end
+
+    context "reloading action" do
+      should "execute the reload action" do
+        @vm.env.actions.expects(:run).with(:reload).once
+        @vm.reload
+      end
+    end
+
+    context "provisioning" do
+      should "execute the provision action" do
+        @vm.env.actions.expects(:run).with(:provision).once
+        @vm.provision
+      end
+    end
+
+    context "destroying" do
+      should "execute the destroy action" do
+        @vm.env.actions.expects(:run).with(:destroy).once
+        @vm.destroy
+      end
+    end
+
+    context "suspending" do
+      should "execute the suspend action" do
+        @vm.env.actions.expects(:run).with(:suspend).once
+        @vm.suspend
+      end
+    end
+
+    context "resuming" do
+      should "execute the resume action" do
+        @vm.env.actions.expects(:run).with(:resume).once
+        @vm.resume
+      end
+    end
+
+    context "starting" do
+      setup do
+        @mock_vm.stubs(:running?).returns(false)
+        @mock_vm.stubs(:saved?).returns(false)
+      end
+
+      should "not do anything if the VM is already running" do
+        @mock_vm.stubs(:running?).returns(true)
+        @vm.expects(:execute!).never
+        @vm.start
+      end
+
+      should "execute the resume action if saved" do
+        @mock_vm.expects(:saved?).returns(true)
+        @vm.expects(:resume).once
+        @vm.env.actions.expects(:run).with(:start).never
+        @vm.start
+      end
+
+      should "execute the start action" do
+        @vm.env.actions.expects(:run).with(:start).once
+        @vm.start
+      end
+    end
+  end
+end