tb 0.1 → 0.2

data/test/test_cmd_gsub.rb

@@ -0,0 +1,103 @@
+require 'test/unit'
+require 'tb/cmdtop'
+require 'tmpdir'
+
+class TestTbCmdGsub < 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
+      foo,bar,baz
+      qux,quuux
+    End
+    Tb::Cmd.main_gsub(['-o', o="o.csv", '[au]', 'YY', i])
+    assert_equal(<<-"End".gsub(/^[ \t]+/, ''), File.read(o))
+      a,b,c
+      foo,bYYr,bYYz
+      qYYx,qYYYYYYx
+    End
+  end
+
+  def test_no_regexp
+    exc = assert_raise(SystemExit) { Tb::Cmd.main_gsub([]) }
+    assert(!exc.success?)
+  end
+
+  def test_no_subst
+    exc = assert_raise(SystemExit) { Tb::Cmd.main_gsub(['foo']) }
+    assert(!exc.success?)
+  end
+
+  def test_opt_e
+    File.open(i="i.csv", "w") {|f| f << <<-"End".gsub(/^[ \t]+/, '') }
+      a,b,c
+      foo,bar,baz
+      qux,quuux
+    End
+    Tb::Cmd.main_gsub(['-o', o="o.csv", '-e', '[au]', 'YY', i])
+    assert_equal(<<-"End".gsub(/^[ \t]+/, ''), File.read(o))
+      a,b,c
+      foo,bYYr,bYYz
+      qYYx,qYYYYYYx
+    End
+  end
+
+  def test_opt_f
+    File.open(i="i.csv", "w") {|f| f << <<-"End".gsub(/^[ \t]+/, '') }
+      a,b,c
+      foo,bar,baz
+      qux,quuux
+    End
+    Tb::Cmd.main_gsub(['-o', o="o.csv", '-f', 'b', '[au]', 'YY', i])
+    assert_equal(<<-"End".gsub(/^[ \t]+/, ''), File.read(o))
+      a,b,c
+      foo,bYYr,baz
+      qux,qYYYYYYx
+    End
+  end
+
+  def test_opt_f_extend
+    File.open(i="i.csv", "w") {|f| f << <<-"End".gsub(/^[ \t]+/, '') }
+      a,b
+      foo,bar,baz
+    End
+    Tb::Cmd.main_gsub(['-o', o="o.csv", '-f', '1', 'baz', 'Y', i])
+    assert_equal(<<-"End".gsub(/^[ \t]+/, ''), File.read(o))
+      a,b
+      foo,bar,Y
+    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_gsub(['-o', o="o.csv", '[46]', 'z', i1, i2])
+    assert_equal(<<-"End".gsub(/^[ \t]+/, ''), File.read(o))
+      a,b
+      1,2
+      3,z
+      z,5
+      8,7
+    End
+  end
+
+end

data/test/test_cmd_help.rb

