add array_include function
This commit is contained in:
parent
d1abd33530
commit
7abdda7b1e
2 changed files with 44 additions and 0 deletions
11
lib/puppet/parser/functions/array_include.rb
Normal file
11
lib/puppet/parser/functions/array_include.rb
Normal 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
|
33
spec/unit/parser/functions/array_include.rb
Normal file
33
spec/unit/parser/functions/array_include.rb
Normal 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
|
Loading…
Reference in a new issue