blob: 444b0e4ec0bfba861458819b732b0f8ed4659b63 (
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
|
module PuppetTestSupport
module Assertions
def assert_rollback_events(events, trans, msg = nil)
run_events(:rollback, events, trans, msg)
end
def assert_events(events, *items)
trans = nil
comp = nil
msg = nil
unless events.is_a? Array
raise Puppet::DevError, "Incorrect call of assert_events"
end
if items[-1].is_a? String
msg = items.pop
end
remove_comp = false
# They either passed a comp or a list of items.
if items[0].is_a? Puppet.type(:component)
comp = items.shift
else
comp = newcomp(items[0].title, *items)
remove_comp = true
end
msg ||= comp.title
assert_nothing_raised("Component %s failed" % [msg]) {
trans = comp.evaluate
}
run_events(:evaluate, trans, events, msg)
if remove_comp
Puppet.type(:component).delete(comp)
end
return trans
end
# A simpler method that just applies what we have.
def assert_apply(*objects)
if objects[0].is_a?(Puppet.type(:component))
comp = objects.shift
unless objects.empty?
objects.each { |o| comp.push o }
end
else
comp = newcomp(*objects)
end
trans = nil
assert_nothing_raised("Failed to create transaction") {
trans = comp.evaluate
}
events = nil
assert_nothing_raised("Failed to evaluate transaction") {
events = trans.evaluate.collect { |e| e.event }
}
Puppet.type(:component).delete(comp)
events
end
end
end
|