@@ -0,0 +1,190 @@
+require 'test/unit'
+require 'tb/cmdtop'
+require 'tmpdir'
+
+class TestTbCmdHelp < 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_stdout(io)
+    save = STDOUT.dup
+    STDOUT.reopen(io)
+    begin
+      yield
+    ensure
+      STDOUT.reopen(save)
+      save.close
+    end
+  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 assert_exit_success(comanndline_words)
+    exc = assert_raise(SystemExit) { Tb::Cmd.main(comanndline_words)  }
+    assert(exc.success?)
+  end
+
+  def assert_exit_fail(comanndline_words)
+    exc = assert_raise(SystemExit) { Tb::Cmd.main(comanndline_words)  }
+    assert(!exc.success?)
+  end
+
+  def test_noarg
+    with_pipe {|r, w|
+      th = reader_thread(r)
+      with_stdout(w) {
+        Tb::Cmd.main([])
+      }
+      w.close
+      msg = th.value
+      r.close
+      assert_match(/Usage:/, msg)
+      assert_match(/ tb csv /, msg)
+      assert_match(/ tb cut /, msg)
+    }
+  end
+
+  def test_h
+    assert_equal(true, Tb::Cmd.main(['-h', '-o', o="msg"]))
+    msg = File.read(o)
+    assert_match(/Usage:/, msg)
+    assert_match(/ tb csv /, msg)
+    assert_match(/ tb cut /, msg)
+  end
+
+  def test_ohelp
+    assert_equal(true, Tb::Cmd.main(['--help', '-o', o="msg"]))
+    msg = File.read(o)
+    assert_match(/Usage:/, msg)
+    assert_match(/ tb csv /, msg)
+    assert_match(/ tb cut /, msg)
+  end
+
+  def test_help
+    assert_equal(true, Tb::Cmd.main(['help', '-o', o="msg"]))
+    msg = File.read(o)
+    assert_match(/Usage:/, msg)
+    assert_match(/ tb csv /, msg)
+    assert_match(/ tb cut /, msg)
+  end
+
+  def test_help_h
+    assert_exit_success(['help', '-o', o="msg", '-h'])
+    msg = File.read(o)
+    assert_match(/Usage: tb help/, msg)
+    assert_no_match(/Example:/, msg)
+  end
+
+  def test_help_hh
+    assert_exit_success(['help', '-o', o="msg", '-hh'])
+    msg = File.read(o)
+    assert_match(/Usage: tb help/, msg)
+    assert_match(/Example:/, msg)
+  end
+
+  def test_help_help
+    assert_exit_success(['help', '-o', o="msg", 'help'])
+    msg = File.read(o)
+    assert_match(/Usage: tb help/, msg)
+    assert_no_match(/Example:/, msg)
+  end
+
+  def test_help_h_help
+    assert_exit_success(['help', '-o', o="msg", '-h', 'help'])
+    msg = File.read(o)
+    assert_match(/Usage: tb help/, msg)
+    assert_match(/Example:/, msg)
+  end
+
+  def test_cat_h
+    assert_exit_success(['cat', '-o', o="msg", '-h'])
+    msg = File.read(o)
+    assert_match(/tb cat /, msg)
+    assert_no_match(/Example:/, msg)
+  end
+
+  def test_cat_hh
+    assert_exit_success(['cat', '-o', o="msg", '-hh'])
+    msg = File.read(o)
+    assert_match(/tb cat /, msg)
+    assert_match(/Example:/, msg)
+  end
+
+  def test_help_cat
+    assert_exit_success(['help', '-o', o="msg", 'cat'])
+    msg = File.read(o)
+    assert_match(/tb cat /, msg)
+    assert_no_match(/Example:/, msg)
+  end
+
+  def test_help_h_cat
+    assert_exit_success(['help', '-o', o="msg", '-h', 'cat'])
+    msg = File.read(o)
+    assert_match(/tb cat /, msg)
+    assert_match(/Example:/, msg)
+  end
+
+  def test_help_unexpected_subcommand1
+    File.open("log", "w") {|log|
+      with_stderr(log) {
+        assert_exit_fail(['help', '-o', "msg", 'foo'])
+      }
+    }
+    assert_match(/unexpected subcommand/, File.read("log"))
+  end
+
+  def test_help_unexpected_subcommand2
+    exc = assert_raise(SystemExit) { Tb::Cmd.main_help(['foo']) }
+    assert(!exc.success?)
+    assert_match(/unexpected subcommand/, exc.message)
+  end
+
+  def test_top_unexpected_subcommand
+    exc = assert_raise(SystemExit) { Tb::Cmd.main_body(['foo']) }
+    assert(!exc.success?)
+    assert_match(/unexpected subcommand/, exc.message)
+  end
+
+end

data/test/test_cmd_join.rb

