summaryrefslogtreecommitdiff
path: root/collect_indices.rb
diff options
context:
space:
mode:
Diffstat (limited to 'collect_indices.rb')
-rw-r--r--collect_indices.rb7
1 files changed, 6 insertions, 1 deletions
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?