diff options
author | Ken Barber <ken@bob.sh> | 2011-07-24 00:39:17 +0100 |
---|---|---|
committer | Ken Barber <ken@bob.sh> | 2011-07-24 00:39:17 +0100 |
commit | fde64f37c98291f4a5b9ee1fcf634288e42b730a (patch) | |
tree | 8142aaf86b7d2bad9a213c1116c1ec37960c491d /lib/puppet/parser/functions/squeeze.rb | |
parent | 07ee33455439d960b9996e8110e011437181b42a (diff) | |
download | puppet-stdlib-fde64f37c98291f4a5b9ee1fcf634288e42b730a.tar.gz puppet-stdlib-fde64f37c98291f4a5b9ee1fcf634288e42b730a.tar.bz2 |
(#1) - fleshed out some more tests.
Diffstat (limited to 'lib/puppet/parser/functions/squeeze.rb')
-rw-r--r-- | lib/puppet/parser/functions/squeeze.rb | 21 |
1 files changed, 19 insertions, 2 deletions
diff --git a/lib/puppet/parser/functions/squeeze.rb b/lib/puppet/parser/functions/squeeze.rb index 02eb00c..9a1dd20 100644 --- a/lib/puppet/parser/functions/squeeze.rb +++ b/lib/puppet/parser/functions/squeeze.rb @@ -7,9 +7,26 @@ module Puppet::Parser::Functions EOS ) do |arguments| - if (arguments.size != 2) then + if ((arguments.size != 2) and (arguments.size != 1)) then raise(Puppet::ParseError, "squeeze(): Wrong number of arguments "+ - "given #{arguments.size} for 2") + "given #{arguments.size} for 2 or 1") + end + + item = arguments[0] + squeezeval = arguments[1] + + if item.is_a?(Array) then + if squeezeval then + item.collect { |i| i.squeeze(squeezeval) } + else + item.collect { |i| i.squeeze } + end + else + if squeezeval then + item.squeeze(squeezeval) + else + item.squeeze + end end end |