Module: PuppetlabsSpec::PuppetInternals

Defined in:
lib/puppetlabs_spec_helper/puppetlabs_spec/puppet_internals.rb

Class Method Summary collapse

Class Method Details

.compiler(parts = {}) ⇒ Object 



41
42
43
44
 
# File 'lib/puppetlabs_spec_helper/puppetlabs_spec/puppet_internals.rb', line 41

def compiler(parts = {})
  compiler_node = parts[:node] || node()
  Puppet::Parser::Compiler.new(compiler_node)
end

.function_method(name, parts = {}) ⇒ Object

Return a method instance for a given function. This is primarily useful for rspec-puppet



58
59
60
61
62
63
64
65
 
# File 'lib/puppetlabs_spec_helper/puppetlabs_spec/puppet_internals.rb', line 58

def function_method(name, parts = {})
  scope = parts[:scope] || scope()
  # Ensure the method instance is defined by side-effect of checking if it
  # exists.  This is a hack, but at least it's a hidden hack and not an
  # exposed hack.
  return nil unless Puppet::Parser::Functions.function(name)
  scope.method("function_#{name}".intern)
end

.node(parts = {}) ⇒ Object 



47
48
49
50
51
52
53
 
# File 'lib/puppetlabs_spec_helper/puppetlabs_spec/puppet_internals.rb', line 47

def node(parts = {})
  node_name = parts[:name] || 'testinghost'
  options = parts[:options] || {}
  node_environment = Puppet::Node::Environment.new(parts[:environment] || 'test')
  options.merge!({:environment => node_environment})
  Puppet::Node.new(node_name, options)
end

.resource(parts = {}) ⇒ Object 



34
35
36
37
38
 
# File 'lib/puppetlabs_spec_helper/puppetlabs_spec/puppet_internals.rb', line 34

def resource(parts = {})
  resource_type = parts[:type] || :hostclass
  resource_name = parts[:name] || "testing"
  Puppet::Resource::Type.new(resource_type, resource_name)
end

.scope(parts = {}) ⇒ Object

parser_scope is intended to return a Puppet::Parser::Scope instance suitable for placing in a test harness with the intent of testing parser functions from modules.



10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
 
# File 'lib/puppetlabs_spec_helper/puppetlabs_spec/puppet_internals.rb', line 10

def scope(parts = {})
  if Puppet.version =~ /^2\.[67]/
    # loadall should only be necessary prior to 3.x
    # Please note, loadall needs to happen first when creating a scope, otherwise
    # you might receive undefined method `function_*' errors
    Puppet::Parser::Functions.autoloader.loadall
  end

  scope_compiler = parts[:compiler] || compiler
  scope_parent = parts[:parent] || scope_compiler.topscope
  scope_resource = parts[:resource] || resource(:type => :node, :title => scope_compiler.node.name)

  if Puppet.version =~ /^2\.[67]/
    scope = Puppet::Parser::Scope.new(:compiler => scope_compiler)
  else
    scope = Puppet::Parser::Scope.new(scope_compiler)
  end

  scope.source = Puppet::Resource::Type.new(:node, "foo")
  scope.parent = scope_parent
  scope
end