From c5467cc507c004d75037d07c70633d1e743825af Mon Sep 17 00:00:00 2001 From: Morgan Haskel Date: Fri, 14 Nov 2014 14:33:59 -0800 Subject: [PATCH] Need to convert strings and fixnums to arrays --- lib/puppet/parser/functions/member.rb | 8 ++++++-- 1 file changed, 6 insertions(+), 2 deletions(-) diff --git a/lib/puppet/parser/functions/member.rb b/lib/puppet/parser/functions/member.rb index bb19a86ce..88609ce5f 100644 --- a/lib/puppet/parser/functions/member.rb +++ b/lib/puppet/parser/functions/member.rb @@ -8,7 +8,7 @@ module Puppet::Parser::Functions newfunction(:member, :type => :rvalue, :doc => <<-EOS This function determines if a variable is a member of an array. -The variable can either be a string or an array. +The variable can be a string, fixnum, or array. *Examples:* @@ -39,7 +39,11 @@ module Puppet::Parser::Functions raise(Puppet::ParseError, 'member(): Requires array to work with') end - if arguments[1].is_a? String + unless arguments[1].is_a? String or arguments[1].is_a? Fixnum or arguments[1].is_a? Array + raise(Puppet::ParseError, 'member(): Item to search for must be a string, fixnum, or array') + end + + if arguments[1].is_a? String or arguments[1].is_a? Fixnum item = Array(arguments[1]) else item = arguments[1]