tb 0.1 → 0.2

data/test/test_cmd_json.rb

@@ -0,0 +1,75 @@
+require 'test/unit'
+require 'tb/cmdtop'
+require 'tmpdir'
+begin
+  require 'json'
+rescue LoadError
+end
+
+class TestTbCmdJSON < Test::Unit::TestCase
+  def setup
+    Tb::Cmd.reset_option
+    @curdir = Dir.pwd
+    @tmpdir = Dir.mktmpdir
+    Dir.chdir @tmpdir
+  end
+  def teardown
+    Tb::Cmd.reset_option
+    Dir.chdir @curdir
+    FileUtils.rmtree @tmpdir
+  end
+
+  def test_basic
+    File.open(i="i.csv", "w") {|f| f << <<-"End".gsub(/^[ \t]+/, '') }
+      a,b,c
+      0,1,2
+      4,5,6
+    End
+    Tb::Cmd.main_json(['-o', o="o.json", i])
+    assert_equal(<<-"End".gsub(/\s/, ''), File.read(o).gsub(/\s/, ''))
+      [{
+        "a": "0",
+        "b": "1",
+        "c": "2"
+      },
+      {
+        "a": "4",
+        "b": "5",
+        "c": "6"
+      }]
+    End
+  end
+
+  def test_twofile
+    File.open(i1="i1.csv", "w") {|f| f << <<-"End".gsub(/^[ \t]+/, '') }
+      a,b
+      1,2
+      3,4
+    End
+    File.open(i2="i2.csv", "w") {|f| f << <<-"End".gsub(/^[ \t]+/, '') }
+      b,a
+      5,6
+      7,8
+    End
+    Tb::Cmd.main_json(['-o', o="o.csv", i1, i2])
+    assert_equal(<<-"End".gsub(/\s/, ''), File.read(o).gsub(/\s/, ''))
+      [{
+        "a": "1",
+        "b": "2"
+      },
+      {
+        "a": "3",
+        "b": "4"
+      },
+      {
+        "b": "5",
+        "a": "6"
+      },
+      {
+        "b": "7",
+        "a": "8"
+      }]
+    End
+  end
+
+end if defined?(JSON)

data/test/test_cmd_ls.rb