@@ -0,0 +1,197 @@
+require 'test/unit'
+require 'tb/cmdtop'
+require 'tmpdir'
+
+class TestTbCmdJoin < 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(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]+/, '') }
+      a,c
+      1,a
+      3,b
+    End
+    Tb::Cmd.main_join(['-o', o="o.csv", i1, i2])
+    assert_equal(<<-"End".gsub(/^[ \t]+/, ''), File.read(o))
+      a,b,c
+      1,2,a
+      3,4,b
+    End
+  end
+
+  def test_noarg
+    exc = assert_raise(SystemExit) { Tb::Cmd.main_join(['-o', "o.csv"]) }
+    assert(!exc.success?)
+  end
+
+  def test_onearg
+    exc = assert_raise(SystemExit) { Tb::Cmd.main_join(['-o', "o.csv", 'foo.csv']) }
+    assert(!exc.success?)
+  end
+
+  def test_outer
+    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,c
+      4,5
+      6,7
+    End
+    Tb::Cmd.main_join(['-o', o="o.csv", '--outer', i1, i2])
+    assert_equal(<<-"End".gsub(/^[ \t]+/, ''), File.read(o))
+      a,b,c
+      1,2,
+      3,4,5
+      ,6,7
+    End
+  end
+
+  def test_outer_missing
+    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,c
+      4,5
+      6,7
+    End
+    Tb::Cmd.main_join(['-o', o="o.csv", '--outer-missing=z', i1, i2])
+    assert_equal(<<-"End".gsub(/^[ \t]+/, ''), File.read(o))
+      a,b,c
+      1,2,z
+      3,4,5
+      z,6,7
+    End
+  end
+
+  def test_outer_left
+    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,c
+      4,5
+      6,7
+    End
+    Tb::Cmd.main_join(['-o', o="o.csv", '--left', i1, i2])
+    assert_equal(<<-"End".gsub(/^[ \t]+/, ''), File.read(o))
+      a,b,c
+      1,2,
+      3,4,5
+    End
+  end
+
+  def test_outer_right
+    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,c
+      4,5
+      6,7
+    End
+    Tb::Cmd.main_join(['-o', o="o.csv", '--right', i1, i2])
+    assert_equal(<<-"End".gsub(/^[ \t]+/, ''), File.read(o))
+      a,b,c
+      3,4,5
+      ,6,7
+    End
+  end
+
+  def test_3file_inner
+    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]+/, '') }
+      a,c
+      1,a
+      3,b
+    End
+    File.open(i3="i3.csv", "w") {|f| f << <<-"End".gsub(/^[ \t]+/, '') }
+      a,d
+      1,x
+      3,z
+    End
+    Tb::Cmd.main_join(['-o', o="o.csv", i1, i2, i3])
+    assert_equal(<<-"End".gsub(/^[ \t]+/, ''), File.read(o))
+      a,b,c,d
+      1,2,a,x
+      3,4,b,z
+    End
+  end
+
+  def test_3file_left
+    File.open(i1="i1.csv", "w") {|f| f << <<-"End".gsub(/^[ \t]+/, '') }
+      a,b
+      1,A
+      2,B
+    End
+    File.open(i2="i2.csv", "w") {|f| f << <<-"End".gsub(/^[ \t]+/, '') }
+      a,c
+      1,C
+      3,D
+    End
+    File.open(i3="i3.csv", "w") {|f| f << <<-"End".gsub(/^[ \t]+/, '') }
+      a,d
+      2,E
+      4,F
+    End
+    Tb::Cmd.main_join(['-o', o="o.csv", '--left', i1, i2, i3])
+    assert_equal(<<-"End".gsub(/^[ \t]+/, ''), File.read(o))
+      a,b,c,d
+      1,A,C,
+      2,B,,E
+    End
+  end
+
+  def test_3file_right
+    File.open(i1="i1.csv", "w") {|f| f << <<-"End".gsub(/^[ \t]+/, '') }
+      a,b
+      1,A
+      2,B
+    End
+    File.open(i2="i2.csv", "w") {|f| f << <<-"End".gsub(/^[ \t]+/, '') }
+      a,c
+      1,C
+      3,D
+    End
+    File.open(i3="i3.csv", "w") {|f| f << <<-"End".gsub(/^[ \t]+/, '') }
+      a,d
+      2,E
+      3,F
+    End
+    Tb::Cmd.main_join(['-o', o="o.csv", '--right', i1, i2, i3])
+    assert_equal(<<-"End".gsub(/^[ \t]+/, ''), File.read(o))
+      a,b,c,d
+      3,,D,F
+      2,,,E
+    End
+  end
+
+end