gri 10.0.0

data/test/test_config.rb

@@ -0,0 +1,62 @@
+require File.expand_path(File.dirname(__FILE__) + '/unittest_helper')
+
+require 'gri/config'
+
+module GRI
+
+class TestConfig < Test::Unit::TestCase
+  def test_config
+    conf = GRI::Config.new
+    conf['key'] = 1
+    ae 1, conf['key']
+    conf['key'] = 'a'
+    ae 'a', conf['key']
+    ae [1, 'a'], conf.getvar('key')
+
+    conf[:s] = 'aaa'
+    ae({"key"=>[1, "a"], :s=>"aaa"}, conf.to_h)
+
+    assert_nil conf['notexist']
+
+    conf['key2'] = false
+    assert !conf['key2']
+    assert conf.has_key?('key2')
+    assert !conf.has_key?('key3')
+
+    Config.init
+    assert_nil Config['key']
+    Config['key'] = 1
+    ae 1, Config['key']
+    Config['key'] = 'a'
+    ae 'a', Config['key']
+    ae [1, 'a'], Config.getvar('key')
+    assert conf.has_key?('key')
+  end
+
+  def test_load_from_file
+    path = File.expand_path(File.dirname(__FILE__) + '/root/test.conf')
+    conf = GRI::Config.load_from_file path
+    ae 1, conf['test-opt-num'].to_i
+    ae 'foo', conf['test-opt-str']
+    ae ["/dir1", "/dir2"], conf.getvar('plugin-dir')
+  end
+
+  def test_get_targets_from_lines
+    lines = ['localhost ver=2c community=public', 'testhost community=xxxxxx']
+    targets = GRI::Config.get_targets_from_lines lines
+    host, options = targets.first
+    ae 'localhost', host
+    ae '2c', options['ver']
+    ae 'public', options['community']
+    assert !options['interfaces']
+
+    lines = ['exhost type=exec cmd="vmstat 1 3|tail -1" ' +
+      'out_keys=procs_r,procs_b,memory_swpd,memory_free,memory_buff']
+    targets = GRI::Config.get_targets_from_lines lines
+    host, options = targets.first
+    ae 'exec', options['type']
+    ae 'vmstat 1 3|tail -1', options['cmd']
+  end
+end
+
+end

data/test/test_ds_list.rb

@@ -0,0 +1,48 @@
+require File.expand_path(File.dirname(__FILE__) + '/unittest_helper')
+
+require 'rack'
+require 'gri/log'
+require 'gri/config'
+require 'gri/ds_list'
+
+module GRI
+
+class TestDSList < Test::Unit::TestCase
+  def setup
+    @root_dir = File.expand_path(File.dirname(__FILE__) + '/root')
+    gra_dir = @root_dir + '/gra'
+    Config.init
+    Config['gra-dir'] = gra_dir
+  end
+
+  def test_ds_list_get_data_hash
+    dlist = DSList.new
+    records = Utils.load_records Config['gra-dir'] + '/testhost'
+    ae 'eth0', records['_eth0']['ifDescr']
+
+    data_hash = dlist.get_data_hash records
+    key, prop, = data_hash[''].first
+    ae '_eth0', key
+    ae 'eth0', prop[:name]
+
+    s = dlist.format_cell '%D', prop
+    ae '<td>testdescr</td>', s
+
+    ae ['<td>eth0</td>', '<td>testdescr</td>'],
+      dlist.format_tr(['%N', '%D'], prop)
+
+    prop[:ipaddr] = '192.168.0.0/255.255.255.0'
+    ae ['<td>192.168.0.0/24</td>'], dlist.format_tr(['%I'], prop)
+  end
+
+  def test_ds_list_mk_comp_links
+    dlist = DSList.new
+    comps = ['s', 'v']
+    ckeys = ['_eth0', '_eth1']
+    res = dlist.mk_comp_links comps, 'testhost', ckeys
+    ae 2, res.size
+    ae "<a href=\"?p=s&amp;r=testhost__eth0&amp;r=testhost__eth1\">stack</a>", res.first
+  end
+end
+
+end

data/test/test_exec_collector.rb

