gri 10.0.0

data/lib/gri/writer.rb

@@ -0,0 +1,184 @@
+require 'gri/ltsv'
+
+module GRI
+  class Writer
+    TYPES = {}
+
+    def self.create wtype, options
+      if (klass = TYPES[wtype])
+        options.select {|k, v| String === k}.each {|k, v|
+          options[k.gsub('-', '_').intern] = v}
+        klass.new options
+      end
+    end
+
+    attr_accessor :loop
+
+    def initialize options={}
+    end
+  end
+
+  class STDOUTWriter < Writer
+    TYPES['stdout'] = self
+
+    def initialize options={}
+      @options = options
+    end
+    def write records
+    end
+  end
+
+  class RRDWriter < Writer
+    TYPES['rrd'] = self
+
+    def initialize options={}
+      @options = options
+      @gra_dir = options[:gra_dir]
+      @sysinfos = {}
+    end
+
+    def write records
+      now = Time.now.to_i
+      hrecords = {}
+      updaters = {}
+      for record in records
+        if (host = record['_host']) and (key = record['_key'])
+          if key == 'SYS'
+            @sysinfos[host] = record
+          else
+            data_name, index = key.scan(/\A([^_\d]*)(_?.*)/).first
+            basename = "#{host}_#{key}"
+            unless (updater = updaters[basename])
+              hdir = "#{@gra_dir}/#{host}"
+              FileUtils.mkdir_p hdir unless File.directory? hdir
+              opts = {:data_name=>data_name, :gra_dir=>@gra_dir,
+                :host=>host, :key=>key,
+                :interval=>(record['_interval'] || @options[:interval])}
+              if @options[:rrdcached_address]
+                opts[:rrdcached_address] = @options[:rrdcached_address]
+              end
+              if @options[:base_dir]
+                opts[:base_dir] = @options[:base_dir]
+              end
+              updater = updaters[basename] = RRDUpdater.new(opts)
+            end
+            time = (record['_time'] || now).to_i
+            updater.update time, record
+            key = updater.key
+          end
+          (r = record.dup)['_key'] = key
+          r['_mtime'] = now
+          (hrecords[host] ||= {})[key] = r
+        end
+      end
+      hrecords.each {|host, h|
+        path = "#{@gra_dir}/#{host}/.records.txt"
+        FileUtils.mkdir_p File.dirname(path) unless File.exist? path
+        Utils.update_ltsv_file path, '_key', h
+      }
+      updaters.each {|k, u| u.close}
+    end
+
+    def finalize
+      unless @sysinfos.empty?
+        if @options[:merge_p]
+          path = "#{@gra_dir}/.sysdb/sysdb.tmp.#{$$}"
+          open(path, 'w') {|f| LTSV.dump_to_io @sysinfos, f}
+        else
+          path = "#{@gra_dir}/.sysdb/sysdb.txt"
+          FileUtils.mkdir_p File.dirname(path) unless File.exist? path
+          Utils.update_ltsv_file path, '_host', @sysinfos
+        end
+      end
+    end
+
+    def merge
+      sysdb_dir = "#{@gra_dir}/.sysdb"
+      path0 = sysdb_dir + '/sysdb.txt'
+      values = LTSV.load_from_file(path0) rescue {}
+      nvalues = values.inject({}) {|h, v| h[v['_host']] = v; h}
+      for path in Dir.glob(sysdb_dir + '/sysdb.tmp.*')
+        (other = LTSV.load_from_file(path); File.unlink path) rescue next
+        other = other.inject({}) {|h, v| h[v['_host']] = v; h}
+        nvalues.merge! other
+      end
+      LTSV.dump_to_file nvalues, path0
+    end
+  end
+
+  class TextWriter < Writer
+    TYPES['text'] = self
+    TYPES['ldb'] = self
+
+    def initialize options={}
+      @options = options
+      @dir = @options[:tra_dir]
+      Dir.mkdir @dir unless File.exist? @dir
+    end
+
+    def write records
+      now = Time.now
+      date = "%04d%02d%02d" % [now.year, now.mon, now.day]
+      time = now.to_i
+      @ios = {}
+      for record in records
+        #record['_time'] = time
+        if (host = record['_host']) and (key = record['_key'])
+          hdir = "#{@dir}/#{host}"
+          Dir.mkdir hdir unless File.exist? hdir
+          data_name, index = key.scan(/\A([^_\d]*)(_?.*)/).first
+          interval = record['_interval']
+          kdir = "#{hdir}/#{data_name}_#{interval}"
+          Dir.mkdir kdir unless File.exist? kdir
+          path = kdir + '/' + date
+          unless io = @ios[path]
+            io = @ios[path] = File.open(path, 'a')
+          end
+          io.puts LTSV.serialize(record)
+        end
+      end
+      @ios.each {|k, io| io.close}
+    end
+
+    def finalize
+    end
+
+    def purge_logs
+      if (day = @options[:tra_expire_day].to_i) > 0
+        self.class.expire @dir, day
+      end
+    end
+
+    def self.expire dir, day
+      run_expire_path = dir + '/.run_expire'
+      now = Time.now
+      if File.exist? run_expire_path
+        last_run = nil
+        open(run_expire_path) {|f| last_run = f.read}
+        last_run = Time.at(last_run.to_i)
+        return if (now - last_run < 24*3600) and !$debug
+      end
+      if (day = day.to_i) > 0
+        t = (now - day * 24 * 3600)
+        expire_date = "%04d%02d%02d" % [t.year, t.mon, t.day]
+        begin
+          Dir.glob(dir + '/*') {|hdir|
+            next if File.symlink? hdir
+            Dir.glob(hdir + '/*') {|path|
+              if File.directory? path
+                Dir.glob(path + '/*') {|path2|
+                  if (basename = File.basename path2) < expire_date
+                    File.unlink path2
+                  end
+                }
+              end
+            }
+          }
+          open(run_expire_path, 'w') {|f| f.print now.to_i.to_s}
+        rescue SystemCallError
+          Log.error "ERROR: #{$!}"
+        end
+      end
+    end
+  end
+end

