#!/usr/bin/env ruby
require File.dirname(__FILE__) + '/../lib/puppettest'
require 'puppet'
require 'puppet/util/metric'
require 'puppettest'
require 'puppet/type'
class TestMetric < PuppetTest::TestCase
confine "Missing RRDtool library" => Puppet.features.rrd?
include PuppetTest
def gendata
totalmax = 1000
changemax = 1000
eventmax = 10
maxdiff = 10
types = [Puppet::Type.type(:file), Puppet::Type.type(:package), Puppet::Type.type(:package)]
data = [:total, :managed, :outofsync, :changed, :totalchanges]
events = [:file_changed, :package_installed, :service_started]
# if this is the first set of data points...
typedata = Hash.new { |typehash,type|
typehash[type] = Hash.new(0)
}
eventdata = Hash.new(0)
typedata = {}
typedata[:total] = rand(totalmax)
typedata[:managed] = rand(typedata[:total])
typedata[:outofsync] = rand(typedata[:managed])
typedata[:changed] = rand(typedata[:outofsync])
typedata[:totalchanges] = rand(changemax)
events.each { |event|
eventdata[event] = rand(eventmax)
}
{:typedata => typedata, :eventdata => eventdata}
end
def rundata(report, time)
assert_nothing_raised {
gendata.each do |name, data|
report.newmetric(name, data)
end
report.metrics.each { |n, m| m.store(time) }
}
end
def test_fakedata
report = Puppet::Transaction::Report.new
time = Time.now.to_i
start = time
10.times {
rundata(report, time)
time += 300
}
rundata(report, time)
report.metrics.each do |n, m| m.graph end
File.open(File.join(Puppet[:rrddir],"index.html"),"w") { |of|
of.puts ""
report.metrics.each { |name, metric|
of.puts "
"
}
}
end
end