summaryrefslogtreecommitdiffstats
diff options
context:
space:
mode:
-rw-r--r--lib/puppet/parser/functions/lookup_hiera_hash.rb22
-rw-r--r--spec/fixtures/hiera.yaml7
-rw-r--r--spec/fixtures/hieradata/default.yaml3
-rw-r--r--spec/functions/lookup_hiera_hash_spec.rb20
-rw-r--r--spec/spec_helper.rb1
5 files changed, 53 insertions, 0 deletions
diff --git a/lib/puppet/parser/functions/lookup_hiera_hash.rb b/lib/puppet/parser/functions/lookup_hiera_hash.rb
new file mode 100644
index 0000000..d96d65f
--- /dev/null
+++ b/lib/puppet/parser/functions/lookup_hiera_hash.rb
@@ -0,0 +1,22 @@
+module Puppet::Parser::Functions
+ newfunction(:lookup_hiera_hash, :arity => 2, :type => :rvalue,
+ :doc => "Lookup a key->value from a Hiera hash") do |args|
+ hash_name = args[0]
+ key_name = args[1]
+ unless hash_name.is_a?(String) and key_name.is_a?(String)
+ raise Puppet::ParseError, "The hash name and the key name must be given as strings."
+ end
+ if defined? call_function
+ hash = call_function('hiera', [hash_name])
+ else
+ hash = function_hiera([hash_name])
+ end
+ unless hash.is_a?(Hash)
+ raise Puppet::ParseError, "The value Hiera returned for #{hash_name} is not a Hash."
+ end
+ unless hash.key?(key_name)
+ raise Puppet::ParseError, "The Hiera hash #{hash_name} does not contain key #{key_name}."
+ end
+ return hash[key_name]
+ end
+end
diff --git a/spec/fixtures/hiera.yaml b/spec/fixtures/hiera.yaml
new file mode 100644
index 0000000..1dc3360
--- /dev/null
+++ b/spec/fixtures/hiera.yaml
@@ -0,0 +1,7 @@
+---
+:backends:
+ - yaml
+:yaml:
+ :datadir: './spec/fixtures/hieradata'
+:hierarchy:
+ - default
diff --git a/spec/fixtures/hieradata/default.yaml b/spec/fixtures/hieradata/default.yaml
new file mode 100644
index 0000000..0d0c944
--- /dev/null
+++ b/spec/fixtures/hieradata/default.yaml
@@ -0,0 +1,3 @@
+my_hash:
+ network: '127.0.0.1'
+not_hash: string
diff --git a/spec/functions/lookup_hiera_hash_spec.rb b/spec/functions/lookup_hiera_hash_spec.rb
new file mode 100644
index 0000000..ffaf8b5
--- /dev/null
+++ b/spec/functions/lookup_hiera_hash_spec.rb
@@ -0,0 +1,20 @@
+require 'spec_helper'
+require 'puppet'
+
+# puppet 4.0 call_function() has no visibility of 3.x functions and will fail anyway
+unless Puppet.version =~ /^4\.0/
+ describe 'lookup_hiera_hash' do
+ # working version
+ it { should run.with_params('my_hash', 'network').and_return('127.0.0.1') }
+ # raise if key does not exist
+ it { should run.with_params('my_hash', 'not_network').and_raise_error(Puppet::ParseError) }
+ # raise if hash value returned by hiera is not a hash
+ it { should run.with_params('not_hash', 'key').and_raise_error(Puppet::ParseError) }
+ # raise if arguments are not two
+ it { should run.with_params('hash', 'key', 'unexpected').and_raise_error(ArgumentError) }
+ it { should run.with_params('hash').and_raise_error(ArgumentError) }
+ # raise if arguments are not strings
+ it { should run.with_params({}, 'key').and_raise_error(Puppet::ParseError) }
+ it { should run.with_params('hash', true).and_raise_error(Puppet::ParseError) }
+ end
+end
diff --git a/spec/spec_helper.rb b/spec/spec_helper.rb
index 15d5eab..251160e 100644
--- a/spec/spec_helper.rb
+++ b/spec/spec_helper.rb
@@ -6,6 +6,7 @@ fixture_path = File.expand_path(File.join(__FILE__, '..', 'fixtures'))
RSpec.configure do |c|
c.alias_it_should_behave_like_to :it_configures, 'configures'
c.alias_it_should_behave_like_to :it_raises, 'raises'
+ c.hiera_config = File.join(fixture_path, 'hiera.yaml')
c.module_path = File.join(fixture_path, 'modules')
c.manifest_dir = File.join(fixture_path, 'manifests')