tb 0.1 → 0.2

data/test/test_cmd_pp.rb

@@ -0,0 +1,62 @@
+require 'test/unit'
+require 'tb/cmdtop'
+require 'tmpdir'
+
+class TestTbCmdPP < 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_pp(['-o', o="o.pp", 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_extend
+    File.open(i="i.csv", "w") {|f| f << <<-"End".gsub(/^[ \t]+/, '') }
+      a,b
+      0,1,2,3
+    End
+    Tb::Cmd.main_pp(['-o', o="o.pp", i])
+    assert_equal(<<-"End".gsub(/\s/, ''), File.read(o).gsub(/\s/, ''))
+      { "a" => "0", "b" => "1", "1" => "2", "2" => "3" }
+    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_pp(['-o', o="o.csv", i1, i2])
+    assert_equal(<<-"End".gsub(/^[ \t]+/, ''), File.read(o))
+      {"a"=>"1", "b"=>"2"}
+      {"a"=>"3", "b"=>"4"}
+      {"b"=>"5", "a"=>"6"}
+      {"b"=>"7", "a"=>"8"}
+    End
+  end
+
+end

data/test/test_cmd_rename.rb

@@ -0,0 +1,91 @@
+require 'test/unit'
+require 'tb/cmdtop'
+require 'tmpdir'
+
+class TestTbCmdRename < 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,d
+      0,1,2,3
+      4,5,6,7
+      8,9,a,b
+      c,d,e,f
+    End
+    Tb::Cmd.main_rename(['-o', o="o.csv", 'b,x,c,b', i])
+    assert_equal(<<-"End".gsub(/^[ \t]+/, ''), File.read(o))
+      a,x,b,d
+      0,1,2,3
+      4,5,6,7
+      8,9,a,b
+      c,d,e,f
+    End
+  end
+
+  def test_no_rename_fields
+    exc = assert_raise(SystemExit) { Tb::Cmd.main_rename([]) }
+    assert(!exc.success?)
+  end
+
+  def test_empty
+    File.open(i="i.csv", "w") {|f| f << <<-"End".gsub(/^[ \t]+/, '') }
+      a,b,c,d
+      0,1,2,3
+      4,5,6,7
+      8,9,a,b
+      c,d,e,f
+    End
+    Tb::Cmd.main_rename(['-o', o="o.csv", '', i])
+    assert_equal(<<-"End".gsub(/^[ \t]+/, ''), File.read(o))
+      a,b,c,d
+      0,1,2,3
+      4,5,6,7
+      8,9,a,b
+      c,d,e,f
+    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_rename(['-o', o="o.csv", 'a,c', i1, i2])
+    assert_equal(<<-"End".gsub(/^[ \t]+/, ''), File.read(o))
+      c,b
+      1,2
+      3,4
+      6,5
+      8,7
+    End
+  end
+
+  def test_field_not_found
+    File.open(i="i.csv", "w") {|f| f << <<-"End".gsub(/^[ \t]+/, '') }
+      a,b
+      1,2
+      3,4
+    End
+    exc = assert_raise(SystemExit) { Tb::Cmd.main_rename(['-o', "o.csv", 'z,c', i]) }
+    assert(!exc.success?)
+    assert_match(/field not found/, exc.message)
+  end
+
+end

data/test/test_cmd_shape.rb

@@ -0,0 +1,50 @@
+require 'test/unit'
+require 'tb/cmdtop'
+require 'tmpdir'
+
+class TestTbCmdShape < 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
+      4,5,6,7
+    End
+    Tb::Cmd.main_shape(['-o', o="o.csv", i])
+    assert_equal(<<-"End".gsub(/^[ \t]+/, ''), File.read(o))
+      header_fields,min_fields,max_fields,records,filename
+      3,2,4,2,i.csv
+    End
+  end
+
+  def test_twofile
+    File.open(i1="i1.csv", "w") {|f| f << <<-"End".gsub(/^[ \t]+/, '') }
+      a
+      1
+      3
+    End
+    File.open(i2="i2.csv", "w") {|f| f << <<-"End".gsub(/^[ \t]+/, '') }
+      b,a
+      5,6
+      7,8
+    End
+    Tb::Cmd.main_shape(['-o', o="o.csv", i1, i2])
+    assert_equal(<<-"End".gsub(/^[ \t]+/, ''), File.read(o))
+      header_fields,min_fields,max_fields,records,filename
+      1,1,1,2,i1.csv
+      2,2,2,2,i2.csv
+    End
+  end
+
+end

data/test/test_cmd_sort.rb

@@ -0,0 +1,105 @@
+require 'test/unit'
+require 'tb/cmdtop'
+require 'tmpdir'
+
+class TestTbCmdSort < 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,4
+      0,3
+      3,2
+    End
+    Tb::Cmd.main_sort(['-o', o="o.csv", i])
+    assert_equal(<<-"End".gsub(/^[ \t]+/, ''), File.read(o))
+      a,b
+      0,3
+      1,4
+      3,2
+    End
+  end
+
+  def test_numeric
+    File.open(i="i.csv", "w") {|f| f << <<-"End".gsub(/^[ \t]+/, '') }
+      a,b
+      1,4
+      0,3
+      3,2
+    End
+    Tb::Cmd.main_sort(['-o', o="o.csv", '-N', i])
+    assert_equal(<<-"End".gsub(/^[ \t]+/, ''), File.read(o))
+      0,3
+      1,4
+      3,2
+      a,b
+    End
+  end
+
+  def test_opt_f
+    File.open(i="i.csv", "w") {|f| f << <<-"End".gsub(/^[ \t]+/, '') }
+      a,b
+      1,4
+      0,3
+      3,2
+    End
+    Tb::Cmd.main_sort(['-o', o="o.csv", '-f', 'b', i])
+    assert_equal(<<-"End".gsub(/^[ \t]+/, ''), File.read(o))
+      a,b
+      3,2
+      0,3
+      1,4
+    End
+  end
+
+  def test_cmp
+    File.open(i="i.csv", "w") {|f| f << <<-"End".gsub(/^[ \t]+/, '') }
+      a,b,c
+      10,a20b0,11
+      1,2e1,3
+      4,,6
+      7,8,9
+    End
+    Tb::Cmd.main_sort(['-o', o="o.csv", '-f', 'b', i])
+    assert_equal(<<-"End".gsub(/^[ \t]+/, ''), File.read(o))
+      a,b,c
+      4,,6
+      7,8,9
+      1,2e1,3
+      10,a20b0,11
+    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,0
+      7,8
+    End
+    Tb::Cmd.main_sort(['-o', o="o.csv", i1, i2])
+    assert_equal(<<-"End".gsub(/^[ \t]+/, ''), File.read(o))
+      a,b
+      0,5
+      1,2
+      3,4
+      8,7
+    End
+  end
+
+end

data/test/test_cmd_tsv.rb

@@ -0,0 +1,67 @@
+require 'test/unit'
+require 'tb/cmdtop'
+require 'tmpdir'
+
+class TestTbCmdTSV < 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_tsv(['-o', o="o.tsv", i])
+    assert_equal(<<-"End".gsub(/^[ \t]+/, ''), File.read(o))
+      a\tb\tc
+      0\t1\t2
+      4\t5\t6
+    End
+  end
+
+  def test_numeric
+    File.open(i="i.csv", "w") {|f| f << <<-"End".gsub(/^[ \t]+/, '') }
+      a,b,c
+      0,1,2
+      4,5,6
+    End
+    Tb::Cmd.main_tsv(['-o', o="o.tsv", '-N', i])
+    assert_equal(<<-"End".gsub(/^[ \t]+/, ''), File.read(o))
+      a\tb\tc
+      0\t1\t2
+      4\t5\t6
+    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_tsv(['-o', o="o.csv", i1, i2])
+    assert_equal(<<-"End".gsub(/^[ \t]+/, ''), File.read(o))
+      a\tb
+      1\t2
+      3\t4
+      6\t5
+      8\t7
+    End
+  end
+
+end

data/test/test_cmd_yaml.rb

@@ -0,0 +1,55 @@
+require 'test/unit'
+require 'tb/cmdtop'
+require 'tmpdir'
+
+class TestTbCmdYAML < 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_yaml(['-o', o="o.yaml", i])
+    assert_equal(
+      [
+        {'a' => '0', 'b' => '1', 'c' => '2'},
+        {'a' => '4', 'b' => '5', 'c' => '6'},
+      ],
+      YAML.load(File.read(o)))
+  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_yaml(['-o', o="o.csv", i1, i2])
+    assert_equal(
+      [
+        {'a' => '1', 'b' => '2'},
+        {'a' => '3', 'b' => '4'},
+        {'a' => '6', 'b' => '5'},
+        {'a' => '8', 'b' => '7'},
+      ],
+      YAML.load(File.read(o)))
+  end
+
+end

data/test/test_cmdtty.rb

@@ -0,0 +1,154 @@
+require 'test/unit'
+require 'tb/cmdtop'
+require 'tmpdir'
+begin
+  require 'pty'
+  require 'io/console'
+rescue LoadError
+end
+
+class TestTbCmdTTY < 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_env(k, v)
+    save = ENV[k]
+    begin
+      ENV[k] = v
+      yield
+    ensure
+      ENV[k] = save
+    end
+  end
+
+  def with_stdout(io)
+    save = STDOUT.dup
+    STDOUT.reopen(io)
+    begin
+      yield
+    ensure
+      STDOUT.reopen(save)
+      save.close
+    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_ttyout_multiscreen
+    File.open(i="i.csv", "w") {|f| f << <<-"End".gsub(/^[ \t]+/, '') }
+      a,b,c
+      0,1,2
+      4,5,6
+    End
+    with_env('PAGER', 'sed "s/^/foo:/"') {
+      PTY.open {|m, s|
+        s.raw!
+        s.winsize = [2, 80]
+        th = reader_thread(m)
+        with_stdout(s) {
+          Tb::Cmd.main_csv([i])
+        }
+        s.close
+        result = th.value
+        assert_equal(<<-"End".gsub(/^[ \t]+/, ''), result)
+          foo:a,b,c
+          foo:0,1,2
+          foo:4,5,6
+        End
+      }
+    }
+  end
+
+  def test_ttyout_singlescreen
+    File.open(i="i.csv", "w") {|f| f << <<-"End".gsub(/^[ \t]+/, '') }
+      a,b,c
+      0,1,2
+      4,5,6
+    End
+    with_env('PAGER', 'sed "s/^/foo:/"') {
+      PTY.open {|m, s|
+        s.raw!
+        s.winsize = [24, 80]
+        th = reader_thread(m)
+        with_stdout(s) {
+          Tb::Cmd.main_csv([i])
+        }
+        s.close
+        result = th.value
+        assert_equal(<<-"End".gsub(/^[ \t]+/, ''), result)
+          a,b,c
+          0,1,2
+          4,5,6
+        End
+      }
+    }
+  end
+
+  def test_ttyout_tab
+    File.open(i="i.csv", "w") {|f| f << <<-"End".gsub(/^[ \t]+/, '') }
+      a,b,c
+      0,\t,2
+    End
+    with_env('PAGER', 'sed "s/^/foo:/"') {
+      PTY.open {|m, s|
+        s.raw!
+        s.winsize = [3, 10]
+        th = reader_thread(m)
+        with_stdout(s) {
+          Tb::Cmd.main_csv([i])
+        }
+        s.close
+        result = th.value
+        assert_equal(<<-"End".gsub(/^[ \t]+/, ''), result)
+          foo:a,b,c
+          foo:0,\t,2
+        End
+      }
+    }
+  end
+
+  def test_ttyout_nottysize
+    File.open(i="i.csv", "w") {|f| f << <<-"End".gsub(/^[ \t]+/, '') }
+      a,b,c
+      0,1,2
+    End
+    with_env('PAGER', 'sed "s/^/foo:/"') {
+      PTY.open {|m, s|
+        s.raw!
+        s.winsize = [0, 0]
+        th = reader_thread(m)
+        with_stdout(s) {
+          Tb::Cmd.main_csv([i])
+        }
+        s.close
+        result = th.value
+        assert_equal(<<-"End".gsub(/^[ \t]+/, ''), result)
+          a,b,c
+          0,1,2
+        End
+      }
+    }
+  end
+
+end if defined?(PTY) && defined?(PTY.open)