tenderloin 0.2.0

data/test/tenderloin/commands_test.rb

@@ -0,0 +1,269 @@
+require File.join(File.dirname(__FILE__), '..', 'test_helper')
+
+class CommandsTest < Test::Unit::TestCase
+  setup do
+    Tenderloin::Env.stubs(:load!)
+
+    @persisted_vm = mock("persisted_vm")
+    @persisted_vm.stubs(:execute!)
+    Tenderloin::Env.stubs(:persisted_vm).returns(@persisted_vm)
+    Tenderloin::Env.stubs(:require_persisted_vm)
+  end
+
+  context "init" do
+    setup do
+      FileUtils.stubs(:cp)
+      @rootfile_path = File.join(Dir.pwd, Tenderloin::Env::ROOTFILE_NAME)
+      @template_path = File.join(PROJECT_ROOT, "templates", Tenderloin::Env::ROOTFILE_NAME)
+    end
+
+    should "error and exit if a rootfile already exists" do
+      File.expects(:exist?).with(@rootfile_path).returns(true)
+      Tenderloin::Commands.expects(:error_and_exit).once
+      Tenderloin::Commands.init
+    end
+
+    should "copy the templated rootfile to the current path" do
+      File.expects(:exist?).with(@rootfile_path).returns(false)
+      FileUtils.expects(:cp).with(@template_path, @rootfile_path).once
+      Tenderloin::Commands.init
+    end
+  end
+
+  context "up" do
+    setup do
+      Tenderloin::Env.stubs(:persisted_vm).returns(nil)
+      Tenderloin::VM.stubs(:execute!)
+      Tenderloin::Env.stubs(:require_box)
+    end
+
+    should "require load the environment" do
+      Tenderloin::Env.expects(:load!).once
+      Tenderloin::Commands.up
+    end
+
+    should "require a box" do
+      Tenderloin::Env.expects(:require_box).once
+      Tenderloin::Commands.up
+    end
+
+    should "call the up action on VM if it doesn't exist" do
+      Tenderloin::VM.expects(:execute!).with(Tenderloin::Actions::VM::Up).once
+      Tenderloin::Commands.up
+    end
+
+    should "call start on the persisted vm if it exists" do
+      Tenderloin::Env.stubs(:persisted_vm).returns(@persisted_vm)
+      @persisted_vm.expects(:start).once
+      Tenderloin::VM.expects(:execute!).never
+      Tenderloin::Commands.up
+    end
+  end
+
+  context "down" do
+    setup do
+      @persisted_vm.stubs(:destroy)
+    end
+
+    should "require a persisted VM" do
+      Tenderloin::Env.expects(:require_persisted_vm).once
+      Tenderloin::Commands.down
+    end
+
+    should "destroy the persisted VM and the VM image" do
+      @persisted_vm.expects(:destroy).once
+      Tenderloin::Commands.down
+    end
+  end
+
+  context "reload" do
+    should "require a persisted VM" do
+      Tenderloin::Env.expects(:require_persisted_vm).once
+      Tenderloin::Commands.reload
+    end
+
+    should "call the `reload` action on the VM" do
+      @persisted_vm.expects(:execute!).with(Tenderloin::Actions::VM::Reload).once
+      Tenderloin::Commands.reload
+    end
+  end
+
+  context "ssh" do
+    setup do
+      Tenderloin::SSH.stubs(:connect)
+    end
+
+    should "require a persisted VM" do
+      Tenderloin::Env.expects(:require_persisted_vm).once
+      Tenderloin::Commands.ssh
+    end
+
+    should "connect to SSH" do
+      Tenderloin::SSH.expects(:connect).once
+      Tenderloin::Commands.ssh
+    end
+  end
+
+  context "halt" do
+    should "require a persisted VM" do
+      Tenderloin::Env.expects(:require_persisted_vm).once
+      Tenderloin::Commands.halt
+    end
+
+    should "call the `halt` action on the VM" do
+      @persisted_vm.expects(:execute!).with(Tenderloin::Actions::VM::Halt).once
+      Tenderloin::Commands.halt
+    end
+  end
+
+  context "suspend" do
+    setup do
+      @persisted_vm.stubs(:suspend)
+      @persisted_vm.stubs(:saved?).returns(false)
+    end
+
+    should "require a persisted VM" do
+      Tenderloin::Env.expects(:require_persisted_vm).once
+      Tenderloin::Commands.suspend
+    end
+
+    should "suspend the VM" do
+      @persisted_vm.expects(:suspend).once
+      Tenderloin::Commands.suspend
+    end
+  end
+
+  context "resume" do
+    setup do
+      @persisted_vm.stubs(:resume)
+      @persisted_vm.stubs(:saved?).returns(true)
+    end
+
+    should "require a persisted VM" do
+      Tenderloin::Env.expects(:require_persisted_vm).once
+      Tenderloin::Commands.resume
+    end
+
+    should "save the state of the VM" do
+      @persisted_vm.expects(:resume).once
+      Tenderloin::Commands.resume
+    end
+  end
+
+  context "package" do
+    setup do
+      @persisted_vm.stubs(:package)
+      @persisted_vm.stubs(:powered_off?).returns(true)
+    end
+
+    should "require a persisted vm" do
+      Tenderloin::Env.expects(:require_persisted_vm).once
+      Tenderloin::Commands.package
+    end
+
+    should "error and exit if the VM is not powered off" do
+      @persisted_vm.stubs(:powered_off?).returns(false)
+      Tenderloin::Commands.expects(:error_and_exit).once
+      @persisted_vm.expects(:package).never
+      Tenderloin::Commands.package
+    end
+
+    should "call package on the persisted VM" do
+      @persisted_vm.expects(:package).once
+      Tenderloin::Commands.package
+    end
+
+    should "pass the out path and include_files to the package method" do
+      out_path = mock("out_path")
+      include_files = mock("include_files")
+      @persisted_vm.expects(:package).with(out_path, include_files).once
+      Tenderloin::Commands.package(out_path, include_files)
+    end
+
+    should "default to an empty array when not include_files are specified" do
+      out_path = mock("out_path")
+      @persisted_vm.expects(:package).with(out_path, []).once
+      Tenderloin::Commands.package(out_path)
+    end
+  end
+
+  context "box" do
+    setup do
+      Tenderloin::Commands.stubs(:box_foo)
+      Tenderloin::Commands.stubs(:box_add)
+      Tenderloin::Commands.stubs(:box_remove)
+    end
+
+    should "load the environment" do
+      Tenderloin::Env.expects(:load!).once
+      Tenderloin::Commands.box(["add"])
+    end
+
+    should "error and exit if the first argument is not 'add' or 'remove'" do
+      Tenderloin::Commands.expects(:error_and_exit).once
+      Tenderloin::Commands.box(["foo"])
+    end
+
+    should "not error and exit if the first argument is 'add' or 'remove'" do
+      commands = ["add", "remove"]
+
+      commands.each do |command|
+        Tenderloin::Commands.expects(:error_and_exit).never
+        Tenderloin::Commands.expects("box_#{command}".to_sym).once
+        Tenderloin::Commands.box([command])
+      end
+    end
+
+    should "forward any additional arguments" do
+      Tenderloin::Commands.expects(:box_add).with(1,2,3).once
+      Tenderloin::Commands.box(["add",1,2,3])
+    end
+  end
+
+  context "box list" do
+    setup do
+      @boxes = ["foo", "bar"]
+
+      Tenderloin::Box.stubs(:all).returns(@boxes)
+      Tenderloin::Commands.stubs(:puts)
+    end
+
+    should "call all on box and sort the results" do
+      @all = mock("all")
+      @all.expects(:sort).returns(@boxes)
+      Tenderloin::Box.expects(:all).returns(@all)
+      Tenderloin::Commands.box_list
+    end
+  end
+
+  context "box add" do
+    setup do
+      @name = "foo"
+      @path = "bar"
+    end
+
+    should "execute the add action with the name and path" do
+      Tenderloin::Box.expects(:add).with(@name, @path).once
+      Tenderloin::Commands.box_add(@name, @path)
+    end
+  end
+
+  context "box remove" do
+    setup do
+      @name = "foo"
+    end
+
+    should "error and exit if the box doesn't exist" do
+      Tenderloin::Box.expects(:find).returns(nil)
+      Tenderloin::Commands.expects(:error_and_exit).once
+      Tenderloin::Commands.box_remove(@name)
+    end
+
+    should "call destroy on the box if it exists" do
+      @box = mock("box")
+      Tenderloin::Box.expects(:find).with(@name).returns(@box)
+      @box.expects(:destroy).once
+      Tenderloin::Commands.box_remove(@name)
+    end
+  end
+end

