microwave 0.1006.33 → 1.0.0

data/spec/unit/search/query_spec.rb

@@ -0,0 +1,99 @@
+#
+# Author:: Adam Jacob (<adam@opscode.com>)
+# Copyright:: Copyright (c) 2009,2010 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 'spec_helper'
+require 'chef/search/query'
+
+describe Chef::Search::Query do 
+  before(:each) do
+    @rest = mock("Chef::REST")
+    Chef::REST.stub!(:new).and_return(@rest)
+    @query = Chef::Search::Query.new
+  end
+
+  describe "search" do
+    before(:each) do
+      @response = {
+        "rows" => [
+          { "id" => "for you" },
+          { "id" => "hip hop" },
+          { "id" => "thought was down by law for you" },
+          { "id" => "kept it hard core for you" },
+        ],
+        "start" => 0,
+        "total" => 4
+      }
+      @rest.stub!(:get_rest).and_return(@response)
+    end
+
+    it "should accept a type as the first argument" do
+      lambda { @query.search("foo") }.should_not raise_error(ArgumentError)
+      lambda { @query.search(:foo) }.should_not raise_error(ArgumentError)
+      lambda { @query.search(Hash.new) }.should raise_error(ArgumentError)
+    end
+
+    it "should query for every object of a type by default" do
+      @rest.should_receive(:get_rest).with("search/foo?q=*:*&sort=X_CHEF_id_CHEF_X%20asc&start=0&rows=1000").and_return(@response)
+      @query = Chef::Search::Query.new
+      @query.search(:foo)
+    end
+
+    it "should allow a custom query" do
+      @rest.should_receive(:get_rest).with("search/foo?q=gorilla:dundee&sort=X_CHEF_id_CHEF_X%20asc&start=0&rows=1000").and_return(@response)
+      @query = Chef::Search::Query.new
+      @query.search(:foo, "gorilla:dundee")
+    end
+
+    it "should let you set a sort order" do
+      @rest.should_receive(:get_rest).with("search/foo?q=gorilla:dundee&sort=id%20desc&start=0&rows=1000").and_return(@response)
+      @query = Chef::Search::Query.new
+      @query.search(:foo, "gorilla:dundee", "id desc")
+    end
+
+    it "should let you set a starting object" do
+      @rest.should_receive(:get_rest).with("search/foo?q=gorilla:dundee&sort=id%20desc&start=2&rows=1000").and_return(@response)
+      @query = Chef::Search::Query.new
+      @query.search(:foo, "gorilla:dundee", "id desc", 2)
+    end
+
+    it "should let you set how many rows to return" do
+      @rest.should_receive(:get_rest).with("search/foo?q=gorilla:dundee&sort=id%20desc&start=2&rows=40").and_return(@response)
+      @query = Chef::Search::Query.new
+      @query.search(:foo, "gorilla:dundee", "id desc", 2, 40)
+    end
+
+    it "should return the raw rows, start, and total if no block is passed" do
+      rows, start, total = @query.search(:foo)
+      rows.should equal(@response["rows"])
+      start.should equal(@response["start"])
+      total.should equal(@response["total"])
+    end
+
+    it "should call a block for each object in the response" do
+      @call_me = mock("blocky")
+      @response["rows"].each { |r| @call_me.should_receive(:do).with(r) }
+      @query.search(:foo) { |r| @call_me.do(r) }
+    end
+
+    it "should page through the responses" do
+      @call_me = mock("blocky")
+      @response["rows"].each { |r| @call_me.should_receive(:do).with(r) }
+      @query.search(:foo, "*:*", nil, 0, 1) { |r| @call_me.do(r) }
+    end
+  end
+end

data/spec/unit/shef/model_wrapper_spec.rb

