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
309
310
311
312
313
314
315
316
317
318
319
320
321
322
323
324
325
326
327
328
329
330
331
332
333
334
335
336
337
338
339
340
341
342
343
344
345
346
347
348
349
350
351
352
353
354
355
356
357
358
359
360
361
362
363
364
365
366
367
368
369
370
371
372
373
374
375
376
377
378
379
380
381
382
383
384
385
386
387
388
389
390
391
392
393
394
395
396
397
398
399
400
401
|
#!/usr/bin/env ruby
require File.dirname(__FILE__) + '/../../spec_helper'
require 'puppet/transaction/resource_harness'
describe Puppet::Transaction::ResourceHarness do
before do
@transaction = Puppet::Transaction.new(Puppet::Resource::Catalog.new)
@resource = Puppet::Type.type(:file).new :path => "/my/file"
@harness = Puppet::Transaction::ResourceHarness.new(@transaction)
@current_state = Puppet::Resource.new(:file, "/my/file")
@resource.stubs(:retrieve).returns @current_state
@status = Puppet::Resource::Status.new(@resource)
Puppet::Resource::Status.stubs(:new).returns @status
end
it "should accept a transaction at initialization" do
harness = Puppet::Transaction::ResourceHarness.new(@transaction)
harness.transaction.should equal(@transaction)
end
it "should delegate to the transaction for its relationship graph" do
@transaction.expects(:relationship_graph).returns "relgraph"
Puppet::Transaction::ResourceHarness.new(@transaction).relationship_graph.should == "relgraph"
end
describe "when copying audited parameters" do
before do
@resource = Puppet::Type.type(:file).new :path => "/foo/bar", :audit => :mode
end
it "should do nothing if no parameters are being audited" do
@resource[:audit] = []
@harness.expects(:cached).never
@harness.copy_audited_parameters(@resource, {}).should == []
end
it "should do nothing if an audited parameter already has a desired value set" do
@resource[:mode] = "755"
@harness.expects(:cached).never
@harness.copy_audited_parameters(@resource, {}).should == []
end
it "should copy any cached values to the 'should' values" do
@harness.cache(@resource, :mode, "755")
@harness.copy_audited_parameters(@resource, {}).should == [:mode]
@resource[:mode].should == 0755
end
it "should cache and log the current value if no cached values are present" do
@resource.expects(:info)
@harness.copy_audited_parameters(@resource, {:mode => "755"}).should == []
@harness.cached(@resource, :mode).should == "755"
end
end
describe "when evaluating a resource" do
it "should create and return a resource status instance for the resource" do
@harness.evaluate(@resource).should be_instance_of(Puppet::Resource::Status)
end
it "should fail if no status can be created" do
Puppet::Resource::Status.expects(:new).raises ArgumentError
lambda { @harness.evaluate(@resource) }.should raise_error
end
it "should retrieve the current state of the resource" do
@resource.expects(:retrieve).returns @current_state
@harness.evaluate(@resource)
end
it "should mark the resource as failed and return if the current state cannot be retrieved" do
@resource.expects(:retrieve).raises ArgumentError
@harness.evaluate(@resource).should be_failed
end
it "should use the status and retrieved state to determine which changes need to be made" do
@harness.expects(:changes_to_perform).with(@status, @resource).returns []
@harness.evaluate(@resource)
end
it "should mark the status as out of sync and apply the created changes if there are any" do
changes = %w{mychanges}
@harness.expects(:changes_to_perform).returns changes
@harness.expects(:apply_changes).with(@status, changes)
@harness.evaluate(@resource).should be_out_of_sync
end
it "should cache the last-synced time" do
changes = %w{mychanges}
@harness.stubs(:changes_to_perform).returns changes
@harness.stubs(:apply_changes)
@harness.expects(:cache).with { |resource, name, time| name == :synced and time.is_a?(Time) }
@harness.evaluate(@resource)
end
it "should flush the resource when applying changes if appropriate" do
changes = %w{mychanges}
@harness.stubs(:changes_to_perform).returns changes
@harness.stubs(:apply_changes)
@resource.expects(:flush)
@harness.evaluate(@resource)
end
it "should use the status and retrieved state to determine which changes need to be made" do
@harness.expects(:changes_to_perform).with(@status, @resource).returns []
@harness.evaluate(@resource)
end
it "should not attempt to apply changes if none need to be made" do
@harness.expects(:changes_to_perform).returns []
@harness.expects(:apply_changes).never
@harness.evaluate(@resource).should_not be_out_of_sync
end
it "should store the resource's evaluation time in the resource status" do
@harness.evaluate(@resource).evaluation_time.should be_instance_of(Float)
end
it "should set the change count to the total number of changes" do
changes = %w{a b c d}
@harness.expects(:changes_to_perform).returns changes
@harness.expects(:apply_changes).with(@status, changes)
@harness.evaluate(@resource).change_count.should == 4
end
end
describe "when creating changes" do
before do
@current_state = Puppet::Resource.new(:file, "/my/file")
@resource.stubs(:retrieve).returns @current_state
Puppet.features.stubs(:root?).returns true
end
it "should retrieve the current values from the resource" do
@resource.expects(:retrieve).returns @current_state
@harness.changes_to_perform(@status, @resource)
end
it "should cache that the resource was checked" do
@harness.expects(:cache).with { |resource, name, time| name == :checked and time.is_a?(Time) }
@harness.changes_to_perform(@status, @resource)
end
it "should create changes with the appropriate property and current value" do
@resource[:ensure] = :present
@current_state[:ensure] = :absent
change = stub 'change'
Puppet::Transaction::Change.expects(:new).with(@resource.parameter(:ensure), :absent).returns change
@harness.changes_to_perform(@status, @resource)[0].should equal(change)
end
it "should not attempt to manage properties that do not have desired values set" do
mode = @resource.newattr(:mode)
@current_state[:mode] = :absent
mode.expects(:insync?).never
@harness.changes_to_perform(@status, @resource)
end
it "should copy audited parameters" do
@resource[:audit] = :mode
@harness.cache(@resource, :mode, "755")
@harness.changes_to_perform(@status, @resource)
@resource[:mode].should == 0755
end
it "should mark changes created as a result of auditing as auditing changes" do
@current_state[:mode] = 0644
@resource[:audit] = :mode
@harness.cache(@resource, :mode, "755")
@harness.changes_to_perform(@status, @resource)[0].must be_auditing
end
describe "and the 'ensure' parameter is present but not in sync" do
it "should return a single change for the 'ensure' parameter" do
@resource[:ensure] = :present
@resource[:mode] = "755"
@current_state[:ensure] = :absent
@current_state[:mode] = :absent
@resource.stubs(:retrieve).returns @current_state
changes = @harness.changes_to_perform(@status, @resource)
changes.length.should == 1
changes[0].property.name.should == :ensure
end
end
describe "and the 'ensure' parameter should be set to 'absent', and is correctly set to 'absent'" do
it "should return no changes" do
@resource[:ensure] = :absent
@resource[:mode] = "755"
@current_state[:ensure] = :absent
@current_state[:mode] = :absent
@harness.changes_to_perform(@status, @resource).should == []
end
end
describe "and the 'ensure' parameter is 'absent' and there is no 'desired value'" do
it "should return no changes" do
@resource.newattr(:ensure)
@resource[:mode] = "755"
@current_state[:ensure] = :absent
@current_state[:mode] = :absent
@harness.changes_to_perform(@status, @resource).should == []
end
end
describe "and non-'ensure' parameters are not in sync" do
it "should return a change for each parameter that is not in sync" do
@resource[:ensure] = :present
@resource[:mode] = "755"
@resource[:owner] = 0
@current_state[:ensure] = :present
@current_state[:mode] = 0444
@current_state[:owner] = 50
mode = stub 'mode_change'
owner = stub 'owner_change'
Puppet::Transaction::Change.expects(:new).with(@resource.parameter(:mode), 0444).returns mode
Puppet::Transaction::Change.expects(:new).with(@resource.parameter(:owner), 50).returns owner
changes = @harness.changes_to_perform(@status, @resource)
changes.length.should == 2
changes.should be_include(mode)
changes.should be_include(owner)
end
end
describe "and all parameters are in sync" do
it "should return an empty array" do
@resource[:ensure] = :present
@resource[:mode] = "755"
@current_state[:ensure] = :present
@current_state[:mode] = 0755
@harness.changes_to_perform(@status, @resource).should == []
end
end
end
describe "when applying changes" do
before do
@change1 = stub 'change1', :apply => stub("event", :status => "success"), :auditing? => false
@change2 = stub 'change2', :apply => stub("event", :status => "success"), :auditing? => false
@changes = [@change1, @change2]
end
it "should apply the change" do
@change1.expects(:apply).returns( stub("event", :status => "success") )
@change2.expects(:apply).returns( stub("event", :status => "success") )
@harness.apply_changes(@status, @changes)
end
it "should mark the resource as changed" do
@harness.apply_changes(@status, @changes)
@status.should be_changed
end
it "should queue the resulting event" do
@harness.apply_changes(@status, @changes)
@status.events.should be_include(@change1.apply)
@status.events.should be_include(@change2.apply)
end
it "should cache the new value if it is an auditing change" do
@change1.expects(:auditing?).returns true
property = stub 'property', :name => "foo", :resource => "myres"
@change1.stubs(:property).returns property
@change1.stubs(:is).returns "myval"
@harness.apply_changes(@status, @changes)
@harness.cached("myres", "foo").should == "myval"
end
end
describe "when determining whether the resource can be changed" do
before do
@resource.stubs(:purging?).returns true
@resource.stubs(:deleting?).returns true
end
it "should be true if the resource is not being purged" do
@resource.expects(:purging?).returns false
@harness.should be_allow_changes(@resource)
end
it "should be true if the resource is not being deleted" do
@resource.expects(:deleting?).returns false
@harness.should be_allow_changes(@resource)
end
it "should be true if the resource has no dependents" do
@harness.relationship_graph.expects(:dependents).with(@resource).returns []
@harness.should be_allow_changes(@resource)
end
it "should be true if all dependents are being deleted" do
dep = stub 'dependent', :deleting? => true
@harness.relationship_graph.expects(:dependents).with(@resource).returns [dep]
@resource.expects(:purging?).returns true
@harness.should be_allow_changes(@resource)
end
it "should be false if the resource's dependents are not being deleted" do
dep = stub 'dependent', :deleting? => false, :ref => "myres"
@resource.expects(:warning)
@harness.relationship_graph.expects(:dependents).with(@resource).returns [dep]
@harness.should_not be_allow_changes(@resource)
end
end
describe "when finding the schedule" do
before do
@catalog = Puppet::Resource::Catalog.new
@resource.catalog = @catalog
end
it "should warn and return nil if the resource has no catalog" do
@resource.catalog = nil
@resource.expects(:warning)
@harness.schedule(@resource).should be_nil
end
it "should return nil if the resource specifies no schedule" do
@harness.schedule(@resource).should be_nil
end
it "should fail if the named schedule cannot be found" do
@resource[:schedule] = "whatever"
@resource.expects(:fail)
@harness.schedule(@resource)
end
it "should return the named schedule if it exists" do
sched = Puppet::Type.type(:schedule).new(:name => "sched")
@catalog.add_resource(sched)
@resource[:schedule] = "sched"
@harness.schedule(@resource).to_s.should == sched.to_s
end
end
describe "when determining if a resource is scheduled" do
before do
@catalog = Puppet::Resource::Catalog.new
@resource.catalog = @catalog
@status = Puppet::Resource::Status.new(@resource)
end
it "should return true if 'ignoreschedules' is set" do
Puppet[:ignoreschedules] = true
@resource[:schedule] = "meh"
@harness.should be_scheduled(@status, @resource)
end
it "should return true if the resource has no schedule set" do
@harness.should be_scheduled(@status, @resource)
end
it "should return the result of matching the schedule with the cached 'checked' time if a schedule is set" do
t = Time.now
@harness.expects(:cached).with(@resource, :checked).returns(t)
sched = Puppet::Type.type(:schedule).new(:name => "sched")
@catalog.add_resource(sched)
@resource[:schedule] = "sched"
sched.expects(:match?).with(t.to_i).returns "feh"
@harness.scheduled?(@status, @resource).should == "feh"
end
end
it "should be able to cache data in the Storage module" do
data = {}
Puppet::Util::Storage.expects(:cache).with(@resource).returns data
@harness.cache(@resource, :foo, "something")
data[:foo].should == "something"
end
it "should be able to retrieve data from the cache" do
data = {:foo => "other"}
Puppet::Util::Storage.expects(:cache).with(@resource).returns data
@harness.cached(@resource, :foo).should == "other"
end
end
|