gri 10.0.0

data/test/test_main.rb

@@ -0,0 +1,19 @@
+require File.expand_path(File.dirname(__FILE__) + '/unittest_helper')
+
+require 'gri/main'
+
+module GRI
+
+class TestMain < Test::Unit::TestCase
+  def test_main_optparse
+    main = Main.new
+    opts = {}
+
+    optparser = main.instance_eval {optparse opts}
+    argv = ['--config-path', '/tmp/gri.conf']
+    optparser.parse argv
+    ae '/tmp/gri.conf', opts[:config_path]
+  end
+end
+
+end

data/test/test_mmsgpack.rb

@@ -0,0 +1,27 @@
+require File.expand_path(File.dirname(__FILE__) + '/unittest_helper')
+
+require 'msgpack'
+require 'gri/mmsgpack'
+
+class TestMMsgpack < Test::Unit::TestCase
+  def eq obj
+    ae obj, MessagePack.unpack(obj.to_msgpack)
+  end
+
+  def test_mmsgpack
+    eq 1
+    eq -1
+    eq 0x7fffffff
+    eq 1000
+    eq 2000000000
+    eq 1000000000000
+
+    eq 'abc'
+
+    assert_nil MessagePack.unpack(nil.to_msgpack)
+
+    eq [1,2,'a']
+
+    eq 1=>'abc'
+  end
+end

data/test/test_msnmp.rb

