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
|
require 'spec_helper'
require 'puppet/face'
require 'puppet/util/command_line'
describe Puppet::Util::CommandLine do
include PuppetSpec::Files
context "#initialize" do
it "should pull off the first argument if it looks like a subcommand" do
command_line = Puppet::Util::CommandLine.new("puppet", %w{ client --help whatever.pp })
expect(command_line.subcommand_name).to eq("client")
expect(command_line.args).to eq(%w{ --help whatever.pp })
end
it "should return nil if the first argument looks like a .pp file" do
command_line = Puppet::Util::CommandLine.new("puppet", %w{ whatever.pp })
expect(command_line.subcommand_name).to eq(nil)
expect(command_line.args).to eq(%w{ whatever.pp })
end
it "should return nil if the first argument looks like a flag" do
command_line = Puppet::Util::CommandLine.new("puppet", %w{ --debug })
expect(command_line.subcommand_name).to eq(nil)
expect(command_line.args).to eq(%w{ --debug })
end
it "should return nil if the first argument is -" do
command_line = Puppet::Util::CommandLine.new("puppet", %w{ - })
expect(command_line.subcommand_name).to eq(nil)
expect(command_line.args).to eq(%w{ - })
end
it "should return nil if the first argument is --help" do
command_line = Puppet::Util::CommandLine.new("puppet", %w{ --help })
expect(command_line.subcommand_name).to eq(nil)
end
it "should return nil if there are no arguments" do
command_line = Puppet::Util::CommandLine.new("puppet", [])
expect(command_line.subcommand_name).to eq(nil)
expect(command_line.args).to eq([])
end
it "should pick up changes to the array of arguments" do
args = %w{subcommand}
command_line = Puppet::Util::CommandLine.new("puppet", args)
args[0] = 'different_subcommand'
expect(command_line.subcommand_name).to eq('different_subcommand')
end
end
context "#execute" do
%w{--version -V}.each do |arg|
it "should print the version and exit if #{arg} is given" do
expect do
described_class.new("puppet", [arg]).execute
end.to output(/^#{Regexp.escape(Puppet.version)}$/).to_stdout
end
end
%w{--help -h help}.each do|arg|
it "should print help and exit if #{arg} is given" do
commandline = Puppet::Util::CommandLine.new("puppet", [arg])
expect(commandline).not_to receive(:exec)
expect {
commandline.execute
}.to exit_with(0)
.and output(/Usage: puppet <subcommand> \[options\] <action> \[options\]/).to_stdout
end
end
it "should fail if the config file isn't readable and we're running a subcommand that requires a readable config file" do
allow(Puppet::FileSystem).to receive(:exist?).with(Puppet[:config]).and_return(true)
allow_any_instance_of(Puppet::Settings).to receive(:read_file).and_return('')
expect_any_instance_of(Puppet::Settings).to receive(:read_file).with(Puppet[:config]).and_raise('Permission denied')
expect{ described_class.new("puppet", ['config']).execute }.to raise_error(SystemExit)
end
it "should not fail if the config file isn't readable and we're running a subcommand that does not require a readable config file" do
allow(Puppet::FileSystem).to receive(:exist?)
allow(Puppet::FileSystem).to receive(:exist?).with(Puppet[:config]).and_return(true)
allow_any_instance_of(Puppet::Settings).to receive(:read_file).and_return('')
expect_any_instance_of(Puppet::Settings).to receive(:read_file).with(Puppet[:config]).and_raise('Permission denied')
commandline = described_class.new("puppet", ['help'])
expect {
commandline.execute
}.to exit_with(0)
.and output(/Usage: puppet <subcommand> \[options\] <action> \[options\]/).to_stdout
end
end
describe "when dealing with puppet commands" do
it "should return the executable name if it is not puppet" do
command_line = Puppet::Util::CommandLine.new("puppetmasterd", [])
expect(command_line.subcommand_name).to eq("puppetmasterd")
end
describe "when the subcommand is not implemented" do
it "should find and invoke an executable with a hyphenated name" do
commandline = Puppet::Util::CommandLine.new("puppet", ['whatever', 'argument'])
expect(Puppet::Util).to receive(:which).with('puppet-whatever').
and_return('/dev/null/puppet-whatever')
expect(Kernel).to receive(:exec).with('/dev/null/puppet-whatever', 'argument')
commandline.execute
end
describe "and an external implementation cannot be found" do
it "should abort and show the usage message" do
expect(Puppet::Util).to receive(:which).with('puppet-whatever').and_return(nil)
commandline = Puppet::Util::CommandLine.new("puppet", ['whatever', 'argument'])
expect(commandline).not_to receive(:exec)
expect {
commandline.execute
}.to exit_with(1)
.and output(/Unknown Puppet subcommand 'whatever'/).to_stdout
end
it "should abort and show the help message" do
expect(Puppet::Util).to receive(:which).with('puppet-whatever').and_return(nil)
commandline = Puppet::Util::CommandLine.new("puppet", ['whatever', 'argument'])
expect(commandline).not_to receive(:exec)
expect {
commandline.execute
}.to exit_with(1)
.and output(/See 'puppet help' for help on available puppet subcommands/).to_stdout
end
%w{--version -V}.each do |arg|
it "should abort and display #{arg} information" do
expect(Puppet::Util).to receive(:which).with('puppet-whatever').and_return(nil)
commandline = Puppet::Util::CommandLine.new("puppet", ['whatever', arg])
expect(commandline).not_to receive(:exec)
expect {
commandline.execute
}.to exit_with(1)
.and output(%r[^#{Regexp.escape(Puppet.version)}$]).to_stdout
end
end
end
end
describe 'when setting process priority' do
let(:command_line) do
Puppet::Util::CommandLine.new("puppet", %w{ agent })
end
before :each do
allow_any_instance_of(Puppet::Util::CommandLine::ApplicationSubcommand).to receive(:run)
end
it 'should never set priority by default' do
expect(Process).not_to receive(:setpriority)
command_line.execute
end
it 'should lower the process priority if one has been specified' do
Puppet[:priority] = 10
expect(Process).to receive(:setpriority).with(0, Process.pid, 10)
command_line.execute
end
it 'should warn if trying to raise priority, but not privileged user' do
Puppet[:priority] = -10
expect(Process).to receive(:setpriority).and_raise(Errno::EACCES, 'Permission denied')
expect(Puppet).to receive(:warning).with("Failed to set process priority to '-10'")
command_line.execute
end
it "should warn if the platform doesn't support `Process.setpriority`" do
Puppet[:priority] = 15
expect(Process).to receive(:setpriority).and_raise(NotImplementedError, 'NotImplementedError: setpriority() function is unimplemented on this machine')
expect(Puppet).to receive(:warning).with("Failed to set process priority to '15'")
command_line.execute
end
end
end
end
|