@@ -0,0 +1,97 @@
+#
+# Author:: Daniel DeLeo (<dan@opscode.com>)
+# Copyright:: Copyright (c) 2010 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 'spec_helper'
+require 'ostruct'
+
+describe Shef::ModelWrapper do
+  before do
+    @model = OpenStruct.new(:name=>"Chef::Node")
+    @wrapper = Shef::ModelWrapper.new(@model)
+  end
+
+  describe "when created with an explicit model_symbol" do
+    before do
+      @model = OpenStruct.new(:name=>"Chef::ApiClient")
+      @wrapper = Shef::ModelWrapper.new(@model, :client)
+    end
+
+    it "uses the explicit model symbol" do
+      @wrapper.model_symbol.should == :client
+    end
+  end
+
+  it "determines the model symbol from the class name" do
+    @wrapper.model_symbol.should == :node
+  end
+
+  describe "when listing objects" do
+    before do
+      @node_1 = Chef::Node.new
+      @node_1.name("sammich")
+      @node_2 = Chef::Node.new
+      @node_2.name("yummy")
+      @server_response = {:node_1 => @node_1, :node_2 => @node_2}
+      @wrapper = Shef::ModelWrapper.new(Chef::Node)
+      Chef::Node.stub(:list).and_return(@server_response)
+    end
+
+    it "lists fully inflated objects without the resource IDs" do
+      @wrapper.all.should have(2).nodes
+      @wrapper.all.should include(@node_1, @node_2)
+    end
+
+    it "maps the listed nodes when given a block" do
+      @wrapper.all {|n| n.name }.sort.reverse.should == %w{yummy sammich}
+    end
+  end
+
+  describe "when searching for objects" do
+    before do
+      @node_1 = Chef::Node.new
+      @node_1.name("sammich")
+      @node_2 = Chef::Node.new
+      @node_2.name("yummy")
+      @server_response = {:node_1 => @node_1, :node_2 => @node_2}
+      @wrapper = Shef::ModelWrapper.new(Chef::Node)
+
+      # Creating a Chef::Search::Query object tries to read the private key...
+      @searcher = mock("Chef::Search::Query #{__FILE__}:#{__LINE__}")
+      Chef::Search::Query.stub!(:new).and_return(@searcher)
+    end
+
+    it "falls back to listing the objects when the 'query' is :all" do
+      Chef::Node.stub(:list).and_return(@server_response)
+      @wrapper.find(:all).should include(@node_1, @node_2)
+    end
+
+    it "searches for objects using the given query string" do
+      @searcher.should_receive(:search).with(:node, 'name:app*').and_yield(@node_1).and_yield(@node_2)
+      @wrapper.find("name:app*").should include(@node_1, @node_2)
+    end
+
+    it "creates a 'AND'-joined query string from a HASH" do
+      # Hash order woes
+      @searcher.should_receive(:search).with(:node, 'name:app* AND name:app*').and_yield(@node_1).and_yield(@node_2)
+      @wrapper.find(:name=>"app*",'name'=>"app*").should include(@node_1, @node_2)
+    end
+
+  end
+
+
+end

data/spec/unit/shef/shef_ext_spec.rb

