spec/unit/parser/functions/template_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

#!/usr/bin/env ruby

require File.dirname(__FILE__) + '/../../../spec_helper'

describe "the template function" do

    before :each do
        @scope = Puppet::Parser::Scope.new()
    end

    it "should exist" do
        Puppet::Parser::Functions.function("template").should == "function_template"
    end

    it "should create a TemplateWrapper when called" do
        tw = stub_everything 'template_wrapper'

        Puppet::Parser::TemplateWrapper.expects(:new).returns(tw)

        @scope.function_template("test")
    end

    it "should give the template filename to the TemplateWrapper" do
        tw = stub_everything 'template_wrapper'
        Puppet::Parser::TemplateWrapper.stubs(:new).returns(tw)

        tw.expects(:file=).with("test")

        @scope.function_template("test")
    end

    it "should return what TemplateWrapper.result returns" do
        tw = stub_everything 'template_wrapper'
        Puppet::Parser::TemplateWrapper.stubs(:new).returns(tw)
        tw.stubs(:file=).with("test")

        tw.expects(:result).returns("template contents evaluated")

        @scope.function_template("test").should == "template contents evaluated"
    end

    it "should concatenate template wrapper outputs for multiple templates" do
        tw1 = stub_everything "template_wrapper1"
        tw2 = stub_everything "template_wrapper2"
        Puppet::Parser::TemplateWrapper.stubs(:new).returns(tw1,tw2)
        tw1.stubs(:file=).with("1")
        tw2.stubs(:file=).with("2")
        tw1.stubs(:result).returns("result1")
        tw2.stubs(:result).returns("result2")

        @scope.function_template(["1","2"]).should == "result1result2"
    end

    it "should raise an error if the template raises an error" do
        tw = stub_everything 'template_wrapper'
        Puppet::Parser::TemplateWrapper.stubs(:new).returns(tw)
        tw.stubs(:result).raises

        lambda { @scope.function_template("1") }.should raise_error(Puppet::ParseError)
    end

end