spec/unit/parser/functions/require_spec.rb


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75

#! /usr/bin/env ruby
require 'spec_helper'
require 'unit/parser/functions/shared'

describe "the require function" do
  before :all do
    Puppet::Parser::Functions.autoloader.loadall
  end

  before :each do
    @catalog = stub 'catalog'

    node      = Puppet::Node.new('localhost')
    compiler  = Puppet::Parser::Compiler.new(node)
    @scope = Puppet::Parser::Scope.new(compiler)

    @scope.stubs(:findresource)
    @klass = stub 'class', :name => "myclass"
    @scope.stubs(:find_hostclass).returns(@klass)

    @resource = Puppet::Parser::Resource.new(:file, "/my/file", :scope => @scope, :source => "source")
    @scope.stubs(:resource).returns @resource
  end

  it "should exist" do
    Puppet::Parser::Functions.function("require").should == "function_require"
  end

  it "should delegate to the 'include' puppet function" do
    @scope.compiler.expects(:evaluate_classes).with(["myclass"], @scope, false)

    @scope.function_require(["myclass"])
  end

  it "should set the 'require' parameter on the resource to a resource reference" do
    @scope.compiler.stubs(:evaluate_classes)
    @scope.function_require(["myclass"])

    @resource["require"].should be_instance_of(Array)
    @resource["require"][0].should be_instance_of(Puppet::Resource)
  end

  it "should lookup the absolute class path" do
    @scope.compiler.stubs(:evaluate_classes)

    @scope.expects(:find_hostclass).with("myclass").returns(@klass)
    @klass.expects(:name).returns("myclass")

    @scope.function_require(["myclass"])
  end

  it "should append the required class to the require parameter" do
    @scope.compiler.stubs(:evaluate_classes)

    one = Puppet::Resource.new(:file, "/one")
    @resource[:require] = one
    @scope.function_require(["myclass"])

    @resource[:require].should be_include(one)
    @resource[:require].detect { |r| r.to_s == "Class[Myclass]" }.should be_instance_of(Puppet::Resource)
  end

  describe "When the future parser is in use" do
    require 'puppet/pops'
    require 'puppet_spec/compiler'
    include PuppetSpec::Compiler

    before(:each) do
      Puppet[:parser] = 'future'
    end

    it_should_behave_like 'all functions transforming relative to absolute names', :function_require
    it_should_behave_like 'an inclusion function, regardless of the type of class reference,', :require
  end
end