summaryrefslogtreecommitdiffstats
path: root/spec/integration/parser/functions/require.rb
blob: 6f169ade13ed7cbf76aaa5c208f0c29d9194f44c (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
#!/usr/bin/env ruby

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

describe "the require function" do

    before :each do
        @parser = Puppet::Parser::Parser.new :Code => ""
        @node = Puppet::Node.new("mynode")
        @compiler = Puppet::Parser::Compiler.new(@node, @parser)

        @compiler.send(:evaluate_main)
        @compiler.catalog.client_version = "0.25"
        @scope = @compiler.topscope
        # preload our functions
        Puppet::Parser::Functions.function(:include)
        Puppet::Parser::Functions.function(:require)
    end

    it "should add a relationship between the 'required' class and our class" do
        @parser.newclass("requiredclass")

        @scope.function_require("requiredclass")
        @scope.resource["require"].should_not be_nil
        ref = @scope.resource["require"].shift
        ref.type.should == "Class"
        ref.title.should == "requiredclass"
    end

    it "should queue relationships between the 'required' class and our classes" do
        @parser.newclass("requiredclass1")
        @parser.newclass("requiredclass2")

        @scope.function_require("requiredclass1")
        @scope.function_require("requiredclass2")

        @scope.resource["require"].should_not be_nil

        (ref1,ref2) = @scope.resource["require"]
        ref1.type.should == "Class"
        ref1.title.should == "requiredclass1"
        ref2.type.should == "Class"
        ref2.title.should == "requiredclass2"
    end

end

describe "the include function" do
    require 'puppet_spec/files'
    include PuppetSpec::Files

    before :each do
        @real_dir = Dir.getwd
        @temp_dir = tmpfile('include_function_integration_test')
        Dir.mkdir @temp_dir
        Dir.chdir @temp_dir
        @parser = Puppet::Parser::Parser.new :Code => ""
        @node = Puppet::Node.new("mynode")
        @compiler = Puppet::Parser::Compiler.new(@node, @parser)
        @compiler.send(:evaluate_main)
        @scope = @compiler.topscope
        # preload our functions
        Puppet::Parser::Functions.function(:include)
        Puppet::Parser::Functions.function(:require)
    end

    after :each do
        Dir.chdir @real_dir
        Dir.rmdir @temp_dir
    end

    def with_file(filename,contents)
        path = File.join(@temp_dir,filename)
        File.open(path, "w") { |f|f.puts contents }
        yield
        File.delete(path)
    end

    it "should add a relationship between the 'included' class and our class" do
        with_file('includedclass',"class includedclass {}") {
            @scope.function_include("includedclass")
            }
        @compiler.catalog.edge?(@scope.resource,@compiler.findresource(:class,"includedclass")).should be_true
    end

end