-#! /usr/bin/env ruby -S rspec
-
require 'spec_helper'
-describe Puppet::Parser::Functions.function(:merge) do
- let(:scope) { PuppetlabsSpec::PuppetInternals.scope }
+describe 'merge' do
+ it { is_expected.not_to eq(nil) }
+ it {
+ is_expected.to run \
+ .with_params({}, 'two') \
+ .and_raise_error(
+ ArgumentError, \
+ Regexp.new(Regexp.escape("rejected: parameter 'args' expects a value of type Undef, Hash, or String[0, 0], got String")),
+ )
+ }
+ it {
+ is_expected.to run \
+ .with_params({}, 1) \
+ .and_raise_error(ArgumentError, %r{parameter 'args' expects a value of type Undef, Hash, or String, got Integer})
+ }
+ it {
+ is_expected.to run \
+ .with_params({ 'one' => 1, 'three' => { 'four' => 4 } }, 'two' => 'dos', 'three' => { 'five' => 5 }) \
+ .and_return('one' => 1, 'three' => { 'five' => 5 }, 'two' => 'dos')
+ }
- describe 'when calling merge from puppet' do
- it "should not compile when no arguments are passed" do
- skip("Fails on 2.6.x, see bug #15912") if Puppet.version =~ /^2\.6\./
- Puppet[:code] = '$x = merge()'
- expect {
- scope.compiler.compile
- }.to raise_error(Puppet::ParseError, /wrong number of arguments/)
- end
+ it { is_expected.to run.with_params.and_return({}) }
+ it { is_expected.to run.with_params({}).and_return({}) }
+ it { is_expected.to run.with_params({}, {}).and_return({}) }
+ it { is_expected.to run.with_params({}, {}, {}).and_return({}) }
- it "should not compile when 1 argument is passed" do
- skip("Fails on 2.6.x, see bug #15912") if Puppet.version =~ /^2\.6\./
- Puppet[:code] = "$my_hash={'one' => 1}\n$x = merge($my_hash)"
- expect {
- scope.compiler.compile
- }.to raise_error(Puppet::ParseError, /wrong number of arguments/)
- end
+ describe 'should accept empty strings as puppet undef' do
+ it { is_expected.to run.with_params({}, '').and_return({}) }
end
- describe 'when calling merge on the scope instance' do
- it 'should require all parameters are hashes' do
- expect { new_hash = scope.function_merge([{}, '2'])}.to raise_error(Puppet::ParseError, /unexpected argument type String/)
- expect { new_hash = scope.function_merge([{}, 2])}.to raise_error(Puppet::ParseError, /unexpected argument type Fixnum/)
- end
-
- it 'should accept empty strings as puppet undef' do
- expect { new_hash = scope.function_merge([{}, ''])}.not_to raise_error
- end
-
- it 'should be able to merge two hashes' do
- new_hash = scope.function_merge([{'one' => '1', 'two' => '1'}, {'two' => '2', 'three' => '2'}])
- expect(new_hash['one']).to eq('1')
- expect(new_hash['two']).to eq('2')
- expect(new_hash['three']).to eq('2')
- end
-
- it 'should merge multiple hashes' do
- hash = scope.function_merge([{'one' => 1}, {'one' => '2'}, {'one' => '3'}])
- expect(hash['one']).to eq('3')
- end
-
- it 'should accept empty hashes' do
- expect(scope.function_merge([{},{},{}])).to eq({})
- end
+ it { is_expected.to run.with_params({ 'key' => 'value' }, {}).and_return('key' => 'value') }
+ it { is_expected.to run.with_params({}, 'key' => 'value').and_return('key' => 'value') }
+ it { is_expected.to run.with_params({ 'key' => 'value1' }, 'key' => 'value2').and_return('key' => 'value2') }
+ it {
+ is_expected.to run \
+ .with_params({ 'key1' => 'value1' }, { 'key2' => 'value2' }, 'key3' => 'value3') \
+ .and_return('key1' => 'value1', 'key2' => 'value2', 'key3' => 'value3')
+ }
+ describe 'should accept iterable and merge produced hashes' do
+ it {
+ is_expected.to run \
+ .with_params([1, 2, 3]) \
+ .with_lambda { |_hsh, val| { val => val } } \
+ .and_return(1 => 1, 2 => 2, 3 => 3)
+ }
+ it {
+ is_expected.to run \
+ .with_params([1, 2, 3]) \
+ .with_lambda { |_hsh, val| { val => val } unless val == 2 } \
+ .and_return(1 => 1, 3 => 3)
+ }
+ it {
+ is_expected.to run \
+ .with_params([1, 2, 3]) \
+ # rubocop:disable Style/Semicolon
+ .with_lambda { |_hsh, val| raise StopIteration if val == 3; { val => val } } \
+ .and_return(1 => 1, 2 => 2)
+ }
+ it {
+ is_expected.to run \
+ .with_params(['a', 'b', 'b', 'c', 'b']) \
+ .with_lambda { |hsh, val| { val => (hsh[val] || 0) + 1 } } \
+ .and_return('a' => 1, 'b' => 3, 'c' => 1)
+ }
+ it {
+ is_expected.to run \
+ .with_params(['a', 'b', 'c']) \
+ .with_lambda { |_hsh, idx, val| { idx => val } } \
+ .and_return(0 => 'a', 1 => 'b', 2 => 'c')
+ }
+ it {
+ is_expected.to run \
+ .with_params('a' => 'A', 'b' => 'B', 'c' => 'C') \
+ .with_lambda { |_hsh, key, val| { key => "#{key}#{val}" } } \
+ .and_return('a' => 'aA', 'b' => 'bB', 'c' => 'cC')
+ }
end
end