pantry-chef 0.1

data/test/acceptance/test_helper.rb

@@ -0,0 +1,25 @@
+require 'pantry/test/acceptance'
+require 'pantry/chef'
+
+Pantry.logger.disable!
+#Pantry.config.log_level = :debug
+
+class Minitest::Test
+
+  def setup
+    Pantry.config.root_dir = File.expand_path("../../root_dir", __FILE__)
+    clean_up_pantry_root
+  end
+
+  # Ensure Pantry.root is always clean for each test.
+  def clean_up_pantry_root
+    Dir["#{Pantry.root}/**/*"].each do |file|
+      FileUtils.rm_rf file
+    end
+  end
+
+  def fixture_path(file_path)
+    File.join(File.dirname(__FILE__), "..", "fixtures", file_path)
+  end
+
+end

data/test/fixtures/cookbooks/bad/recipes/default.rb

@@ -0,0 +1 @@
+# Do nothing

data/test/fixtures/cookbooks/mini/metadata.rb

@@ -0,0 +1,5 @@
+name          "mini"
+description   "The smallest cookbook possible"
+version       "1.0.0"
+
+recipe        "mini", "The basic recipe"

data/test/fixtures/cookbooks/mini/recipes/default.rb

@@ -0,0 +1 @@
+# Do nothing

data/test/fixtures/environments/test.rb

@@ -0,0 +1,2 @@
+name "test"
+description "Pantry test env"

data/test/fixtures/roles/app.rb

@@ -0,0 +1,2 @@
+name "app"
+description "Application test role"

data/test/fixtures/roles/app1.rb

@@ -0,0 +1,2 @@
+name "app1"
+description "Application test role"

data/test/root_dir/applications/pantry/chef/roles/app.rb

@@ -0,0 +1,2 @@
+name "app"
+description "Application test role"

data/test/unit/chef/configure_chef_test.rb

@@ -0,0 +1,64 @@
+require 'unit/test_helper'
+
+describe Pantry::Chef::ConfigureChef do
+
+  fake_fs!
+
+  it "creates expected directory structure for chef file storage" do
+    command = Pantry::Chef::ConfigureChef.new
+    command.perform(Pantry::Message.new)
+
+    %w(cache cookbooks environments).each do |dir|
+      assert File.directory?(Pantry.root.join("chef", dir)),
+        "Did not create the chef #{dir} directory"
+    end
+  end
+
+  it "writes out chef/solo.rb pointing chef to the right locations" do
+    command = Pantry::Chef::ConfigureChef.new
+    command.perform(Pantry::Message.new)
+
+    solo_rb = Pantry.root.join("etc", "chef", "solo.rb")
+    assert File.exists?(solo_rb), "Did not write out a solo.rb file"
+
+    solo_contents = File.read(solo_rb)
+
+    assert_match %r|cookbook_path\s*"#{Pantry.root.join("chef", "cookbooks")}|,
+      solo_contents
+    assert_match %r|file_cache_path\s*"#{Pantry.root.join("chef", "cache")}|,
+      solo_contents
+    assert_match %r|role_path\s*"#{Pantry.root.join("chef", "roles")}|,
+      solo_contents
+    assert_match %r|environment_path\s*"#{Pantry.root.join("chef", "environments")}|,
+      solo_contents
+    assert_match %r|json_attribs\s*"#{Pantry.root.join("etc", "chef", "node.json")}"|,
+      solo_contents
+  end
+
+  it "writes out the node.json file with run list based on client roles" do
+    client = Pantry::Client.new(roles: %w(app db))
+
+    command = Pantry::Chef::ConfigureChef.new
+    command.client = client
+    command.perform(Pantry::Message.new)
+
+    node_json = File.read(Pantry.root.join("etc", "chef", "node.json"))
+
+    assert_match "role[app]", node_json
+    assert_match "role[db]", node_json
+  end
+
+  it "writes out the current environment of the Client if one exists" do
+    client = Pantry::Client.new(environment: "staging")
+
+    command = Pantry::Chef::ConfigureChef.new
+    command.client = client
+    command.perform(Pantry::Message.new)
+
+    solo_rb = Pantry.root.join("etc/chef/solo.rb")
+    solo_contents = File.read(solo_rb)
+
+    assert_match %|environment "staging"|, solo_contents
+  end
+
+end