@@ -0,0 +1,203 @@
+require 'test/unit'
+require 'tb/cmdtop'
+require 'tmpdir'
+
+class TestTbCmdLs < Test::Unit::TestCase
+  def setup
+    Tb::Cmd.reset_option
+    @curdir = Dir.pwd
+    @tmpdir = Dir.mktmpdir
+    Dir.chdir @tmpdir
+  end
+  def teardown
+    Tb::Cmd.reset_option
+    Dir.chdir @curdir
+    FileUtils.rmtree @tmpdir
+  end
+
+  def with_stderr(io)
+    save = STDERR.dup
+    STDERR.reopen(io)
+    begin
+      yield
+    ensure
+      STDERR.reopen(save)
+      save.close
+    end
+  end
+
+  def with_pipe
+    r, w = IO.pipe
+    begin
+      yield r, w
+    ensure
+      r.close if !r.closed?
+      w.close if !w.closed?
+    end
+  end
+
+  def reader_thread(io)
+    Thread.new {
+      r = ''
+      loop {
+        begin
+          r << io.readpartial(4096)
+        rescue EOFError, Errno::EIO
+          break
+        end
+      }
+      r
+    }
+  end
+
+  def test_basic
+    Dir.mkdir("d")
+    File.open("d/a", "w") {}
+    Tb::Cmd.main_ls(['-o', o="o.csv", "d"])
+    assert_equal(<<-"End".gsub(/^[ \t]+/, ''), File.read(o))
+      filename
+      d/a
+    End
+  end
+
+  def test_opt_l_single
+    File.open("a", "w") {}
+    File.chmod(0754, "a")
+    Tb::Cmd.main_ls(['-o', o="o.csv", '-l', 'a'])
+    tb = Tb.load_csv(o)
+    assert_equal(1, tb.size)
+    assert_equal(%w[filemode nlink user group size mtime filename symlink], tb.list_fields)
+    rec = tb.get_record(0)
+    assert_equal("-rwxr-xr--", rec["filemode"])
+    assert_equal("1", rec["nlink"])
+    assert_match(/\A\d+\z/, rec["size"])
+    assert_match(/\A\d+-\d\d-\d\dT\d\d:\d\d:\d\d/, rec["mtime"])
+    assert_equal("a", rec["filename"])
+    assert_equal(nil, rec["symlink"])
+  end
+
+  def test_opt_l
+    Dir.mkdir("d")
+    File.open("d/a", "w") {}
+    Dir.mkdir("d/d2")
+    Tb::Cmd.main_ls(['-o', o="o.csv", '-l', 'd'])
+    tb = Tb.load_csv(o)
+    assert_equal(2, tb.size)
+    assert_equal("d/a", tb.get_record(0)["filename"])
+    assert_equal("d/d2", tb.get_record(1)["filename"])
+  end
+
+  def test_opt_ll_single
+    File.open("a", "w") {}
+    File.chmod(0754, "a")
+    Tb::Cmd.main_ls(['-o', o="o.csv", '-ll', 'a'])
+    tb = Tb.load_csv(o)
+    assert_equal(1, tb.size)
+    assert_equal(%w[dev ino mode filemode nlink uid user gid group rdev size blksize blocks atime mtime ctime filename symlink], tb.list_fields)
+    rec = tb.get_record(0)
+    assert_match(/\A0x[0-9a-f]+\z/, rec["dev"])
+    assert_match(/\A\d+\z/, rec["ino"])
+    assert_match(/\A0[0-7]+\z/, rec["mode"])
+    assert_equal("-rwxr-xr--", rec["filemode"])
+    assert_equal("1", rec["nlink"])
+    assert_match(/\A\d+\z/, rec["uid"])
+    assert_match(/\A\d+\z/, rec["gid"])
+    assert_match(/\A0x[0-9a-f]+\z/, rec["rdev"])
+    assert_match(/\A\d+\z/, rec["size"])
+    assert_match(/\A\d+\z/, rec["blksize"])
+    assert_match(/\A\d+\z/, rec["blocks"])
+    assert_match(/\A\d+-\d\d-\d\dT\d\d:\d\d:\d\d/, rec["atime"])
+    assert_match(/\A\d+-\d\d-\d\dT\d\d:\d\d:\d\d/, rec["mtime"])
+    assert_match(/\A\d+-\d\d-\d\dT\d\d:\d\d:\d\d/, rec["ctime"])
+    assert_equal("a", rec["filename"])
+    assert_equal(nil, rec["symlink"])
+  end
+
+  def test_opt_a
+    Dir.mkdir("d")
+    File.open("d/.foo", "w") {}
+    Tb::Cmd.main_ls(['-o', o="o.csv", '-a', 'd'])
+    tb = Tb.load_csv(o)
+    assert_equal(3, tb.size)
+    assert_equal("d", tb.get_record(0)["filename"]) # d/.
+    assert_equal(".", tb.get_record(1)["filename"]) # d/..
+    assert_equal("d/.foo", tb.get_record(2)["filename"])
+  end
+
+  def test_opt_A
+    Dir.mkdir("d")
+    File.open("d/.foo", "w") {}
+    Tb::Cmd.main_ls(['-o', o="o.csv", '-A', 'd'])
+    tb = Tb.load_csv(o)
+    assert_equal(1, tb.size)
+    assert_equal("d/.foo", tb.get_record(0)["filename"])
+  end
+
+  def test_opt_R
+    Dir.mkdir("d")
+    File.open("d/a", "w") {}
+    Dir.mkdir("d/d2")
+    File.open("d/d2/b", "w") {}
+    File.open("d/d2/c", "w") {}
+    Tb::Cmd.main_ls(['-o', o="o.csv", '-R', 'd'])
+    tb = Tb.load_csv(o)
+    assert_equal(4, tb.size)
+    assert_equal("d/a", tb.get_record(0)["filename"])
+    assert_equal("d/d2", tb.get_record(1)["filename"])
+    assert_equal("d/d2/b", tb.get_record(2)["filename"])
+    assert_equal("d/d2/c", tb.get_record(3)["filename"])
+  end
+
+  def test_opt_Ra
+    Dir.mkdir("d")
+    File.open("d/a", "w") {}
+    Dir.mkdir("d/d2")
+    File.open("d/d2/b", "w") {}
+    File.open("d/d2/c", "w") {}
+    Tb::Cmd.main_ls(['-o', o="o.csv", '-Ra', 'd'])
+    tb = Tb.load_csv(o)
+    assert_equal(8, tb.size)
+    assert_equal("d/.", tb.get_record(0)["filename"])
+    assert_equal("d/..", tb.get_record(1)["filename"])
+    assert_equal("d/a", tb.get_record(2)["filename"])
+    assert_equal("d/d2", tb.get_record(3)["filename"])
+    assert_equal("d/d2/.", tb.get_record(4)["filename"])
+    assert_equal("d/d2/..", tb.get_record(5)["filename"])
+    assert_equal("d/d2/b", tb.get_record(6)["filename"])
+    assert_equal("d/d2/c", tb.get_record(7)["filename"])
+  end
+
+  def test_symlink
+    File.symlink("a", "b")
+    Tb::Cmd.main_ls(['-o', o="o.csv", '-l', 'b'])
+    tb = Tb.load_csv(o)
+    assert_equal(1, tb.size)
+    assert_equal("b", tb.get_record(0)["filename"])
+    assert_equal("a", tb.get_record(0)["symlink"])
+  end
+
+  def test_not_found
+    File.open("log", "w") {|log|
+      with_stderr(log) {
+        exc = assert_raise(SystemExit) { Tb::Cmd.main_ls(['-o', "o.csv", '-l', 'a']) }
+        assert(!exc.success?)
+      }
+    }
+  end
+
+  def test_dir_entries_fail
+    Dir.mkdir("d")
+    begin
+      File.chmod(0, "d")
+      File.open("log", "w") {|log|
+        with_stderr(log) {
+          exc = assert_raise(SystemExit) { Tb::Cmd.main_ls(['-o', "o.csv", '-l', 'd']) }
+          assert(!exc.success?)
+        }
+      }
+    ensure
+      File.chmod(0700, "d")
+    end
+  end
+
+end

