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
|
#! /usr/bin/env ruby
require 'spec_helper'
require 'puppet/application/master'
require 'puppet/daemon'
require 'puppet/network/server'
describe Puppet::Application::Master, :unless => Puppet.features.microsoft_windows? do
before :each do
@master = Puppet::Application[:master]
@daemon = stub_everything 'daemon'
Puppet::Daemon.stubs(:new).returns(@daemon)
Puppet::Util::Log.stubs(:newdestination)
Puppet::Node.indirection.stubs(:terminus_class=)
Puppet::Node.indirection.stubs(:cache_class=)
Puppet::Node::Facts.indirection.stubs(:terminus_class=)
Puppet::Node::Facts.indirection.stubs(:cache_class=)
Puppet::Transaction::Report.indirection.stubs(:terminus_class=)
Puppet::Resource::Catalog.indirection.stubs(:terminus_class=)
Puppet::SSL::Host.stubs(:ca_location=)
end
it "should operate in master run_mode" do
@master.class.run_mode.name.should equal(:master)
end
it "should declare a main command" do
@master.should respond_to(:main)
end
it "should declare a compile command" do
@master.should respond_to(:compile)
end
it "should declare a preinit block" do
@master.should respond_to(:preinit)
end
describe "during preinit" do
before :each do
@master.stubs(:trap)
end
it "should catch INT" do
@master.stubs(:trap).with { |arg,block| arg == :INT }
@master.preinit
end
it "should create a Puppet Daemon" do
Puppet::Daemon.expects(:new).returns(@daemon)
@master.preinit
end
it "should give ARGV to the Daemon" do
argv = stub 'argv'
ARGV.stubs(:dup).returns(argv)
@daemon.expects(:argv=).with(argv)
@master.preinit
end
end
[:debug,:verbose].each do |option|
it "should declare handle_#{option} method" do
@master.should respond_to("handle_#{option}".to_sym)
end
it "should store argument value when calling handle_#{option}" do
@master.options.expects(:[]=).with(option, 'arg')
@master.send("handle_#{option}".to_sym, 'arg')
end
end
describe "when applying options" do
before do
@master.command_line.stubs(:args).returns([])
end
it "should set the log destination with --logdest" do
Puppet::Log.expects(:newdestination).with("console")
@master.handle_logdest("console")
end
it "should put the setdest options to true" do
@master.options.expects(:[]=).with(:setdest,true)
@master.handle_logdest("console")
end
it "should parse the log destination from ARGV" do
@master.command_line.stubs(:args).returns(%w{--logdest /my/file})
Puppet::Util::Log.expects(:newdestination).with("/my/file")
@master.parse_options
end
it "should support dns alt names from ARGV" do
Puppet.settings.initialize_global_settings(["--dns_alt_names", "foo,bar,baz"])
@master.preinit
@master.parse_options
Puppet[:dns_alt_names].should == "foo,bar,baz"
end
end
describe "during setup" do
before :each do
Puppet::Log.stubs(:newdestination)
Puppet.stubs(:settraps)
Puppet::SSL::CertificateAuthority.stubs(:instance)
Puppet::SSL::CertificateAuthority.stubs(:ca?)
Puppet.settings.stubs(:use)
@master.options.stubs(:[]).with(any_parameters)
end
it "should abort stating that the master is not supported on Windows" do
Puppet.features.stubs(:microsoft_windows?).returns(true)
expect { @master.setup }.to raise_error(Puppet::Error, /Puppet master is not supported on Microsoft Windows/)
end
it "should set log level to debug if --debug was passed" do
@master.options.stubs(:[]).with(:debug).returns(true)
@master.setup
Puppet::Log.level.should == :debug
end
it "should set log level to info if --verbose was passed" do
@master.options.stubs(:[]).with(:verbose).returns(true)
@master.setup
Puppet::Log.level.should == :info
end
it "should set console as the log destination if no --logdest and --daemonize" do
@master.stubs(:[]).with(:daemonize).returns(:false)
Puppet::Log.expects(:newdestination).with(:syslog)
@master.setup
end
it "should set syslog as the log destination if no --logdest and not --daemonize" do
Puppet::Log.expects(:newdestination).with(:syslog)
@master.setup
end
it "should set syslog as the log destination if --rack" do
@master.options.stubs(:[]).with(:rack).returns(:true)
Puppet::Log.expects(:newdestination).with(:syslog)
@master.setup
end
it "should print puppet config if asked to in Puppet config" do
Puppet.settings.stubs(:print_configs?).returns(true)
Puppet.settings.expects(:print_configs).returns(true)
expect { @master.setup }.to exit_with 0
end
it "should exit after printing puppet config if asked to in Puppet config" do
Puppet.settings.stubs(:print_configs?).returns(true)
expect { @master.setup }.to exit_with 1
end
it "should tell Puppet.settings to use :main,:ssl,:master and :metrics category" do
Puppet.settings.expects(:use).with(:main,:master,:ssl,:metrics)
@master.setup
end
describe "with no ca" do
it "should set the ca_location to none" do
Puppet::SSL::Host.expects(:ca_location=).with(:none)
@master.setup
end
end
describe "with a ca configured" do
before :each do
Puppet::SSL::CertificateAuthority.stubs(:ca?).returns(true)
end
it "should set the ca_location to local" do
Puppet::SSL::Host.expects(:ca_location=).with(:local)
@master.setup
end
it "should tell Puppet.settings to use :ca category" do
Puppet.settings.expects(:use).with(:ca)
@master.setup
end
it "should instantiate the CertificateAuthority singleton" do
Puppet::SSL::CertificateAuthority.expects(:instance)
@master.setup
end
end
end
describe "when running" do
before do
@master.preinit
end
it "should dispatch to compile if called with --compile" do
@master.options[:node] = "foo"
@master.expects(:compile)
@master.run_command
end
it "should dispatch to main otherwise" do
@master.options[:node] = nil
@master.expects(:main)
@master.run_command
end
describe "the compile command" do
before do
Puppet[:manifest] = "site.pp"
Puppet.stubs(:err)
@master.stubs(:puts)
end
it "should compile a catalog for the specified node" do
@master.options[:node] = "foo"
Puppet::Resource::Catalog.indirection.expects(:find).with("foo").returns Puppet::Resource::Catalog.new
expect { @master.compile }.to exit_with 0
end
it "should convert the catalog to a pure-resource catalog and use 'PSON::pretty_generate' to pretty-print the catalog" do
catalog = Puppet::Resource::Catalog.new
PSON.stubs(:pretty_generate)
Puppet::Resource::Catalog.indirection.expects(:find).returns catalog
catalog.expects(:to_resource).returns("rescat")
@master.options[:node] = "foo"
PSON.expects(:pretty_generate).with('rescat', :allow_nan => true, :max_nesting => false)
expect { @master.compile }.to exit_with 0
end
it "should exit with error code 30 if no catalog can be found" do
@master.options[:node] = "foo"
Puppet::Resource::Catalog.indirection.expects(:find).returns nil
$stderr.expects(:puts)
expect { @master.compile }.to exit_with 30
end
it "should exit with error code 30 if there's a failure" do
@master.options[:node] = "foo"
Puppet::Resource::Catalog.indirection.expects(:find).raises ArgumentError
$stderr.expects(:puts)
expect { @master.compile }.to exit_with 30
end
end
describe "the main command" do
before :each do
@master.preinit
@server = stub_everything 'server'
Puppet::Network::Server.stubs(:new).returns(@server)
@app = stub_everything 'app'
Puppet::SSL::Host.stubs(:localhost)
Puppet::SSL::CertificateAuthority.stubs(:ca?)
Process.stubs(:uid).returns(1000)
Puppet.stubs(:service)
Puppet[:daemonize] = false
Puppet.stubs(:notice)
Puppet.stubs(:start)
end
it "should create a Server" do
Puppet::Network::Server.expects(:new)
@master.main
end
it "should give the server to the daemon" do
@daemon.expects(:server=).with(@server)
@master.main
end
it "should generate a SSL cert for localhost" do
Puppet::SSL::Host.expects(:localhost)
@master.main
end
it "should make sure to *only* hit the CA for data" do
Puppet::SSL::CertificateAuthority.stubs(:ca?).returns(true)
Puppet::SSL::Host.expects(:ca_location=).with(:only)
@master.main
end
it "should drop privileges if running as root" do
Puppet.features.stubs(:root?).returns true
Puppet::Util.expects(:chuser)
@master.main
end
it "should daemonize if needed" do
Puppet[:daemonize] = true
@daemon.expects(:daemonize)
@master.main
end
it "should start the service" do
@daemon.expects(:start)
@master.main
end
describe "with --rack", :if => Puppet.features.rack? do
before do
require 'puppet/network/http/rack'
Puppet::Network::HTTP::Rack.stubs(:new).returns(@app)
end
it "it should not start a daemon" do
@master.options.stubs(:[]).with(:rack).returns(:true)
@daemon.expects(:start).never
@master.main
end
it "it should return the app" do
@master.options.stubs(:[]).with(:rack).returns(:true)
app = @master.main
app.should equal(@app)
end
end
end
end
end
|