sensu-extensions 0.0.2 → 0.0.4

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: e9b7abb0a8e576a835a2162c0b1e6f070e5c340a
-  data.tar.gz: 46547b38a70a2182b7894a3d27042b872075942f
+  metadata.gz: a70c9420e9ef4c2dab2a6cf6779ee760e0c5a0ef
+  data.tar.gz: 0321f2582670a20f127d149b769c1f8e7397a2c9
 SHA512:
-  metadata.gz: 11bd8e02a097a891b41f42c5f85296fbc6d71188099e0140b8bdde1a843b257de8b46c9b5ef11446da7a7acb67e4a54927e13f4bfd67462615e70e356e8b839e
-  data.tar.gz: 0e4f8e91af10254dea3fa887f82d5f0d56017d09aef5284bef0529ae34082aa47c3489e5c4275c1ac423ef793a11580aaa9488697f3e3bca8f7470b8eea65432
+  metadata.gz: 7cbdd74714cf3278ba63688c574e4b5de7797a402ff7b178dc68c9e303a9c2253041573f8a79a857be10e1fa5c86a62fe4316c6dceac26b4894e93cadd197de6
+  data.tar.gz: d0799591f7564b4059604201ad14807c78c7370e7e319d262e4034663ff3a82fb50d91adeec8046b9efe049baa9b658b68462f4ccafc74fc588e5a87cd7d0c1d

data/lib/sensu/extensions/loader.rb

@@ -1,4 +1,6 @@
 require "sensu/extension"
+require "sensu/extensions/mutators/json"
+require "sensu/extensions/mutators/ruby_hash"
 require "sensu/extensions/mutators/only_check_output"
 require "sensu/extensions/handlers/debug"
 
@@ -84,8 +86,11 @@ module Sensu
       end
 
       # Load instances of the loaded extensions.
-      def load_instances
-        Extension::CATEGORIES.each do |category|
+      #
+      # @param [String] sensu service to load extensions for.
+      def load_instances(service=nil)
+        service ||= sensu_service_name
+        categories_to_load(service).each do |category|
           extension_type = category.to_s.chop
           Extension.const_get(extension_type.capitalize).descendants.each do |klass|
             extension = klass.new
@@ -120,6 +125,29 @@ module Sensu
         @extensions[category].has_key?(name)
       end
 
+      # Retrieve Sensu service name.
+      #
+      # @return [String] service name.
+      def sensu_service_name
+        File.basename($0).split("-").last
+      end
+
+      # Determine which extension categories to load for the given
+      # sensu service.
+      #
+      # @param [String] sensu service to load extensions for.
+      # @return [Array] extension categories.
+      def categories_to_load(service)
+        case service
+        when "client"
+          [:checks]
+        when "server"
+          Extension::CATEGORIES.reject { |category| category == :checks }
+        else
+          Extension::CATEGORIES
+        end
+      end
+
       # Record a warning.
       #
       # @param message [String] warning message.

data/lib/sensu/extensions/mutators/json.rb

@@ -0,0 +1,20 @@
+require "sensu/extension"
+require "multi_json"
+
+module Sensu
+  module Extension
+    class Json < Mutator
+      def name
+        "json"
+      end
+
+      def description
+        "returns JSON formatted event data"
+      end
+
+      def run(event)
+        yield MultiJson.dump(event), 0
+      end
+    end
+  end
+end

data/lib/sensu/extensions/mutators/ruby_hash.rb

@@ -0,0 +1,19 @@
+require "sensu/extension"
+
+module Sensu
+  module Extension
+    class RubyHash < Mutator
+      def name
+        "ruby_hash"
+      end
+
+      def description
+        "returns ruby hash event data"
+      end
+
+      def run(event)
+        yield event, 0
+      end
+    end
+  end
+end

data/sensu-extensions.gemspec

@@ -2,7 +2,7 @@
 
 Gem::Specification.new do |spec|
   spec.name          = "sensu-extensions"
-  spec.version       = "0.0.2"
+  spec.version       = "0.0.4"
   spec.authors       = ["Sean Porter"]
   spec.email         = ["portertech@gmail.com"]
   spec.summary       = "The Sensu extension loader library"