data/test/test_cmd_mheader.rb

@@ -0,0 +1,86 @@
+require 'test/unit'
+require 'tb/cmdtop'
+require 'tmpdir'
+
+class TestTbCmdMheader < Test::Unit::TestCase
+  def setup
+    Tb::Cmd.reset_option
+    @curdir = Dir.pwd
+    @tmpdir = Dir.mktmpdir
+    Dir.chdir @tmpdir
+  end
+  def teardown
+    Tb::Cmd.reset_option
+    Dir.chdir @curdir
+    FileUtils.rmtree @tmpdir
+  end
+
+  def test_basic
+    File.open(i="i.csv", "w") {|f| f << <<-"End".gsub(/^[ \t]+/, '') }
+          ,2000,2000,2001,2001
+      name,aaaa,bbbb,aaaa,bbbb
+      x,1,2,3,4
+      y,5,6,7,8
+    End
+    Tb::Cmd.main_mheader(['-o', o="o.csv", i])
+    assert_equal(<<-"End".gsub(/^[ \t]+/, ''), File.read(o))
+      name,2000 aaaa,2000 bbbb,2001 aaaa,2001 bbbb
+      x,1,2,3,4
+      y,5,6,7,8
+    End
+  end
+
+  def test_opt_c
+    File.open(i="i.csv", "w") {|f| f << <<-"End".gsub(/^[ \t]+/, '') }
+      a,b,c
+      1,2,3
+      4,5,6
+    End
+    Tb::Cmd.main_mheader(['-o', o="o.csv", '-c', '2', i])
+    assert_equal(<<-"End".gsub(/^[ \t]+/, ''), File.read(o))
+      a 1,b 2,c 3
+      4,5,6
+    End
+  end
+
+  def test_no_unique_header
+    File.open(i="i.csv", "w") {|f| f << <<-"End".gsub(/^[ \t]+/, '') }
+      a,a
+      1,1
+    End
+    save = STDERR.dup
+    logf = File.open(log="log", "w")
+    STDERR.reopen(logf)
+    logf.close
+    Tb::Cmd.main_mheader(['-o', o="o.csv", i])
+    STDERR.reopen(save)
+    save.close
+    assert_equal('', File.read(o))
+    assert_match(/no header found/, File.read(log))
+  ensure
+    save.close if save && !save.closed?
+    logf.close if logf && !logf.closed?
+  end
+
+  def test_twofile
+    File.open(i1="i1.csv", "w") {|f| f << <<-"End".gsub(/^[ \t]+/, '') }
+      a,b
+      1,2
+      3,4
+    End
+    File.open(i2="i2.csv", "w") {|f| f << <<-"End".gsub(/^[ \t]+/, '') }
+      b,a
+      5,6
+      7,8
+    End
+    Tb::Cmd.main_mheader(['-o', o="o.csv", '-c', '2', i1, i2])
+    assert_equal(<<-"End".gsub(/^[ \t]+/, ''), File.read(o))
+      a 1,b 2
+      3,4
+      b,a
+      5,6
+      7,8
+    End
+  end
+
+end