@@ -0,0 +1,33 @@
+require File.expand_path(File.dirname(__FILE__) + '/unittest_helper')
+
+require 'mock'
+require 'gri/plugin/exec_collector'
+
+module GRI
+
+class MockExecCollector < ExecCollector
+  def popen cmd
+    return MockIO.new, MockIO.new, MockIO.new
+  end
+end
+
+class TestExecCollector < Test::Unit::TestCase
+  def test_exec_collector
+    Collector::TYPES['exec'] = MockExecCollector
+    host = 'exhost'
+    options = {'type'=>'exec', 'cmd'=>'vmstat 1 1|tail -1',
+      'out_keys'=>'procs_r,procs_b,memory_swpd,memory_free,memory_buff'}
+    records = []
+    collector = Collector.create(options['type'], host, options) {|rs|
+      records += rs
+    }
+    assert_kind_of ExecCollector, collector
+
+    collector.on_read "0 0"
+    assert records.empty?
+    collector.on_read " 2000 3000\n0 0 4000 5000"
+    ae 1, records.size
+  end
+end
+
+end

data/test/test_format_helper.rb

@@ -0,0 +1,68 @@
+require File.expand_path(File.dirname(__FILE__) + '/unittest_helper')
+
+require 'gri/format_helper'
+
+module GRI
+
+class TestFormatHelper < Test::Unit::TestCase
+  include FormatHelper
+
+  def test_format_helper_h
+    ae "&lt;&#x2F;br&gt;", h('</br>')
+    ae "[nil]", h([nil])
+  end
+
+  def test_format_helper_escape_once
+    ae '&amp;', escape_once('&')
+    ae '&amp;', escape_once('&amp;')
+  end
+
+  def test_format_helper_u
+    ae "_%3A%3C%3E%5C%25", u('_:<>\%')
+  end
+
+  def test_format_helper_url_to
+    ae '/aaa', url_to('aaa')
+    ae '?a=1', url_to('?a=1')
+    assert_match %r{\A/aaa\?[ab]=[^&]+&}, url_to('aaa', :a=>1, 'b'=>'"')
+  end
+
+  def test_format_helper_mk_query
+    ae '?a=1', mk_query(:a=>1)
+    assert_match /\bb=a\+%26\b/, mk_query(:a=>1, :b=>'a &')
+    assert_not_match /\bc=/, mk_query(:c=>nil)
+  end
+
+  def test_format_helper_td
+    ae '<td>item</td>', td('item')
+    ae '<th>item</th>', td('item', :head=>true)
+    ae '<td colspan=2>item</td>', td('item', :colspan=>2)
+  end
+
+  def test_format_helper_mk_tag
+    ae '<a href="&#x2F;">top</a>', mk_tag('a', {:href=>'/'}, 'top')
+  end
+
+  def test_format_helper_text_field
+    ae '<input type="text" name="text" value="aaa" size="40"/>',
+      text_field('text', 'aaa', 40, nil, nil)
+  end
+
+  def test_format_helper_popup_menu
+    a = [[1, 'a'], [2, 'b'], [3, 'c', true]]
+    ae '<select name="popup"><option value="1">a</option><option value="2">b</option><option value="3" selected>c</option></select>',
+      popup_menu('popup', nil, *a)
+  end
+
+  def test_format_helper_to_scalestr
+    assert_nil to_scalestr(nil)
+    ae '0.0', to_scalestr(0)
+    ae '1.024K', to_scalestr(1024)
+    ae '1K', to_scalestr(1024, 1024)
+    ae '1.234M', to_scalestr(1234000)
+    ae '1.234G', to_scalestr(1234000000)
+    ae '1.234T', to_scalestr(1234000000000)
+  end
+end
+
+end

data/test/test_graph.rb

