gri 10.0.0

data/test/test_request.rb

@@ -0,0 +1,26 @@
+require File.expand_path(File.dirname(__FILE__) + '/unittest_helper')
+
+require 'rack'
+require 'gri/request'
+
+module GRI
+
+class TestRequest < Test::Unit::TestCase
+  def test_request
+    env = Rack::MockRequest.env_for '/?r=1&r=2'
+    req = GRI::Request.new env
+    ae({'r'=>'2'}, req.params)
+    ae ['1', '2'], req.gparams.getvar('r')
+  end
+
+  def test_request_parse_query
+    s = 'r=192.168.0.1__GigabitEthernet1-2&stime=-108000&z=&tz=&y=&g=0'
+    res = Rack::Utils.parse_query s
+    ae '192.168.0.1__GigabitEthernet1-2', res['r']
+    ae '-108000', res['stime']
+    ae '', res['z']
+    ae '0', res['g']
+  end
+end
+
+end

data/test/test_rrd.rb

@@ -0,0 +1,53 @@
+require File.expand_path(File.dirname(__FILE__) + '/unittest_helper')
+
+require 'gri/rrd'
+
+class TestRRD < Test::Unit::TestCase
+  def setup
+    @test_dir = File.expand_path(File.dirname(__FILE__))
+  end
+
+  def create_test_rrd path
+
+  end
+
+  def test_rrd_version
+    rrd = RRD.new
+    assert_match /\A1\.\d\z/, rrd.version
+  end
+
+  def test_remove_base_dir
+    args = 'foo', 'bar', '/usr/local/gri/gra/host/host.rrd'
+    rrd = RRD.new nil, nil
+    res = rrd.remove_base_dir *args
+    ae args, res
+    base_dir = '/usr/local/gri/gra'
+    rrd = RRD.new nil, base_dir
+    res = rrd.remove_base_dir *args
+    ae 'host/host.rrd', res[2]
+    base_dir = '/usr/local/gri/gra'
+    rrd = RRD.new nil, base_dir
+    res = rrd.remove_base_dir *args
+    ae 'host/host.rrd', res[2]
+  end
+
+  def test_rrd_create_and_graph
+    s = Time.local(2012,12,19,9)
+    rrd_path = '/var/tmp/test.rrd'
+    File.unlink rrd_path rescue nil
+    rrd = RRD.new
+    rrd.set_create_args rrd_path, s - 300, 300,
+      'DS:number:GAUGE:600:U:U RRA:MAX:0.5:1:1000'
+    s.to_i.step((s+24*3600).to_i, 300) {|i|
+      rrd.buffered_update "#{i}:#{i%3600/300}"
+    }
+    rrd.flush_buffer
+
+    stime = Time.local(2012,12,19,9,0)
+    etime = Time.local(2012,12,20,9,0)
+    img = rrd.graphgen stime, etime,
+      ["DEF:number=#{rrd_path}:number:MAX", "LINE1:number#ff0000:id"]
+    open('/var/tmp/test.png', 'w') {|f| f.write img}
+    assert(img.size > 0)
+  end
+end

data/test/test_rrd_updater.rb

