add array_include function

This commit is contained in:
mh 2010-12-12 13:23:53 +01:00
parent d1abd33530
commit 7abdda7b1e
2 changed files with 44 additions and 0 deletions

View file

@ -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

View file

@ -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