Initial Chef repository

cookbooks/chef-solo-search/tests/data/data_bags/node/alpha.json

@@ -0,0 +1,10 @@
+{
+  "id":       "alpha",
+  "name":     "alpha.example.com",
+  "json_class": "Chef::Node",
+  "run_list": ["role[test_server]"],
+  "chef_environment": "default",
+  "automatic": {
+    "hostname": "alpha.example.com"
+  }
+}

cookbooks/chef-solo-search/tests/data/data_bags/node/beta.json

@@ -0,0 +1,10 @@
+{
+  "id":       "beta",
+  "name":     "beta.example.com",
+  "json_class": "Chef::Node",
+  "run_list": ["role[test_server]","role[beta_server]"],
+  "chef_environment": "default",
+  "automatic": {
+    "hostname": "beta.example.com"
+  }
+}

cookbooks/chef-solo-search/tests/data/data_bags/node/without_json_class.json

@@ -0,0 +1,7 @@
+{
+  "id": "without_json_class",
+  "name": "wjc.example.com",
+  "chef_environment": "default",
+  "hostname": "wjc.example.com",
+  "run_list": ["role[test_server]"]
+}

cookbooks/chef-solo-search/tests/data/data_bags/users/jerry.json

@@ -0,0 +1,8 @@
+{
+    "id": "jerry",
+    "username": "speedy",
+    "age": 22,
+    "gender": "male",
+    "married": true,
+    "color": "green"
+}

cookbooks/chef-solo-search/tests/data/data_bags/users/lea.json

@@ -0,0 +1,10 @@
+{
+    "id": "lea",
+    "username": "lea",
+    "age": 35,
+    "gender": "female",
+    "married": true,
+    "children": ["tom"],
+    "tag": "tag::test",
+    "tags": ["tag::first", "tag::second"]
+}

cookbooks/chef-solo-search/tests/data/data_bags/users/mike.json

@@ -0,0 +1,13 @@
+{
+    "id": "mike",
+    "username": "mike the hammer",
+    "age": 42,
+    "gender": "male",
+    "married": true,
+    "children": ["tom", "jerry"],
+    "address": {
+        "street": {
+            "floor": 1
+        }
+    }
+}

cookbooks/chef-solo-search/tests/data/data_bags/users/tom.json

@@ -0,0 +1,10 @@
+{
+    "id": "tom",
+    "username": "tom von homme",
+    "age": 13,
+    "gender": "male",
+    "married": false,
+    "address": {
+      "street": "wilhelmsstrasse"
+    }
+}

cookbooks/chef-solo-search/tests/gemfiles/Gemfile.10

@@ -0,0 +1,4 @@
+source "https://rubygems.org"
+
+gem "chef", "~> 10.0"
+gem "rake"

cookbooks/chef-solo-search/tests/gemfiles/Gemfile.11

@@ -0,0 +1,5 @@
+source "https://rubygems.org"
+
+gem "chef", "~> 11.0"
+gem "rake"
+gem "treetop"

cookbooks/chef-solo-search/tests/test_data_bags.rb

@@ -0,0 +1,45 @@
+#
+# Copyright 2011, edelight GmbH
+#
+# Authors:
+#       Markus Korn <markus.korn@edelight.de>
+#
+# 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 "test/unit"
+require "chef"
+
+# mocking chef such that it thinks it's running as chef-solo and knows about
+# the location of the data_bag
+Chef::Config[:solo] = true
+Chef::Config[:data_bag_path] = "tests/data/data_bags"
+
+def data_bag_item(bag, item)
+  # wrapper around creating a new Recipe instance and calling data_bag on it
+  node = Chef::Node.new()
+  events = Chef::EventDispatch::Dispatcher.new
+  cookbooks = Chef::CookbookCollection.new()
+  run_context = Chef::RunContext.new(node, cookbooks, events)
+  return Chef::Recipe.new("test_cookbook", "test_recipe", run_context).data_bag_item(bag, item)
+end
+
+class TestDataBags < Test::Unit::TestCase
+
+  def test_data_bag
+    item = data_bag_item("users", "mike")
+    assert_equal item["age"], 42
+    assert_equal item[:age], nil    #upstream code for chef-solo does not use mashes
+  end
+
+end

