microwave 0.1006.33 → 1.0.0

data/spec/unit/couchdb_spec.rb

@@ -0,0 +1,274 @@
+#
+# Author:: Adam Jacob (<adam@opscode.com>)
+# Copyright:: Copyright (c) 2008 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 Chef::CouchDB do
+  before(:each) do
+    Chef::Config[:couchdb_database] = "chef"
+    @rest = mock("Chef::REST")
+    @rest.stub!(:run_request).and_return({"couchdb" => "Welcome", "version" =>"0.9.0"})
+    @rest.stub!(:url).and_return("http://localhost:5984")
+    Chef::REST.stub!(:new).and_return(@rest)
+    @couchdb = Chef::CouchDB.new
+  end
+
+  describe "new" do
+    it "should create a new Chef::REST object from the default url" do
+      old_url = Chef::Config[:couchdb_url]
+      Chef::Config[:couchdb_url] = "http://monkey"
+      Chef::REST.should_receive(:new).with("http://monkey", nil, nil)
+      Chef::CouchDB.new
+      Chef::Config[:couchdb_url] = old_url
+    end
+
+    it "should create a new Chef::REST object from a provided url" do
+      Chef::REST.should_receive(:new).with("http://monkeypants", nil, nil)
+      Chef::CouchDB.new("http://monkeypants")
+    end
+  end
+
+  describe "create_db" do
+    before(:each) do
+      @couchdb.stub!(:create_design_document).and_return(true)
+    end
+
+    it "should get a list of current databases" do
+      @rest.should_receive(:get_rest).and_return(["chef"])
+      @couchdb.create_db
+    end
+
+    it "should create the chef database if it does not exist" do
+      @rest.stub!(:get_rest).and_return([])
+      @rest.should_receive(:put_rest).with("chef", {}).and_return(true)
+      @couchdb.create_db
+    end
+
+    it "should not create the chef database if it does exist" do
+      @rest.stub!(:get_rest).and_return(["chef"])
+      @rest.should_not_receive(:put_rest)
+      @couchdb.create_db
+    end
+
+    it "should return 'chef'" do
+      @rest.should_receive(:get_rest).with("_all_dbs").and_return(%w{chef})
+      @couchdb.create_db.should eql("chef")
+    end
+  end
+
+  describe "create_design_document" do
+    before(:each) do
+      @mock_design = {
+        "version" => 1,
+        "_rev" => 1
+      }
+      @mock_data = {
+        "version" => 1,
+        "language" => "javascript",
+        "views" => {
+          "all" => {
+            "map" => <<-EOJS
+            function(doc) {
+              if (doc.chef_type == "node") {
+                emit(doc.name, doc);
+              }
+            }
+            EOJS
+          },
+        }
+      }
+      @rest.stub!(:get_rest).and_return(@mock_design)
+      @rest.stub!(:put_rest).and_return(true)
+      @couchdb.stub!(:create_db).and_return(true)
+    end
+
+    def do_create_design_document
+      @couchdb.create_design_document("bob", @mock_data)
+    end
+
+    it "should fetch the existing design document" do
+      @rest.should_receive(:get_rest).with("chef/_design/bob")
+      do_create_design_document
+    end
+
+    it "should populate the _rev in the new design if the versions dont match" do
+      @mock_data["version"] = 2
+      do_create_design_document
+      @mock_data["_rev"].should eql(1)
+    end
+
+    it "should create the view if it requires updating" do
+      @mock_data["version"] = 2
+      @rest.should_receive(:put_rest).with("chef/_design%2Fbob", @mock_data)
+      do_create_design_document
+    end
+
+    it "should not create the view if it does not require updating" do
+      @mock_data["version"] = 1
+      @rest.should_not_receive(:put_rest)
+      do_create_design_document
+    end
+  end
+
+  describe "store" do
+    before(:each) do
+      @mock_results = {
+        "rows" => [
+          "id" => 'a0934635-e111-45d9-8223-cb58e1c9434c'
+        ]
+      }
+      @couchdb.stub!(:get_view).with("id_map", "name_to_id", :key => [ "node", "bob" ]).and_return(@mock_results)
+    end
+
+    it "should put the object into couchdb with a pre-existing GUID" do
+      item_to_store = {}
+      item_to_store.should_receive(:add_to_index)
+      @rest.should_receive(:put_rest).with("chef/#{@mock_results["rows"][0]["id"]}", item_to_store).and_return(true)
+      @couchdb.store("node", "bob", item_to_store)
+    end
+
+    it "should put the object into couchdb with a new GUID" do
+      @mock_results = { "rows" => [] }
+      item_to_store = {}
+      item_to_store.should_receive(:add_to_index).with(:database => "chef", :id => "aaaaaaaa-xxxx-xxxx-xxxx-xxxxxxxxxxx", :type => "node")
+      @couchdb.stub!(:get_view).with("id_map", "name_to_id", :key => [ "node", "bob" ]).and_return(@mock_results)
+      UUIDTools::UUID.stub!(:random_create).and_return("aaaaaaaa-xxxx-xxxx-xxxx-xxxxxxxxxxx")
+      @rest.should_receive(:put_rest).with("chef/aaaaaaaa-xxxx-xxxx-xxxx-xxxxxxxxxxx", item_to_store).and_return(true)
+      @couchdb.store("node", "bob", item_to_store)
+    end
+
+  end
+
+  describe "when fetching the database status" do
+    it "gets couchdb's version string'" do
+      @rest.should_receive(:get_rest).with('/').and_return({"couchdb" => "Welcome","version" => "1.0.1"})
+      @couchdb.server_stats.should == {"couchdb" => "Welcome","version" => "1.0.1"}
+    end
+
+    it "gets database stats" do
+      db_stats = {"db_name" => "opscode_account","doc_count" => 206,"doc_del_count" => 1,"update_seq" => 208,"purge_seq" => 0,
+                  "compact_running" => false,"disk_size" => 122969,"instance_start_time" => "1298070021394804","disk_format_version" => 5,"committed_update_seq" => 208}
+      @rest.should_receive(:get_rest).with('/chef').and_return(db_stats)
+      @couchdb.db_stats.should == db_stats
+    end
+
+  end
+
+  describe "load" do
+    before(:each) do
+      @mock_node = Chef::Node.new()
+      @mock_node.name("bob")
+      @couchdb.stub!(:find_by_name).with("node", "bob").and_return(@mock_node)
+    end
+
+    it "should load the object from couchdb" do
+      @couchdb.load("node", "bob").should eql(@mock_node)
+    end
+  end
+
+  describe "delete" do
+    before(:each) do
+      @mock_current = {
+        "version" => 1,
+        "_rev" => 1
+      }
+      @rest.stub!(:get_rest).and_return(@mock_current)
+      @rest.stub!(:delete_rest).and_return(true)
+      @node = Chef::Node.new()
+      @node.name("bob")
+      @node.couchdb_rev = 15
+      @couchdb.stub!(:find_by_name).with("node", "bob", true).and_return([ @node, "ax" ])
+    end
+
+    def do_delete(rev=nil)
+      @couchdb.delete("node", "bob", rev)
+    end
+
+    it "should remove the object from couchdb with a specific revision" do
+      @node.should_receive(:delete_from_index)
+      @rest.should_receive(:delete_rest).with("chef/ax?rev=1")
+      do_delete(1)
+    end
+
+    it "should remove the object from couchdb based on the couchdb_rev of the current obj" do
+      @node.should_receive(:delete_from_index)
+      @rest.should_receive(:delete_rest).with("chef/ax?rev=15")
+      do_delete
+    end
+  end
+
+  describe "list" do
+    before(:each) do
+      Chef::Config.stub!(:[]).with(:couchdb_database).and_return("chef")
+      @mock_response = {"rows" => []}
+    end
+
+    describe "on couchdb 0.9+" do
+      before do
+        Chef::Config.stub!(:[]).with(:couchdb_version).and_return(0.9)
+      end
+
+      it "should get the view for all objects if inflate is true" do
+        @rest.should_receive(:get_rest).with("chef/_design/node/_view/all").and_return(@mock_response)
+        @couchdb.list("node", true)
+      end
+
+      it "should get the view for just the object id's if inflate is false" do
+        @rest.should_receive(:get_rest).with("chef/_design/node/_view/all_id").and_return(@mock_response)
+        @couchdb.list("node", false)
+      end
+    end
+  end
+
+  describe "has_key?" do
+    it "should return true if the object exists" do
+      @couchdb.stub!(:find_by_name).with("node", "bob").and_return(true)
+      @couchdb.has_key?("node", "bob").should eql(true)
+    end
+
+    it "should return false if the object does not exist" do
+      @couchdb.stub!(:find_by_name).and_raise(Chef::Exceptions::CouchDBNotFound)
+      @couchdb.has_key?("node", "bob").should eql(false)
+    end
+  end
+
+  describe "get_view" do
+    it "should construct a call to the view for the proper design document" do
+      @rest.should_receive(:get_rest).with("chef/_design/nodes/_view/mastodon")
+      @couchdb.get_view("nodes", "mastodon")
+    end
+
+    it "should allow arguments to the view" do
+      @rest.should_receive(:get_rest).with("chef/_design/nodes/_view/mastodon?startkey=%22dont%20stay%22")
+      @couchdb.get_view("nodes", "mastodon", :startkey => "dont stay")
+    end
+
+  end
+
+  describe "view_uri" do
+    it "should output an appropriately formed view URI" do
+      @couchdb.should_receive(:view_uri).with("nodes", "all").and_return("chef/_design/nodes/_view/all")
+      @couchdb.view_uri("nodes", "all")
+    end
+  end
+
+end
+
+
+
+