@@ -0,0 +1,139 @@
+require File.expand_path(File.dirname(__FILE__) + '/unittest_helper')
+
+require 'fileutils'
+require 'gri/updater'
+require 'gri/utils'
+
+module GRI
+
+class TestRRDUpdater < Test::Unit::TestCase
+  class MockRRD
+    attr_accessor :daemon
+    attr_reader :init_args, :updates
+
+    def initialize *args
+      @init_args = args
+      @updates = []
+    end
+
+    def set_create_args *args
+      @rrdname = args.shift
+      @args = args
+    end
+
+    def buffered_update s
+      @updates.push s
+    end
+  end
+
+  def setup
+    Config.init
+    @tmp_dir = File.expand_path(File.dirname(__FILE__) + '/.t')
+    Dir.mkdir @tmp_dir unless File.directory? @tmp_dir
+  end
+
+  def test_updater_get_tag_ary
+    Config['tag-rule'] = '/^((AA\d+)\s+\S+)/ aa $2.downcase $2'
+    Config['tag-rule'] = '/^((BB\d+)\s+\S+)/ bb $2.downcase $2'
+    res = Updater.get_tag_ary 'BB01 aaa'
+    ae 'bb01', res[0][1]
+  end
+
+  def test_updater_mk_rra_args
+    updater = GRI::RRDUpdater.new :data_name=>''
+    res = updater.mk_rra_args
+    ae ["RRA:AVERAGE:0.5:12:9000", "RRA:MAX:0.5:1:20000",
+      "RRA:MAX:0.5:12:9000", "RRA:MAX:0.5:144:2000"], res
+    Config['rra-max'] = '300 9999'
+    res = updater.mk_rra_args
+    ae ["RRA:MAX:0.5:300:9999"], res
+  end
+
+  def test_update_sysdb
+    path = "/tmp/.sysdb/sysdb.txt"
+    FileUtils.rm_r File.dirname(path), :force=>true
+    FileUtils.mkdir_p File.dirname(path)
+
+    sysdb = {}
+    Utils.update_ltsv_file path, '_host', sysdb
+    ae 0, File.size(path)
+    sysdb = {'host1'=>{'sysDescr'=>'Linux', '_host'=>'host1'}}
+    Utils.update_ltsv_file path, '_host', sysdb
+    sysdb = {'host2'=>{'sysDescr'=>'FreeBSD', '_host'=>'host2'}}
+    Utils.update_ltsv_file path, '_host', sysdb
+    FileUtils.rm_r File.dirname(path), :force=>true
+  end
+
+  def test_updater_create
+    data_name = 'num'
+    updater = GRI::RRDUpdater.new :data_name=>data_name
+    ds_specs = GRI::DEFS.get_specs(data_name)[:ds]
+    now_i = Time.now.to_i
+    time = now_i - now_i % 300
+    rrd = updater.create_rrd '/tmp/foo.rrd', time, 300, ds_specs, nil
+    assert_kind_of RRD, rrd
+  end
+
+  def test_updater_mk_ds_args
+    data_name = ''
+    updater = RRDUpdater.new :data_name=>data_name
+    specs = DEFS.get_specs data_name
+    ds_specs = specs[:ds]
+    ae 'ifInOctets,inoctet,DERIVE,MAX,AREA,#90f090,in,8', ds_specs.first
+    ds_args, record_keys = updater.mk_ds_args ds_specs, 300
+    ae 'DS:inoctet:DERIVE:750:U:U', ds_args.first
+    ae 'ifInOctets', record_keys.first
+  end
+
+  def test_updater_update1
+    dir = @tmp_dir
+    basename = 'foo_num_test'
+    rrd_path = "#{dir}/#{basename}.rrd"
+    File.unlink rrd_path rescue nil
+    data_name = 'num'
+    updater = GRI::RRDUpdater.new :data_name=>data_name,
+      :dir=>@tmp_dir, :host=>'foo', :index=>'test'
+
+    # mk_update_str
+    s = updater.mk_update_str [1, 2, 3, 4]
+    ae '1:2:3:4', s
+    s = updater.mk_update_str [1, 2, nil, 4]
+    ae '1:2:U:4', s
+    s = updater.mk_update_str [1, 'notnum', nil, 4]
+    ae '1:U:U:4', s
+
+    # update
+    time = Time.now.to_i
+    updater.update time, :number=>10
+    updater.close
+    assert File.exist?(rrd_path)
+  end
+
+  def test_updater_update2
+    updater = RRDUpdater.new :data_name=>'', :dir=>'/var/tmp',
+      :host=>'testhost', :index=>'eth0'
+    updater.rrd_class = MockRRD
+    time = Time.local(2013,4,1).to_i
+    record = {'ifDescr'=>'eth0', 'ifInOctets'=>10, 'ifOutOctets'=>20,
+      'ifInErrors'=>1, 'ifOutErrors'=>0, 'ifOperStatus'=>1, 'ifSpeed'=>1}
+    updater.update time, record
+    ae '1364742000:10:20:U:U:1:0:U:U:U:U', updater.rrd.updates.first
+  end
+
+  def test_update_records_file
+    test_dir = File.expand_path(File.dirname(__FILE__))
+    org_path = test_dir + '/root/gra/testhost/.records.txt'
+    path = '/var/tmp/.records.txt'
+    FileUtils.cp org_path, path
+
+    h = {'new'=>{'_key'=>'new', 'a'=>1}}
+    Utils.update_ltsv_file path, '_key', h
+
+    rs = GRI::Utils.load_records File.dirname(path)
+    ae '1', rs['new']['a']
+
+    File.unlink path
+  end
+end
+
+end

