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
|
require 'puppettest/servertest'
module PuppetTest::ExeTest
include PuppetTest::ServerTest
def setup
super
setbindir
setlibdir
end
def bindir
File.join(basedir, "bin")
end
def sbindir
File.join(basedir, "sbin")
end
def setbindir
unless ENV["PATH"].split(":").include?(bindir)
ENV["PATH"] = [bindir, ENV["PATH"]].join(":")
end
unless ENV["PATH"].split(":").include?(sbindir)
ENV["PATH"] = [sbindir, ENV["PATH"]].join(":")
end
end
def setlibdir
ENV["RUBYLIB"] = $LOAD_PATH.find_all { |dir|
dir =~ /puppet/ or dir =~ /\.\./
}.join(":")
end
# Run a ruby command. This explicitly uses ruby to run stuff, since we
# don't necessarily know where our ruby binary is, dernit.
# Currently unused, because I couldn't get it to work.
def rundaemon(*cmd)
@ruby ||= %x{which ruby}.chomp
cmd = cmd.unshift(@ruby).join(" ")
out = nil
Dir.chdir(bindir()) {
out = %x{#{@ruby} #{cmd}}
}
return out
end
def startmasterd(args = "")
output = nil
manifest = mktestmanifest()
args += " --manifest %s" % manifest
args += " --confdir %s" % Puppet[:confdir]
args += " --rundir %s" % File.join(Puppet[:vardir], "run")
args += " --vardir %s" % Puppet[:vardir]
args += " --certdnsnames %s" % Puppet[:certdnsnames]
args += " --masterport %s" % @@port
args += " --user %s" % Puppet::Util::SUIDManager.uid
args += " --group %s" % Puppet::Util::SUIDManager.gid
args += " --autosign true"
#if Puppet[:debug]
# args += " --debug"
#end
cmd = "puppetmasterd %s" % args
assert_nothing_raised {
output = %x{#{cmd}}.chomp
}
assert_equal("", output, "Puppetmasterd produced output %s" % output)
assert($CHILD_STATUS == 0, "Puppetmasterd exit status was %s" % $CHILD_STATUS)
sleep(1)
cleanup do
stopmasterd
sleep(1)
end
return manifest
end
def stopmasterd(running = true)
ps = Facter["ps"].value || "ps -ef"
pidfile = File.join(Puppet[:vardir], "run", "puppetmasterd.pid")
pid = nil
if FileTest.exists?(pidfile)
pid = File.read(pidfile).chomp.to_i
File.unlink(pidfile)
end
return unless running
if running or pid
runningpid = nil
%x{#{ps}}.chomp.split(/\n/).each { |line|
if line =~ /ruby.+puppetmasterd/
next if line =~ /\.rb/ # skip the test script itself
next if line =~ /^puppet/ # skip masters running as 'puppet'
ary = line.sub(/^\s+/, '').split(/\s+/)
pid = ary[1].to_i
end
}
end
# we default to mandating that it's running, but teardown
# doesn't require that
if pid
if pid == $PID
raise Puppet::Error, "Tried to kill own pid"
end
begin
Process.kill(:INT, pid)
rescue
# ignore it
end
end
end
def teardown
stopmasterd(false)
super
end
end
|