cookbooks/chef-solo-search/tests/test_search.rb

@@ -0,0 +1,244 @@
+#
+# Copyright 2011, edelight GmbH
+#
+# Authors:
+#       Markus Korn <markus.korn@edelight.de>
+#
+# 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 "rubygems"
+require "test/unit"
+require "chef"
+
+# mocking chef such that it thinks it's running as chef-solo and knows about
+# the location of the data_bag
+Chef::Config[:solo] = true
+Chef::Config[:data_bag_path] = "#{File.dirname(__FILE__)}/data/data_bags"
+
+# load the extension
+require File.expand_path('../../libraries/search', __FILE__)
+
+module SearchDbTests
+  
+  def test_search_all
+    # try to get data of all users
+    nodes = search(:users, "*:*")
+    assert_equal nodes.length, 4
+    nodes = search(:users)
+    assert_equal nodes.length, 4
+    nodes = search(:users, nil)
+    assert_equal nodes.length, 4
+  end
+  
+  def test_search_exact_match
+    nodes = search(:users, "username:speedy")
+    assert_equal nodes.length, 1
+    assert_equal nodes[0]["username"], "speedy"
+  end
+  
+  def test_get_all_with_field
+    nodes = search(:users, "username:*")
+    assert nodes.length > 0
+    assert nodes.all?{|x| !x["username"].nil?}
+  end
+  
+  def test_get_all_without_field
+    nodes = search(:users, "(NOT username:*)")
+    assert nodes.length == 0
+    nodes = search(:users, "(NOT color:*)")
+    assert nodes.length == 3
+    assert nodes.all?{|x| x["color"].nil?}
+  end
+  
+  def test_get_all_but_speedy
+    nodes = search(:users, "NOT username:speedy")
+    assert nodes.length > 0
+    assert nodes.all?{|x| x["username"] != "speedy"}
+  end
+  
+  def test_array_includes
+    nodes = search(:users, "children:tom")
+    assert nodes.length == 2
+    assert nodes.all?{ |x| x["children"].include?("tom") }
+    nodes = search(:users, "children:jerry")
+    assert nodes.length == 1
+    assert nodes.all?{ |x| x["children"].include?("jerry") }
+  end
+  
+  def test_boolean
+    nodes = search(:users, "married:true")
+    assert nodes.length == 3
+    assert nodes.all?{ |x| x["married"] == true }
+    nodes = search(:users, "married:false")
+    assert nodes.length == 1
+    assert nodes[0]["married"] == false
+  end
+  
+  def test_integer
+    nodes = search(:users, "age:35")
+    assert nodes.length == 1
+    assert nodes[0]["age"] == 35
+  end
+  
+  def test_AND_condition
+    nodes = search(:users, "married:true AND age:35")
+    assert nodes.length == 1
+    assert nodes[0]["username"] == "lea"
+  end
+  
+  def test_OR_condition
+    nodes = search(:users, "age:42 OR age:22")
+    assert nodes.length == 2
+  end
+  
+  def test_NOT_condition
+    nodes = search(:users, "children:tom AND (NOT gender:female)")
+    assert nodes.length == 1
+    nodes = search(:users, "children:tom AND (NOT gender:female) AND age:42")
+    assert nodes.length == 1
+    nodes = search(:users, "children:tom AND (NOT gender:female) AND (NOT age:42)")
+    assert nodes.length == 0
+  end
+  
+  def test_any_value
+    nodes = search(:users, "children:*")
+    assert nodes.length == 2
+  end
+  
+  def test_any_value_lucene_range
+    nodes = search(:users, "address:[* TO *]")
+    assert nodes.length == 2
+  end
+  
+  def test_general_lucene_range_fails
+    assert_raises RuntimeError do
+      nodes = search(:users, "children:[aaa TO zzz]")
+    end
+  end
+  
+  def test_block_usage
+    # bracket syntax
+    result = []
+    search(:users, "*:*") {|x| result << x["id"]}
+    assert result.length == 4
+    
+    # do...end syntax
+    result = []
+    search(:users) do |x|
+      result << x["id"]
+    end
+    assert result.length == 4
+  end
+  
+  def test_check_escaped_chars
+    nodes = search(:users, 'tag:tag\:\:test')
+    assert nodes.length == 1
+    nodes = search(:users, "tag:tag\\:\\:test")
+    assert nodes.length == 1
+    nodes = search(:users, 'tags:tag\:\:first')
+    assert nodes.length == 1
+    nodes = search(:users, "tags:tag\\:\\:first")
+    assert nodes.length == 1
+    nodes = search(:users, 'tags:tag\:\:*')
+    assert nodes.length == 1
+    nodes = search(:users, "tags:tag\\:\\:*")
+    assert nodes.length == 1
+  end
+  
+  def test_wildcards
+    nodes = search(:users, "gender:f??ale")
+    assert nodes.length == 1
+    nodes = search(:users, "username:spee?y")
+    assert nodes.length == 1
+    nodes = search(:users, "username:spee*")
+    assert nodes.length == 1
+  end
+  
+  def test_empty_field_value
+    assert_raise(RuntimeError) {
+      search(:users, "gender:#{nil} AND age:35")
+    }
+    assert_raise(RuntimeError) {
+      search(:users, "gender: AND age:35")
+    }
+    assert_raise(RuntimeError) {
+      search(:users, "gender:\"\" AND age:35")
+    }
+  end
+  
+  def test_OR_group
+    nodes = search(:users, "id:(mike OR tom)")
+    assert nodes.length == 2
+  end
+  
+  def test_nested_fieldnames
+    nodes = search(:users, "address_street:wilhelmsstrasse")
+    assert nodes.length == 1
+    nodes = search(:users, "address_street_floor:1")
+    assert nodes.length == 1
+  end
+end
+
+module SearchNodeTests
+  def test_list_nodes
+    nodes = search(:node)
+    assert_equal Chef::Node, nodes.find{ |n| n["hostname"] == "alpha.example.com" }.class
+    assert_equal Chef::Node, nodes.find{ |n| n["hostname"] == "beta.example.com" }.class
+    assert_equal Hash, nodes.find{ |n| n["hostname"] == "wjc.example.com" }.class
+    assert_equal 3, nodes.length
+  end
+
+  def test_search_node_with_wide_filter
+    nodes = search(:node, "role:test_server AND chef_environment:default")
+    assert_equal 2, nodes.length
+  end
+
+  def test_search_node_with_narrow_filter
+    nodes = search(:node, "role:beta_server")
+    assert_equal 1, nodes.length
+  end
+
+  def test_search_node_with_attr_filter
+    nodes = search(:node, "hostname:beta.example.com")
+    assert_equal 1, nodes.length
+  end
+
+  def test_search_node_without_json_class
+    nodes = search(:node, "chef_environment:default")
+    assert_equal 3, nodes.length
+  end
+end
+
+class TestImplicitSearchDB < Test::Unit::TestCase
+  include SearchDbTests
+  include SearchNodeTests
+
+  def search(*args, &block)
+    # wrapper around creating a new Recipe instance and calling search on it
+    node = Chef::Node.new()
+    cookbooks = Chef::CookbookCollection.new()
+    run_context = Chef::RunContext.new(node, cookbooks, nil)
+    return Chef::Recipe.new("test_cookbook", "test_recipe", run_context).search(*args, &block)
+  end
+end
+
+class TestExplicitSearchDB < Test::Unit::TestCase
+  include SearchDbTests
+  include SearchNodeTests
+
+  def search(*args, &block)
+    Chef::Search::Query.new.search(*args, &block)
+  end
+end
+