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
|
#
# Author:: Daniel DeLeo (<dan@opscode.com>)
# Copyright:: Copyright (c) 2012 Opscode, Inc.
# 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.
#
require 'functional/resource/base'
require 'chef/version'
require 'chef/shell'
require 'chef/mixin/command/unix'
describe Shell do
# chef-shell's unit tests are by necessity very mock-heavy, and frequently do
# not catch cases where chef-shell fails to boot because of changes in
# chef/client.rb
describe "smoke tests", :unix_only => true do
include Chef::Mixin::Command::Unix
def read_until(io, expected_value)
start = Time.new
buffer = ""
until buffer.include?(expected_value)
begin
buffer << io.read_nonblock(1)
rescue Errno::EWOULDBLOCK, Errno::EAGAIN, Errno::EIO, EOFError
sleep 0.01
end
if Time.new - start > 30
STDERR.puts "did not read expected value `#{expected_value}' within 15s"
STDERR.puts "Buffer so far: `#{buffer}'"
break
end
end
buffer
end
def wait_or_die(pid)
start = Time.new
until exitstatus = Process.waitpid2(pid, Process::WNOHANG)
if Time.new - start > 5
STDERR.puts("chef-shell tty did not exit cleanly, killing it")
Process.kill(:KILL, pid)
end
sleep 0.01
end
exitstatus[1]
end
def run_chef_shell_with(options)
case ohai[:platform]
when "aix"
config = File.expand_path("shef-config.rb", CHEF_SPEC_DATA)
path_to_chef_shell = File.expand_path("../../../bin/chef-shell", __FILE__)
output = ''
status = popen4("#{path_to_chef_shell} -c #{config} #{options}", :waitlast => true) do |pid, stdin, stdout, stderr|
read_until(stdout, "chef >")
yield stdout, stdin if block_given?
stdin.write("'done'\n")
output = read_until(stdout, '=> "done"')
stdin.print("exit\n")
read_until(stdout, "\n")
end
[output, status.exitstatus]
else
# Windows ruby installs don't (always?) have PTY,
# so hide the require here
begin
require 'pty'
config = File.expand_path("shef-config.rb", CHEF_SPEC_DATA)
path_to_chef_shell = File.expand_path("../../../bin/chef-shell", __FILE__)
reader, writer, pid = PTY.spawn("#{path_to_chef_shell} -c #{config} #{options}")
read_until(reader, "chef >")
yield reader, writer if block_given?
writer.puts('"done"')
output = read_until(reader, '=> "done"')
writer.print("exit\n")
read_until(reader, "exit")
read_until(reader, "\n")
read_until(reader, "\n")
writer.close
exitstatus = wait_or_die(pid)
[output, exitstatus]
rescue PTY::ChildExited => e
[output, e.status]
end
end
end
it "boots correctly with -lauto" do
output, exitstatus = run_chef_shell_with("-lauto")
output.should include("done")
expect(exitstatus).to eq(0)
end
it "sets the log_level from the command line" do
output, exitstatus = run_chef_shell_with("-lfatal") do |out, keyboard|
show_log_level_code = %q[puts "===#{Chef::Log.level}==="]
keyboard.puts(show_log_level_code)
read_until(out, show_log_level_code)
end
output.should include("===fatal===")
expect(exitstatus).to eq(0)
end
end
end
|