data/mcollector

@@ -0,0 +1,47 @@
+# coding: us-ascii
+
+if Object.const_defined? :RUBY_VERSION
+  lib_dir = File.dirname(__FILE__) + '/lib'
+  $LOAD_PATH.push lib_dir
+  require 'rubygems'
+  require 'socket'
+else
+  # mruby
+  GC.disable #XXX
+  lib_dir = File.dirname(__FILE__) + '/lib'
+  $LOAD_PATH.push lib_dir
+end
+
+require 'gri/mlog'
+require 'gri/app_collector'
+require 'gri/writer'
+require 'gri/config'
+require 'gri/builtindefs'
+require 'gri/plugin'
+
+config_path = '/usr/local/gri/gri.conf'
+gritab_path = nil
+wtype = nil
+while arg = ARGV.shift
+  case arg
+  when /\A--debug/
+    $debug = true
+  when /\A--config-path(?:=(\w+))?/
+    config_path = ($1 || ARGV.shift)
+  when /\A--gritab-path(?:=(\w+))?/
+    gritab_path = ($1 || ARGV.shift)
+  when /\A--writer(?:=(\w+))?/
+    wtype = ($1 || ARGV.shift)
+  end
+end
+
+config = GRI::Config.init config_path
+config['gritab-path'] = gritab_path if gritab_path
+GRI::Plugin.load_plugins [], config
+config['root-dir'] ||= GRI::Config::ROOT_PATH
+config['tra-dir'] ||= '/usr/local/gri/tra'
+wtype ||= 'fluent'
+writer = GRI::Writer.create wtype, config.to_h
+app = GRI::AppCollector.new(config)
+app.writers.push writer
+app.run

data/test/mock.rb

@@ -0,0 +1,60 @@
+class MockLoop
+  def initialize
+    @handlers = {}
+    @rs = []
+  end
+
+  def attach collector
+    collector.attach self
+  end
+
+  def detach collector
+    collector.on_detach
+  end
+
+  def on_detach &on_detach
+    @on_detach = on_detach
+  end
+
+  def watch io, flag, to, handler
+    @handlers[io] = handler
+    @rs.push io
+  end
+
+  def run
+  end
+
+  def run_handler ind
+    io = @rs[ind]
+    @handlers[io].on_readable io
+  end
+end
+
+class MockIO
+  def close; end
+end
+
+module MockFork
+  def fork &cb
+    1
+  end
+end
+
+class MockUDPSocket
+  attr_accessor :data
+
+  def connect host, port
+    @host = host
+    @port = port
+  end
+
+  def send msg, flags
+    @msg = msg
+  end
+
+  def recv maxlen, flags
+    @data
+  end
+
+  def close; end
+end

