monitoring_protocols 0.0.4

data/specs/factories.rb

@@ -0,0 +1,44 @@
+require 'factory_girl'
+
+FactoryGirl.define do
+  factory :collectd_data_common do
+    time { Time.now }
+    host "localhost"
+    plugin "memory"
+    plugin_instance nil
+    
+    type "memory"
+    type_instance "active"
+  end
+  
+  factory :collectd_data_point, :parent => :collectd_data_common, :class => "MonitoringProtocols::Collectd::NetworkMessage" do
+    values { [rand(200)] }
+    interval 10
+  end
+  
+  factory :collectd_notification, :parent => :collectd_data_common, :class => "MonitoringProtocols::Collectd::NetworkMessage" do
+    severity 1
+    message "notification message"
+  end
+  
+  
+  factory :data_common do
+    time { Time.now }
+    host "localhost"
+    app_name "system"
+    res_name "memory"
+    metric_name "active"
+    
+    initialize_with{ new(time: time) }
+  end
+  
+  factory :data_point, :parent => :data_common, :class => "MonitoringProtocols::DataPoint" do
+    value { rand(200) }
+  end
+  
+  factory :notification, :parent => :data_common, :class => "MonitoringProtocols::Notification" do
+    severity :error
+    message "notification message"
+  end
+  
+end

data/specs/spec_helper.rb

@@ -0,0 +1,15 @@
+require 'rubygems'
+require 'bundler/setup'
+
+require 'eetee'
+
+$LOAD_PATH.unshift( File.expand_path('../../lib' , __FILE__) )
+require 'monitoring_protocols'
+
+require 'eetee/ext/mocha'
+require 'eetee/ext/time'
+# require 'eetee/ext/em'
+
+require File.expand_path('../factories', __FILE__)
+
+# Thread.abort_on_exception = true

data/specs/unit/collectd/builder_spec.rb

@@ -0,0 +1,58 @@
+require File.expand_path('../../../spec_helper', __FILE__)
+
+describe 'Collectd Builder' do
+  before do
+    @builder = MonitoringProtocols::Collectd::Builder.new
+    
+    @builder.host = "localhost"
+    @builder.time = 1
+    @builder.interval = 10
+    @builder.plugin = "plugin"
+    @builder.plugin_instance = "plugin_instance"
+    @builder.type = "type"
+    @builder.type_instance = "type_instance"
+    @builder.add_value(:counter, 42)
+    @builder.add_value(:gauge, 5.24)
+  end
+  
+  it 'can generate a packet' do
+    expected = [
+        "\x00\x00\x00\x0elocalhost\x00",                    # host
+        "\x00\x01\x00\x0c\x00\x00\x00\x00\x00\x00\x00\x01", # time
+        "\x00\x07\x00\x0c\x00\x00\x00\x00\x00\x00\x00\x0a", # interval
+        "\x00\x02\x00\x0bplugin\x00",                       # plugin
+        "\x00\x03\x00\x14plugin_instance\x00",              # plugin_instance
+        "\x00\x04\x00\x09type\x00",                         # type
+        "\x00\x05\x00\x12type_instance\x00",                # type_instance
+        "\x00\x06\x00\x18\x00\x02", # value headers
+        "\00\01", # types
+        "\x00\x00\x00\x00\x00\x00\x00\x2a",  # value
+        "\xf6\x28\x5c\x8f\xc2\xf5\x14\x40"   # value2
+      ]
+    
+    if "".respond_to?(:encode)
+      expected = expected.map{|s| s.force_encoding('ASCII-8BIT') }
+    end
+    
+    data = @builder.build_packet
+    
+    data[0,14].should == expected[0]
+    data[14,12].should == expected[1]
+    data[26,12].should == expected[2]
+    data[38,11].should == expected[3]
+    data[49,20].should == expected[4]
+    data[69, 9].should == expected[5]
+    data[78,18].should == expected[6]
+        
+    data[96,6].should  == expected[7]
+    
+    # types
+    data[102,2].should == expected[8]
+    
+    # value1
+    data[104,8].should == expected[9]
+    # value2
+    data[112,8].should == expected[10]
+  end
+  
+end

data/specs/unit/collectd/msg_spec.rb

