summaryrefslogtreecommitdiff
path: root/spec/unit/parser/templatewrapper_spec.rb
blob: 81e0d846da35d0c95ae05dd6305f99ef789b810f (plain)
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
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
#! /usr/bin/env ruby
require 'spec_helper'
require 'puppet/parser/templatewrapper'

describe Puppet::Parser::TemplateWrapper do
  let(:known_resource_types) { Puppet::Resource::TypeCollection.new("env") }
  let(:scope) do
    compiler = Puppet::Parser::Compiler.new(Puppet::Node.new("mynode"))
    compiler.environment.stubs(:known_resource_types).returns known_resource_types
    Puppet::Parser::Scope.new compiler
  end

  let(:tw) { Puppet::Parser::TemplateWrapper.new(scope) }

  it "marks the file for watching" do
    full_file_name = given_a_template_file("fake_template", "content")

    known_resource_types.expects(:watch_file).with(full_file_name)
    tw.file = "fake_template"
  end

  it "fails if a template cannot be found" do
    Puppet::Parser::Files.expects(:find_template).returns nil

    expect { tw.file = "fake_template" }.to raise_error(Puppet::ParseError)
  end

  it "stringifies as template[<filename>] for a file based template" do
    Puppet::Parser::Files.stubs(:find_template).returns("/tmp/fake_template")
    tw.file = "fake_template"
    tw.to_s.should eql("template[/tmp/fake_template]")
  end

  it "stringifies as template[inline] for a string-based template" do
    tw.to_s.should eql("template[inline]")
  end

  it "reads and evaluates a file-based template" do
    given_a_template_file("fake_template", "template contents")

    tw.file = "fake_template"
    tw.result.should eql("template contents")
  end

  it "provides access to the name of the template via #file" do
    full_file_name = given_a_template_file("fake_template", "<%= file %>")

    tw.file = "fake_template"
    tw.result.should == full_file_name
  end

  it "evaluates a given string as a template" do
    tw.result("template contents").should eql("template contents")
  end

  it "provides the defined classes with #classes" do
    catalog = mock 'catalog', :classes => ["class1", "class2"]
    scope.expects(:catalog).returns( catalog )
    tw.classes.should == ["class1", "class2"]
  end

  it "provides all the tags with #all_tags" do
    catalog = mock 'catalog', :tags => ["tag1", "tag2"]
    scope.expects(:catalog).returns( catalog )
    tw.all_tags.should == ["tag1","tag2"]
  end

  it "provides the tags defined in the current scope with #tags" do
    scope.expects(:tags).returns( ["tag1", "tag2"] )
    tw.tags.should == ["tag1","tag2"]
  end

  it "warns about deprecated access to in-scope variables via method calls" do
    Puppet.expects(:deprecation_warning).with("Variable access via 'in_scope_variable' is deprecated. Use '@in_scope_variable' instead. template[inline]:1")
    scope["in_scope_variable"] = "is good"
    tw.result("<%= in_scope_variable %>")
  end

  it "provides access to in-scope variables via method calls" do
    scope["in_scope_variable"] = "is good"
    tw.result("<%= in_scope_variable %>").should == "is good"
  end

  it "errors if accessing via method call a variable that does not exist" do
    expect { tw.result("<%= does_not_exist %>") }.to raise_error(Puppet::ParseError)
  end

  it "reports that variable is available when it is in scope" do
    scope["in_scope_variable"] = "is good"
    tw.result("<%= has_variable?('in_scope_variable') %>").should == "true"
  end

  it "reports that a variable is not available when it is not in scope" do
    tw.result("<%= has_variable?('not_in_scope_variable') %>").should == "false"
  end

  it "provides access to in-scope variables via instance variables" do
    scope["one"] = "foo"
    tw.result("<%= @one %>").should == "foo"
  end

  %w{! . ; :}.each do |badchar|
    it "translates #{badchar} to _ in instance variables" do
      scope["one#{badchar}"] = "foo"
      tw.result("<%= @one_ %>").should == "foo"
    end
  end

  def given_a_template_file(name, contents)
    full_name = "/full/path/to/#{name}"
    Puppet::Parser::Files.stubs(:find_template).
      with(name, anything()).
      returns(full_name)
    File.stubs(:read).with(full_name).returns(contents)

    full_name
  end
end