@@ -0,0 +1,147 @@
+# coding: us-ascii
+require File.expand_path(File.dirname(__FILE__) + '/unittest_helper')
+
+require 'gri/msnmp'
+
+class TestMSNMP < Test::Unit::TestCase
+  def test_ber_enc
+    # tag 0x02 == INT
+    ae "\x02\x01\x00", BER.enc(0x02, 0)
+    ae "\x02\x01\xFF", BER.enc(0x02, 255)
+    ae "\x02\x02\x01\x00", BER.enc(0x02, 256)
+    ae "\x02\x01\xff", BER.enc(0x02, -1)
+    ae "\x02\x01\x80", BER.enc(0x02, -128)
+    ae "\x02\x02\xff\x7f", BER.enc(0x02, -129)
+
+    # tag 0x04 == STR
+    ae "\x04\x01a", BER.enc(0x04, 'a')
+    ae "\x04\x08abcdefgh", BER.enc(0x04, 'abcdefgh')
+  end
+
+  def test_ber_enc_len
+    ae "\x1", BER.enc_len(1)
+    ae "\x7f", BER.enc_len(127)
+    ae "\x81\x80", BER.enc_len(128)
+    ae "\x81\xff", BER.enc_len(255)
+    ae "\x82\x01\x00", BER.enc_len(256)
+    ae "\x82\xff\xff", BER.enc_len(65535)
+  end
+
+  def test_ber_enc_oid
+    ae "+\x06\x01\x02\x01\x02", BER.enc_v_oid('1.3.6.1.2.1.2')
+    ae "\x06\x06+\x06\x01\x02\x01\x02", BER.enc_oid('1.3.6.1.2.1.2')
+  end
+
+  def test_ber_enc_oid_list
+    list = ['sysDescr', 'sysName'].map {|n| SNMP::OIDS[n] + '.0'}
+    s = BER.enc_oid_list list
+    t, val, remain = BER.tlv(s)
+    ae 0x30, t
+    assert remain.empty?
+  end
+
+  def test_ber_cat_enoid
+    oids = [BER.enc_v_oid('1.3.6.1.2.1.1'), BER.enc_v_oid('1.3.6.1.2.1.2')]
+    varbind = BER.cat_enoid oids
+    t, val, remain = BER.tlv(varbind)
+    ae 0x30, t
+    assert remain.empty?
+  end
+
+  def test_dec_oid
+    ae [1,3,6,1,2,1,2], BER.dec_oid(BER.enc_v_oid('1.3.6.1.2.1.2'))
+  end
+
+  def test_ber_dec_int
+    ae 1, BER.dec_int("\x01")
+    ae 127, BER.dec_int("\x7f")
+    ae -128, BER.dec_int("\x80")
+    ae -129, BER.dec_int("\xff\x7f")
+  end
+
+  def test_ber_dec_cnt
+    ae 1, BER.dec_cnt("\x00\x00\x00\x01")
+    ae(2**31+1, BER.dec_cnt("\x80\x00\x00\x01"))
+  end
+
+  def test_ber_tlv
+    tag, val, remain = BER.tlv "\x04\x06public\x04\x08abcdefgh"
+    ae 4, tag
+    ae 'public', val
+    ae "\x04\x08abcdefgh", remain
+
+    s = 'a' * 200
+    tag, val, remain = BER.tlv(BER.enc_str(s))
+    ae 4, tag
+    ae s, val
+    assert remain.empty?
+
+    s = 'a' * 300
+    tag, val, remain = BER.tlv(BER.enc_str(s))
+    ae 4, tag
+    ae s, val
+    assert remain.empty?
+  end
+
+  def test_ber_dec_msg
+    varbind = "\0" * 160
+    s = BER.enc_int(1234) + BER.enc_int(0) + BER.enc_int(0) + varbind
+    s = "\xA2" + BER.enc_len(s.size) + s
+    s = BER.enc_int(1) + BER.enc_str('public') + s
+    msg = "\x30" + BER.enc_len(s.size) + s
+
+    enc_request_id, error_status, error_index, varbind = BER.dec_msg msg
+    ae BER.enc_int(1234), enc_request_id
+    ae 0, error_status
+    ae 0, error_index
+  end
+
+  def test_ber_dec_varbind_block
+    oid = '1.3.6.1.2.1.6.9.0'
+    s = BER.enc_oid(oid) + "B\x01\x10" # GAUGE 1 16
+    msg = "\x30" + BER.enc_len(s.size) + s
+    res, = BER.dec_varbind(msg)
+    ae oid, BER.dec_oid(res[0]).join('.')
+    ae 0x42, res[1] # GAUGE
+    ae 16, res[2]
+
+    oid = SNMP::OIDS['sysDescr']+'.0'
+    s = BER.enc_oid(oid) + BER.enc_str("test")
+    s = "\x30" + BER.enc_len(s.size) + s
+    res, = BER.dec_varbind(s)
+    ae oid, BER.dec_oid(res[0]).join('.')
+    ae 0x04, res[1]
+    ae 'test', res[2]
+  end
+
+  def test_snmp_enoid2name
+    snmp = SNMP.new '127.0.0.1'
+    enoid = BER.enc_v_oid SNMP::OIDS['system']
+    ae 'system', snmp.enoid2name(enoid)
+    enoid = BER.enc_v_oid '1.3.6.1.2.1.1.1.0'
+    ae 'sysDescr.0', snmp.enoid2name(enoid)
+  end
+
+  def test_c64
+    c = BER::C64.new "\0"
+    ae "0x0000000000000000", c.to_s
+
+    c = BER::C64.new "\x01\x00\x00"
+    ae "0x0000000000010000", c.to_s
+    ae 65536, c.to_i
+
+    n = 0xffffffffffffffff
+    c = BER::C64.new [n].pack('Q')
+    ae "0xffffffffffffffff", c.to_s
+
+    c = BER::C64.new "\x12\x34\x56\x78\xfe\xdc\xba\x98"
+    ae "0x12345678fedcba98", c.to_s
+    ae "\x124Vx\xFE\xDC\xBA\x98", c.to_binary
+    ae "\xcf\x124Vx\xfe\xdc\xba\x98", c.to_msgpack
+
+    c = BER::C64.new "\x01\x23\x45\x67\x89"
+    ae "\xcf\x00\x00\x00\x01#Eg\x89", c.to_msgpack
+
+    c = BER::C64.new "\x00\x03\x00\x00\x00\x00\x00\x00"
+  end
+end

data/test/test_page.rb

@@ -0,0 +1,51 @@
+require File.expand_path(File.dirname(__FILE__) + '/unittest_helper')
+
+require 'rack'
+require 'gri/page'
+
+module GRI
+
+class TestPage < Test::Unit::TestCase
+  def setup
+    @root_dir = File.dirname(__FILE__) + '/root'
+    @page = Page.new
+  end
+
+  def test_page_mk_page_title
+    rs = ['testhost__eth0']
+    res = @page.mk_page_title([@root_dir + '/gra'], rs, {})
+    ae 3, res.size
+    ae 'testdescr', res[0]
+    assert_match(/__eth0/, res[1])
+    assert_match(/ eth0/, res[2])
+  end
+
+  def test_page_mk_param_str
+    rs = ['r0.example.com', 'r 1?']
+    params = {}
+    assert_match /\br=r\+1%3F/, @page.mk_param_str(0, 0, rs, nil, params)
+  end
+
+  def test_page_mk_graph_tag
+    stime = Time.local(2013,10,1)
+    etime = stime + 24*3600
+    rs = ['testhost__eth0']
+    params = {'p'=>'s'}
+    res = @page.mk_graph_tag stime, etime, rs, params
+    ae 2, res.scan(/inoctet/).size
+    ae 2, res.scan(/indiscard/).size
+    ae 2, res.scan(/inerror/).size
+  end
+
+  def test_page_parse_request
+    s = '/?r=testhost__eth0&cs=2013-10-01&ce=2013-10-02'
+    env = Rack::MockRequest.env_for s
+    req = GRI::Request.new env
+    stime, etime, params = @page.parse_request req
+    ae 1380553200, stime.to_i
+    ae 1380639600, etime.to_i
+    ae 'testhost__eth0', params['r']
+  end
+end
+
+end

