add array union function
This commit is contained in:
parent
521bad6447
commit
3827e73d67
2 changed files with 49 additions and 0 deletions
10
lib/puppet/parser/functions/array_union.rb
Normal file
10
lib/puppet/parser/functions/array_union.rb
Normal file
|
@ -0,0 +1,10 @@
|
||||||
|
Puppet::Parser::Functions::newfunction(
|
||||||
|
:array_union,
|
||||||
|
:type => :rvalue,
|
||||||
|
:doc => "Unions two arrays
|
||||||
|
|
||||||
|
Example: array_union(['a','b'],['c']) -> ['a','b','c']"
|
||||||
|
) do |args|
|
||||||
|
raise Puppet::ParseError, 'array_union() needs two arguments' if args.length != 2
|
||||||
|
args[0].to_a | args[1].to_a
|
||||||
|
end
|
39
spec/unit/parser/functions/array_union.rb
Normal file
39
spec/unit/parser/functions/array_union.rb
Normal file
|
@ -0,0 +1,39 @@
|
||||||
|
#! /usr/bin/env ruby
|
||||||
|
|
||||||
|
require File.dirname(__FILE__) + '/../../../spec_helper'
|
||||||
|
|
||||||
|
describe "the array_union function" do
|
||||||
|
|
||||||
|
before :each do
|
||||||
|
@scope = Puppet::Parser::Scope.new
|
||||||
|
end
|
||||||
|
|
||||||
|
it "should exist" do
|
||||||
|
Puppet::Parser::Functions.function("array_union").should == "function_array_union"
|
||||||
|
end
|
||||||
|
|
||||||
|
it "should raise a ParseError if there is less than 2 arguments" do
|
||||||
|
lambda { @scope.function_array_union([["foo"]]) }.should( raise_error(Puppet::ParseError))
|
||||||
|
end
|
||||||
|
|
||||||
|
it "should raise a ParseError if there is more than 2 arguments" do
|
||||||
|
lambda { @scope.function_array_union([["foo", "bar", "gazonk"]]) }.should( raise_error(Puppet::ParseError))
|
||||||
|
end
|
||||||
|
|
||||||
|
it "should union to arrays" do
|
||||||
|
result = @scope.function_array_union([['a','b'],['c']])
|
||||||
|
result.should(eql(['a','b','c']))
|
||||||
|
end
|
||||||
|
|
||||||
|
it "should do nothing if an item is present" do
|
||||||
|
result = @scope.function_array_union([['a','b'],['b']])
|
||||||
|
result.should(eql(['a','b']))
|
||||||
|
end
|
||||||
|
|
||||||
|
it "should leave the argument untouched" do
|
||||||
|
a = ['a','b']
|
||||||
|
result = @scope.function_array_union([a,['c']])
|
||||||
|
a.should(eql(['a','b']))
|
||||||
|
end
|
||||||
|
|
||||||
|
end
|
Loading…
Reference in a new issue