@@ -17,6 +17,7 @@ Gem::Specification.new do |spec|
 
   spec.add_dependency "sensu-em"
   spec.add_dependency "sensu-extension"
+  spec.add_dependency "multi_json"
 
   spec.add_development_dependency "bundler", "~> 1.6"
   spec.add_development_dependency "rake"

data/spec/assets/extensions/error.rb

@@ -9,8 +9,8 @@ module Sensu
         "error extension"
       end
 
-      def run(event, &callback)
-        block.call(event, 0)
+      def run(event)
+        yield event, 0
       end
     end
   end

data/spec/assets/extensions/mock_check.rb

@@ -0,0 +1,17 @@
+module Sensu
+  module Extension
+    class MockCheck < Check
+      def name
+        "mock_check"
+      end
+
+      def description
+        "returns 'foo'"
+      end
+
+      def run
+        yield 'foo', 0
+      end
+    end
+  end
+end

data/spec/assets/extensions/test.rb

@@ -9,8 +9,8 @@ module Sensu
         "test extension"
       end
 
-      def run(event, &callback)
-        block.call(event, 0)
+      def run(event)
+        yield event, 0
       end
     end
   end

data/spec/extensions_spec.rb

@@ -28,18 +28,22 @@ describe "Sensu::Extensions" do
 
   it "can load extensions from a directory" do
     extensions = Sensu::Extensions.load(:extension_dir => @extension_dir)
-    extensions.loaded_files.size.should eq(1)
+    extensions.loaded_files.size.should eq(2)
     extensions.handler_exists?("test").should be_true
+    extensions.check_exists?("mock_check").should be_true
   end
 
   it "can load extensions from one or more directories" do
     extensions = Sensu::Extensions.load(:extension_dirs => [@extension_dir])
-    extensions.loaded_files.size.should eq(1)
+    extensions.loaded_files.size.should eq(2)
     extensions.handler_exists?("test").should be_true
+    extensions.check_exists?("mock_check").should be_true
   end
 
   it "can load the built-in extensions" do
     extensions = Sensu::Extensions.load
+    extensions.mutator_exists?("json").should be_true
+    extensions.mutator_exists?("ruby_hash").should be_true
     extensions.mutator_exists?("only_check_output").should be_true
     extensions.handler_exists?("debug").should be_true
   end

data/spec/loader_spec.rb

@@ -14,7 +14,7 @@ describe "Sensu::Extensions::Loader" do
   it "can provide the extensions loader API" do
     @loader.should respond_to(:load_file, :load_directory, :load_instances,
                               :warnings, :loaded_files, :[], :all,
-                              :generics, :bridges, :checks, :mutators, :handlers)
+                              :bridges, :checks, :mutators, :handlers)
   end
 
   it "can load an extension from a file" do
@@ -41,10 +41,12 @@ describe "Sensu::Extensions::Loader" do
 
   it "can load extensions from a directory" do
     @loader.load_directory(@extension_dir)
-    @loader.warnings.size.should eq(5)
-    @loader.loaded_files.size.should eq(1)
+    @loader.warnings.size.should eq(6)
+    @loader.loaded_files.size.should eq(2)
     extension = Sensu::Extension::Test.new
     extension.should respond_to(:name, :description, :definition, :safe_run, :stop, :has_key?, :[])
+    extension = Sensu::Extension::MockCheck.new
+    extension.should respond_to(:name, :description, :definition, :safe_run, :stop, :has_key?, :[])
   end
 
   it "can attempt to load extensions from a nonexistent directory" do
@@ -61,6 +63,8 @@ describe "Sensu::Extensions::Loader" do
     extension.should respond_to(:name, :description, :definition, :safe_run, :stop, :has_key?, :[])
     @loader.handlers.should include(extension.definition)
     @loader.all.should include(extension)
+    @loader.mutator_exists?("json").should be_true
+    @loader.mutator_exists?("ruby_hash").should be_true
     @loader.mutator_exists?("only_check_output").should be_true
   end
 