data/test/tenderloin/config_test.rb

@@ -0,0 +1,123 @@
+require File.join(File.dirname(__FILE__), '..', 'test_helper')
+
+class ConfigTest < Test::Unit::TestCase
+  context "resetting" do
+    setup do
+      Tenderloin::Config.run { |config| }
+      Tenderloin::Config.execute!
+    end
+
+    should "return the same config object typically" do
+      config = Tenderloin::Config.config
+      assert config.equal?(Tenderloin::Config.config)
+    end
+
+    should "create a new object if cleared" do
+      config = Tenderloin::Config.config
+      Tenderloin::Config.reset!
+      assert !config.equal?(Tenderloin::Config.config)
+    end
+
+    should "empty the runners" do
+      assert !Tenderloin::Config.config_runners.empty?
+      Tenderloin::Config.reset!
+      assert Tenderloin::Config.config_runners.empty?
+    end
+  end
+
+  context "accessing configuration" do
+    setup do
+      Tenderloin::Config.run { |config| }
+      Tenderloin::Config.execute!
+    end
+
+    should "forward config to the class method" do
+      assert_equal Tenderloin.config, Tenderloin::Config.config
+    end
+  end
+
+  context "initializing" do
+    teardown do
+      Tenderloin::Config.instance_variable_set(:@config_runners, nil)
+      Tenderloin::Config.instance_variable_set(:@config, nil)
+    end
+
+    should "not run the blocks right away" do
+      obj = mock("obj")
+      obj.expects(:foo).never
+      Tenderloin::Config.run { |config| obj.foo }
+      Tenderloin::Config.run { |config| obj.foo }
+      Tenderloin::Config.run { |config| obj.foo }
+    end
+
+    should "run the blocks when execute! is ran" do
+      obj = mock("obj")
+      obj.expects(:foo).times(2)
+      Tenderloin::Config.run { |config| obj.foo }
+      Tenderloin::Config.run { |config| obj.foo }
+      Tenderloin::Config.execute!
+    end
+
+    should "run the blocks with the same config object" do
+      Tenderloin::Config.run { |config| assert config }
+      Tenderloin::Config.run { |config| assert config }
+      Tenderloin::Config.execute!
+    end
+
+    should "not be loaded, initially" do
+      assert !Tenderloin::Config.config.loaded?
+    end
+
+    should "be loaded after running" do
+      Tenderloin::Config.run {}
+      Tenderloin::Config.execute!
+      assert Tenderloin::Config.config.loaded?
+    end
+  end
+
+  context "base class" do
+    setup do
+      @base = Tenderloin::Config::Base.new
+    end
+
+    should "forward [] access to methods" do
+      @base.expects(:foo).once
+      @base[:foo]
+    end
+
+    should "return a hash of instance variables" do
+      data = { :foo => "bar", :bar => "baz" }
+
+      data.each do |iv, value|
+        @base.instance_variable_set("@#{iv}".to_sym, value)
+      end
+
+      result = @base.instance_variables_hash
+      assert_equal data.length, result.length
+
+      data.each do |iv, value|
+        assert_equal value, result[iv]
+      end
+    end
+
+    should "convert instance variable hash to json" do
+      @json = mock("json")
+      @iv_hash = mock("iv_hash")
+      @iv_hash.expects(:to_json).once.returns(@json)
+      @base.expects(:instance_variables_hash).returns(@iv_hash)
+      assert_equal @json, @base.to_json
+    end
+  end
+
+  context "chef config" do
+    setup do
+      @config = Tenderloin::Config::ChefConfig.new
+      @config.json = "HEY"
+    end
+
+    should "not include the 'json' key in the config dump" do
+      result = JSON.parse(@config.to_json)
+      assert !result.has_key?("json")
+    end
+  end
+end