data/test/test_cmd_newfield.rb

@@ -0,0 +1,63 @@
+require 'test/unit'
+require 'tb/cmdtop'
+require 'tmpdir'
+
+class TestTbCmdNewfield < Test::Unit::TestCase
+  def setup
+    Tb::Cmd.reset_option
+    @curdir = Dir.pwd
+    @tmpdir = Dir.mktmpdir
+    Dir.chdir @tmpdir
+  end
+  def teardown
+    Tb::Cmd.reset_option
+    Dir.chdir @curdir
+    FileUtils.rmtree @tmpdir
+  end
+
+  def test_basic
+    File.open(i="i.csv", "w") {|f| f << <<-"End".gsub(/^[ \t]+/, '') }
+      a,b
+      1,2
+      3,4
+    End
+    Tb::Cmd.main_newfield(['-o', o="o.csv", 'c', '_["a"].to_i + _["b"].to_i', i])
+    assert_equal(<<-"End".gsub(/^[ \t]+/, ''), File.read(o))
+      c,a,b
+      3,1,2
+      7,3,4
+    End
+  end
+
+  def test_no_new_field_name
+    exc = assert_raise(SystemExit) { Tb::Cmd.main_newfield([]) }
+    assert(!exc.success?)
+  end
+
+  def test_no_newfield_ruby_exp
+    exc = assert_raise(SystemExit) { Tb::Cmd.main_newfield(['foo']) }
+    assert(!exc.success?)
+  end
+
+  def test_twofile
+    File.open(i1="i1.csv", "w") {|f| f << <<-"End".gsub(/^[ \t]+/, '') }
+      a,b
+      1,2
+      3,4
+    End
+    File.open(i2="i2.csv", "w") {|f| f << <<-"End".gsub(/^[ \t]+/, '') }
+      b,a
+      5,6
+      7,8
+    End
+    Tb::Cmd.main_newfield(['-o', o="o.csv", 'c', '_["a"].to_i - _["b"].to_i', i1, i2])
+    assert_equal(<<-"End".gsub(/^[ \t]+/, ''), File.read(o))
+      c,a,b
+      -1,1,2
+      -1,3,4
+      1,6,5
+      1,8,7
+    End
+  end
+
+end

data/test/test_cmd_pnm.rb

@@ -0,0 +1,35 @@
+require 'test/unit'
+require 'tb/cmdtop'
+require 'tmpdir'
+
+class TestTbCmdPNM < Test::Unit::TestCase
+  def setup
+    Tb::Cmd.reset_option
+    @curdir = Dir.pwd
+    @tmpdir = Dir.mktmpdir
+    Dir.chdir @tmpdir
+  end
+  def teardown
+    Tb::Cmd.reset_option
+    Dir.chdir @curdir
+    FileUtils.rmtree @tmpdir
+  end
+
+  def test_basic
+    File.open(i="i.ppm", "w") {|f| f << <<-"End".gsub(/^[ \t]+/, '') }
+      P1
+      2 3
+      10
+      11
+      01
+    End
+    Tb::Cmd.main_pnm(['-o', o="o.ppm", i])
+    assert_equal(<<-"End".gsub(/^[ \t]+/, ''), File.read(o))
+      P1
+      2 3
+      10
+      11
+      01
+    End
+  end
+end