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
|
#
# Author:: John Keiser <jkeiser@chef.io>
# Copyright:: Copyright (c) 2015 John Keiser.
# License:: Apache License, Version 2.0
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
# http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.
#
#
# Mocks File and ShellOut.
#
# world.files["/etc/rc.d/init.d"] = "blah" means that /etc/rc.d/init.d is a file with contents "blah."
# world.commands["shellme"] = "result" means that when shell_out! calls "shellme", it will return "result" in stdout.
# world.commands["shellme"] = 10 means that when shell_out! calls "shellme", it will return exit code 10
# world.commands["shellme"] = { code: 10, stdout: "result", stderr: "result2"}
#
shared_context "double world" do
let(:world) { DoubleWorld.new }
before do
allow(::File).to receive(:exist?) { |f| world.files.has_key?(f) }
allow(::File).to receive(:exists?) { |f| world.files.has_key?(f) }
allow(::File).to receive(:executable?) { |f| world.files.has_key?(f) }
allow(::File).to receive(:open) { |f| StringIO.new(world.files[f]) }
allow(::Mixlib::ShellOut).to receive(:new) do |c|
result = world.commands[c]
case result
when String
result = { stdout: result }
when Integer
result = { exitstatus: result }
when Hash
else
raise ArgumentError, "Result for command #{c} unexpected or undefined: #{result.inspect}"
end
DoubleShellout.new(result)
end
end
private
class DoubleWorld
def initialize
@files = {}
@commands = {}
end
attr_reader :files
attr_reader :commands
end
class DoubleShellout
def initialize(properties)
@properties = {
stdout: "",
stderr: "",
exitstatus: 0
}.merge(properties)
end
def method_missing(name, *args)
@properties[name.to_sym]
end
def error?
exitstatus != 0
end
def error!
raise Mixlib::ShellOut::ShellCommandFailed, "Expected process to exit with 0, but received #{exitstatus}" if error?
end
end
end
|