@@ -0,0 +1,153 @@
+# Author:: Daniel DeLeo (<dan@kallistec.com>)
+# Copyright:: Copyright (c) 2009 Daniel DeLeo
+# Copyright:: Copyright (c) 2010 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 'spec_helper'
+
+describe Shef::Extensions do
+  describe "extending object for top level methods" do
+
+    before do
+      @shef_client = TestableShefSession.instance
+      Shef.stub!(:session).and_return(@shef_client)
+      @job_manager = TestJobManager.new
+      @root_context = Object.new
+      @root_context.instance_eval(&ObjectTestHarness)
+      Shef::Extensions.extend_context_object(@root_context)
+      @root_context.conf = mock("irbconf")
+    end
+
+    it "finds a subsession in irb for an object" do
+      target_context_obj = Chef::Node.new
+
+      irb_context = mock("context", :main => target_context_obj)
+      irb_session = mock("irb session", :context => irb_context)
+      @job_manager.jobs = [[:thread, irb_session]]
+      @root_context.stub!(:jobs).and_return(@job_manager)
+      @root_context.ensure_session_select_defined
+      @root_context.jobs.select_shef_session(target_context_obj).should == irb_session
+      @root_context.jobs.select_shef_session(:idontexist).should be_nil
+    end
+
+    it "finds, then switches to a session" do
+      @job_manager.jobs = []
+      @root_context.stub!(:ensure_session_select_defined)
+      @root_context.stub!(:jobs).and_return(@job_manager)
+      @job_manager.should_receive(:select_shef_session).and_return(:the_shef_session)
+      @job_manager.should_receive(:switch).with(:the_shef_session)
+      @root_context.find_or_create_session_for(:foo)
+    end
+
+    it "creates a new session if an existing one isn't found" do
+      @job_manager.jobs = []
+      @root_context.stub!(:jobs).and_return(@job_manager)
+      @job_manager.stub!(:select_shef_session).and_return(nil)
+      @root_context.should_receive(:irb).with(:foo)
+      @root_context.find_or_create_session_for(:foo)
+    end
+
+    it "switches to recipe context" do
+      @root_context.should respond_to(:recipe)
+      @shef_client.recipe = :monkeyTime
+      @root_context.should_receive(:find_or_create_session_for).with(:monkeyTime)
+      @root_context.recipe
+    end
+
+    it "switches to attribute context" do
+      @root_context.should respond_to(:attributes)
+      @shef_client.node = "monkeyNodeTime"
+      @root_context.should_receive(:find_or_create_session_for).with("monkeyNodeTime")
+      @root_context.attributes
+    end
+
+    it "has a help command" do
+      @root_context.should respond_to(:help)
+    end
+
+    it "turns irb tracing on and off" do
+      @root_context.should respond_to(:trace)
+      @root_context.conf.should_receive(:use_tracer=).with(true)
+      @root_context.stub!(:tracing?)
+      @root_context.tracing :on
+    end
+
+    it "says if tracing is on or off" do
+      @root_context.conf.stub!(:use_tracer).and_return(true)
+      @root_context.should_receive(:puts).with("tracing is on")
+      @root_context.tracing?
+    end
+
+    it "prints node attributes" do
+      node = mock("node", :attribute => {:foo => :bar})
+      @shef_client.node = node
+      @root_context.should_receive(:pp).with({:foo => :bar})
+      @root_context.ohai
+      @root_context.should_receive(:pp).with(:bar)
+      @root_context.ohai(:foo)
+    end
+
+    it "resets the recipe and reloads ohai data" do
+      @shef_client.should_receive(:reset!)
+      @root_context.reset
+    end
+
+    it "turns irb echo on and off" do
+      @root_context.conf.should_receive(:echo=).with(true)
+      @root_context.echo :on
+    end
+
+    it "says if echo is on or off" do
+      @root_context.conf.stub!(:echo).and_return(true)
+      @root_context.should_receive(:puts).with("echo is on")
+      @root_context.echo?
+    end
+
+    it "gives access to the stepable iterator" do
+      Shef::StandAloneSession.instance.stub!(:reset!)
+      Shef.session.stub!(:rebuild_context)
+      events = Chef::EventDispatch::Dispatcher.new
+      run_context = Chef::RunContext.new(Chef::Node.new, {}, events)
+      run_context.resource_collection.instance_variable_set(:@iterator, :the_iterator)
+      Shef.session.run_context = run_context
+      @root_context.chef_run.should == :the_iterator
+    end
+
+    it "lists directory contents" do
+      entries = %w{. .. someFile}
+      Dir.should_receive(:entries).with("/tmp").and_return(entries)
+      @root_context.ls "/tmp"
+    end
+
+  end
+
+  describe "extending the recipe object" do
+
+    before do
+      @events = Chef::EventDispatch::Dispatcher.new
+      @run_context = Chef::RunContext.new(Chef::Node.new, {}, @events)
+      @recipe_object = Chef::Recipe.new(nil, nil, @run_context)
+      Shef::Extensions.extend_context_recipe(@recipe_object)
+    end
+
+    it "gives a list of the resources" do
+      resource = @recipe_object.file("foo")
+      @recipe_object.should_receive(:pp).with(["file[foo]"])
+      @recipe_object.resources
+    end
+
+  end
+end

data/spec/unit/shef/shef_session_spec.rb

