support multiple ports in gen tlsa

[dsa-puppet.git] / modules / puppetmaster / lib / puppet / parser / functions / has_role.rb

diff --git a/modules/puppetmaster/lib/puppet/parser/functions/has_role.rb b/modules/puppetmaster/lib/puppet/parser/functions/has_role.rb
index a8f9fd3a30d1d7ccaca9d39e49994aad09ab720f..c8b537f796ea85c62d7603b6cc76653db38605e4 100644 (file)
--- a/modules/puppetmaster/lib/puppet/parser/functions/has_role.rb
+++ b/modules/puppetmaster/lib/puppet/parser/functions/has_role.rb
@@ -1,10 +1,12 @@
 module Puppet::Parser::Functions
   newfunction(:has_role, :type => :rvalue) do |args|
-    begin
-      role = args.shift
+      role = args[0]
       roles = lookupvar('site::roles')
       fqdn = lookupvar('fqdn')
+      if not roles.include?(role)
+        err "Failed to look up missing role #{role}"
+        return false
+      end
       return roles[role].include?(fqdn)
     end
-  end
 end