data/test/unit/chef/list_cookbooks_test.rb

@@ -0,0 +1,49 @@
+require 'unit/test_helper'
+
+describe Pantry::Chef::ListCookbooks do
+
+  describe "#perform" do
+    fake_fs!
+
+    it "returns the list of cookbooks and latest version known" do
+      cookbooks = [
+        Pantry.root.join("chef", "cookbook-cache", "mini.tgz"),
+        Pantry.root.join("chef", "cookbook-cache", "ruby.tgz"),
+        Pantry.root.join("chef", "cookbook-cache", "pantry.tgz")
+      ]
+
+      cookbooks.each do|c|
+        FileUtils.mkdir_p(File.dirname(c))
+        FileUtils.touch(c)
+      end
+
+      Pantry.stubs(:file_checksum).returns("deadbeef")
+
+      command = Pantry::Chef::ListCookbooks.new
+      cookbook_list = command.perform(Pantry::Message.new)
+
+      assert_equal [
+        ["mini",   0, "deadbeef"],
+        ["pantry", 0, "deadbeef"],
+        ["ruby",   0, "deadbeef"]
+      ], cookbook_list
+    end
+  end
+
+  describe "#receive_server_response" do
+    mock_ui!
+
+    it "displays the list alphabetical order" do
+      message = Pantry::Message.new
+      message << ["pantry", 0, "deadbeef"]
+      message << ["ruby", 0, "deadbeef"]
+      message << ["mini", 0, "deadbeef"]
+
+      command = Pantry::Chef::ListCookbooks.new
+      command.receive_server_response(message)
+
+      assert_equal "mini\npantry\nruby\n", stdout
+    end
+  end
+
+end

data/test/unit/chef/run_chef_solo_test.rb

@@ -0,0 +1,29 @@
+require 'unit/test_helper'
+
+describe Pantry::Chef::RunChefSolo do
+
+  it "executes the chef-solo command, returning outputs" do
+    config_file = Pantry.root.join(*%w(etc chef solo.rb))
+    Open3.expects(:capture3).with("chef-solo --config #{config_file}").
+      returns(["chef ran", "error", 0])
+
+    command = Pantry::Chef::RunChefSolo.new
+    stdout, stderr, status = command.perform(Pantry::Message.new)
+
+    assert_equal "chef ran", stdout
+    assert_equal "error", stderr
+    assert_equal 0, status
+  end
+
+  it "returns error message if chef-solo not found on the system" do
+    Open3.expects(:capture3).raises(Errno::ENOENT, "Can't find LOLZ")
+
+    command = Pantry::Chef::RunChefSolo.new
+    stdout, stderr, status = command.perform(Pantry::Message.new)
+
+    assert_equal "", stdout
+    assert_equal "No such file or directory - Can't find LOLZ", stderr
+    assert_equal 1, status
+  end
+
+end

data/test/unit/chef/run_test.rb

@@ -0,0 +1,5 @@
+require 'unit/test_helper'
+
+describe Pantry::Chef::Run do
+
+end

data/test/unit/chef/send_cookbooks_test.rb

@@ -0,0 +1,44 @@
+require 'unit/test_helper'
+
+describe Pantry::Chef::SendCookbooks do
+
+  fake_fs!
+
+  describe "#perform" do
+
+    it "takes the list of receivers and builds senders, passing in the proper cookbook file" do
+      message = Pantry::Message.new
+      message << ["mini",   "receiver_ident",  "file_uuid"]
+      message << ["pantry", "receiver_ident2", "file_uuid2"]
+
+      cookbooks = [
+        Pantry.root.join("chef", "cookbook-cache", "mini.tgz"),
+        Pantry.root.join("chef", "cookbook-cache", "pantry.tgz")
+      ]
+
+      cookbooks.each do|c|
+        FileUtils.mkdir_p(File.dirname(c))
+        FileUtils.touch(c)
+      end
+
+      server = mock
+      server.expects(:send_file).with(
+        Pantry.root.join("chef", "cookbook-cache", "mini.tgz"),
+        "receiver_ident",
+        "file_uuid"
+      )
+
+      server.expects(:send_file).with(
+        Pantry.root.join("chef", "cookbook-cache", "pantry.tgz"),
+        "receiver_ident2",
+        "file_uuid2"
+      )
+
+      command = Pantry::Chef::SendCookbooks.new
+      command.server = server
+      command.perform(message)
+    end
+
+  end
+
+end