@@ -76,4 +80,18 @@ describe "Sensu::Extensions::Loader" do
     @loader.handler_exists?("debug").should be_true
     @loader.mutator_exists?("only_check_output").should be_true
   end
+
+  it "can load specific extension categories for a sensu client" do
+    @loader.load_instances("client")
+    @loader.handler_exists?("debug").should be_false
+    @loader.mutator_exists?("only_check_output").should be_false
+    @loader.check_exists?("mock_check").should be_true
+  end
+
+  it "can load specific extension categories for a sensu server" do
+    @loader.load_instances("server")
+    @loader.handler_exists?("debug").should be_true
+    @loader.mutator_exists?("only_check_output").should be_true
+    @loader.check_exists?("mock_check").should be_false
+  end
 end

data/spec/mutators/json_spec.rb

@@ -0,0 +1,24 @@
+require File.join(File.dirname(__FILE__), "..", "helpers")
+require "sensu/extensions/mutators/json"
+
+describe "Sensu::Extension::Json" do
+  include Helpers
+
+  before do
+    @extension = Sensu::Extension::Json.new
+  end
+
+  it "can run, returning JSON event data" do
+    event = {
+      :client => {},
+      :check => {
+        :output => "foo",
+        :status => 0
+      }
+    }
+    @extension.safe_run(event) do |output, status|
+      output.should eq(MultiJson.dump(event))
+      status.should eq(0)
+    end
+  end
+end

data/spec/mutators/ruby_hash_spec.rb

@@ -0,0 +1,24 @@
+require File.join(File.dirname(__FILE__), "..", "helpers")
+require "sensu/extensions/mutators/ruby_hash"
+
+describe "Sensu::Extension::RubyHash" do
+  include Helpers
+
+  before do
+    @extension = Sensu::Extension::RubyHash.new
+  end
+
+  it "can run, returning event data as a ruby hash" do
+    event = {
+      :client => {},
+      :check => {
+        :output => "foo",
+        :status => 0
+      }
+    }
+    @extension.safe_run(event) do |output, status|
+      output.should eq(event)
+      status.should eq(0)
+    end
+  end
+end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: sensu-extensions
 version: !ruby/object:Gem::Version
-  version: 0.0.2
+  version: 0.0.4
 platform: ruby
 authors:
 - Sean Porter
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2014-05-23 00:00:00.000000000 Z
+date: 2014-05-25 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: sensu-em
@@ -38,6 +38,20 @@ dependencies:
     - - ">="
       - !ruby/object:Gem::Version
         version: '0'
+- !ruby/object:Gem::Dependency
+  name: multi_json
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
 - !ruby/object:Gem::Dependency
   name: bundler
   requirement: !ruby/object:Gem::Requirement
@@ -110,15 +124,20 @@ files:
 - lib/sensu/extensions.rb
 - lib/sensu/extensions/handlers/debug.rb
 - lib/sensu/extensions/loader.rb
+- lib/sensu/extensions/mutators/json.rb
 - lib/sensu/extensions/mutators/only_check_output.rb
+- lib/sensu/extensions/mutators/ruby_hash.rb
 - sensu-extensions.gemspec
 - spec/assets/extensions/error.rb
+- spec/assets/extensions/mock_check.rb
 - spec/assets/extensions/test.rb
 - spec/extensions_spec.rb
 - spec/handlers/debug_spec.rb
 - spec/helpers.rb
 - spec/loader_spec.rb
+- spec/mutators/json_spec.rb
 - spec/mutators/only_check_output_spec.rb
+- spec/mutators/ruby_hash_spec.rb
 homepage: https://github.com/sensu/sensu-extensions
 licenses:
 - MIT
@@ -145,9 +164,12 @@ specification_version: 4
 summary: The Sensu extension loader library
 test_files:
 - spec/assets/extensions/error.rb
+- spec/assets/extensions/mock_check.rb
 - spec/assets/extensions/test.rb
 - spec/extensions_spec.rb
 - spec/handlers/debug_spec.rb
 - spec/helpers.rb
 - spec/loader_spec.rb
+- spec/mutators/json_spec.rb
 - spec/mutators/only_check_output_spec.rb
+- spec/mutators/ruby_hash_spec.rb