data/test/test_scheduler.rb

@@ -0,0 +1,31 @@
+require File.expand_path(File.dirname(__FILE__) + '/unittest_helper')
+
+require 'socket'
+require 'mock'
+require 'gri/log'
+require 'gri/scheduler'
+require 'gri/collector'
+
+module GRI
+
+class MockCollector < Collector
+  TYPES['mock'] = self
+end
+
+class TestScheduler < Test::Unit::TestCase
+  def setup
+    loop = MockLoop.new
+    metrics = Hash.new 0
+    @scheduler = Scheduler.new loop, metrics
+  end
+
+  def test_scheduler_process1
+    options = {}
+    metrics = @scheduler.instance_eval {@metrics}
+    ae 0, metrics[:run_count]
+    @scheduler.process1 'mock', 'testhost', options
+    ae 1, metrics[:run_count]
+  end
+end
+
+end

data/test/test_tra_collector.rb

@@ -0,0 +1,40 @@
+require File.expand_path(File.dirname(__FILE__) + '/unittest_helper')
+
+require 'mock'
+require 'gri/tra_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 TestTraCollector < Test::Unit::TestCase
+  def setup
+    @root_dir = File.dirname(__FILE__) + '/root'
+  end
+
+  def test_tra_collector
+    config = {}
+    writer = MockWriter.new
+    collector = Collector.create('tra', 'testhost', {}) {|records|
+      writer.write records
+    }
+    dir = @root_dir + '/gra/testhost'
+    Dir.glob(dir + '/.lu_*') {|path| File.unlink path}
+    db = LocalLDB.new "#{@root_dir}/tra/testhost"
+    options = {}
+    ae 0, writer.record_count
+    collector.update_rrd_dir dir, db, options
+    ae 20, writer.record_count
+  end
+end
+
+end

data/test/test_trad.rb

@@ -0,0 +1,33 @@
+require File.expand_path(File.dirname(__FILE__) + '/unittest_helper')
+
+require 'rack'
+require 'gri/trad'
+
+module GRI
+
+class TestTrad < Test::Unit::TestCase
+  def setup
+    Config.init
+    test_root = File.expand_path(File.dirname(__FILE__) + '/root')
+    Config['tra-dir'] = test_root + '/tra'
+    Config['gritab-path'] = test_root + '/testtab'
+    @trad = Trad.new
+    Rack::MockRequest::DEFAULT_ENV['REMOTE_ADDR'] = '127.0.0.1'
+    @app = Rack::MockRequest.new @trad
+  end
+
+  def test_trad_get
+    res = @app.get '/get?h=testhost&s=&t=1371178800'
+    ae 200, res.status
+    a = res.body.scan(/ifInOctets:(\d+)/)
+    ae a.first.first, '27497623151'
+    ae a.last.first, '27498895868'
+  end
+
+  def test_trad_gritab
+    res = @app.get '/gritab'
+    assert_match /\Alocalhost\s/, res.body.split(/\n/).first
+  end
+end
+
+end

data/test/test_util_marshal.rb

@@ -0,0 +1,17 @@
+require File.expand_path(File.dirname(__FILE__) + '/unittest_helper')
+
+require 'gri/util_marshal'
+
+module GRI
+
+class TestUtilMarshal < Test::Unit::TestCase
+  def test_util_marshal
+    obj = {'a'=>10, :a=>[nil, 1, 2]}
+    path = '/tmp/test.dump'
+    Marshal.dump_to_file obj, path
+    newobj = Marshal.load_from_file path
+    ae obj, newobj
+  end
+end
+
+end