@@ -0,0 +1,69 @@
+require File.expand_path(File.dirname(__FILE__) + '/unittest_helper')
+
+require 'gri/graph'
+
+module GRI
+
+class TestGraph < Test::Unit::TestCase
+  def setup
+    @root_dir = File.dirname(__FILE__) + '/root'
+    @graph = Graph.new :dir=>'/var/tmp'
+  end
+
+  def test_graph_get_rrdpaths_and_ub
+    gra_dirs = [@root_dir + '/gra']
+    rs = ['testhost__eth0']
+    graph = Graph.new :dirs=>gra_dirs
+    res = graph.get_rrdpaths_and_ub rs
+    ae 1000000000, res[1]
+  end
+
+  def test_graph_mk_graph_title
+    gra_dirs = [@root_dir + '/gra']
+    rs = ['testhost__eth0']
+    graph = Graph.new :dirs=>gra_dirs
+    params = {}
+    res = graph.mk_graph_title rs, params
+    assert_nil res
+
+    params = {'p'=>'t'}
+    res = graph.mk_graph_title rs, params
+    ae '"testhost eth0 testdescr"', res
+  end
+
+  def test_graph_mk_graph_args
+    rrdpaths = ['/usr/local/gri/gra/host00__eth0.rrd',
+      '/usr/local/gri/gra/host01__eth0.rrd',]
+    params = {'p'=>''}
+    specs = DEFS.get_specs ''
+    res = @graph.mk_graph_args specs, rrdpaths, params
+    assert_match /\A-v\s+"bps"/, res[0]
+    assert_match /\A--base\s+1000/, res[1]
+    assert_match /\A--lower-limit\s+0/, res[2]
+    exprs = res[3].split
+    assert_match /\ADEF:v1inoctet=/, exprs[0]
+    assert_match /\ADEF:v2inoctet=/, exprs[1]
+    assert_match /\ACDEF:inoctet0=/, exprs[2]
+    assert_match /\AAREA:inoctet0#/, exprs[3]
+    assert_match /\AGPRINT:inoctet0:/, exprs[4]
+    exprs = res[4].split
+    assert_match /\ADEF:v1outoctet=/, exprs[0]
+    assert_match /\ADEF:v2outoctet=/, exprs[1]
+    assert_match /\ACDEF:outoctet0=/, exprs[2]
+    assert_match /\ALINE1:outoctet0#/, exprs[3]
+    assert_match /\AGPRINT:outoctet0:/, exprs[4]
+  end
+
+  def test_graph_mk_defstr
+    rrdpaths = ['/usr/local/gri/gra/host00__eth0.rrd',
+      '/usr/local/gri/gra/host01__eth0.rrd',]
+    params = {'fmt'=>'json'}
+    specs = DEFS.get_specs ''
+    ds_specs = specs[:ds].grep /Octets/
+    res = @graph.mk_defstr ds_specs, rrdpaths, params, nil
+    assert_match /XPORT:inoctet0:"in"/, res[0]
+    assert_match /XPORT:outoctet0:"out"/, res[1]
+  end
+end
+
+end

data/test/test_ldb.rb

@@ -0,0 +1,29 @@
+require File.expand_path(File.dirname(__FILE__) + '/unittest_helper')
+
+require 'gri/ldb'
+
+module GRI
+
+class TestLDB < Test::Unit::TestCase
+  def setup
+    root_dir = File.expand_path(File.dirname(__FILE__) + '/root')
+    dir = root_dir + "/tra/testhost"
+    @ldb = LocalLDB.new dir
+  end
+
+  def test_local_ldb
+    time = Time.local(2013, 6, 14, 12, 0)
+    a = []
+    @ldb.get_after('', time) {|t, h|
+      a.push h
+    }
+    ae 7, a.size
+  end
+
+  def test_get_data_names
+    data_names = @ldb.get_data_names
+    ae [''], data_names.keys
+  end
+end
+
+end

data/test/test_list.rb

