vagrant 0.3.4 → 0.4.0

data/test/vagrant/systems/linux_test.rb

@@ -0,0 +1,174 @@
+require File.join(File.dirname(__FILE__), '..', '..', 'test_helper')
+
+class LinuxSystemTest < Test::Unit::TestCase
+  setup do
+    @klass = Vagrant::Systems::Linux
+    @ssh = mock("ssh")
+    @mock_env = mock_environment
+    @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
+
+  context "preparing sync" do
+    setup do
+      @ssh.stubs(:exec!)
+      @vm.stubs(:ssh).returns(@ssh)
+      @vm.ssh.stubs(:upload!)
+    end
+
+    should "upload the sync template" do
+      @vm.ssh.expects(:upload!).with do |string_io, guest_path|
+        string_io.string =~ /#!\/bin\/sh/ && guest_path == @mock_env.config.vm.sync_script
+      end
+
+      @instance.prepare_sync(@ssh)
+    end
+
+    should "remove old crontab entries file" do
+      @ssh.expects(:exec!).with("sudo rm #{@mock_env.config.vm.sync_crontab_entry_file}", :error_check => false)
+      @instance.prepare_sync(@ssh)
+    end
+
+    should "prepare the sync template for execution" do
+      @ssh.expects(:exec!).with("sudo chmod +x #{@mock_env.config.vm.sync_script}")
+      @instance.prepare_sync(@ssh)
+    end
+  end
+
+  context "setting up an sync folder" do
+    setup do
+      @ssh.stubs(:exec!)
+    end
+
+    should "create the new rysnc destination directory" do
+      sync_path = 'foo'
+      @ssh.expects(:exec!).with("sudo mkdir -p #{sync_path}")
+      @instance.create_sync(@ssh, :syncpath => "foo")
+    end
+
+    should "add an entry to the crontab file" do
+      @instance.expects(:render_crontab_entry).returns('foo')
+      @ssh.expects(:exec!).with do |cmd|
+        cmd =~ /echo/ && cmd =~ /foo/ && cmd =~ /#{@mock_env.config.vm.sync_crontab_entry_file}/
+      end
+      @instance.create_sync(@ssh, {})
+    end
+
+    should "use the crontab entry file to define vagrant users cron entries" do
+      @ssh.expects(:exec!).with("crontab #{@mock_env.config.vm.sync_crontab_entry_file}")
+      @instance.create_sync(@ssh, {})
+    end
+
+    should "chown the sync directory" do
+      @instance.expects(:chown).with(@ssh, "foo")
+      @instance.create_sync(@ssh, :syncpath => "foo")
+    end
+
+    should "return provide a formatted crontab entry that runs every minute" do
+      assert @instance.render_crontab_entry({}).include?("* * * * *")
+    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=#{@vm.env.config.ssh.username},gid=#{@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::Actions::ActionException) {
+        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 = mock_environment do |config|
+        config.vm.shared_folder_uid = uid
+        config.vm.shared_folder_gid = gid
+      end
+
+      @vm.stubs(:env).returns(env)
+
+      @ssh.expects(:exec!).with("sudo mount -t vboxsf -o uid=#{uid},gid=#{gid} #{@name} #{@guestpath}").returns(@success_return)
+      mount_folder
+    end
+  end
+end

data/test/vagrant/util/error_helper_test.rb

@@ -0,0 +1,5 @@
+require File.join(File.dirname(__FILE__), '..', '..', 'test_helper')
+
+class ErrorHelperUtilTest < Test::Unit::TestCase
+  # TODO
+end

data/test/vagrant/util/output_helper_test.rb

@@ -0,0 +1,5 @@
+require File.join(File.dirname(__FILE__), '..', '..', 'test_helper')
+
+class OutputHelperUtilTest < Test::Unit::TestCase
+  # TODO
+end

data/test/vagrant/util/plain_logger_test.rb

@@ -0,0 +1,17 @@
+require File.join(File.dirname(__FILE__), '..', '..', '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 File.join(File.dirname(__FILE__), '..', '..', '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/{errors_test.rb → translator_test.rb}

@@ -1,56 +1,60 @@
 require File.join(File.dirname(__FILE__), '..', '..', 'test_helper')
 
-class ErrorsUtilTest < Test::Unit::TestCase
+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)
-      Errors.reset!
+      @klass.reset!
     end
 
     should "load the file initially, then never again unless reset" do
-      YAML.expects(:load_file).with(File.join(PROJECT_ROOT, "templates", "errors.yml")).once
-      Errors.errors
-      Errors.errors
-      Errors.errors
-      Errors.errors
+      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", "errors.yml")).twice
-      Errors.errors
-      Errors.reset!
-      Errors.errors
+      YAML.expects(:load_file).with(File.join(PROJECT_ROOT, "templates", "strings.yml")).twice
+      @klass.strings
+      @klass.reset!
+      @klass.strings
     end
   end
 
-  context "getting the error string" do
+  context "getting the string translated" do
     setup do
-      @errors = {}
-      @errors[:foo] = "foo bar baz"
-      Errors.stubs(:errors).returns(@errors)
+      @strings = {}
+      @strings[:foo] = "foo bar baz"
+      @klass.stubs(:strings).returns(@strings)
     end
 
     should "render the error string" do