data/test/test_pcollector.rb

@@ -0,0 +1,71 @@
+require File.expand_path(File.dirname(__FILE__) + '/unittest_helper')
+
+require 'gri/config'
+require 'gri/pcollector'
+require 'mock'
+
+module GRI
+
+class TestPCollector < Test::Unit::TestCase
+  def setup
+    path = File.expand_path(File.dirname(__FILE__) + '/root/testtab')
+    config = Config.new
+    config['gritab-path'] = path
+    @app = AppCollector.new config
+    lines = @app.load_target_lines config
+    @targets = @app.get_targets_from_lines lines, config
+  end
+
+  def test_pcollector_get_ptargets
+    targets = @targets
+
+    basetime = Time.local(2013,4,1,10,5).to_i
+    ptargets = @app.get_ptargets targets, basetime, 0
+    ae 1, ptargets.size
+    ae basetime.to_i, ptargets.keys.first
+
+    ptargets = @app.get_ptargets targets, basetime, 1
+    t = Time.local(2013,4,1,10,5).to_i
+    hosts = ptargets[t].map {|n| targets[n].first}.sort
+    ae ["host01", "host02", "host03", "host04", "host05", "localhost"], hosts
+
+    basetime = Time.local(2013,4,1,10,0).to_i
+    ptargets = @app.get_ptargets targets, basetime, 300
+    t = Time.local(2013,4,1,10,0).to_i
+    hosts = ptargets[t].map {|n| targets[n].first}.sort
+    ae ["host00", "host01", "host02", "host03", "host04", "host05", "host06",
+      "localhost"], hosts
+
+    basetime = Time.local(2013,4,1,10,7).to_i
+    ptargets = @app.get_ptargets targets, basetime, 300
+    t = Time.local(2013,4,1,10,10).to_i
+    hosts = ptargets[t].map {|n| targets[n].first}.sort
+    ae ["host01", "host02", "host03", "host04", "host05", "localhost"], hosts
+
+    basetime = Time.local(2013,4,1,10,7).to_i
+    ptargets = @app.get_ptargets targets, basetime, 300, 120
+
+    #for t, inds in ptargets.sort_by {|t,| t}
+    #  p [Time.at(t), inds.map {|ind|targets[ind].first}]
+    #end
+  end
+
+  def test_pcollector_server_loop
+    @app.server_loop([], {}, nil, nil)
+  end
+
+  def test_pcollector_get_max_queue_size
+    config = Config.new
+    app = AppCollector.new config
+    assert_kind_of Integer, @app.get_max_queue_size
+  end
+
+  def test_pcollector_fork_child
+    @app.extend MockFork
+    nproc = 2
+    pids = @app.fork_child nil, nil, nproc, nil, nil, nil, nil
+    ae nproc, pids.size
+  end
+end
+
+end

data/test/test_plugin.rb

@@ -0,0 +1,62 @@
+# coding: us-ascii
+require File.expand_path(File.dirname(__FILE__) + '/unittest_helper')
+
+require 'gri/config'
+require 'gri/plugin'
+
+module GRI
+
+class TestPlugin < Test::Unit::TestCase
+  def test_plugin_load_plugin_dir
+    lib_dir = File.dirname(__FILE__) + '/../lib/gri/plugin'
+    files = GRI::Plugin.get_plugin_files lib_dir
+    assert(files.size > 5)
+
+    config = Config.new
+    config['enable-plugin'] = 'ucdavis'
+    files = GRI::Plugin.get_plugin_files lib_dir, config
+    ae 1, files.size
+
+    config['enable-plugin'] = 'netsnmp'
+    files = GRI::Plugin.get_plugin_files lib_dir, config
+    ae 2, files.size
+
+    config['disable-plugin'] = 'ucdavis'
+    files = GRI::Plugin.get_plugin_files lib_dir, config
+    ae 1, files.size
+  end
+
+require 'gri/msnmp'
+require 'gri/builtindefs'
+require 'gri/vendor'
+require 'gri/polling_unit'
+require 'gri/plugin/ucdavis'
+
+  DATA = [
+    ["+\x06\x01\x04\x01\x8Fe\t\x01\x01\x01", 2, 1],
+    ["+\x06\x01\x04\x01\x8Fe\t\x01\x02\x01", 4, "/"],
+    ["+\x06\x01\x04\x01\x8Fe\t\x01\x06\x01", 2, 40_000_000],
+    ["+\x06\x01\x04\x01\x8Fe\t\x01\a\x01", 2, 25_000_000],
+    ["+\x06\x01\x04\x01\x8Fe\t\x01\b\x01", 2, 9_000_000],
+  ]
+
+  def test_plugin_ucdavis_feed
+    SNMP.new 'host'
+    vendor = Vendor.check({}, {'ucd-dsk'=>true})
+    pu = vendor.get_punits.detect {|pu| pu.name == 'ucd-dsk'}
+
+    wh = {}
+    DATA.each {|enoid, tag, val|
+      pu.feed wh, enoid, tag, val}
+    h = wh[1]
+    ae '/', h['dskPath']
+    ae 40000000, h['dskTotal']
+    ae 9000000, h['dskUsed']
+
+    pu.fix_workhash :disk=>wh
+    ae 40000000*1024, h['dskTotal']
+    ae 9000000*1024, h['dskUsed']
+  end
+end
+
+end