data/test/root/gra/.sysdb/sysdb.txt

@@ -0,0 +1,3 @@
+_host:192.168.0.1	sysUpTime:2555041303	sysDescr:Linux host.example.com 2.6.18-164.6.1.el5 #1 SMP Tue Nov 3 16:18:27 EST 2009 i686
+_host:testhost	sysUpTime:999999	sysDescr:Linux host.example.com 2.6.18-164.6.1.el5 #1 SMP Tue Nov 3 16:18:27 EST 2009 i686
+_host:r00.example.com	sysDescr:IOS XR Software (Cisco CRS-8/S), Version 9.9.9[00]

data/test/root/gra/testhost/.records.txt

@@ -0,0 +1,2 @@
+_key:_eth0	ifDescr:eth0	ifAlias:testdescr	ifSpeed:1000000000
+_key:l1	laLoad:0.01	laNames:Load-1

data/test/root/if.def

@@ -0,0 +1,2 @@
+testhost__eth0		Test eth0 Description
+testhost__eth1		test_eth1

data/test/root/test.conf

@@ -0,0 +1,4 @@
+test-opt-num    1
+test-opt-str    foo
+plugin-dir	/dir1
+plugin-dir	/dir2

data/test/root/testtab

@@ -0,0 +1,9 @@
+# gritab for test
+localhost       ver=2c community=public no-interfaces
+host00          interval=3600
+host01          interval=60
+host02          interval=10
+host03          interval=12
+host04          interval=60
+host05          interval=10
+host06          interval=180

data/test/root/testtab2

@@ -0,0 +1,2 @@
+# gritab for test
+host00          hostname=host99

data/test/root/tra/testhost/_/20130614