@@ -0,0 +1,65 @@
+require File.expand_path(File.dirname(__FILE__) + '/unittest_helper')
+
+require 'rack'
+require 'gri/log'
+require 'gri/config'
+require 'gri/builtindefs'
+require 'gri/grapher'
+require 'gri/list'
+require 'gri/msnmp'
+require 'gri/vendor'
+require 'gri/plugin/ucdavis'
+
+module GRI
+
+class TestList < Test::Unit::TestCase
+  def setup
+    @root_dir = File.expand_path(File.dirname(__FILE__) + '/root')
+    gra_dir = @root_dir + '/gra'
+    Config.init
+    Config['gra-dir'] = gra_dir
+    @list = List.new
+    @app = Rack::MockRequest.new @list
+  end
+
+  def test_mk_regexp
+    s = '1a.*#'
+    list = List.new
+    ae "1a\\.\\*\\#", list.mk_regexp(s).source
+    list = List.new :use_regexp_search=>'on'
+    ae "1a.*\\#", list.mk_regexp(s).source
+  end
+
+  def test_load_sysdb
+    sysdb = @list.load_sysdb Config.getvar('gra-dir')
+    assert_match /Linux host.example.com/, sysdb['192.168.0.1']['sysDescr']
+  end
+
+  def test_list
+    res = @app.get '/'
+    ae 200, res.status
+  end
+
+  def test_list_sysdb_list
+    dirs = Config.getvar 'gra-dir'
+    sysdb = {}
+    params = {}
+    lines = @list.sysdb_list dirs, sysdb, params
+    ae [], lines
+    sysdb = {'host1'=>{}, 'host2'=>{}}
+    lines = @list.sysdb_list dirs, sysdb, params
+    ae ['host1', 'host2'], lines.map {|h, s| h}
+  end
+
+  def test_list_grep_graph
+    dirs = Config.getvar 'gra-dir'
+    sysdb = @list.load_sysdb dirs
+    hlines = @list.sysdb_list dirs, sysdb, {}
+    hosts = hlines.map {|h, l| h}
+    res = @list.grep_graph dirs, hosts, {}
+    ae 2, res.values.first.size
+    ae '', res.values.first.first.first
+  end
+end
+
+end

data/test/test_log.rb

@@ -0,0 +1,16 @@
+require File.expand_path(File.dirname(__FILE__) + '/unittest_helper')
+
+require 'gri/log'
+
+module GRI
+
+class TestLog < Test::Unit::TestCase
+  def test_log
+    time = Time.local(2014,1,2,3,4,56)
+    formatter = Log::Formatter.new
+    s = formatter.call nil, time, nil, 'test message'
+    ae "2014-01-02 03:04:56 test message\n", s
+  end
+end
+
+end

data/test/test_loop.rb

@@ -0,0 +1,35 @@
+require File.expand_path(File.dirname(__FILE__) + '/unittest_helper')
+
+require 'gri/loop'
+require 'gri/collector'
+
+module GRI
+
+class TestLoop < Test::Unit::TestCase
+  def setup
+    @loop = Loop.new
+  end
+
+  def test_loop_attach_detach
+    collector = GRI::Collector.new('testhost', {}, {})
+    detached = false
+    @loop.on_detach {detached = true}
+
+    @loop.attach collector
+    ae 1, @loop.collectors.size
+    io = Object.new
+    @loop.watch io, :r, 5, collector
+    @loop.watch io, :w, 5, collector
+    @loop.watch io, :rw, 5, collector
+    @loop.detach collector
+    ae 0, @loop.collectors.size
+    assert detached
+  end
+
+  def test_loop_has_active_watchers
+    assert !@loop.has_active_watchers?
+    @loop.run
+  end
+end
+
+end

data/test/test_ltsv.rb

@@ -0,0 +1,38 @@
+require File.expand_path(File.dirname(__FILE__) + '/unittest_helper')
+
+require 'gri/ltsv'
+
+module GRI
+
+class TestLTSV < Test::Unit::TestCase
+  def test_ltsv_escape
+    ae 'a\tb', LTSV.escape("a\tb")
+    ae 'a\n\r\tb', LTSV.escape("a\n\r\tb")
+  end
+
+  def test_ltsv_serialize
+    value = {'a'=>'1', 'b'=>'2', 'zzz'=>'10'}
+    line = LTSV.serialize value
+    ae value, LTSV.parse_string(line)
+  end
+
+  def test_ltsv_dump_to_file
+    path = '/var/tmp/ltsv_test.txt'
+    values = [{'a'=>'1', 'b'=>'2'}, {'x'=>'10', 'y'=>'20'}]
+    LTSV.dump_to_file values, path
+    vs = LTSV.load_from_file path
+    ae values, vs
+    File.unlink path
+  end
+
+  def test_parse_string
+    line = "a:b\tb:1\tc:\td:"
+    h = LTSV.parse_string line
+    ae 'b', h['a']
+    ae '1', h['b']
+    assert_nil h['c']
+    assert_nil h['d']
+  end
+end
+
+end