Skip to content
Permalink
Browse files

Add serverspec test Sensu plugin

  • Loading branch information
tayzlor committed Jun 16, 2014
1 parent c55bb3d commit 2f4c665be2c56983d4df85981f5e89f389d0083f
Showing with 109 additions and 0 deletions.
  1. +109 −0 plugins/serverspec/check-serverspec.rb
@@ -0,0 +1,109 @@
#!/usr/bin/env ruby
#
# Check Serverspec tests plugin
# ===
#
# Runs http://serverspec.org/ spec tests against your servers.
# Fails with a critical if tests are failing.
#
# Examples:
#
# # Run entire suite of testd
# check-serverspec -d /etc/my_tests_dir
#
# # Run only one set of tests
# check-serverspec -d /etc/my_tests_dir -t spec/test_one

require 'rubygems' if RUBY_VERSION < '1.9.0'
require 'json'
require 'socket'
require 'serverspec'
require 'sensu-plugin/check/cli'

class CheckServerspec < Sensu::Plugin::Check::CLI

option :tests_dir,
:short => '-d /tmp/dir',
:long => '--tests-dir /tmp/dir',
:required => true

option :spec_tests,
:short => '-t spec/test',
:long => '--spec-tests spec/test',
:default => nil

option :handler,
:short => '-l HANDLER',
:long => '--handler HANDLER',
:default => 'default'

def sensu_client_socket(msg)
u = UDPSocket.new
u.send(msg + "\n", 0, '127.0.0.1', 3030)
end

def send_ok(check_name, msg)
d = { 'name' => check_name, 'status' => 0, 'output' => 'OK: ' + msg, 'handler' => config[:handler] }
sensu_client_socket d.to_json
end

def send_warning(check_name, msg)
d = { 'name' => check_name, 'status' => 1, 'output' => 'WARNING: ' + msg, 'handler' => config[:handler] }
sensu_client_socket d.to_json
end

def send_critical(check_name, msg)
d = { 'name' => check_name, 'status' => 2, 'output' => 'CRITICAL: ' + msg, 'handler' => config[:handler] }
sensu_client_socket d.to_json
end

def run
serverspec_results = `cd #{config[:tests_dir]} ; ruby -S rspec #{config[:spec_tests]} --format json`
parsed = JSON.parse(serverspec_results)

parsed["examples"].each do |serverspec_test|
test_name = serverspec_test["file_path"].split('/')[-1] + "_" + serverspec_test["line_number"].to_s
output = serverspec_test["full_description"].gsub!(/\"/, '')

if serverspec_test["status"] == "passed"
send_ok(
test_name
output
)
else
send_warning(
test_name
output
)
end

end

puts parsed["summary_line"]
failures = parsed["summary_line"].split[2]
if failures != '0'
exit_with(
:critical,
parsed["summary_line"]
)
else
exit_with(
:ok,
parsed["summary_line"]
)
end

end

def exit_with(sym, message)
case sym
when :ok
ok message
when :critical
critical message
else
unknown message
end
end

end

0 comments on commit 2f4c665

Please sign in to comment.
You can’t perform that action at this time.