data/test/test_utils.rb

@@ -0,0 +1,15 @@
+require File.expand_path(File.dirname(__FILE__) + '/unittest_helper')
+
+require 'gri/utils'
+
+module GRI
+
+class TestUtils < Test::Unit::TestCase
+  def test_key_encode
+    s = '1/1'
+    r = GRI::Utils.key_encode s
+    ae '1-1', r
+  end
+end
+
+end

data/test/test_vendor.rb

@@ -0,0 +1,40 @@
+require File.expand_path(File.dirname(__FILE__) + '/unittest_helper')
+
+require 'gri/builtindefs'
+require 'gri/msnmp'
+require 'gri/vendor'
+require 'gri/plugin/ucdavis'
+require 'gri/plugin/netsnmp'
+
+module GRI
+
+class TestVendor < Test::Unit::TestCase
+  def test_vendor
+    sysinfo = {}
+    vendor = Vendor.check sysinfo
+    ae 'unknown', vendor.name
+    sysinfo = {'sysObjectID'=>''}
+    vendor = Vendor.check sysinfo
+    ae 'unknown', vendor.name
+
+    sysinfo = {"sysName"=>"testhost.example.com",
+      "sysObjectID"=>"+\006\001\004\001\277\b\003\002\n",
+      "sysLocation"=>"Right here, right now.",
+      "sysDescr"=>"Linux testhost.example.com 2.6.18-274.12.1.el5PAE #1 SMP Tue Nov 29 14:16:58 EST 2011 i686",
+      "sysUpTime"=>1000000000}
+
+    vendor = Vendor.check sysinfo
+    assert vendor.options['interfaces']
+    assert vendor.options['ifMIB']
+    assert vendor.options['ucd-la']
+    assert vendor.options['ucd-memory']
+    assert_nil vendor.options['notexist']
+    ae '2c', vendor.options['ver']
+    ae 'Linux', sysinfo['_firm']
+
+    vendor = Vendor.check sysinfo, 'interfaces'=>false
+    assert !vendor.options['interfaces']
+  end
+end
+
+end

data/test/test_writer.rb

@@ -0,0 +1,33 @@
+require File.expand_path(File.dirname(__FILE__) + '/unittest_helper')
+
+require 'fileutils'
+
+require 'gri/writer'
+
+module GRI
+
+class Writer
+  attr_reader :options
+end
+
+class TestWriter < Test::Unit::TestCase
+  def test_writer
+    tra_dir = '/tmp'
+    writer = Writer.create 'ldb', :tra_dir=>tra_dir, :tra_expire_day=>1
+    assert_kind_of TextWriter, writer
+
+    hdir = tra_dir + '/testhost'
+    FileUtils.rm_r hdir, :force=>true
+    records = [{'_host'=>'testhost', '_key'=>'SYS', '_sysdescr'=>'TEST'}]
+    writer.write records
+    writer.finalize
+    sysdir = hdir + '/SYS_'
+    fname = Dir.glob(sysdir + '/*').first
+    str = File.read fname
+    assert_match /_sysdescr:TEST/, str
+    writer.purge_logs
+    FileUtils.rm_r hdir, :force=>true
+  end
+end
+
+end

data/test/unittest_helper.rb

@@ -0,0 +1,27 @@
+require 'rubygems'
+begin
+  require 'simplecov'
+  SimpleCov.start
+rescue LoadError
+end
+require 'test/unit'
+require 'pp'
+
+$test_dir = File.expand_path File.dirname(__FILE__)
+$:.unshift $test_dir
+$:.unshift($test_dir + '/../lib')
+require 'gri/q'
+$test = true
+#$debug = true
+
+module Test
+  module Unit
+    module Assertions
+      alias ae assert_equal
+    end
+  end
+end
+
+unless ARGV.grep(/--show-log/).empty?
+  $show_log = true
+end