data/test/tenderloin/downloaders/base_test.rb

@@ -0,0 +1,20 @@
+require File.join(File.dirname(__FILE__), '..', '..', 'test_helper')
+
+class BaseDownloaderTest < Test::Unit::TestCase
+  should "include the util class so subclasses have access to it" do
+    assert Tenderloin::Downloaders::Base.include?(Tenderloin::Util)
+  end
+
+  context "base instance" do
+    setup do
+      @base = Tenderloin::Downloaders::Base.new
+    end
+
+    should "implement prepare which does nothing" do
+      assert_nothing_raised do
+        assert @base.respond_to?(:download!)
+        @base.download!("source", "destination")
+      end
+    end
+  end
+end

data/test/tenderloin/downloaders/file_test.rb

@@ -0,0 +1,32 @@
+require File.join(File.dirname(__FILE__), '..', '..', 'test_helper')
+
+class FileDownloaderTest < Test::Unit::TestCase
+  setup do
+    @downloader, @tempfile = mock_downloader(Tenderloin::Downloaders::File)
+    @uri = "foo.box"
+  end
+
+  context "downloading" do
+    setup do
+      @file = mock("file")
+      @file.stubs(:read)
+      @file.stubs(:eof?).returns(false)
+      @downloader.stubs(:open).yields(@file)
+    end
+
+    should "open with the given uri" do
+      @downloader.expects(:open).with(@uri).once
+      @downloader.download!(@uri, @tempfile)
+    end
+
+    should "buffer the read from the file and write to the tempfile" do
+      data = mock("data")
+      write_seq = sequence("write_seq")
+      @file.stubs(:eof?).returns(false).in_sequence(write_seq)
+      @file.expects(:read).returns(data).in_sequence(write_seq)
+      @tempfile.expects(:write).with(data).in_sequence(write_seq)
+      @file.stubs(:eof?).returns(true).in_sequence(write_seq)
+      @downloader.download!(@uri, @tempfile)
+    end
+  end
+end