@@ -0,0 +1,20 @@
+ifOutDiscards:0	ifInNUcastPkts:136951	_key:_eth0	ifInUcastPkts:116663644	ifHighSpeed:1000	ifInErrors:0	ifAdminStatus:1	ifSpeed:1000000000	ifHCInOctets:27494649894	_time:1371175021	ifOutNUcastPkts:0	ifInDiscards:0	ifOperStatus:1	ifHCOutOctets:12602582932	_mtime:1371175021	ifOutOctets:12602582932	ifInOctets:27494649894	ifOutErrors:0	ifAlias:	_host:testhost	ipaddr:198.51.100.208/255.255.255.0	ifOutUcastPkts:83951137	ifDescr:eth0
+ifOutOctets:12602738085	ifOutUcastPkts:83952397	ifInNUcastPkts:136956	_host:testhost	ifInDiscards:0	ifDescr:eth0	ifOutDiscards:0	ifHCInOctets:27494875934	ipaddr:198.51.100.208/255.255.255.0	ifAlias:	ifInUcastPkts:116665729	ifInErrors:0	_mtime:1371175321	ifOutNUcastPkts:0	ifOutErrors:0	ifOperStatus:1	ifAdminStatus:1	ifHCOutOctets:12602738085	_time:1371175322	_key:_eth0	ifInOctets:27494875934	ifSpeed:1000000000	ifHighSpeed:1000
+ifInNUcastPkts:136961	ifDescr:eth0	ipaddr:198.51.100.208/255.255.255.0	ifOutUcastPkts:83953757	ifInOctets:27495118285	ifOperStatus:1	ifHCInOctets:27495118285	_host:testhost	ifOutOctets:12602899810	ifHighSpeed:1000	ifAlias:	ifOutNUcastPkts:0	_key:_eth0	ifInUcastPkts:116667880	ifInErrors:0	ifInDiscards:0	_mtime:1371175621	ifOutDiscards:0	ifHCOutOctets:12602899810	_time:1371175621	ifOutErrors:0	ifAdminStatus:1	ifSpeed:1000000000
+ifOutDiscards:0	ifInOctets:27495328828	ifInUcastPkts:116669910	ifInDiscards:0	ifAdminStatus:1	ifHCInOctets:27495328828	_time:1371175921	ifDescr:eth0	_mtime:1371175921	ifOutOctets:12603046499	ifOutUcastPkts:83954969	ifOperStatus:1	ifSpeed:1000000000	ifOutNUcastPkts:0	ifInNUcastPkts:136966	ifAlias:	ifHighSpeed:1000	_key:_eth0	_host:testhost	ifOutErrors:0	ipaddr:198.51.100.208/255.255.255.0	ifInErrors:0	ifHCOutOctets:12603046499
+ifOutNUcastPkts:0	ifSpeed:1000000000	ifOutUcastPkts:83956472	ifInUcastPkts:116672186	ifOutErrors:0	_mtime:1371176222	_host:testhost	ifOperStatus:1	ifDescr:eth0	ifAlias:	ifInErrors:0	ifOutOctets:12603224788	_key:_eth0	ifInDiscards:0	ipaddr:198.51.100.208/255.255.255.0	ifHighSpeed:1000	ifHCInOctets:27495585909	ifInNUcastPkts:136971	ifOutDiscards:0	ifInOctets:27495585909	ifHCOutOctets:12603224788	_time:1371176222	ifAdminStatus:1
+_mtime:1371176521	_host:testhost	ifOutNUcastPkts:0	ifInUcastPkts:116674397	ifInErrors:0	ipaddr:198.51.100.208/255.255.255.0	ifOutOctets:12603382781	ifHCOutOctets:12603382781	ifAdminStatus:1	ifSpeed:1000000000	ifOutDiscards:0	ifInDiscards:0	_key:_eth0	_time:1371176521	ifInNUcastPkts:136976	ifOperStatus:1	ifDescr:eth0	ifOutErrors:0	ifInOctets:27495830850	ifOutUcastPkts:83957856	ifAlias:	ifHighSpeed:1000	ifHCInOctets:27495830850
+_key:_eth0	_host:testhost	ifHCInOctets:27496048401	ifOutDiscards:0	_mtime:1371176821	ipaddr:198.51.100.208/255.255.255.0	ifInErrors:0	ifOutOctets:12603528168	ifOutNUcastPkts:0	ifAlias:	ifInNUcastPkts:136981	ifOutUcastPkts:83959049	ifOperStatus:1	ifHighSpeed:1000	ifAdminStatus:1	ifDescr:eth0	_time:1371176821	ifInUcastPkts:116676373	ifInDiscards:0	ifSpeed:1000000000	ifHCOutOctets:12603528168	ifInOctets:27496048401	ifOutErrors:0
+ifOutErrors:0	_key:_eth0	ifOutUcastPkts:83960396	ifOperStatus:1	ifHighSpeed:1000	ifInUcastPkts:116678564	ifOutOctets:12603694511	ifAdminStatus:1	ifAlias:	ifInNUcastPkts:136985	ifInErrors:0	ifSpeed:1000000000	ifDescr:eth0	ipaddr:198.51.100.208/255.255.255.0	ifHCOutOctets:12603694511	_mtime:1371177121	ifOutNUcastPkts:0	ifHCInOctets:27496309820	_time:1371177121	_host:testhost	ifOutDiscards:0	ifInDiscards:0	ifInOctets:27496309820
+ifHighSpeed:1000	ifOutOctets:12603839868	_mtime:1371177421	ifHCInOctets:27496535042	ifHCOutOctets:12603839868	ifInErrors:0	ifInUcastPkts:116680537	ifOperStatus:1	ifSpeed:1000000000	_key:_eth0	ifInNUcastPkts:136990	ifInDiscards:0	ifDescr:eth0	_time:1371177421	ipaddr:198.51.100.208/255.255.255.0	ifOutDiscards:0	ifAlias:	ifOutErrors:0	ifInOctets:27496535042	ifOutUcastPkts:83961590	ifAdminStatus:1	ifOutNUcastPkts:0	_host:testhost
+ifHighSpeed:1000	ifHCInOctets:27496761294	ifOutNUcastPkts:0	ifOutUcastPkts:83962783	ifOutDiscards:0	_host:testhost	ifInDiscards:0	ifAdminStatus:1	ipaddr:198.51.100.208/255.255.255.0	ifInNUcastPkts:136996	ifInUcastPkts:116682517	ifAlias:	ifHCOutOctets:12603985303	ifOutErrors:0	_time:1371177721	_key:_eth0	ifInErrors:0	ifOperStatus:1	ifSpeed:1000000000	_mtime:1371177721	ifOutOctets:12603985303	ifInOctets:27496761294	ifDescr:eth0
+ifAdminStatus:1	_mtime:1371178022	ifInNUcastPkts:137000	ifInUcastPkts:116684520	_key:_eth0	ifOutNUcastPkts:0	ifInDiscards:0	_time:1371178022	ifAlias:	ifOutDiscards:0	ifSpeed:1000000000	ifHighSpeed:1000	ifHCInOctets:27496981613	ifOperStatus:1	ifDescr:eth0	ifHCOutOctets:12604130342	_host:testhost	ifOutUcastPkts:83963970	ifOutOctets:12604130342	ipaddr:198.51.100.208/255.255.255.0	ifInOctets:27496981613	ifOutErrors:0	ifInErrors:0
+ifOutErrors:0	ifOutOctets:12604275519	_time:1371178322	ifInOctets:27497194428	_key:_eth0	ifInUcastPkts:116686502	ifInDiscards:0	_mtime:1371178322	ifInNUcastPkts:137005	ifAdminStatus:1	ifAlias:	ifHighSpeed:1000	_host:testhost	ifOutUcastPkts:83965158	ifOutDiscards:0	ifHCOutOctets:12604275519	ifOutNUcastPkts:0	ifOperStatus:1	ifDescr:eth0	ifHCInOctets:27497194428	ipaddr:198.51.100.208/255.255.255.0	ifInErrors:0	ifSpeed:1000000000
+ifInOctets:27497416028	ifOutUcastPkts:83966348	ifOutDiscards:0	_host:testhost	ifInDiscards:0	ifAdminStatus:1	ifInErrors:0	ifOutOctets:12604420636	ifOperStatus:1	_mtime:1371178621	ifHCInOctets:27497416028	ifInUcastPkts:116688523	ifDescr:eth0	ifOutNUcastPkts:0	ifSpeed:1000000000	ipaddr:198.51.100.208/255.255.255.0	ifAlias:	_time:1371178621	_key:_eth0	ifOutErrors:0	ifHCOutOctets:12604420636	ifInNUcastPkts:137010	ifHighSpeed:1000
+ifOutErrors:0	ifInNUcastPkts:137016	ifOutUcastPkts:83967540	ifInErrors:0	ifSpeed:1000000000	ifHighSpeed:1000	_time:1371178922	_key:_eth0	ifDescr:eth0	ifOperStatus:1	ifInUcastPkts:116690529	ifAdminStatus:1	ifAlias:	_mtime:1371178922	_host:testhost	ifOutDiscards:0	ifOutOctets:12604565981	ifHCOutOctets:12604565981	ifHCInOctets:27497623151	ifInDiscards:0	ipaddr:198.51.100.208/255.255.255.0	ifOutNUcastPkts:0	ifInOctets:27497623151
+ipaddr:198.51.100.208/255.255.255.0	ifDescr:eth0	ifOutDiscards:0	ifInDiscards:0	ifInOctets:27497835613	ifHCInOctets:27497835613	_key:_eth0	_host:testhost	ifOutNUcastPkts:0	ifAdminStatus:1	ifOutErrors:0	ifHighSpeed:1000	ifInUcastPkts:116692506	ifInErrors:0	ifSpeed:1000000000	ifAlias:	ifHCOutOctets:12604711242	ifOperStatus:1	_time:1371179221	_mtime:1371179221	ifOutUcastPkts:83968730	ifOutOctets:12604711242	ifInNUcastPkts:137019
+_host:testhost	ifDescr:eth0	ifOutOctets:12604856191	ifAlias:	ipaddr:198.51.100.208/255.255.255.0	ifOutNUcastPkts:0	ifOperStatus:1	ifHCInOctets:27498054760	ifInErrors:0	ifAdminStatus:1	_time:1371179522	ifInUcastPkts:116694492	ifInOctets:27498054760	ifHCOutOctets:12604856191	_mtime:1371179522	ifInNUcastPkts:137025	ifInDiscards:0	ifSpeed:1000000000	ifOutDiscards:0	_key:_eth0	ifOutUcastPkts:83969916	ifOutErrors:0	ifHighSpeed:1000
+_mtime:1371179822	ifAdminStatus:1	ifHCOutOctets:12605008314	ifOutNUcastPkts:0	ifOutOctets:12605008314	_host:testhost	ifInDiscards:0	ifOperStatus:1	ifInUcastPkts:116696540	ifAlias:	_time:1371179822	ipaddr:198.51.100.208/255.255.255.0	ifHCInOctets:27498292406	ifInErrors:0	ifOutUcastPkts:83971184	ifOutErrors:0	ifInOctets:27498292406	ifSpeed:1000000000	ifDescr:eth0	ifHighSpeed:1000	_key:_eth0	ifInNUcastPkts:137030	ifOutDiscards:0
+_host:testhost	ifAlias:	ifInErrors:0	ifAdminStatus:1	ifHCOutOctets:12605153527	_mtime:1371180121	ifOutDiscards:0	ifSpeed:1000000000	ifHighSpeed:1000	ifOutNUcastPkts:0	ifOutUcastPkts:83972374	ifInOctets:27498512600	ifOperStatus:1	ifDescr:eth0	ipaddr:198.51.100.208/255.255.255.0	ifInNUcastPkts:137036	ifOutOctets:12605153527	_time:1371180121	ifHCInOctets:27498512600	ifInDiscards:0	ifOutErrors:0	_key:_eth0	ifInUcastPkts:116698540
+ifOutDiscards:0	_host:testhost	ipaddr:198.51.100.208/255.255.255.0	ifHCInOctets:27498717329	ifInNUcastPkts:137041	ifInOctets:27498717329	ifOutOctets:12605298644	_mtime:1371180421	ifHighSpeed:1000	ifInErrors:0	ifInDiscards:0	ifDescr:eth0	_time:1371180421	ifOutNUcastPkts:0	ifOutUcastPkts:83973564	ifOutErrors:0	ifAdminStatus:1	ifAlias:	ifHCOutOctets:12605298644	ifInUcastPkts:116700513	ifOperStatus:1	ifSpeed:1000000000	_key:_eth0
+ifHCOutOctets:12605443941	ifHCInOctets:27498895868	_host:testhost	ifInUcastPkts:116702539	ifOperStatus:1	ifDescr:eth0	ifOutNUcastPkts:0	ifInDiscards:0	ifAdminStatus:1	ifAlias:	_key:_eth0	ipaddr:198.51.100.208/255.255.255.0	ifOutErrors:0	_time:1371180722	_mtime:1371180722	ifInNUcastPkts:137045	ifInOctets:27498895868	ifSpeed:1000000000	ifInErrors:0	ifOutDiscards:0	ifOutOctets:12605443941	ifHighSpeed:1000	ifOutUcastPkts:83974756

