summaryrefslogtreecommitdiffstats
path: root/test/other/events.rb
blob: 3f02575d9badaf396d8fc8dfc1a0c256a6e9682f (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
119
120
121
122
123
124
#!/usr/bin/env ruby

require File.dirname(__FILE__) + '/../lib/puppettest'

require 'puppet'
require 'puppettest'


class TestEvents < Test::Unit::TestCase
	include PuppetTest

    def test_simplesubscribe
        name = tempfile()
        file = Puppet::Type.type(:file).new(
            :name => name,
            :ensure => "file"
        )
        exec = Puppet::Type.type(:exec).new(
            :name => "echo true",
            :path => "/usr/bin:/bin",
            :refreshonly => true,
            :subscribe => Puppet::Resource::Reference.new(file.class.name, file.name)
        )

        comp = mk_catalog("eventtesting", file, exec)

        trans = assert_events([:file_created, :triggered], comp)

        assert_equal(1, trans.triggered?(exec, :refresh))
    end

    def test_simplerequire
        name = tempfile()
        file = Puppet::Type.type(:file).new(
            :name => name,
            :ensure => "file"
        )
        exec = Puppet::Type.type(:exec).new(
            :name => "echo true",
            :path => "/usr/bin:/bin",
            :refreshonly => true,
            :require => Puppet::Resource::Reference.new(file.class.name, file.name)
        )


        config = mk_catalog
        config.add_resource file
        config.add_resource exec
        trans = config.apply

        assert_equal(1, trans.events.length)

        assert_equal(0, trans.triggered?(exec, :refresh))
    end

    def test_multiplerefreshes
        files = []

        4.times { |i|
            files << Puppet::Type.type(:file).new(
                :name => tempfile(),
                :ensure => "file"
            )
        }

        fname = tempfile()
        exec = Puppet::Type.type(:exec).new(
            :name => "touch %s" % fname,
            :path => "/usr/bin:/bin",
            :refreshonly => true
        )

        exec[:subscribe] = files.collect { |f|
            Puppet::Resource::Reference.new(:file, f.name)
        }

        comp = mk_catalog(exec, *files)

        assert_apply(comp)
        assert(FileTest.exists?(fname), "Exec file did not get created")
    end

    # Make sure refreshing happens mid-transaction, rather than at the end.
    def test_refreshordering
        file = tempfile()

        exec1 = Puppet::Type.type(:exec).new(
            :title => "one",
            :name => "echo one >> %s" % file,
            :path => "/usr/bin:/bin"
        )

        exec2 = Puppet::Type.type(:exec).new(
            :title => "two",
            :name => "echo two >> %s" % file,
            :path => "/usr/bin:/bin",
            :refreshonly => true,
            :subscribe => exec1
        )

        exec3 = Puppet::Type.type(:exec).new(
            :title => "three",
            :name => "echo three >> %s" % file,
            :path => "/usr/bin:/bin",
            :require => exec2
        )
        execs = [exec1, exec2, exec3]

        config = mk_catalog(exec1,exec2,exec3)
        
        trans = Puppet::Transaction.new(config)
        execs.each do |e| assert(config.vertex?(e), "%s is not in graph" % e.title) end
        trans.prepare
        execs.each do |e| assert(config.vertex?(e), "%s is not in relgraph" % e.title) end
        reverse = trans.relationship_graph.reversal
        execs.each do |e| assert(reverse.vertex?(e), "%s is not in reversed graph" % e.title) end
        
        config.apply

        assert(FileTest.exists?(file), "File does not exist")

        assert_equal("one\ntwo\nthree\n", File.read(file))
    end
end