sensu-plugin 0.0.3 → 0.0.5

data/lib/sensu-handler.rb

@@ -1,29 +1,22 @@
+require 'net/http'
 require 'json'
-require 'sensu/config'
 
 module Sensu
   class Handler
 
     # Implementing classes should override this.
 
-    def handle(event)
+    def handle
       puts 'ignoring event -- no handler defined'
     end
 
-    # Overriding filtering logic is optional. Returns truthy if the
-    # event should be handled and falsy if it should not.
+    # Filters exit the proccess if the event should not be handled.
+    # Implementation of the default filters is below.
 
-    def filter(event)
-      if event['check']['alert'] == false
-        puts 'alert disabled -- filtered event ' + short_name(event)
-        exit 0
-      end
-      refresh = (60.fdiv(event['check']['interval']) * 30).to_i
-      event['occurrences'] == 1 || event['occurrences'] % refresh == 0
-    end
-
-    def short_name(event)
-      event['client']['name'] + '/' + event['check']['name']
+    def filter
+      filter_disabled
+      filter_repeated
+      filter_silenced
     end
 
     # This works just like Plugin::CLI's autorun.
@@ -50,11 +43,68 @@ module Sensu
       @settings ||= CONFIGS.map {|f| load_config(f) }.reduce {|a, b| a.deep_merge(b) }
     end
 
+    def read_event(file)
+      begin
+        @event = ::JSON.parse(file.read)
+        @event['occurrences'] ||= 1
+        @event['check'] ||= Hash.new
+        @event['client'] ||= Hash.new
+      rescue
+        puts 'Error reading event'
+        exit 0
+      end
+    end
+
     at_exit do
       handler = @@autorun.new
-      event = ::JSON.parse(STDIN.read)
-      if handler.filter(event)
-        handler.handle(event)
+      handler.read_event(STDIN)
+      handler.filter
+      handler.handle
+    end
+
+    # Helpers and filters
+
+    def bail(msg)
+      puts msg + ': ' + @event['client']['name'] + '/' + @event['check']['name']
+      exit 0
+    end
+
+    def api_request(*path)
+      http = Net::HTTP.new(settings['api']['host'], settings['api']['port'])
+      http.request(Net::HTTP::Get.new(path.join('/')))
+    end
+
+    def filter_disabled
+      if @event['check']['alert'] == false
+        bail 'alert disabled'
+      end
+    end
+
+    def filter_repeated
+      occurrences = @event['check']['occurrences'] || 1
+      interval = @event['check']['interval'] || 30
+      refresh = @event['check']['refresh'] || 1800
+      if @event['occurrences'] < occurrences
+        bail 'not enough occurrences'
+      end
+      if @event['occurrences'] > occurrences
+        n = refresh.fdiv(interval).to_i
+        bail 'only repeating alert every' + n + 'occurrences' unless @event['occurrences'] % n == 0
+      end
+    end
+
+    def filter_silenced
+      begin
+        timeout(3) do
+          if api_request('/stash/silence', @event['client']['name']).code == '200'
+            bail 'client alerts silenced'
+          end
+          if api_request('/stash/silence', @event['client']['name'], @event['check']['name']).code == '200'
+            bail 'check alerts silenced'
+          end
+        end
+      rescue Timeout::Error
+        puts 'Timed out while attempting to query the Sensu API for stashes'
       end
     end
 

data/lib/sensu-plugin.rb

@@ -1,6 +1,6 @@
 module Sensu
   module Plugin
-    VERSION = "0.0.3"
+    VERSION = "0.0.5"
     EXIT_CODES = {
       'OK' => 0,
       'WARNING' => 1,

data/test/{check.rb → external_check_test.rb}

@@ -3,34 +3,40 @@
 require 'rubygems'
 require 'minitest/autorun'
 
-class TestCheck < MiniTest::Unit::TestCase
+class TestCheckExternal < MiniTest::Unit::TestCase
 
   def setup
-    @script = File.join(File.dirname(__FILE__), 'checks/check-options')
+    @script = File.join(File.dirname(__FILE__), 'external/check-options')
+  end
+
+  def run_script(*args)
+    IO.popen([@script] + args, 'r+') do |child|
+      child.read
+    end
   end
 
   def test_ok
-    system @script, '-o'
+    run_script '-o'
     assert $?.exitstatus == 0
   end
 
   def test_warning
-    system @script, '-w'
+    run_script '-w'
     assert $?.exitstatus == 1
   end
 
   def test_critical
-    system @script, '-c'
+    run_script '-c'
     assert $?.exitstatus == 2
   end
 
   def test_unknown
-    system @script, '-u'
+    run_script '-u'
     assert $?.exitstatus == 3
   end
 
   def test_fallthrough
-    system @script
+    run_script
     assert $?.exitstatus == 1
   end
 

data/test/external_handler_test.rb

@@ -0,0 +1,37 @@
+#!/usr/bin/env ruby
+
+require 'rubygems'
+require 'minitest/autorun'
+require 'json'
+
+class TestHandlerExternal < MiniTest::Unit::TestCase
+
+  def setup
+    @script = File.join(File.dirname(__FILE__), 'external/handle-nofilter')
+  end
+
+  def run_script(event)
+    output = IO.popen(@script, 'r+') do |child|
+      child.puts JSON.generate(event)
+      child.close_write
+      child.read
+    end
+  end
+
+  def test_handled
+    event = {
+      'client' => { 'name' => 'test' },
+      'check' => { 'name' => 'test' },
+      'occurrences' => 1,
+    }
+    output = run_script(event)
+    assert $?.exitstatus == 0 && output =~ /Event:.*test/
+  end
+
+  def test_missing_keys
+    event = {}
+    output = run_script(event)
+    assert $?.exitstatus == 0 && output =~ /Event:/
+  end
+
+end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: sensu-plugin
 version: !ruby/object:Gem::Version
-  version: 0.0.3
+  version: 0.0.5
   prerelease: 
 platform: ruby
 authors:
@@ -9,11 +9,11 @@ authors:
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2011-12-19 00:00:00.000000000 Z
+date: 2011-12-22 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: json
-  requirement: &70160334903060 !ruby/object:Gem::Requirement
+  requirement: &70106722973320 !ruby/object:Gem::Requirement
     none: false
     requirements:
     - - ! '>='
@@ -21,10 +21,10 @@ dependencies:
         version: '0'
   type: :runtime
   prerelease: false
-  version_requirements: *70160334903060
+  version_requirements: *70106722973320
 - !ruby/object:Gem::Dependency
   name: mixlib-cli
-  requirement: &70160334902480 !ruby/object:Gem::Requirement
+  requirement: &70106722972800 !ruby/object:Gem::Requirement
     none: false
     requirements:
     - - ! '>='
@@ -32,7 +32,7 @@ dependencies:
         version: 1.1.0
   type: :runtime
   prerelease: false
-  version_requirements: *70160334902480
+  version_requirements: *70106722972800
 description: Plugins and helper libraries for Sensu, a monitoring framework
 email:
 - decklin@red-bean.com
@@ -46,7 +46,8 @@ files:
 - lib/sensu-plugin/metric/cli.rb
 - lib/sensu-plugin/util/procs.rb
 - lib/sensu-plugin.rb
-- test/check.rb
+- test/external_check_test.rb
+- test/external_handler_test.rb
 homepage: https://github.com/sonian/sensu-plugin
 licenses:
 - MIT
@@ -73,4 +74,5 @@ signing_key:
 specification_version: 3
 summary: Sensu Plugins
 test_files:
-- test/check.rb
+- test/external_check_test.rb
+- test/external_handler_test.rb