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
|
#!/usr/bin/env ruby
require File.dirname(__FILE__) + '/../lib/puppettest'
require 'puppet'
require 'puppet/reports'
require 'puppet/transaction/report'
require 'puppettest'
require 'puppettest/reporttesting'
class TestReports < Test::Unit::TestCase
include PuppetTest
include PuppetTest::Reporttesting
def mkreport
# First do some work
objects = []
6.times do |i|
file = tempfile()
# Make every third file
File.open(file, "w") { |f| f.puts "" } if i % 3 == 0
objects << Puppet::Type.type(:file).new(
:path => file,
:ensure => "file"
)
end
config = mk_catalog(*objects)
# So the report works out.
config.retrieval_duration = 0.001
trans = config.apply
report = Puppet::Transaction::Report.new
trans.add_metrics_to_report(report)
return report
end
# Make sure we can use reports as log destinations.
def test_reports_as_log_destinations
report = fakereport
assert_nothing_raised {
Puppet::Util::Log.newdestination(report)
}
# Now make a file for testing logging
file = Puppet::Type.type(:file).new(:path => tempfile(), :ensure => "file")
file.finish
log = nil
assert_nothing_raised {
log = file.log "This is a message, yo"
}
assert(report.logs.include?(log), "Report did not get log message")
assert_nothing_raised {
Puppet::Util::Log.close(report)
}
log = file.log "This is another message, yo"
assert(! report.logs.include?(log), "Report got log message after close")
end
def test_store_report
# Create a bunch of log messages in an array.
report = Puppet::Transaction::Report.new
# We have to reuse reporting here because of something going on in the
# server/report.rb file
Puppet.settings.use(:main, :master)
3.times { |i|
log = Puppet.warning("Report test message %s" % i)
report << log
}
assert_nothing_raised do
report.extend(Puppet::Reports.report(:store))
end
yaml = YAML.dump(report)
file = report.process
assert(FileTest.exists?(file), "report file did not get created")
assert_equal(yaml, File.read(file), "File did not get written")
end
if Puppet.features.rrd?
def test_rrdgraph_report
Puppet.settings.use(:main, :metrics)
report = mkreport
assert(! report.metrics.empty?, "Did not receive any metrics")
assert_nothing_raised do
report.extend(Puppet::Reports.report(:rrdgraph))
end
assert_nothing_raised {
report.process
}
hostdir = nil
assert_nothing_raised do
hostdir = report.hostdir
end
assert(hostdir, "Did not get hostdir back")
assert(FileTest.directory?(hostdir), "Host rrd dir did not get created")
index = File.join(hostdir, "index.html")
assert(FileTest.exists?(index), "index file was not created")
# Now make sure it creaets each of the rrd files
%w{changes resources time}.each do |type|
file = File.join(hostdir, "%s.rrd" % type)
assert(FileTest.exists?(file), "Did not create rrd file for %s" % type)
daily = file.sub ".rrd", "-daily.png"
assert(FileTest.exists?(daily),
"Did not make daily graph for %s" % type)
end
end
else
$stderr.puts "Install RRD for metric reporting tests"
end
end
|