data/test/tenderloin/downloaders/http_test.rb

@@ -0,0 +1,40 @@
+require File.join(File.dirname(__FILE__), '..', '..', 'test_helper')
+
+class HttpDownloaderTest < Test::Unit::TestCase
+  setup do
+    @downloader, @tempfile = mock_downloader(Tenderloin::Downloaders::HTTP)
+    @downloader.stubs(:report_progress)
+    @downloader.stubs(:complete_progress)
+    @uri = "foo.box"
+  end
+
+  context "downloading" do
+    setup do
+      @parsed_uri = mock("parsed")
+      URI.stubs(:parse).with(@uri).returns(@parsed_uri)
+    end
+
+    should "parse the URI and use that parsed URI for Net::HTTP" do
+      URI.expects(:parse).with(@uri).returns(@parsed_uri).once
+      Net::HTTP.expects(:get_response).with(@parsed_uri).once
+      @downloader.download!(@uri, @tempfile)
+    end
+
+    should "read the body of the response and place each segment into the file" do
+      response = mock("response")
+      response.stubs(:content_length)
+      segment = mock("segment")
+      segment.stubs(:length).returns(7)
+
+      Net::HTTP.stubs(:get_response).yields(response)
+      response.expects(:read_body).once.yields(segment)
+      @tempfile.expects(:write).with(segment).once
+
+      @downloader.download!(@uri, @tempfile)
+    end
+  end
+
+  context "reporting progress" do
+    # TODO: Testing for this, probably
+  end
+end