data/test/test_app.rb

@@ -0,0 +1,58 @@
+require File.expand_path(File.dirname(__FILE__) + '/unittest_helper')
+
+require 'gri/app_collector'
+
+module GRI
+
+class MockWriter
+  attr_reader :record_count
+  def initialize
+    @record_count = 0
+  end
+
+  def write records
+    @record_count += records.size
+  end
+end
+
+class TestApp < Test::Unit::TestCase
+  def setup
+    @root_dir = File.dirname(__FILE__) + '/root'
+  end
+
+  def test_app_load_target_lines
+    config = Config.new
+    path = File.expand_path(@root_dir + '/testtab')
+    config['gritab-path'] = path
+    app = AppCollector.new config
+    lines = app.load_target_lines config
+    assert(lines.size >= 9)
+
+    targets = app.get_targets_from_lines lines, config
+    host, options = targets.first
+    ae 'localhost', host
+    ae '2c', options['ver']
+    ae 'public', options['community']
+    assert !options['interfaces']
+
+    config['host-pat'] = '00'
+    targets = app.get_targets_from_lines lines, config
+    ae 1, targets.size
+    ae 'host00', targets.first.first
+
+    config['host-pat'] = '02'
+    targets = app.get_targets_from_lines lines, config
+    ae 2, targets.size
+    ae 'host02', targets.last.first
+  end
+
+  def test_load_fake_descr_files
+    config = Config.new
+    app = AppCollector.new config
+    files = [@root_dir + '/if.def']
+    h = app.load_fake_descr_files files
+    ae 'Test eth0 Description', h['testhost']['_eth0']
+  end
+end
+
+end