data/spec/unit/daemon_spec.rb

@@ -0,0 +1,281 @@
+#
+# Author:: AJ Christensen (<aj@junglist.gen.nz>)
+# Copyright:: Copyright (c) 2008 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 Chef::Daemon do
+  before do
+    @original_config = Chef::Config.configuration
+    if windows?
+      mock_struct = #Struct::Passwd.new(nil, nil, 111, 111)
+      mock_struct = OpenStruct.new(:uid => 2342, :gid => 2342)
+      Etc.stub!(:getpwnam).and_return mock_struct
+      Etc.stub!(:getgrnam).and_return mock_struct
+      # mock unimplemented methods
+      Process.stub!(:initgroups).and_return nil
+      Process::GID.stub!(:change_privilege).and_return 11
+      Process::UID.stub!(:change_privilege).and_return 11
+    end
+  end
+
+  after do
+    Chef::Config.configuration.replace(@original_config)
+  end
+
+  describe ".running?" do
+
+    before do
+      Chef::Daemon.name = "spec"
+    end
+
+    describe "when a pid file exists" do
+
+      before do
+        Chef::Daemon.stub!(:pid_from_file).and_return(1337)
+      end
+
+      it "should check that there is a process matching the pidfile" do
+        Process.should_receive(:kill).with(0, 1337)
+        Chef::Daemon.running?
+      end
+
+    end
+
+    describe "when the pid file is nonexistent" do
+
+      before do
+        Chef::Daemon.stub!(:pid_from_file).and_return(nil)
+      end
+
+      it "should return false" do
+        Chef::Daemon.running?.should be_false
+      end
+
+    end
+  end
+
+  describe ".pid_file" do
+
+    describe "when the pid_file option has been set" do
+
+      before do
+        Chef::Config[:pid_file] = "/var/run/chef/chef-client.pid"
+      end
+
+      after do
+        Chef::Config.configuration.replace(@original_config)
+      end
+
+      it "should return the supplied value" do
+        Chef::Daemon.pid_file.should eql("/var/run/chef/chef-client.pid")
+      end
+    end
+
+    describe "without the pid_file option set" do
+
+      before do
+        Chef::Config[:pid_file] = nil
+        Chef::Daemon.name = "chef-client"
+      end
+
+      it "should return a valued based on @name" do
+        Chef::Daemon.pid_file.should eql("/tmp/chef-client.pid")
+      end
+
+    end
+  end
+
+  describe ".pid_from_file" do
+
+    before do
+      Chef::Config[:pid_file] = "/var/run/chef/chef-client.pid"
+    end
+
+    it "should suck the pid out of pid_file" do
+      File.should_receive(:read).with("/var/run/chef/chef-client.pid").and_return("1337")
+      Chef::Daemon.pid_from_file
+    end
+  end
+
+  describe ".save_pid_file" do
+
+    before do
+      Process.stub!(:pid).and_return(1337)
+      Chef::Config[:pid_file] = "/var/run/chef/chef-client.pid"
+      Chef::Application.stub!(:fatal!).and_return(true)
+      @f_mock = mock(File, { :print => true, :close => true, :write => true })
+      File.stub!(:open).with("/var/run/chef/chef-client.pid", "w").and_yield(@f_mock)
+    end
+
+    it "should try and create the parent directory" do
+      FileUtils.should_receive(:mkdir_p).with("/var/run/chef")
+      Chef::Daemon.save_pid_file
+    end
+
+    it "should open the pid file for writing" do
+      File.should_receive(:open).with("/var/run/chef/chef-client.pid", "w")
+      Chef::Daemon.save_pid_file
+    end
+
+    it "should write the pid, converted to string, to the pid file" do
+      @f_mock.should_receive(:write).with("1337").once.and_return(true)
+      Chef::Daemon.save_pid_file
+    end
+
+  end
+
+  describe ".remove_pid_file" do
+    before do
+      Chef::Config[:pid_file] = "/var/run/chef/chef-client.pid"
+    end
+
+    describe "when the pid file exists" do
+
+      before do
+        File.stub!(:exists?).with("/var/run/chef/chef-client.pid").and_return(true)
+      end
+
+      it "should remove the file" do
+        FileUtils.should_receive(:rm).with("/var/run/chef/chef-client.pid")
+        Chef::Daemon.remove_pid_file
+      end
+
+
+    end
+
+    describe "when the pid file does not exist" do
+
+      before do
+        File.stub!(:exists?).with("/var/run/chef/chef-client.pid").and_return(false)
+      end
+
+      it "should not remove the file" do
+        FileUtils.should_not_receive(:rm)
+        Chef::Daemon.remove_pid_file
+      end
+
+    end
+  end
+
+  describe ".change_privilege" do
+
+    before do
+      Chef::Application.stub!(:fatal!).and_return(true)
+      Chef::Config[:user] = 'aj'
+      Dir.stub!(:chdir)
+    end
+
+    it "changes the working directory to root" do
+      Dir.rspec_reset
+      Dir.should_receive(:chdir).with("/").and_return(0)
+      Chef::Daemon.change_privilege
+    end
+
+    describe "when the user and group options are supplied" do
+
+      before do
+        Chef::Config[:group] = 'staff'
+      end
+
+      it "should log an appropriate info message" do
+        Chef::Log.should_receive(:info).with("About to change privilege to aj:staff")
+        Chef::Daemon.change_privilege
+      end
+
+      it "should call _change_privilege with the user and group" do
+        Chef::Daemon.should_receive(:_change_privilege).with("aj", "staff")
+        Chef::Daemon.change_privilege
+      end
+    end
+
+    describe "when just the user option is supplied" do
+      before do
+        Chef::Config[:group] = nil
+      end
+
+      it "should log an appropriate info message" do
+        Chef::Log.should_receive(:info).with("About to change privilege to aj")
+        Chef::Daemon.change_privilege
+      end
+
+      it "should call _change_privilege with just the user" do
+        Chef::Daemon.should_receive(:_change_privilege).with("aj")
+        Chef::Daemon.change_privilege
+      end
+    end
+  end
+
+  describe "._change_privilege" do
+
+    before do
+      Process.stub!(:euid).and_return(0)
+      Process.stub!(:egid).and_return(0)
+
+      Process::UID.stub!(:change_privilege).and_return(nil)
+      Process::GID.stub!(:change_privilege).and_return(nil)
+
+      @pw_user = mock("Struct::Passwd", :uid => 501)
+      @pw_group = mock("Struct::Group", :gid => 20)
+
+      Process.stub!(:initgroups).and_return(true)
+
+      Etc.stub!(:getpwnam).and_return(@pw_user)
+      Etc.stub!(:getgrnam).and_return(@pw_group)
+    end
+
+    describe "with sufficient privileges" do
+      before do
+        Process.stub!(:euid).and_return(0)
+        Process.stub!(:egid).and_return(0)
+      end
+
+      it "should initialize the supplemental group list" do
+        Process.should_receive(:initgroups).with("aj", 20)
+        Chef::Daemon._change_privilege("aj")
+      end
+
+      it "should attempt to change the process GID" do
+        Process::GID.should_receive(:change_privilege).with(20).and_return(20)
+        Chef::Daemon._change_privilege("aj")
+      end
+
+      it "should attempt to change the process UID" do
+        Process::UID.should_receive(:change_privilege).with(501).and_return(501)
+        Chef::Daemon._change_privilege("aj")
+      end
+    end
+
+    describe "with insufficient privileges" do
+      before do
+        Process.stub!(:euid).and_return(999)
+        Process.stub!(:egid).and_return(999)
+      end
+
+      it "should log an appropriate error message and fail miserably" do
+        Process.stub!(:initgroups).and_raise(Errno::EPERM)
+        error = "Operation not permitted"
+        if RUBY_PLATFORM.match("solaris2")
+          error = "Not owner"
+        end
+        Chef::Application.should_receive(:fatal!).with("Permission denied when trying to change 999:999 to 501:20. #{error}")
+        Chef::Daemon._change_privilege("aj")
+      end
+    end
+
+  end
+end