data/test/unit/chef/sync_cookbooks_test.rb

@@ -0,0 +1,40 @@
+require 'unit/test_helper'
+
+describe Pantry::Chef::SyncCookbooks do
+
+  describe "#perform" do
+
+    it "requests to the server a list of cookbooks it should have" do
+      client = stub_everything
+      client.expects(:send_request).with do |message|
+        assert_equal "Chef::ListCookbooks", message.type
+      end.returns(mock(:value => Pantry::Message.new))
+
+      command = Pantry::Chef::SyncCookbooks.new
+      command.client = client
+      command.perform(Pantry::Message.new)
+    end
+
+    it "fires off receivers for each cookbook in the list and waits for downloads to finish" do
+      client = stub_everything
+      message = Pantry::Message.new
+      message << ["cookbook1", 0, "checksum1"]
+
+      client.stubs(:send_request).returns(stub(:value => message)).times(2)
+
+      client.expects(:receive_file).with(0, "checksum1").returns(
+        receive_info = Pantry::Communication::FileService::ReceivingFile.new(0, "checksum1", 1, 1)
+      )
+
+      receive_info.expects(:wait_for_finish)
+
+      command = Pantry::Chef::SyncCookbooks.new
+      command.client = client
+      command.perform(Pantry::Message.new)
+    end
+
+    it "deletes cookbooks stored locally not in the list (?)"
+
+  end
+
+end

data/test/unit/chef/sync_data_bags_test.rb

@@ -0,0 +1,21 @@
+require 'unit/test_helper'
+
+describe Pantry::Chef::SyncDataBags do
+
+  it "reads from the server directory for chef" do
+    cmd = Pantry::Chef::SyncDataBags.new
+    cmd.client = Pantry::ClientInfo.new(application: "pantry")
+
+    dir = cmd.server_directory(Pathname.new(""))
+    assert_equal "applications/pantry/chef/data_bags", dir.to_s
+  end
+
+  it "writes to the client's local chef dir" do
+    cmd = Pantry::Chef::SyncDataBags.new
+
+    dir = cmd.client_directory(Pathname.new(""))
+    assert_equal "chef/data_bags", dir.to_s
+  end
+
+end
+

data/test/unit/chef/sync_environments_test.rb

@@ -0,0 +1,21 @@
+require 'unit/test_helper'
+
+describe Pantry::Chef::SyncEnvironments do
+
+  it "reads from the server directory for chef" do
+    cmd = Pantry::Chef::SyncEnvironments.new
+    cmd.client = Pantry::ClientInfo.new(application: "pantry")
+
+    dir = cmd.server_directory(Pathname.new(""))
+    assert_equal "applications/pantry/chef/environments", dir.to_s
+  end
+
+  it "writes to the client's local chef dir" do
+    cmd = Pantry::Chef::SyncEnvironments.new
+
+    dir = cmd.client_directory(Pathname.new(""))
+    assert_equal "chef/environments", dir.to_s
+  end
+
+end
+

data/test/unit/chef/sync_roles_test.rb

@@ -0,0 +1,21 @@
+require 'unit/test_helper'
+
+describe Pantry::Chef::SyncRoles do
+
+  it "reads from the server directory for chef" do
+    cmd = Pantry::Chef::SyncRoles.new
+    cmd.client = Pantry::ClientInfo.new(application: "pantry")
+
+    dir = cmd.server_directory(Pathname.new(""))
+    assert_equal "applications/pantry/chef/roles", dir.to_s
+  end
+
+  it "writes to the client's local chef dir" do
+    cmd = Pantry::Chef::SyncRoles.new
+
+    dir = cmd.client_directory(Pathname.new(""))
+    assert_equal "chef/roles", dir.to_s
+  end
+
+end
+

