vagrant-fixed-ssh 1.0.7

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

@@ -0,0 +1,16 @@
+require File.expand_path("../../../base", __FILE__)
+
+require "vagrant/util/line_ending_helpers"
+
+describe Vagrant::Util::LineEndingHelpers do
+  let(:klass) do
+    Class.new do
+      extend Vagrant::Util::LineEndingHelpers
+    end
+  end
+
+  it "should convert DOS to unix-style line endings" do
+    klass.dos_to_unix("foo\r\nbar\r\n").should == "foo\nbar\n"
+  end
+end
+

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

@@ -0,0 +1,17 @@
+require File.expand_path("../../../base", __FILE__)
+
+require "vagrant/util/network_ip"
+
+describe Vagrant::Util::NetworkIP do
+  let(:klass) do
+    Class.new do
+      extend Vagrant::Util::NetworkIP
+    end
+  end
+
+  describe "network address" do
+    it "calculates it properly" do
+      klass.network_address("192.168.2.234", "255.255.255.0").should == "192.168.2.0"
+    end
+  end
+end

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

@@ -0,0 +1,106 @@
+require File.expand_path("../../../base", __FILE__)
+
+require "vagrant/util/retryable"
+
+describe Vagrant::Util::Retryable do
+  let(:klass) do
+    Class.new do
+      extend Vagrant::Util::Retryable
+    end
+  end
+
+  it "doesn't retry by default" do
+    tries = 0
+
+    block = lambda do
+      tries += 1
+      raise RuntimeError, "Try"
+    end
+
+    # It should re-raise the error
+    expect { klass.retryable(&block) }.
+      to raise_error(RuntimeError)
+
+    # It should've tried once
+    tries.should == 1
+  end
+
+  it "retries the set number of times" do
+    tries = 0
+
+    block = lambda do
+      tries += 1
+      raise RuntimeError, "Try"
+    end
+
+    # It should re-raise the error
+    expect { klass.retryable(:tries => 5, &block) }.
+      to raise_error(RuntimeError)
+
+    # It should've tried all specified times
+    tries.should == 5
+  end
+
+  it "only retries on the given exception" do
+    tries = 0
+
+    block = lambda do
+      tries += 1
+      raise StandardError, "Try"
+    end
+
+    # It should re-raise the error
+    expect { klass.retryable(:tries => 5, :on => RuntimeError, &block) }.
+      to raise_error(StandardError)
+
+    # It should've never tried since it was a different kind of error
+    tries.should == 1
+  end
+
+  it "can retry on multiple types of errors" do
+    tries = 0
+
+    foo_error = Class.new(StandardError)
+    bar_error = Class.new(StandardError)
+
+    block = lambda do
+      tries += 1
+      raise foo_error, "Try" if tries == 1
+      raise bar_error, "Try" if tries == 2
+      raise RuntimeError, "YAY"
+    end
+
+    # It should re-raise the error
+    expect { klass.retryable(:tries => 5, :on => [foo_error, bar_error], &block) }.
+      to raise_error(RuntimeError)
+
+    # It should've never tried since it was a different kind of error
+    tries.should == 3
+  end
+
+  it "doesn't sleep between tries by default" do
+    block = lambda do
+      raise RuntimeError, "Try"
+    end
+
+    # Sleep should never be called
+    klass.should_not_receive(:sleep)
+
+    # Run it.
+    expect { klass.retryable(:tries => 5, &block) }.
+      to raise_error(RuntimeError)
+  end
+
+  it "sleeps specified amount between retries" do
+    block = lambda do
+      raise RuntimeError, "Try"
+    end
+
+    # Sleep should be called between each retry
+    klass.should_receive(:sleep).with(10).exactly(4).times
+
+    # Run it.
+    expect { klass.retryable(:tries => 5, :sleep => 10, &block) }.
+      to raise_error(RuntimeError)
+  end
+end

data/test/unit/vagrant_test.rb

@@ -0,0 +1,27 @@
+require File.expand_path("../base", __FILE__)
+
+describe Vagrant do
+  it "has the path to the source root" do
+    described_class.source_root.should == Pathname.new(File.expand_path("../../../", __FILE__))
+  end
+
+  it "has a registry for commands" do
+    described_class.commands.should be_a(Vagrant::Registry)
+  end
+
+  it "has a registry for config keys" do
+    described_class.config_keys.should be_a(Vagrant::Registry)
+  end
+
+  it "has a registry for hosts" do
+    described_class.hosts.should be_a(Vagrant::Registry)
+  end
+
+  it "has a registry for guests" do
+    described_class.guests.should be_a(Vagrant::Registry)
+  end
+
+  it "has a registry for provisioners" do
+    described_class.provisioners.should be_a(Vagrant::Registry)
+  end
+end

data/test/unit_legacy/locales/en.yml