data/test/test_builtindefs.rb

@@ -0,0 +1,24 @@
+require File.expand_path(File.dirname(__FILE__) + '/unittest_helper')
+
+require 'gri/builtindefs'
+require 'gri/msnmp'
+require 'gri/vendor'
+require 'gri/plugin/ucdavis'
+
+module GRI
+
+class TestBuiltinDEFS < Test::Unit::TestCase
+  def setup
+  end
+
+  def test_get_spec
+    specs = DEFS.get_specs 'num'
+    ae '_index', specs[:prop][:name]
+    ae 4, specs[:rra].size
+    specs = DEFS.get_specs ''
+    ae 'ifInOctets,inoctet,DERIVE,MAX,AREA,#90f090,in,8', specs[:ds].first
+    specs = DEFS.get_specs :l
+  end
+end
+
+end

data/test/test_collector.rb

@@ -0,0 +1,58 @@
+# coding: us-ascii
+
+require File.expand_path(File.dirname(__FILE__) + '/unittest_helper')
+
+require 'socket'
+require 'mock'
+require 'gri/log'
+require 'gri/collector'
+
+module GRI
+
+class MockSNMP < SNMP
+  def connect
+    @sock = MockUDPSocket.new
+    @sock.connect @host, @port
+  end
+end
+
+class TestCollector < Test::Unit::TestCase
+  TEST_SYS = "0\x81\xF2\x02\x01\x01\x04\bpublic00\xA2\x81\xE2\x02" +
+    "\x02\x03\xE9\x02\x01\x00\x02\x01\x000\x81\xD50d\x06\b+\x06\x01" +
+    "\x02\x01\x01\x01\x00\x04XLinux xxxxx.yyyyy.zzz.com " +
+    "X.X.XX-XXX.XX.X.XXX #1 SMP Thu Oct 31 12:00:00 EDT 2013 x86_64" +
+    "0\x1F\x06\b+\x06\x01\x02\x01\x01\x05\x00\x04\x13xxxxx.yyyyy.zzz.com" +
+    "0\x16\x06\b+\x06\x01\x02\x01\x01\x02\x00\x06\n+\x06\x01\x04\x01" +
+    "\xBF\b\x03\x02\n0\x10\x06\b+\x06\x01\x02\x01\x01\x03\x00C\x04" +
+    "\x7F\xC9i\xA10\"\x06\b+\x06\x01\x02\x01\x01\x06\x00" +
+    "\x04\x16xxxxx xxxx, xxxxx xxx."
+
+  def test_snmp_collecotr
+    c = Collector.create('snmp', 'testhost', {}) {|records|}
+    snmp = MockSNMP.new 'testhost'
+    c.instance_eval {
+      @loop = MockLoop.new
+      @snmp = snmp
+    }
+
+    rs = {}
+    c.get(SNMPCollector::SYSOIDS) {|results|
+      results.each {|enoid, tag, val| rs[SNMP::ROIDS[enoid]] = val}
+    }
+    ae :GET_REQ, snmp.state
+
+    ae 0, c.instance_eval('@retry_count')
+    c.retry
+    ae 1, c.instance_eval('@retry_count')
+
+    snmp.sock.data = TEST_SYS
+    c.on_readable
+    ae rs['sysName.0'], 'xxxxx.yyyyy.zzz.com'
+
+    enoid = BER.enc_v_oid SNMP::OIDS['ipAdEntIfIndex']
+    c.walk(enoid) {|enoid, tag, val|}
+    ae :GETNEXT_REQ, snmp.state
+  end
+end
+
+end