diff options
Diffstat (limited to 'spec/unit/knife/cookbook_upload_spec.rb')
-rw-r--r-- | spec/unit/knife/cookbook_upload_spec.rb | 183 |
1 files changed, 183 insertions, 0 deletions
diff --git a/spec/unit/knife/cookbook_upload_spec.rb b/spec/unit/knife/cookbook_upload_spec.rb new file mode 100644 index 0000000000..4659e60371 --- /dev/null +++ b/spec/unit/knife/cookbook_upload_spec.rb @@ -0,0 +1,183 @@ +# +# Author:: Matthew Kent (<mkent@magoazul.com>) +# Author:: Steven Danna (<steve@opscode.com>) +# Copyright:: Copyright (c) 2012 Opscode, Inc. +# License:: Apache License, Version 2.0 +# +# Licensed under the Apache License, Version 2.0 (the "License"); +# you may not use this file except in compliance with the License. +# You may obtain a copy of the License at +# +# http://www.apache.org/licenses/LICENSE-2.0 +# +# Unless required by applicable law or agreed to in writing, software +# distributed under the License is distributed on an "AS IS" BASIS, +# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. +# See the License for the specific language governing permissions and +# limitations under the License. +# + +require File.expand_path(File.join(File.dirname(__FILE__), "..", "..", "spec_helper")) + +require 'chef/cookbook_uploader' +require 'timeout' + +describe Chef::Knife::CookbookUpload do + before(:each) do + @knife = Chef::Knife::CookbookUpload.new + @knife.name_args = ['test_cookbook'] + + @cookbook = Chef::CookbookVersion.new('test_cookbook') + + @cookbook_loader = {} + @cookbook_loader.stub!(:[]).and_return(@cookbook) + @cookbook_loader.stub!(:merged_cookbooks).and_return([]) + @cookbook_loader.stub!(:load_cookbooks).and_return(@cookbook_loader) + Chef::CookbookLoader.stub!(:new).and_return(@cookbook_loader) + + @output = StringIO.new + @knife.ui.stub!(:stdout).and_return(@output) + @knife.ui.stub!(:stderr).and_return(@output) + end + + describe 'run' do + before(:each) do + @knife.stub!(:upload).and_return(true) + Chef::CookbookVersion.stub(:list_all_versions).and_return({}) + end + + it 'should print usage and exit when a cookbook name is not provided' do + @knife.name_args = [] + @knife.should_receive(:show_usage) + @knife.ui.should_receive(:fatal) + lambda { @knife.run }.should raise_error(SystemExit) + end + + describe 'when specifying a cookbook name' do + it 'should upload the cookbook' do + @knife.should_receive(:upload).once + @knife.run + end + + it 'should report on success' do + @knife.should_receive(:upload).once + @knife.ui.should_receive(:info).with(/Uploaded 1 cookbook/) + @knife.run + end + end + + describe 'when specifying the same cookbook name twice' do + it 'should upload the cookbook only once' do + @knife.name_args = ['test_cookbook', 'test_cookbook'] + @knife.should_receive(:upload).once + @knife.run + end + end + + describe 'when specifying a cookbook name among many' do + before(:each) do + @knife.name_args = ['test_cookbook1'] + @cookbooks = { + 'test_cookbook1' => Chef::CookbookVersion.new('test_cookbook1'), + 'test_cookbook2' => Chef::CookbookVersion.new('test_cookbook2'), + 'test_cookbook3' => Chef::CookbookVersion.new('test_cookbook3') + } + @cookbook_loader = {} + @cookbook_loader.stub!(:merged_cookbooks).and_return([]) + @cookbook_loader.stub(:[]) { |ckbk| @cookbooks[ckbk] } + Chef::CookbookLoader.stub!(:new).and_return(@cookbook_loader) + end + + it "should read only one cookbook" do + @cookbook_loader.should_receive(:[]).once.with('test_cookbook1') + @knife.run + end + + it "should not read all cookbooks" do + @cookbook_loader.should_not_receive(:load_cookbooks) + @knife.run + end + + it "should upload only one cookbook" do + @knife.should_receive(:upload).exactly(1).times + @knife.run + end + end + + # This is testing too much. We should break it up. + describe 'when specifying a cookbook name with dependencies' do + it "should upload all dependencies once" do + @knife.name_args = ["test_cookbook2"] + @knife.config[:depends] = true + @test_cookbook1 = Chef::CookbookVersion.new('test_cookbook1') + @test_cookbook2 = Chef::CookbookVersion.new('test_cookbook2') + @test_cookbook3 = Chef::CookbookVersion.new('test_cookbook3') + @test_cookbook2.metadata.depends("test_cookbook3") + @test_cookbook3.metadata.depends("test_cookbook1") + @test_cookbook3.metadata.depends("test_cookbook2") + @cookbook_loader.stub!(:[]) do |ckbk| + { "test_cookbook1" => @test_cookbook1, + "test_cookbook2" => @test_cookbook2, + "test_cookbook3" => @test_cookbook3 }[ckbk] + end + @knife.stub!(:cookbook_names).and_return(["test_cookbook1", "test_cookbook2", "test_cookbook3"]) + @knife.should_receive(:upload).exactly(3).times + Timeout::timeout(5) do + @knife.run + end.should_not raise_error(Timeout::Error) + end + end + + it "should freeze the version of the cookbooks if --freeze is specified" do + @knife.config[:freeze] = true + @cookbook.should_receive(:freeze_version).once + @knife.run + end + + describe 'with -a or --all' do + before(:each) do + @knife.config[:all] = true + @test_cookbook1 = Chef::CookbookVersion.new('test_cookbook1') + @test_cookbook2 = Chef::CookbookVersion.new('test_cookbook2') + @cookbook_loader.stub!(:each).and_yield("test_cookbook1", @test_cookbook1).and_yield("test_cookbook2", @test_cookbook2) + @cookbook_loader.stub!(:cookbook_names).and_return(["test_cookbook1", "test_cookbook2"]) + end + + it 'should upload all cookbooks' do + @knife.should_receive(:upload).once + @knife.run + end + + it 'should report on success' do + @knife.should_receive(:upload).once + @knife.ui.should_receive(:info).with(/Uploaded all cookbooks/) + @knife.run + end + + it 'should update the version constraints for an environment' do + @knife.stub!(:assert_environment_valid!).and_return(true) + @knife.config[:environment] = "production" + @knife.should_receive(:update_version_constraints).once + @knife.run + end + end + + describe 'when a frozen cookbook exists on the server' do + it 'should fail to replace it' do + @knife.stub!(:upload).and_raise(Chef::Exceptions::CookbookFrozen) + @knife.ui.should_receive(:error).with(/Failed to upload 1 cookbook/) + lambda { @knife.run }.should raise_error(SystemExit) + end + + it 'should not update the version constraints for an environment' do + @knife.stub!(:assert_environment_valid!).and_return(true) + @knife.config[:environment] = "production" + @knife.stub!(:upload).and_raise(Chef::Exceptions::CookbookFrozen) + @knife.ui.should_receive(:error).with(/Failed to upload 1 cookbook/) + @knife.ui.should_receive(:warn).with(/Not updating version constraints/) + @knife.should_not_receive(:update_version_constraints) + lambda { @knife.run }.should raise_error(SystemExit) + end + end + end # run +end # Chef::Knife::CookbookUpload |