add array_include function

2 files changed, 44 insertions, 0 deletions

diff --git a/lib/puppet/parser/functions/array_include.rb b/lib/puppet/parser/functions/array_include.rb
new file mode 100644
index 0000000..ce4748d
--- /dev/null
+++ b/lib/puppet/parser/functions/array_include.rb
@@ -0,0 +1,11 @@
+Puppet::Parser::Functions::newfunction(
+  :array_include,
+  :type => :rvalue,
+  :doc => "Checks whether an item is included or not
+
+    Example: array_include(['a','b'],'b') -> true
+    Example: array_include(['a','b'],'c') -> false"
+) do |args|
+  raise Puppet::ParseError, 'array_include() needs two arguments' if args.length != 2
+  args[0].include?(args[1])
+end
diff --git a/spec/unit/parser/functions/array_include.rb b/spec/unit/parser/functions/array_include.rb
new file mode 100644
index 0000000..fecf495
--- /dev/null
+++ b/spec/unit/parser/functions/array_include.rb
@@ -0,0 +1,33 @@
+#! /usr/bin/env ruby
+
+require File.dirname(__FILE__) + '/../../../spec_helper'
+
+describe "the array_include function" do
+
+  before :each do
+    @scope = Puppet::Parser::Scope.new
+  end
+
+  it "should exist" do
+    Puppet::Parser::Functions.function("array_include").should == "function_array_include"
+  end
+
+  it "should raise a ParseError if there is less than 2 arguments" do
+    lambda { @scope.function_array_include(["foo"]) }.should( raise_error(Puppet::ParseError))
+  end
+
+  it "should raise a ParseError if there is more than 2 arguments" do
+    lambda { @scope.function_array_include(["foo", "bar", "gazonk"]) }.should( raise_error(Puppet::ParseError))
+  end
+
+  it "should return true if an item is present in the array" do
+    result = @scope.function_array_include(['a','b'],'b')
+    result.should == true
+  end
+
+  it "should return false if an item is not present" do
+    result = @scope.function_array_include(['a','b'],'c')
+    result.should == false
+  end
+
+end