Set arity on all functions

.travis.yml

@@ -12,7 +12,6 @@ matrix:
   allow_failures:
     - rvm: 2.0.0
     - rvm: ruby-head
-  include:
     - rvm: 1.8.7
       env: PUPPET_GEM_VERSION="~> 2.7"
 notifications:

lib/puppet/parser/functions/abs.rb

@@ -3,15 +3,12 @@
 #
 
 module Puppet::Parser::Functions
-  newfunction(:abs, :type => :rvalue, :doc => <<-EOS
+  newfunction(:abs, :type => :rvalue, :arity => 1, :doc => <<-EOS
     Returns the absolute value of a number, for example -34.56 becomes
     34.56. Takes a single integer and float value as an argument.
     EOS
   ) do |arguments|
 
-    raise(Puppet::ParseError, "abs(): Wrong number of arguments " +
-      "given (#{arguments.size} for 1)") if arguments.size < 1
-
     value = arguments[0]
 
     # Numbers in Puppet are often string-encoded which is troublesome ...

lib/puppet/parser/functions/any2array.rb

@@ -3,7 +3,7 @@
 #
 
 module Puppet::Parser::Functions
-  newfunction(:any2array, :type => :rvalue, :doc => <<-EOS
+  newfunction(:any2array, :type => :rvalue, :arity => -1, :doc => <<-EOS
 This converts any object to an array containing that object. Empty argument
 lists are converted to an empty array. Arrays are left untouched. Hashes are
 converted to arrays of alternating keys and values.

lib/puppet/parser/functions/base64.rb

@@ -1,6 +1,6 @@
 module Puppet::Parser::Functions
 
-  newfunction(:base64, :type => :rvalue, :doc => <<-'ENDHEREDOC') do |args|
+  newfunction(:base64, :type => :rvalue, :arity => 2, :doc => <<-'ENDHEREDOC') do |args|
 
     Base64 encode or decode a string based on the command and the string submitted
 
@@ -13,8 +13,6 @@ module Puppet::Parser::Functions
 
     require 'base64'
 
-    raise Puppet::ParseError, ("base64(): Wrong number of arguments (#{args.length}; must be = 2)") unless args.length == 2
-
     actions = ['encode','decode']
 
     unless actions.include?(args[0])

lib/puppet/parser/functions/bool2num.rb

@@ -3,17 +3,14 @@
 #
 
 module Puppet::Parser::Functions
-  newfunction(:bool2num, :type => :rvalue, :doc => <<-EOS
+  newfunction(:bool2num, :type => :rvalue, :arity => 1, :doc => <<-EOS
     Converts a boolean to a number. Converts the values:
       false, f, 0, n, and no to 0
       true, t, 1, y, and yes to 1
     Requires a single boolean or string as an input.
     EOS
   ) do |arguments|
 
-    raise(Puppet::ParseError, "bool2num(): Wrong number of arguments " +
-      "given (#{arguments.size} for 1)") if arguments.size < 1
-
     value = arguments[0]
     klass = value.class
 

lib/puppet/parser/functions/capitalize.rb

@@ -3,15 +3,12 @@
 #
 
 module Puppet::Parser::Functions
-  newfunction(:capitalize, :type => :rvalue, :doc => <<-EOS
+  newfunction(:capitalize, :type => :rvalue, :arity => 1, :doc => <<-EOS
     Capitalizes the first letter of a string or array of strings.
     Requires either a single string or an array as an input.
     EOS
   ) do |arguments|
 
-    raise(Puppet::ParseError, "capitalize(): Wrong number of arguments " +
-      "given (#{arguments.size} for 1)") if arguments.size < 1
-
     value = arguments[0]
     klass = value.class
 

lib/puppet/parser/functions/chomp.rb

@@ -3,16 +3,13 @@
 #
 
 module Puppet::Parser::Functions
-  newfunction(:chomp, :type => :rvalue, :doc => <<-'EOS'
+  newfunction(:chomp, :type => :rvalue, :arity => 1, :doc => <<-'EOS'
     Removes the record separator from the end of a string or an array of
     strings, for example `hello\n` becomes `hello`.
     Requires a single string or array as an input.
     EOS
   ) do |arguments|
 
-    raise(Puppet::ParseError, "chomp(): Wrong number of arguments " +
-      "given (#{arguments.size} for 1)") if arguments.size < 1
-
     value = arguments[0]
     klass = value.class
 

lib/puppet/parser/functions/chop.rb

@@ -3,7 +3,7 @@
 #
 
 module Puppet::Parser::Functions
-  newfunction(:chop, :type => :rvalue, :doc => <<-'EOS'
+  newfunction(:chop, :type => :rvalue, :arity => 1, :doc => <<-'EOS'
     Returns a new string with the last character removed. If the string ends
     with `\r\n`, both characters are removed. Applying chop to an empty
     string returns an empty string. If you wish to merely remove record
@@ -12,9 +12,6 @@ module Puppet::Parser::Functions
     EOS
   ) do |arguments|
 
-    raise(Puppet::ParseError, "chop(): Wrong number of arguments " +
-      "given (#{arguments.size} for 1)") if arguments.size < 1
-
     value = arguments[0]
     klass = value.class
 

lib/puppet/parser/functions/concat.rb

@@ -3,7 +3,7 @@
 #
 
 module Puppet::Parser::Functions
-  newfunction(:concat, :type => :rvalue, :doc => <<-EOS
+  newfunction(:concat, :type => :rvalue, :arity => 2, :doc => <<-EOS
 Appends the contents of array 2 onto array 1.
 
 *Example:*
@@ -16,10 +16,6 @@ module Puppet::Parser::Functions
     EOS
   ) do |arguments|
 
-    # Check that 2 arguments have been given ...
-    raise(Puppet::ParseError, "concat(): Wrong number of arguments " +
-      "given (#{arguments.size} for 2)") if arguments.size != 2
-
     a = arguments[0]
     b = arguments[1]
 

lib/puppet/parser/functions/defined_with_params.rb

@@ -3,6 +3,7 @@
 
 Puppet::Parser::Functions.newfunction(:defined_with_params,
                                       :type => :rvalue,
+                                      :arity => -1,
                                       :doc => <<-'ENDOFDOC'
 Takes a resource reference and an optional hash of attributes.
 

lib/puppet/parser/functions/delete.rb

@@ -5,7 +5,7 @@
 # TODO(Krzysztof Wilczynski): We need to add support for regular expression ...
 
 module Puppet::Parser::Functions
-  newfunction(:delete, :type => :rvalue, :doc => <<-EOS
+  newfunction(:delete, :type => :rvalue, :arity => 2, :doc => <<-EOS
 Deletes all instances of a given element from an array, substring from a
 string, or key from a hash.
 
@@ -22,11 +22,6 @@ module Puppet::Parser::Functions
     EOS
   ) do |arguments|
 
-    if (arguments.size != 2) then
-      raise(Puppet::ParseError, "delete(): Wrong number of arguments "+
-        "given #{arguments.size} for 2.")
-    end
-
     collection = arguments[0]
     item = arguments[1]
 

lib/puppet/parser/functions/delete_at.rb

@@ -3,7 +3,7 @@
 #
 
 module Puppet::Parser::Functions
-  newfunction(:delete_at, :type => :rvalue, :doc => <<-EOS
+  newfunction(:delete_at, :type => :rvalue, :arity => 2, :doc => <<-EOS
 Deletes a determined indexed value from an array.
 
 *Examples:*
@@ -14,9 +14,6 @@ module Puppet::Parser::Functions
     EOS
   ) do |arguments|
 
-    raise(Puppet::ParseError, "delete_at(): Wrong number of arguments " +
-      "given (#{arguments.size} for 2)") if arguments.size < 2
-
     array = arguments[0]
 
     unless array.is_a?(Array)

lib/puppet/parser/functions/delete_undef_values.rb

@@ -1,5 +1,5 @@
 module Puppet::Parser::Functions
-  newfunction(:delete_undef_values, :type => :rvalue, :doc => <<-EOS
+  newfunction(:delete_undef_values, :type => :rvalue, :arity => 1, :doc => <<-EOS
 Returns a copy of input hash or array with all undefs deleted.
 
 *Examples:*
@@ -15,10 +15,6 @@ module Puppet::Parser::Functions
       EOS
     ) do |args|
 
-    raise(Puppet::ParseError,
-          "delete_undef_values(): Wrong number of arguments given " +
-          "(#{args.size})") if args.size < 1
-
     result = args[0]
     if result.is_a?(Hash)
       result.delete_if {|key, val| val.equal? :undef}

lib/puppet/parser/functions/delete_values.rb

@@ -1,5 +1,5 @@
 module Puppet::Parser::Functions
-  newfunction(:delete_values, :type => :rvalue, :doc => <<-EOS
+  newfunction(:delete_values, :type => :rvalue, :arity => 2, :doc => <<-EOS
 Deletes all instances of a given value from a hash.
 
 *Examples:*
@@ -11,10 +11,6 @@ module Puppet::Parser::Functions
       EOS
     ) do |arguments|
 
-    raise(Puppet::ParseError,
-          "delete_values(): Wrong number of arguments given " +
-          "(#{arguments.size} of 2)") if arguments.size != 2
-
     hash, item = arguments
 
     if not hash.is_a?(Hash)

lib/puppet/parser/functions/difference.rb

@@ -3,7 +3,7 @@
 #
 
 module Puppet::Parser::Functions
-  newfunction(:difference, :type => :rvalue, :doc => <<-EOS
+  newfunction(:difference, :type => :rvalue, :arity => 2, :doc => <<-EOS
 This function returns the difference between two arrays.
 The returned array is a copy of the original array, removing any items that
 also appear in the second array.
@@ -16,10 +16,6 @@ module Puppet::Parser::Functions
     EOS
   ) do |arguments|
 
-    # Two arguments are required
-    raise(Puppet::ParseError, "difference(): Wrong number of arguments " +
-      "given (#{arguments.size} for 2)") if arguments.size != 2
-
     first = arguments[0]
     second = arguments[1]
 

lib/puppet/parser/functions/dirname.rb

@@ -1,12 +1,9 @@
 module Puppet::Parser::Functions
-  newfunction(:dirname, :type => :rvalue, :doc => <<-EOS
+  newfunction(:dirname, :type => :rvalue, :arity => 1, :doc => <<-EOS
     Returns the dirname of a path.
     EOS
   ) do |arguments|
 
-    raise(Puppet::ParseError, "dirname(): Wrong number of arguments " +
-      "given (#{arguments.size} for 1)") if arguments.size < 1
-
     path = arguments[0]
     return File.dirname(path)
   end

lib/puppet/parser/functions/downcase.rb

@@ -3,14 +3,11 @@
 #
 
 module Puppet::Parser::Functions
-  newfunction(:downcase, :type => :rvalue, :doc => <<-EOS
+  newfunction(:downcase, :type => :rvalue, :arity => 1, :doc => <<-EOS
 Converts the case of a string or all strings in an array to lower case.
     EOS
   ) do |arguments|
 
-    raise(Puppet::ParseError, "downcase(): Wrong number of arguments " +
-      "given (#{arguments.size} for 1)") if arguments.size < 1
-
     value = arguments[0]
     klass = value.class
 

lib/puppet/parser/functions/empty.rb

@@ -3,7 +3,7 @@
 #
 
 module Puppet::Parser::Functions
-  newfunction(:empty, :type => :rvalue, :doc => <<-EOS
+  newfunction(:empty, :type => :rvalue, :arity => 1, :doc => <<-EOS
 Returns true if the variable is empty.
     EOS
   ) do |arguments|

lib/puppet/parser/functions/ensure_packages.rb

@@ -4,13 +4,11 @@
 require 'puppet/parser/functions'
 
 module Puppet::Parser::Functions
-  newfunction(:ensure_packages, :type => :statement, :doc => <<-EOS
+  newfunction(:ensure_packages, :type => :statement, :arity => 1, :doc => <<-EOS
 Takes a list of packages and only installs them if they don't already exist.
     EOS
   ) do |arguments|
 
-    raise(Puppet::ParseError, "ensure_packages(): Wrong number of arguments " +
-      "given (#{arguments.size} for 1)") if arguments.size != 1
     raise(Puppet::ParseError, "ensure_packages(): Requires array " +
       "given (#{arguments[0].class})") if !arguments[0].kind_of?(Array)
 

lib/puppet/parser/functions/ensure_resource.rb

@@ -3,6 +3,7 @@
 
 Puppet::Parser::Functions.newfunction(:ensure_resource,
                                       :type => :statement,
+                                      :arity => -3,
                                       :doc => <<-'ENDOFDOC'
 Takes a resource type, title, and a list of attributes that describe a
 resource.
@@ -27,8 +28,6 @@
 ENDOFDOC
 ) do |vals|
   type, title, params = vals
-  raise(ArgumentError, 'Must specify a type') unless type
-  raise(ArgumentError, 'Must specify a title') unless title
   params ||= {}
 
   items = [title].flatten

lib/puppet/parser/functions/flatten.rb

@@ -3,7 +3,7 @@
 #
 
 module Puppet::Parser::Functions
-  newfunction(:flatten, :type => :rvalue, :doc => <<-EOS
+  newfunction(:flatten, :type => :rvalue, :arity => 1, :doc => <<-EOS
 This function flattens any deeply nested arrays and returns a single flat array
 as a result.
 
@@ -15,9 +15,6 @@ module Puppet::Parser::Functions
     EOS
   ) do |arguments|
 
-    raise(Puppet::ParseError, "flatten(): Wrong number of arguments " +
-      "given (#{arguments.size} for 1)") if arguments.size != 1
-
     array = arguments[0]
 
     unless array.is_a?(Array)

lib/puppet/parser/functions/floor.rb

@@ -1,13 +1,10 @@
 module Puppet::Parser::Functions
-  newfunction(:floor, :type => :rvalue, :doc => <<-EOS
+  newfunction(:floor, :type => :rvalue, :arity => 1, :doc => <<-EOS
     Returns the largest integer less or equal to the argument.
     Takes a single numeric value as an argument.
     EOS
   ) do |arguments|
 
-    raise(Puppet::ParseError, "floor(): Wrong number of arguments " +
-          "given (#{arguments.size} for 1)") if arguments.size != 1
-
     arg = arguments[0]
 
     raise(Puppet::ParseError, "floor(): Wrong argument type " +

lib/puppet/parser/functions/fqdn_rotate.rb

@@ -3,14 +3,11 @@
 #
 
 module Puppet::Parser::Functions
-  newfunction(:fqdn_rotate, :type => :rvalue, :doc => <<-EOS
+  newfunction(:fqdn_rotate, :type => :rvalue, :arity => 1, :doc => <<-EOS
 Rotates an array a random number of times based on a nodes fqdn.
     EOS
   ) do |arguments|
 
-    raise(Puppet::ParseError, "fqdn_rotate(): Wrong number of arguments " +
-      "given (#{arguments.size} for 1)") if arguments.size < 1
-
     value = arguments[0]
     klass = value.class
     require 'digest/md5'

lib/puppet/parser/functions/get_module_path.rb

@@ -1,13 +1,12 @@
 module Puppet::Parser::Functions
-  newfunction(:get_module_path, :type =>:rvalue, :doc => <<-EOT
+  newfunction(:get_module_path, :type =>:rvalue, :arity => 1, :doc => <<-EOT
     Returns the absolute path of the specified module for the current
     environment.
 
     Example:
       $module_path = get_module_path('stdlib')
   EOT
   ) do |args|
-    raise(Puppet::ParseError, "get_module_path(): Wrong number of arguments, expects one") unless args.size == 1
     if module_path = Puppet::Module.find(args[0], compiler.environment.to_s)
       module_path.path
     else

lib/puppet/parser/functions/getparam.rb

@@ -3,6 +3,7 @@
 
 Puppet::Parser::Functions.newfunction(:getparam,
                                       :type => :rvalue,
+                                      :arity => 2,
                                       :doc => <<-'ENDOFDOC'
 Takes a resource reference and name of the parameter and
 returns value of resource's parameter.
@@ -22,7 +23,6 @@
 ENDOFDOC
 ) do |vals|
   reference, param = vals
-  raise(ArgumentError, 'Must specify a reference') unless reference
   raise(ArgumentError, 'Must specify name of a parameter') unless param and param.instance_of? String
 
   return '' if param.empty?