@@ -0,0 +1,93 @@
+require File.expand_path('../../../spec_helper', __FILE__)
+
+describe 'Collectd Network Message' do
+  
+  describe 'data point' do
+    before do
+      @time_obj = Time.now
+      @time = @time_obj.to_i
+      
+      @point = FactoryGirl.build(:collectd_data_point,
+          time: @time,
+          type: 'memory',
+          type_instance: 'active'
+        )
+      
+      @point.time.should == @time
+    end
+
+    should 'return formatted plugin' do
+      @point.plugin = "plugin"
+      @point.plugin_instance = nil
+
+      @point.plugin_display.should == "plugin"
+
+      @point.plugin_instance = "instance"
+      @point.plugin_display.should == "plugin/instance"
+    end
+
+    should 'return formatted type' do
+      @point.type = "type"
+      @point.type_instance = nil
+
+      @point.type_display.should == "type"
+
+      @point.type_instance = "instance"
+      @point.type_display.should == "type/instance"
+    end
+    
+    
+    should 'convert datapoint to DataPoint' do
+      d = @point.convert_content()
+      d.class.should == Array
+      d.size.should == 1
+      
+      d = d.first
+      d.class.should == MonitoringProtocols::DataPoint
+      
+      d.first.should != true
+      d.host.should == @point.host
+      d.time.gmt_offset.should == 0
+      d.time.iso8601().should == @time_obj.getutc().iso8601()
+      
+      d.app_name.should == @point.plugin
+      d.res_name.should == @point.type
+      d.metric_name.should == @point.type_instance
+      d.value.should == d.value
+    end
+    
+  end
+  
+  
+  describe 'notification' do
+    before do
+      @point = FactoryGirl.build(:collectd_notification,
+          type: 'memory',
+          type_instance: 'active',
+          
+          severity: 1,
+          message: 'ahhhhhhh'
+        )
+    end
+
+
+    should 'convert notification to Notification' do
+      d = @point.convert_content()
+      d.class.should == Array
+      d.size.should == 1
+      d = d.first
+      d.class.should == MonitoringProtocols::Notification
+      
+      d.host.should == @point.host
+      d.app_name.should == @point.plugin
+      d.res_name.should == @point.type
+      d.metric_name.should == @point.type_instance
+      
+      d.severity.should == :warn
+      d.message.should == "ahhhhhhh"
+    end
+    
+  end  
+  
+end
+

data/specs/unit/collectd/parser_spec.rb