@@ -0,0 +1,8 @@
+en:
+  vagrant:
+    test:
+      errors:
+        test_key: This is a test key
+        test_key_with_interpolation: This is a test key that says %{key}
+      alternate:
+        test_key: This is an alternate

data/test/unit_legacy/test_helper.rb

@@ -0,0 +1,32 @@
+# Add this folder to the load path for "test_helper"
+$:.unshift(File.dirname(__FILE__))
+
+require 'vagrant'
+require 'contest'
+require 'mocha'
+
+# Try to load ruby debug since its useful if it is available.
+# But not a big deal if its not available (probably on a non-MRI
+# platform)
+begin
+  require 'ruby-debug'
+rescue LoadError
+end
+
+# Do not buffer output for tests
+$stdout.sync = true
+$stderr.sync = true
+
+# Set the home directory to some temporary directory
+ENV["HOME"] = Vagrant.source_root.join("test", "tmp", "home").to_s
+
+# Set the log output to nothing
+ENV["VAGRANT_LOG"] = "NULL"
+
+# Add the I18n locale for tests
+I18n.load_path << File.expand_path("../locales/en.yml", __FILE__)
+
+class Test::Unit::TestCase
+  include Vagrant::TestHelpers
+end
+

data/test/unit_legacy/vagrant/action/box/destroy_test.rb

@@ -0,0 +1,18 @@
+require "test_helper"
+
+class DestroyBoxActionTest < Test::Unit::TestCase
+  setup do
+    @klass = Vagrant::Action::Box::Destroy
+    @app, @env = action_env
+    @env["box"] = Vagrant::Box.new(vagrant_env, "foo")
+
+    @instance = @klass.new(@app, @env)
+  end
+
+  should "delete the box directory" do
+    seq = sequence("seq")
+    FileUtils.expects(:rm_rf).with(@env["box"].directory).in_sequence(seq)
+    @app.expects(:call).with(@env).once.in_sequence(seq)
+    @instance.call(@env)
+  end
+end

data/test/unit_legacy/vagrant/action/box/download_test.rb

@@ -0,0 +1,125 @@
+require "test_helper"
+
+class DownloadBoxActionTest < Test::Unit::TestCase
+  setup do
+    @klass = Vagrant::Action::Box::Download
+    @app, @env = action_env
+    @env["box"] = Vagrant::Box.new(vagrant_env, "foo")
+    @env["box"].uri = "http://google.com"
+  end
+
+  context "initializing" do
+    should "initialize download classes" do
+      @klass.new(@app, @env)
+      assert_equal [Vagrant::Downloaders::HTTP, Vagrant::Downloaders::File], @env["download.classes"]
+    end
+  end
+
+  context "with an instance" do
+    setup do
+      @instance = @klass.new(@app, @env)
+    end
+
+    context "calling" do
+      should "call the proper methods in sequence" do
+        seq = sequence("seq")
+        @instance.expects(:instantiate_downloader).in_sequence(seq).returns(true)
+        @instance.expects(:download).in_sequence(seq)
+        @app.expects(:call).with(@env).in_sequence(seq)
+        @instance.expects(:recover).with(@env).in_sequence(seq)
+        @instance.call(@env)
+      end
+    end
+
+    context "instantiating downloader" do
+      should "instantiate the proper class" do
+        instance = mock("instance")
+        Vagrant::Downloaders::HTTP.expects(:new).with(@env).returns(instance)
+        instance.expects(:prepare).with(@env["box"].uri).once
+        assert @instance.instantiate_downloader
+      end
+
+      should "complain that the file doesn't exist if the URI is invalid for any downloaders" do
+        @env["box"].uri = "foobar"
+        assert_raises(Vagrant::Errors::DownloaderFileDoesntExist) {
+          @instance.instantiate_downloader
+        }
+      end
+    end
+
+    context "downloading" do
+      setup do
+        @path = "foo"
+
+        @tempfile = mock("tempfile")
+        @tempfile.stubs(:path).returns(@path)
+
+        @instance.stubs(:with_tempfile).yields(@tempfile)
+        @instance.stubs(:download_to)
+      end
+
+      should "make a tempfile and copy the URI contents to it" do
+        @instance.expects(:with_tempfile).yields(@tempfile)
+        @instance.expects(:download_to).with(@tempfile)
+        @instance.download
+      end
+
+      should "save the tempfile path" do
+        @instance.download
+        assert @env.has_key?("download.temp_path")
+        assert_equal @tempfile.path, @env["download.temp_path"]
+        assert_equal @tempfile.path, @instance.temp_path
+      end
+    end
+
+    context "tempfile" do
+      should "create a tempfile in the vagrant tmp directory" do
+        File.expects(:open).with { |name, bitmask|
+          name.to_s =~ /#{Vagrant::Action::Box::Download::BASENAME}/ &&  name.to_s =~ /#{@env.env.tmp_path}/
+        }.once
+        @instance.with_tempfile
+      end
+
+      should "yield the tempfile object" do
+        @tempfile = mock("tempfile")
+        File.expects(:open).yields(@tempfile)
+
+        @instance.with_tempfile do |otherfile|
+          assert @tempfile.equal?(otherfile)
+        end
+      end
+    end
+
+    context "cleaning up" do
+      setup do
+        @temp_path = "foo"
+        @instance.stubs(:temp_path).returns(@temp_path)
+        File.stubs(:exist?).returns(true)
+      end
+
+      should "delete the temporary file if it exists" do
+        File.expects(:unlink).with(@temp_path).once
+        @instance.recover(@env)
+      end
+
+      should "not delete anything if it doesn't exist" do
+        File.stubs(:exist?).returns(false)
+        File.expects(:unlink).never
+        @instance.recover(@env)
+      end
+    end
+
+    context "downloading to" do
+      setup do
+        @downloader = mock("downloader")
+        @instance.instance_variable_set(:@downloader, @downloader)
+      end
+
+      should "call download! on the download with the URI and tempfile" do
+        tempfile = "foo"
+        @downloader.expects(:download!).with(@env["box"].uri, tempfile)
+        @instance.download_to(tempfile)
+      end
+    end
+  end
+end

