From fa68d78b200befb286b3e18415dd7f820369f5e5 Mon Sep 17 00:00:00 2001 From: Krzysztof Wilczynski Date: Sat, 23 Apr 2011 18:11:08 +0100 Subject: Small changes. Added better error checking etc ... Signed-off-by: Krzysztof Wilczynski --- collect_indices.rb | 7 ++++++- 1 file changed, 6 insertions(+), 1 deletion(-) (limited to 'collect_indices.rb') diff --git a/collect_indices.rb b/collect_indices.rb index 65abfce..e96c3e9 100644 --- a/collect_indices.rb +++ b/collect_indices.rb @@ -10,7 +10,12 @@ module Puppet::Parser::Functions raise(Puppet::ParseError, "Wrong number of arguments " + "given (#{arguments.size} for 2)") if arguments.size < 2 - array = arguments.shift + array = arguments.shift + + if not array.is_a?(Array) + raise(Puppet::ParseError, 'Requires an array to work with') + end + indices = *arguments # Get them all ... Pokemon ... if not indices or indices.empty? -- cgit v1.2.3