puppetlabs-stdlib/spec/unit/puppet/parser/functions/has_key_spec.rb
Dan Bode b1beaec2da (#8708) Add has_key function
It is difficult to use existance of keys in a hash
as a boolean condition in Puppet (see #8705)

This function provides a working solution until
the underlying issue in Puppet can be resolved.

Reviewed-by: Jeff McCune
2011-07-29 15:11:10 -07:00

46 lines
1.8 KiB
Ruby

require 'puppet'
require 'mocha'
describe Puppet::Parser::Functions.function(:has_key) do
# Pulled from Dan's create_resources function
# TODO - this should be moved to spec_helper since the
# logic is likely to be applied to multiple rspec files.
let(:compiler) {
topscope = Puppet::Parser::Scope.new
# This is necessary so we don't try to use the compiler to discover our parent.
topscope.parent = nil
my_scope = Puppet::Parser::Scope.new
my_scope.compiler = Puppet::Parser::Compiler.new(Puppet::Node.new("floppy", :environment => 'production'))
my_scope.parent = topscope
compiler = my_scope.compiler
}
let(:scope) {
scope = Puppet::Parser::Scope.new
scope.stubs(:environment).returns(Puppet::Node::Environment.new('production'))
scope
}
describe 'when calling has_key from puppet' do
it "should not compile when no arguments are passed" do
Puppet[:code] = 'has_key()'
expect { compiler.compile }.should raise_error(Puppet::ParseError, /wrong number of arguments/)
end
it "should not compile when 1 argument is passed" do
Puppet[:code] = "has_key('foo')"
expect { compiler.compile }.should raise_error(Puppet::ParseError, /wrong number of arguments/)
end
it "should require the first value to be a Hash" do
Puppet[:code] = "has_key('foo', 'bar')"
expect { compiler.compile }.should raise_error(Puppet::ParseError, /expects the first argument to be a hash/)
end
end
describe 'when calling the function has_key from a scope instance' do
it 'should detect existing keys' do
scope.function_has_key([{'one' => 1}, 'one']).should be_true
end
it 'should detect existing keys' do
scope.function_has_key([{'one' => 1}, 'two']).should be_false
end
end
end