data/test/unit_legacy/vagrant/action/box/package_test.rb

@@ -0,0 +1,25 @@
+require "test_helper"
+
+class PackageBoxActionTest < Test::Unit::TestCase
+  setup do
+    @klass = Vagrant::Action::Box::Package
+    @app, @env = action_env
+    @env["box"] = Vagrant::Box.new(vagrant_env, "foo")
+
+    @instance = @klass.new(@app, @env)
+  end
+
+  should "be a subclass of general packaging middleware" do
+    assert @instance.is_a?(Vagrant::Action::General::Package)
+  end
+
+  should "set the package directory then call parent" do
+    @instance.expects(:general_call).once.with() do |env|
+      assert env["package.directory"]
+      assert_equal env["package.directory"], @env["box"].directory
+      true
+    end
+
+    @instance.call(@env)
+  end
+end

data/test/unit_legacy/vagrant/action/box/unpackage_test.rb

@@ -0,0 +1,84 @@
+require "test_helper"
+
+class UnpackageBoxActionTest < Test::Unit::TestCase
+  setup do
+    @klass = Vagrant::Action::Box::Unpackage
+    @app, @env = action_env
+    @env["box"] = Vagrant::Box.new(vagrant_env, "foo")
+
+    @instance = @klass.new(@app, @env)
+  end
+
+  context "calling" do
+    should "call the proper chain" do
+      seq = sequence("sequence")
+      @instance.expects(:setup_box_directory).in_sequence(seq).returns(true)
+      @instance.expects(:decompress).in_sequence(seq)
+      @app.expects(:call).with(@env)
+      @instance.call(@env)
+    end
+  end
+
+  context "cleaning up" do
+    setup do
+      @instance.stubs(:box_directory).returns("foo")
+      File.stubs(:directory?).returns(false)
+      FileUtils.stubs(:rm_rf)
+    end
+
+    should "do nothing if box directory is not set" do
+      @instance.stubs(:box_directory).returns(nil)
+      File.expects(:directory?).never
+      FileUtils.expects(:rm_rf).never
+      @instance.recover(nil)
+    end
+
+    should "do nothing if not a directory" do
+      FileUtils.expects(:rm_rf).never
+      @instance.recover(nil)
+    end
+
+    should "remove the directory if exists" do
+      File.expects(:directory?).with(@instance.box_directory).once.returns(true)
+      FileUtils.expects(:rm_rf).with(@instance.box_directory).once
+      @instance.recover(nil)
+    end
+  end
+
+  context "setting up the box directory" do
+    setup do
+      File.stubs(:directory?).returns(false)
+      FileUtils.stubs(:mkdir_p)
+    end
+
+    should "error the environment if the box already exists" do
+      File.expects(:directory?).returns(true)
+      assert_raises(Vagrant::Errors::BoxAlreadyExists) {
+        @instance.setup_box_directory
+      }
+    end
+
+    should "create the directory" do
+      FileUtils.expects(:mkdir_p).with(@env["box"].directory).once
+      @instance.setup_box_directory
+    end
+  end
+
+  context "decompressing" do
+    setup do
+      @env["download.temp_path"] = "bar"
+
+      Dir.stubs(:chdir).yields
+    end
+
+    should "change to the box directory" do
+      Dir.expects(:chdir).with(@env["box"].directory)
+      @instance.decompress
+    end
+
+    should "open the tar file within the new directory, and extract it all" do
+      Archive::Tar::Minitar.expects(:unpack).with(@env["download.temp_path"], @env["box"].directory.to_s).once
+      @instance.decompress
+    end
+  end
+end