data/test/test_plugin_cisco.rb

@@ -0,0 +1,23 @@
+require File.expand_path(File.dirname(__FILE__) + '/unittest_helper')
+
+require 'gri/msnmp'
+require 'gri/builtindefs'
+require 'gri/vendor'
+require 'gri/plugin/cisco'
+
+module GRI
+
+class TestPluginCisco < Test::Unit::TestCase
+  def test_plugin_cisco
+    sysinfo = {'sysObjectID'=>"+\x06\x01\x04\x01\t\x01",
+      'sysDescr'=>"Cisco IOS Software, c9999x00000_rp Software (c9999x00000_xx), Version 12.3(45)X678, RELEASE SOFTWARE ()",
+      }
+    vendor = Vendor.check sysinfo
+    assert vendor.options['interfaces']
+    assert vendor.options['lcpu']
+    ae '2c', vendor.options['ver']
+    ae 'c9999x00000_xx', sysinfo['_firm']
+  end
+end
+
+end

data/test/test_polling_unit.rb

@@ -0,0 +1,58 @@
+# coding: us-ascii
+require File.expand_path(File.dirname(__FILE__) + '/unittest_helper')
+
+require 'gri/msnmp'
+require 'gri/builtindefs'
+require 'gri/polling_unit'
+
+module GRI
+  class TestPollingUnit < Test::Unit::TestCase
+    def test_polling_unit_all_units
+      units = PollingUnit.all_units
+      assert_kind_of PollingUnit, units['interfaces']
+      assert_kind_of PollingUnit, units['ifMIB']
+      assert_kind_of PollingUnit, units['ipaddr']
+    end
+
+    def test_polling_unit
+      pu = PollingUnit.new 'test', :test
+      ae 'test', pu.name
+      pu.dhash = {}
+      oids = ['ifDescr', 'ifSpeed', 'ifAdminStatus', 'ifOperStatus']
+      pu.set_oids oids
+      ae 4, pu.oids.size
+      ae "+\x06\x01\x02\x01\x02\x02\x01\x02", pu.oids.first
+    end
+
+    def test_polling_unit_feed
+      snmp = SNMP.new 'host'
+
+      workhash = {''=>{}}
+
+      units = PollingUnit.all_units
+      pu = units['interfaces']
+      wh = workhash[pu.cat]
+      enoid = BER.enc_v_oid(SNMP::OIDS['ifDescr'] + '.1')
+      pu.feed wh, enoid, 4, 'lo'
+      ae 'lo', wh[1]['ifDescr']
+      enoid = BER.enc_v_oid(SNMP::OIDS['ifDescr'] + '.2')
+      pu.feed wh, enoid, 4, 'eth0'
+      ae 'eth0', wh[2]['ifDescr']
+
+      pu = units['ifMIB']
+      wh = workhash[pu.cat]
+      enoid = BER.enc_v_oid(SNMP::OIDS['ifHighSpeed'] + '.2')
+      pu.feed wh, enoid, 2, 10000
+      ae 10000, wh[2]['ifHighSpeed']
+      pu.fix_workhash workhash
+      ae 10000000000, wh[2]['ifSpeed']
+
+      workhash[:ipaddr] = {
+        '192.168.0.1'=>{'ifIndex'=>2, 'mask'=>'255.255.255.0'}
+      }
+      pu = units['ipaddr']
+      pu.fix_workhash workhash
+      ae '192.168.0.1/255.255.255.0', wh[2]['ipaddr']
+    end
+  end
+end