-      TemplateRenderer.expects(:render_string).with(@errors[:foo], anything).once
-      Errors.error_string(:foo)
+      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(@errors[:foo], data).once
-      Errors.error_string(:foo, 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, Errors.error_string(:foo)
+      assert_equal result, @klass.t(:foo)
     end
 
     should "return an unknown if the key doesn't exist" do
-      result = Errors.error_string(:unknown)
+      result = @klass.t(:unknown)
       assert result =~ /Unknown/i
     end
   end

data/test/vagrant/util_test.rb

@@ -1,64 +1,27 @@
 require File.join(File.dirname(__FILE__), '..', 'test_helper')
 
 class UtilTest < Test::Unit::TestCase
-  class RegUtil
-    extend Vagrant::Util
+  class UtilIncludeTest
+    include Vagrant::Util
   end
 
-  context "erroring" do
-    # TODO: Any way to stub Kernel.exit? Can't test nicely
-    # otherwise
+  setup do
+    @klass = UtilIncludeTest
   end
 
-  context "logger" do
-    class OtherUtil
-      extend Vagrant::Util
+  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
-      @config = Vagrant::Config::Top.new
-      @config.stubs(:loaded?).returns(true)
-      @config.vagrant.log_output = STDOUT
-      Vagrant::Config.stubs(:config).returns(@config)
-      Vagrant::Logger.reset_logger!
-    end
-
-    teardown do
-      Vagrant::Logger.reset_logger!
-    end
-
-    should "return a logger to nil if config is not loaded" do
-      @config.expects(:loaded?).returns(false)
-      logger = RegUtil.logger
-      assert_nil logger.instance_variable_get(:@logdev)
-    end
-
-    should "return a logger using the configured output" do
-      logger = RegUtil.logger
-      logdev = logger.instance_variable_get(:@logdev)
-      assert logger
-      assert !logdev.nil?
-      assert_equal STDOUT, logdev.dev
-    end
-
-    should "only instantiate a logger once" do
-      Vagrant::Logger.expects(:new).once.returns("GOOD")
-      RegUtil.logger
-      RegUtil.logger
+      @instance = @klass.new
     end
 
-    should "be able to reset the logger" do
-      Vagrant::Logger.expects(:new).twice
-      RegUtil.logger
-      Vagrant::Logger.reset_logger!
-      RegUtil.logger
-    end
-
-    should "return the same logger across classes" do
-      logger = RegUtil.logger
-      other = OtherUtil.logger
-
-      assert logger.equal?(other)
+    should "have the util methods" do
+      assert @instance.respond_to?(:error_and_exit)
     end
   end
 end

data/test/vagrant/vm_test.rb

@@ -14,20 +14,22 @@ class VMTest < Test::Unit::TestCase
 
   context "being an action runner" do
     should "be an action runner" do
-      vm = Vagrant::VM.new
+      vm = Vagrant::VM.new(:env => @env)
       assert vm.is_a?(Vagrant::Actions::Runner)
     end
   end
 
   context "finding a VM" do
-    should "return nil if the VM is not found" do
+    should "return return an uncreated VM object if the VM is not found" do
       VirtualBox::VM.expects(:find).returns(nil)
-      assert_nil Vagrant::VM.find("foo")
+      result = Vagrant::VM.find("foo")
+      assert result.is_a?(Vagrant::VM)
+      assert !result.created?
     end
 
-    should "return a Vagrant::VM object for that VM otherwise" do
+    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")
+      result = Vagrant::VM.find("foo", mock_environment)
       assert result.is_a?(Vagrant::VM)
       assert_equal "bar", result.vm
     end
@@ -35,10 +37,98 @@ class VMTest < Test::Unit::TestCase
 
   context "vagrant VM instance" do
     setup do
-      @vm = Vagrant::VM.new(@mock_vm)
+      @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")
@@ -63,12 +153,44 @@ class VMTest < Test::Unit::TestCase
 
     context "packaging" do
       should "queue up the actions and execute" do
-        out_path = mock("out_path")
-        action_seq = sequence("actions")
-        @vm.expects(:add_action).with(Vagrant::Actions::VM::Export).once.in_sequence(action_seq)
-        @vm.expects(:add_action).with(Vagrant::Actions::VM::Package, out_path, []).once.in_sequence(action_seq)
+        action_seq = sequence("action_seq")
+        @vm.expects(:add_action).with(Vagrant::Actions::VM::Export, nil).once.in_sequence(action_seq)
+        @vm.expects(:add_action).with(Vagrant::Actions::VM::Package, nil).once.in_sequence(action_seq)
         @vm.expects(:execute!).in_sequence(action_seq)
-        @vm.package(out_path)
+        @vm.package
+      end
+    end
+
+    context "upping" do
+      should "execute the up action" do
+        @vm.expects(:execute!).with(Vagrant::Actions::VM::Up, nil).once
+        @vm.up
+      end
+    end
+
+    context "halting" do
+      should "execute the halt action" do
+        @vm.expects(:execute!).with(Vagrant::Actions::VM::Halt, nil).once
+        @vm.halt
+      end
+
+      should "force if specified" do
+        @vm.expects(:execute!).with(Vagrant::Actions::VM::Halt, {:foo => :bar}).once
+        @vm.halt({:foo => :bar})
+      end
+    end
+
+    context "reloading action" do
+      should "execute the reload action" do
+        @vm.expects(:execute!).with(Vagrant::Actions::VM::Reload).once
+        @vm.reload
+      end
+    end
+
+    context "provisioning" do
+      should "execute the provision action" do
+        @vm.expects(:execute!).with(Vagrant::Actions::VM::Provision).once
+        @vm.provision
       end
     end