Initial Chef repository

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
+