data/test/unit/chef/upload_cookbook_test.rb

@@ -0,0 +1,132 @@
+require 'unit/test_helper'
+
+describe Pantry::Chef::UploadCookbook do
+
+  mock_ui!
+
+  def build_command(cookbook_name)
+    @command ||=
+      Pantry::Chef::UploadCookbook.new(fixture_path("cookbooks/#{cookbook_name}"))
+  end
+
+  describe "#prepare_message" do
+
+    after do
+      File.unlink(@command.cookbook_tarball) if @command && @command.cookbook_tarball
+    end
+
+    it "figures out name of the requested cookbook" do
+      command = build_command("mini")
+      message = command.prepare_message({})
+
+      assert_not_nil message, "Did not return a message"
+      assert_equal "mini", message[:cookbook_name]
+    end
+
+    it "tars up the cookbook, noting the size and a checksum of the file" do
+      command = build_command("mini")
+      message = command.prepare_message({})
+
+      assert message[:cookbook_size] > 0, "Did not calculate a size of the tarball"
+      assert_not_nil message[:cookbook_checksum], "Did not calculate a checksum"
+
+      assert_not_nil command.cookbook_tarball, "Did not save a pointer to the tarball"
+      assert File.exists?(command.cookbook_tarball), "No tarball found on the file system"
+    end
+
+    it "errors out if no metadata file" do
+      assert_raises Pantry::Chef::MissingMetadata do
+        command = build_command("bad")
+        command.prepare_message({})
+      end
+    end
+
+    it "errors if it can't find the cookbook" do
+      assert_raises Pantry::Chef::UnknownCookbook do
+        command = build_command("nonexist")
+        command.prepare_message({})
+      end
+    end
+  end
+
+  describe "#perform" do
+
+    let(:incoming_message) do
+      m = command.to_message
+      m[:cookbook_name]     = "testing"
+      m[:cookbook_size]     = 100
+      m[:cookbook_checksum] = "123abc"
+      m
+    end
+
+    let(:command) { build_command("mini") }
+
+    let(:receiver_info) {
+      stub(
+        :file_uuid => "abc123",
+        :receiver_uuid => "receiver",
+        :on_complete => nil
+      )
+    }
+
+    it "ensures a place exists for the uploaded cookbook to go" do
+      command.server_or_client = stub(:receive_file => receiver_info)
+      response = command.perform(incoming_message)
+
+      assert File.directory?(Pantry.root.join("chef", "cookbooks")),
+        "Did not create directory for the testing cookbook"
+    end
+
+    it "responds successfully and fires off a file upload receiver" do
+      server_mock = mock
+      server_mock.expects(:receive_file).with(100, "123abc").returns(receiver_info)
+
+      command.server_or_client = server_mock
+
+      response = command.perform(incoming_message)
+
+      assert_equal [true, "receiver", "abc123"], response
+    end
+
+  end
+
+  describe "#receive_server_response" do
+
+    let(:command) { build_command("mini") }
+
+    it "triggers a file upload actor with the cookbook tarball and message UUID" do
+      client = mock
+      client.expects(:send_file).with do |file, receiver_uuid|
+        assert File.exists?(file), "Did not find the file"
+        assert_equal "abc123", receiver_uuid
+      end.returns(mock(:wait_for_finish))
+
+      command.server_or_client = client
+      command.prepare_message({})
+
+      response_message = Pantry::Message.new
+      response_message.body << "true"
+      response_message.body << "abc123"
+
+      command.receive_server_response(response_message)
+    end
+
+    it "fails out with a message and cleans up if the server response with an error" do
+      client = mock
+      client.expects(:send_file).never
+
+      command.server_or_client = client
+      command.prepare_message({})
+
+      response_message = Pantry::Message.new
+      response_message << "false"
+      response_message << "Unable to Upload Reason"
+
+      command.receive_server_response(response_message)
+
+      assert_match /ERROR: Unable to Upload Reason/, stdout
+    end
+
+  end
+
+end