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
|
if __FILE__ == $0
$:.unshift '../../lib'
$:.unshift '..'
$puppetbase = "../.."
$debug = true
else
$debug = false
end
require 'puppet'
require 'test/unit'
require 'puppettest.rb'
require 'base64'
class TestBucket < ServerTest
# run through all of the files and exercise the filebucket methods
def checkfiles(client)
files = filelist()
# iterate across all of the files
files.each { |file|
spin
name = File.basename(file)
tmppath = File.join(tmpdir,name)
# copy the files to our tmp directory so we can modify them...
File.open(tmppath,File::WRONLY|File::TRUNC|File::CREAT) { |wf|
File.open(file) { |rf|
wf.print(rf.read)
}
}
# make sure the copy worked
assert(FileTest.exists?(tmppath))
# backup both the orig file and the tmp file
osum = nil
tsum = nil
nsum = nil
spin
assert_nothing_raised {
osum = client.backup(file)
}
spin
assert_nothing_raised {
tsum = client.backup(tmppath)
}
# verify you got the same sum back for both
assert(tsum == osum)
# modify our tmp file
File.open(tmppath,File::WRONLY|File::TRUNC) { |wf|
wf.print "This is some test text\n"
}
# back it up
spin
assert_nothing_raised {
#STDERR.puts("backing up %s" % tmppath) if $debug
nsum = client.backup(tmppath)
}
# and verify the sum changed
assert(tsum != nsum)
# restore the orig
spin
assert_nothing_raised {
nsum = client.restore(tmppath,tsum)
}
# and verify it actually got restored
spin
contents = File.open(tmppath) { |rf|
#STDERR.puts("reading %s" % tmppath) if $debug
rf.read
}
csum = Digest::MD5.hexdigest(contents)
assert(tsum == csum)
}
end
# a list of files that should be on the system
# just something to test moving files around
def filelist
if defined? @files
return @files
else
@files = []
end
%w{
who bash vim sh uname /etc/passwd /etc/syslog.conf /etc/hosts
}.each { |file|
# if it's fully qualified, just add it
if file =~ /^\//
if FileTest.exists?(file)
@files.push file
end
else
# else if it's unqualified, look for it in our path
begin
path = %x{which #{file}}
rescue => detail
#STDERR.puts "Could not search for binaries: %s" % detail
next
end
if path != ""
@files.push path.chomp
end
end
}
return @files
end
def setup
super
@bucket = File.join(Puppet[:puppetconf], "buckettesting")
@@tmpfiles << @bucket
end
# test operating against the local filebucket object
# this calls the direct server methods, which are different than the
# Dipper methods
def test_localserver
files = filelist()
server = nil
assert_nothing_raised {
server = Puppet::Server::FileBucket.new(
:Bucket => @bucket
)
}
# iterate across them...
files.each { |file|
spin
contents = File.open(file) { |of| of.read }
md5 = nil
# add a file to the repository
assert_nothing_raised {
#STDERR.puts("adding %s" % file) if $debug
md5 = server.addfile(Base64.encode64(contents),file)
}
# and get it back again
newcontents = nil
assert_nothing_raised {
#STDERR.puts("getting %s" % file) if $debug
newcontents = Base64.decode64(server.getfile(md5))
}
# and then make sure they're still the same
assert(
contents == newcontents
)
}
end
# test with a server and a Dipper
def test_localboth
files = filelist()
tmpdir = File.join(tmpdir(),"tmpfiledir")
FileUtils.mkdir_p(tmpdir)
bucket = nil
client = nil
threads = []
assert_nothing_raised {
bucket = Puppet::Server::FileBucket.new(
:Bucket => @bucket
)
}
#sleep(30)
assert_nothing_raised {
client = Puppet::Client::Dipper.new(
:Bucket => bucket
)
}
checkfiles(client)
end
# test that things work over the wire
def test_webxmlmix
files = filelist()
tmpdir = File.join(tmpdir(),"tmpfiledir")
FileUtils.mkdir_p(tmpdir)
Puppet[:autosign] = true
client = nil
port = Puppet[:masterport]
pid = mkserver(:CA => nil, :FileBucket => { :Bucket => @bucket})
assert_nothing_raised {
client = Puppet::Client::Dipper.new(
:Server => "localhost",
:Port => @@port
)
}
checkfiles(client)
unless pid
raise "Uh, we don't have a child pid"
end
system("kill %s" % pid)
end
end
# $Id$
|