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
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
249
250
251
252
253
254
255
256
257
258
259
260
261
262
263
264
265
266
267
268
269
270
271
272
273
274
275
276
277
278
279
280
281
282
283
284
285
286
287
288
289
290
291
292
293
294
295
296
297
298
299
300
301
302
303
304
305
306
307
308
|
#!/usr/bin/env ruby
$:.unshift("../lib").unshift("../../lib") if __FILE__ =~ /\.rb$/
require 'puppet/rails'
require 'puppettest'
require 'puppettest/parsertesting'
require 'puppettest/resourcetesting'
require 'puppettest/railstesting'
class TestCollector < Test::Unit::TestCase
include PuppetTest
include PuppetTest::ParserTesting
include PuppetTest::ResourceTesting
include PuppetTest::RailsTesting
Parser = Puppet::Parser
AST = Parser::AST
def setup
super
Puppet[:trace] = false
@interp, @scope, @source = mkclassframing
end
# Test just collecting a specific resource. This is used by the 'realize'
# function, and it's much faster than iterating over all of the resources.
def test_collect_resource
# Make a couple of virtual resources
one = mkresource(:type => "file", :title => "/tmp/virtual1",
:virtual => true, :params => {:owner => "root"})
two = mkresource(:type => "file", :title => "/tmp/virtual2",
:virtual => true, :params => {:owner => "root"})
@scope.setresource one
@scope.setresource two
# Now make a collector
coll = nil
assert_nothing_raised do
coll = Puppet::Parser::Collector.new(@scope, "file", nil, nil, :virtual)
end
# Now set the resource in the collector
assert_nothing_raised do
coll.resources = one.ref
end
# Now run the collector
assert_nothing_raised do
coll.evaluate
end
# And make sure the resource is no longer virtual
assert(! one.virtual?,
"Resource is still virtual")
# But the other still is
assert(two.virtual?,
"Resource got realized")
end
def test_virtual
# Make a virtual resource
virtual = mkresource(:type => "file", :title => "/tmp/virtual",
:virtual => true, :params => {:owner => "root"})
@scope.setresource virtual
# And a non-virtual
real = mkresource(:type => "file", :title => "/tmp/real",
:params => {:owner => "root"})
@scope.setresource real
# Now make a collector
coll = nil
# Make a fake query
code = proc do |res|
true
end
assert_nothing_raised do
coll = Puppet::Parser::Collector.new(@scope, "file", nil, code, :virtual)
end
# Set it in our scope
@scope.newcollection(coll)
# Make sure it's in the collections
assert_equal([coll], @scope.collections)
# And try to collect the virtual resources.
ret = nil
assert_nothing_raised do
ret = coll.collect_virtual
end
assert_equal([virtual], ret)
# Now make sure evaluate does the right thing.
assert_nothing_raised do
ret = coll.evaluate
end
# Make sure it got deleted from the collection list
assert_equal([], @scope.collections)
# And make sure our virtual object is no longer virtual
assert(! virtual.virtual?, "Virtual object did not get realized")
# Now make a new collector of a different type and make sure it
# finds nothing.
assert_nothing_raised do
coll = Puppet::Parser::Collector.new(@scope, "exec", nil, nil, :virtual)
end
# Remark this as virtual
virtual.virtual = true
assert_nothing_raised do
ret = coll.evaluate
end
assert_equal([], ret)
end
if Puppet.features.rails?
def test_collect_exported
railsinit
# Set a hostname
@scope.host = Facter.value(:hostname)
# make an exported resource
exported = mkresource(:type => "file", :title => "/tmp/exported",
:exported => true, :params => {:owner => "root"})
@scope.setresource exported
assert(exported.exported?, "Object was not marked exported")
assert(exported.virtual?, "Object was not marked virtual")
# And a non-exported
real = mkresource(:type => "file", :title => "/tmp/real",
:params => {:owner => "root"})
@scope.setresource real
# Now make a collector
coll = nil
assert_nothing_raised do
coll = Puppet::Parser::Collector.new(@scope, "file", nil, nil, :exported)
end
# Set it in our scope
@scope.newcollection(coll)
# Make sure it's in the collections
assert_equal([coll], @scope.collections)
# And try to collect the virtual resources.
ret = nil
assert_nothing_raised do
ret = coll.collect_exported
end
assert_equal([exported], ret)
# Now make sure evaluate does the right thing.
assert_nothing_raised do
ret = coll.evaluate
end
# Make sure it got deleted from the collection list
assert_equal([], @scope.collections)
# And make sure our exported object is no longer exported
assert(! exported.virtual?, "Virtual object did not get realized")
# But it should still be marked exported.
assert(exported.exported?, "Resource got un-exported")
# Now make a new collector of a different type and make sure it
# finds nothing.
assert_nothing_raised do
coll = Puppet::Parser::Collector.new(@scope, "exec", nil, nil, :exported)
end
# Remark this as virtual
exported.virtual = true
assert_nothing_raised do
ret = coll.evaluate
end
assert_equal([], ret)
# Now create a whole new scope and make sure we can actually retrieve
# the resource from the database, not just from the scope.
# First create a host object and store our resource in it.
# Now collect our facts
facts = {}
Facter.each do |fact, value| facts[fact] = value end
# Now try storing our crap
# Remark this as exported
exported.exported = true
host = Puppet::Rails::Host.store(
:resources => [exported],
:facts => facts,
:name => facts["hostname"]
)
assert(host, "did not get rails host")
host.save
# And make sure it's in there
newres = host.resources.find_by_restype_and_title_and_exported("file", "/tmp/exported", true)
assert(newres, "Did not find resource in db")
interp, scope, source = mkclassframing
scope.host = "two"
# Now make a collector
coll = nil
assert_nothing_raised do
coll = Puppet::Parser::Collector.new(scope, "file", nil, nil, :exported)
end
# Set it in our scope
scope.newcollection(coll)
# Make sure it's in the collections
assert_equal([coll], scope.collections)
# And try to collect the virtual resources.
ret = nil
assert_nothing_raised do
ret = coll.collect_exported
end
assert_equal(["/tmp/exported"], ret.collect { |f| f.title })
# Make sure we can evaluate the same collection multiple times and
# that later collections do nothing
assert_nothing_raised do
ret = coll.evaluate
end
# Make sure it got deleted from the collection list
assert_equal([], scope.collections)
end
def test_collection_conflicts
railsinit
# First make a railshost we can conflict with
host = Puppet::Rails::Host.new(:name => "myhost")
host.resources.build(:title => "/tmp/conflicttest", :restype => "file",
:exported => true)
host.save
# Now make a normal resource
normal = mkresource(:type => "file", :title => "/tmp/conflicttest",
:params => {:owner => "root"})
@scope.setresource normal
@scope.host = "otherhost"
# Now make a collector
coll = nil
assert_nothing_raised do
coll = Puppet::Parser::Collector.new(@scope, "file", nil, nil, :exported)
end
# And try to collect the virtual resources.
assert_raise(Puppet::ParseError) do
ret = coll.collect_exported
end
end
# Make sure we do not collect resources from the host we're on
def test_no_resources_from_me
railsinit
# Make our configuration
host = Puppet::Rails::Host.new(:name => "myhost")
host.resources.build(:title => "/tmp/hosttest", :type => "PuppetFile",
:exported => true)
host.save
@scope.host = "myhost"
# Now make a collector
coll = nil
assert_nothing_raised do
coll = Puppet::Parser::Collector.new(@scope, "file", nil, nil, :exported)
end
# And make sure we get nada back
ret = nil
assert_nothing_raised do
ret = coll.collect_exported
end
assert(ret.empty?, "Found exports from our own host")
end
end
end
# $Id$
|