@@ -0,0 +1,212 @@
+require File.expand_path('../../../spec_helper', __FILE__)
+
+class EEtee::Context
+  def builder
+    MonitoringProtocols::Collectd::Builder
+  end
+end
+
+describe 'Collectd Ruby parser' do
+  before do
+    @builder_class = MonitoringProtocols::Collectd::Builder
+    @packet = ->(cmd, *args){
+      @builder_class.send(cmd, *args)
+    }
+    
+    @parser_class = MonitoringProtocols::Collectd::Parser
+  end
+    
+  describe 'Simple packets' do
+    it 'can parse numbers' do
+      type, val, buffer = @parser_class.parse_part( builder.number(1, 122) )
+      buffer.should == ""
+      val.should == 122
+      
+      type, val, _ = @parser_class.parse_part( builder.number(1, 2500) )
+      val.should == 2500
+      
+      type, val, _ = @parser_class.parse_part( builder.number(1, 356798) )
+      val.should == 356798
+    end
+    
+    should 'parse strings' do
+      type, str, _ = @parser_class.parse_part( builder.string(0, "hostname1") )
+      str.should == 'hostname1'
+      
+      type, str, _ = @parser_class.parse_part( builder.string(0, "string with spaces") )
+      str.should == 'string with spaces'
+      
+      type, str, _ = @parser_class.parse_part( builder.string(0, "a really long string with many words in it") )
+      str.should == 'a really long string with many words in it'
+    end
+    
+    should 'parse values' do
+      buffer = builder.values(
+          [builder::COUNTER, builder::DERIVE, builder::DERIVE],
+          [1034, -4567, 34]
+        )
+        
+      type, values, rest = @parser_class.parse_part( buffer )
+      rest.should == ""
+      values.should == [1034, -4567, 34]
+    end
+  end
+  
+  describe 'One notification in buffer' do
+    before do
+      @now = Time.new.to_i
+      
+      @pkt = ""
+      @pkt << builder.number(1, @now)
+      @pkt << builder.string(0, 'hostname')
+         
+      @pkt << builder.string(2, 'plugin')
+      @pkt << builder.string(3, 'plugin_inst')
+      @pkt << builder.string(4, 'type')
+      @pkt << builder.string(5, 'type_inst')
+      @pkt << builder.number(257, 2) # severity
+      @pkt << builder.string(256, 'a message')
+    end
+    
+    should 'parse the notification' do
+      data, rest = @parser_class.parse_packet(@pkt)
+      
+      rest.should == ""
+      
+      data.class.should             == MonitoringProtocols::Collectd::NetworkMessage
+      data.host.should              == 'hostname'
+      data.time.should              == @now
+      data.plugin.should            == 'plugin'
+      data.plugin_instance.should   == 'plugin_inst'
+      data.type.should              == 'type'
+      data.type_instance.should     == 'type_inst'
+      data.message.should           == 'a message'
+      data.severity.should          == 2
+      
+    end
+  end
+  
+  describe 'One packet in buffer' do
+    before do
+      @now = Time.new.to_i
+      @interval = 10
+      
+      pkt = builder.new
+      pkt.time = @now
+      pkt.host = 'hostname'
+      pkt.interval = @interval
+      pkt.plugin = 'plugin'
+      pkt.plugin_instance = 'plugin_inst'
+      pkt.type = 'type'
+      pkt.type_instance = 'type_inst'
+      
+      pkt.add_value(:counter, 1034)
+      pkt.add_value(:gauge, 3.45)
+               
+      @pkt = pkt.build_packet
+    end
+    
+    should 'parse buffer' do
+      data, rest = @parser_class.parse_packet(@pkt)
+      
+      data.class.should            == MonitoringProtocols::Collectd::NetworkMessage
+      data.host.should             == 'hostname'
+      data.time.should             == @now
+      data.interval.should         == @interval
+      data.plugin.should           == 'plugin'
+      data.plugin_instance.should  == 'plugin_inst'
+      data.type.should             == 'type'
+      data.type_instance.should    == 'type_inst'
+      
+      data.values.size.should      == 2
+      data.values[0].should        == 1034
+      data.values[1].should        == 3.45
+      
+      rest.should == ""
+    end
+  end
+  
+  describe "Multiple packets in buffer" do
+     before do
+       @now = Time.new.to_i
+       @interval = 10
+   
+       pkt = builder.new
+       pkt.time = @now
+       pkt.host = 'hostname'
+       pkt.interval = @interval
+       pkt.plugin = 'plugin'
+       pkt.plugin_instance = 'plugin_inst'
+       pkt.type = 'type'
+       pkt.type_instance = 'type_inst'
+       
+       pkt.add_value(:counter, 1034)
+       pkt.add_value(:gauge, 3.45)
+                
+       @pkt = pkt.build_packet
+       
+       @pkt << builder.string(2, 'plugin2')
+       @pkt << builder.string(3, 'plugin2_inst')
+       @pkt << builder.string(4, 'type2')
+       @pkt << builder.string(5, 'type2_inst')
+       
+       @pkt << builder.values([builder::COUNTER], [42])
+  
+       
+       @pkt << builder.string(5, 'type21_inst')
+       @pkt << builder.values([builder::GAUGE], [3.1415927])
+     end
+     
+     should 'parse buffer' do
+       data = @parser_class.parse(@pkt)
+       
+       data.size.should == 3
+       
+       data[0].class.should == MonitoringProtocols::Collectd::NetworkMessage
+       
+       data[0].host.should              == 'hostname'
+       data[0].time.should              == @now
+       data[0].interval.should          == @interval
+       data[0].plugin.should            == 'plugin'
+       data[0].plugin_instance.should   == 'plugin_inst'
+       data[0].type.should              == 'type'
+       data[0].type_instance.should     == 'type_inst'
+       data[0].values.size.should       == 2
+       data[0].values[0].should         == 1034
+       data[0].values[1].should         == 3.45
+       
+       data[1].host.should              == 'hostname'
+       data[1].time.should              == @now
+       data[1].interval.should          == @interval
+       data[1].plugin.should            == 'plugin2'
+       data[1].plugin_instance.should   == 'plugin2_inst'
+       data[1].type.should              == 'type2'
+       data[1].type_instance.should     == 'type2_inst'
+       data[1].values.size.should       == 1
+       data[1].values[0].should         == 42
+       
+       data[2].host.should              == 'hostname'
+       data[2].time.should              == @now
+       data[2].interval.should          == @interval
+       data[2].plugin.should            == 'plugin2'
+       data[2].plugin_instance.should   == 'plugin2_inst'
+       data[2].type.should              == 'type2'
+       data[2].type_instance.should     == 'type21_inst'
+       data[2].values.size.should       == 1
+       data[2].values[0].should         == 3.1415927
+       
+     end
+     
+     # should 'parse using feed interface' do
+     #  parser = @parser.new
+      
+     #  ret = parser.feed(@pkt[0,20])
+     #  ret.size.should == []
+      
+     #  ret = parser.feed(@pkt[21..-1])
+     #  ret.size.should == 2
+     # end
+     
+   end
+  
+end

data/specs/unit/core_spec.rb

@@ -0,0 +1,19 @@
+require File.expand_path('../../spec_helper', __FILE__)
+
+describe 'Core' do
+  should 'return parser for collectd' do
+    MonitoringProtocols.get_parser(:collectd).class.should ==
+      MonitoringProtocols::Collectd::Parser
+  end
+  
+  should 'return builder for collectd' do
+    MonitoringProtocols.get_builder(:collectd).class.should ==
+      MonitoringProtocols::Collectd::Builder
+  end
+
+  
+  should 'return nil for unknown protocol' do
+    MonitoringProtocols.get_parser(:i_am_invalid).should == nil
+  end
+    
+end