@@ -0,0 +1,141 @@
+# Author:: Daniel DeLeo (<dan@kallistec.com>)
+# Copyright:: Copyright (c) 2009 Daniel DeLeo
+# 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 'spec_helper'
+require "ostruct"
+
+
+class TestableShefSession < Shef::ShefSession
+
+  def rebuild_node
+    nil
+  end
+
+  def rebuild_collection
+    nil
+  end
+
+  def loading
+    nil
+  end
+
+  def loading_complete
+    nil
+  end
+
+end
+
+describe Shef::ShefSession do
+
+  it "is a singleton object" do
+    Shef::ShefSession.should include(Singleton)
+  end
+
+end
+
+describe Shef::StandAloneSession do
+  before do
+    @session = Shef::StandAloneSession.instance
+    @node = @session.node = Chef::Node.new
+    @events = Chef::EventDispatch::Dispatcher.new
+    @run_context = @session.run_context = Chef::RunContext.new(@node, {}, @events)
+    @recipe = @session.recipe = Chef::Recipe.new(nil, nil, @run_context)
+    Shef::Extensions.extend_context_recipe(@recipe)
+  end
+
+  it "has a run_context" do
+    @session.run_context.should equal(@run_context)
+  end
+
+  it "returns a collection based on it's standalone recipe file" do
+    @session.resource_collection.should == @recipe.run_context.resource_collection
+  end
+
+  it "gives nil for the definitions (for now)" do
+    @session.definitions.should be_nil
+  end
+
+  it "gives nil for the cookbook_loader" do
+    @session.cookbook_loader.should be_nil
+  end
+
+  it "runs chef with the standalone recipe" do
+    @session.stub!(:node_built?).and_return(true)
+    Chef::Log.stub!(:level)
+    chef_runner = mock("Chef::Runner.new", :converge => :converged)
+    # pre-heat resource collection cache
+    @session.resource_collection
+
+    Chef::Runner.should_receive(:new).with(@session.recipe.run_context).and_return(chef_runner)
+    @recipe.run_chef.should == :converged
+  end
+
+end
+
+describe Shef::SoloSession do
+  before do
+    Chef::Config[:shef_solo] = true
+    @session = Shef::SoloSession.instance
+    @node = Chef::Node.new
+    @events = Chef::EventDispatch::Dispatcher.new
+    @run_context = @session.run_context = Chef::RunContext.new(@node, {}, @events)
+    @session.node = @node
+    @recipe = @session.recipe = Chef::Recipe.new(nil, nil, @run_context)
+    Shef::Extensions.extend_context_recipe(@recipe)
+  end
+
+  after do
+    Chef::Config[:shef_solo] = nil
+  end
+
+  it "returns a collection based on it's compilation object and the extra recipe provided by shef" do
+    @session.stub!(:node_built?).and_return(true)
+    kitteh = Chef::Resource::Cat.new("keyboard")
+    @recipe.run_context.resource_collection << kitteh
+    @session.resource_collection.should include(kitteh)
+  end
+
+  it "returns definitions from it's compilation object" do
+    @session.definitions.should == @run_context.definitions
+  end
+
+  it "keeps json attribs and passes them to the node for consumption" do
+    @session.node_attributes = {"besnard_lakes" => "are_the_dark_horse"}
+    @session.node.besnard_lakes.should == "are_the_dark_horse"
+    #pending "1) keep attribs in an ivar 2) pass them to the node 3) feed them to the node on reset"
+  end
+
+  it "generates it's resource collection from the compiled cookbooks and the ad hoc recipe" do
+    @session.stub!(:node_built?).and_return(true)
+    kitteh_cat = Chef::Resource::Cat.new("kitteh")
+    @run_context.resource_collection << kitteh_cat
+    keyboard_cat = Chef::Resource::Cat.new("keyboard_cat")
+    @recipe.run_context.resource_collection << keyboard_cat
+    #@session.rebuild_collection
+    @session.resource_collection.should include(kitteh_cat, keyboard_cat)
+  end
+
+  it "runs chef with a resource collection from the compiled cookbooks" do
+    @session.stub!(:node_built?).and_return(true)
+    Chef::Log.stub!(:level)
+    chef_runner = mock("Chef::Runner.new", :converge => :converged)
+    Chef::Runner.should_receive(:new).with(an_instance_of(Chef::RunContext)).and_return(chef_runner)
+
+    @recipe.run_chef.should == :converged
+  end
+
+end