add missing sudo entry for debsources

[mirror/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 e5f99d5..c8b537f 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')
-      return fqdn in roles[role]
+      if not roles.include?(role)
+        err "Failed to look up missing role #{role}"
+        return false
+      end
+      return roles[role].include?(fqdn)
     end
-  end
 end