1 files changed, 212 insertions, 0 deletions
diff --git a/chef/spec/unit/couchdb_spec.rb b/chef/spec/unit/couchdb_spec.rb
new file mode 100644
index 0000000000..621439df51
--- /dev/null
+++ b/chef/spec/unit/couchdb_spec.rb
@@ -0,0 +1,212 @@
+#
+# Author:: Adam Jacob (<adam@hjksolutions.com>)
+# Copyright:: Copyright (c) 2008 HJK Solutions, LLC
+# 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"))
+
+describe Chef::CouchDB, "new" do
+  it "should create a new Chef::REST object from the default url" do
+    Chef::Config[:couchdb_url] = "http://monkey"
+    Chef::REST.should_receive(:new).with("http://monkey").and_return(true)
+    Chef::CouchDB.new
+  end
+  
+  it "should create a new Chef::REST object from a provided url" do
+    Chef::REST.should_receive(:new).with("http://monkeypants").and_return(true)
+    Chef::CouchDB.new("http://monkeypants")
+  end
+end
+
+describe Chef::CouchDB, "create_db" do
+  before(:each) do
+    @mock_rest = mock("Chef::REST", :null_object => true)
+    @mock_rest.stub!(:get_rest).and_return([ "chef" ])
+    @mock_rest.stub!(:put_rest).and_return(true)
+    Chef::REST.stub!(:new).and_return(@mock_rest)
+  end
+  
+  def do_create_db
+    couch = Chef::CouchDB.new
+    couch.create_db
+  end
+  
+  it "should get a list of current databases" do
+    @mock_rest.should_receive(:get_rest).and_return(["chef"])
+    do_create_db
+  end
+  
+  it "should create the chef database if it does not exist" do
+    @mock_rest.stub!(:get_rest).and_return([])
+    @mock_rest.should_receive(:put_rest).with("chef", {}).and_return(true)
+    do_create_db
+  end
+  
+  it "should not create the chef database if it does exist" do
+    @mock_rest.stub!(:get_rest).and_return(["chef"])
+    @mock_rest.should_not_receive(:put_rest)
+    do_create_db
+  end
+  
+  it "should return 'chef'" do
+    do_create_db.should eql("chef")
+  end
+end
+
+describe Chef::CouchDB, "create_design_document" do
+  before(:each) do
+    @mock_rest = mock("Chef::REST", :null_object => true)
+    @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
+        },
+      }
+    }
+    @mock_rest.stub!(:get_rest).and_return(@mock_design)
+    @mock_rest.stub!(:put_rest).and_return(true)
+    Chef::REST.stub!(:new).and_return(@mock_rest)
+  end
+  
+  def do_create_design_document
+    couchdb = Chef::CouchDB.new
+    couchdb.create_design_document("bob", @mock_data)
+  end
+  
+  it "should fetch the existing design document" do
+    @mock_rest.should_receive(:get_rest).with("chef/_design%2Fbob")
+    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
+    @mock_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
+    @mock_rest.should_not_receive(:put_rest)
+    do_create_design_document
+  end
+end
+
+describe Chef::CouchDB, "store" do
+  it "should put the object into couchdb" do
+    @mock_rest = mock("Chef::REST", :null_object => true)
+    @mock_rest.should_receive(:put_rest).with("chef/node_bob", {}).and_return(true)
+    Chef::REST.stub!(:new).and_return(@mock_rest)
+    Chef::CouchDB.new.store("node", "bob", {})
+  end
+end
+
+describe Chef::CouchDB, "load" do
+  it "should load the object from couchdb" do
+    @mock_rest = mock("Chef::REST", :null_object => true)
+    @mock_rest.should_receive(:get_rest).with("chef/node_bob").and_return(true)
+    Chef::REST.stub!(:new).and_return(@mock_rest)
+    Chef::CouchDB.new.load("node", "bob").should eql(true)
+  end
+end
+
+describe Chef::CouchDB, "delete" do
+  before(:each) do
+    @mock_current = {
+      "version" => 1,
+      "_rev" => 1
+    }
+    @mock_rest = mock("Chef::REST", :null_object => true)
+    @mock_rest.stub!(:get_rest).and_return(@mock_current)
+    @mock_rest.stub!(:delete_rest).and_return(true)
+    Chef::REST.stub!(:new).and_return(@mock_rest)
+  end
+  
+  def do_delete(rev=nil)
+    Chef::REST.stub!(:new).and_return(@mock_rest)
+    Chef::CouchDB.new.delete("node", "bob", rev)
+  end
+  
+  it "should remove the object from couchdb with a specific revision" do
+    @mock_rest.should_receive(:delete_rest).with("chef/node_bob?rev=1")
+    do_delete(1)  
+  end
+  
+  it "should remove the object from couchdb based on the couchdb_rev of the current obj" do
+    mock_real = mock("Inflated Object")
+    mock_real.stub!(:respond_to?).and_return(true)
+    mock_real.stub!(:couchdb_rev).and_return(2)
+    @mock_rest.should_receive(:get_rest).with("chef/node_bob").and_return(mock_real)
+    @mock_rest.should_receive(:delete_rest).with("chef/node_bob?rev=2")
+    do_delete
+  end
+  
+  it "should remove the object from couchdb based on the current objects rev" do
+    @mock_rest.should_receive(:delete_rest).with("chef/node_bob?rev=1")
+    do_delete
+  end
+end
+
+describe Chef::CouchDB, "list" do
+  before(:each) do
+    @mock_rest = mock("Chef::REST", :null_object => true)
+    Chef::REST.stub!(:new).and_return(@mock_rest)
+  end
+  
+  it "should get the view for all objects if inflate is true" do
+    @mock_rest.should_receive(:get_rest).with("chef/_view/node/all").and_return(true)
+    Chef::CouchDB.new.list("node", true)
+  end
+  
+  it "should get the view for just the object id's if inflate is false" do
+    @mock_rest.should_receive(:get_rest).with("chef/_view/node/all_id").and_return(true)
+    Chef::CouchDB.new.list("node", false)
+  end
+end
+
+describe Chef::CouchDB, "has_key?" do
+  before(:each) do
+    @mock_rest = mock("Chef::REST", :null_object => true)
+    Chef::REST.stub!(:new).and_return(@mock_rest)
+  end
+  
+  it "should return true if the object exists" do
+    @mock_rest.should_receive(:get_rest).and_return(true)
+    Chef::CouchDB.new.has_key?("node", "bob").should eql(true)
+  end
+  
+  it "should return false if the object does not exist" do
+    @mock_rest.should_receive(:get_rest).and_raise(ArgumentError)
+    Chef::CouchDB.new.has_key?("node", "bob").should eql(false)
+  end
+end