require File.join(File.expand_path(File.dirname(__FILE__)), '../..', 'test_helper.rb')
require 'rbbt/tsv'
require 'rbbt/tsv/filter'

class TestTSVFilters < Test::Unit::TestCase
  def _test_collect
    content1 =<<-EOF
#: :sep=/\\s+/#:case_insensitive=false
#Id    ValueA    ValueB
row1    a|aa|aaa    b
row2    A    B
    EOF

    TmpFile.with_file(content1) do |filename|
      tsv = TSV.open filename
      tsv.filter
      tsv.add_filter "field:ValueA", ["A"]
      assert_equal 1, tsv.collect.size
    end
  end

  def _test_through
    content1 =<<-EOF
#: :sep=/\\s+/#:case_insensitive=false
#Id    ValueA    ValueB
row1    a|aa|aaa    b
row2    A    B
    EOF

    TmpFile.with_file(content1) do |filename|
      tsv = TSV.open filename
      tsv.filter
      tsv.add_filter "field:ValueA", ["A"]
      elem = []
      tsv.through do |k,v| elem << k end
      assert_equal 1, elem.size
    end
  end

  def _test_2_filters
    content1 =<<-EOF
#: :sep=/\\s+/#:case_insensitive=false
#Id    ValueA    ValueB
row1    a|aa|aaa    b
row2    A    B
row3    A    C
    EOF

    TmpFile.with_file(content1) do |filename|
      tsv = TSV.open filename
      tsv.filter
      tsv.add_filter "field:ValueA", ["A"]

      elem = []
      tsv.through do |k,v| elem << k end
      assert_equal 2, elem.size
      assert_equal ["row2", "row3"], elem.sort

      tsv.add_filter "field:ValueB", ["C"]

      elem = []
      tsv.through do |k,v| elem << k end
      assert_equal 1, elem.size
    end
  end

  def _test_filter_persistence
    content1 =<<-EOF
#: :sep=/\\s+/#:case_insensitive=false
#Id    ValueA    ValueB
row1    a|aa|aaa    b
row2    A    B
row3    A    C
    EOF

    TmpFile.with_file(content1) do |filename|
      tsv = TSV.open filename
      tsv.filter
      tsv.add_filter "field:ValueA", ["A"], Rbbt.tmp.test.Filter.fieldValueA.find(:user)

      elem = []
      tsv.through do |k,v| elem << k end
      assert_equal 2, elem.size

      tsv.add_filter "field:ValueB", ["C"]

      elem = []
      tsv.through do |k,v| elem << k end
      assert_equal 1, elem.size

      tsv = TSV.open filename
      tsv.filter
      tsv.add_filter "field:ValueA", ["A"], Rbbt.tmp.test.Filter.fieldValueA.find(:user)

      elem = []
      tsv.through do |k,v| elem << k end
      assert_equal 2, elem.size
    end
  end

  def __test_filter_persistence_update
    content1 =<<-EOF
#: :sep=/\\s+/#:case_insensitive=false
#Id    ValueA    ValueB
row1    a|aa|aaa    b
row2    A    B
    EOF

    TmpFile.with_file(content1) do |filename|
      tsv = TSV.open filename
      tsv.filter
      tsv.add_filter "field:ValueA", ["A"], Rbbt.tmp.test.Filter.fieldValueA.find(:user)

      elem = []
      tsv.through do |k,v| elem << k end
      assert_equal 1, elem.size

      tsv["row3"] = [["A"], ["C"]]

      elem = []
      tsv.through do |k,v| elem << k end
      assert_equal 2, elem.size

      tsv.add_filter "field:ValueB", ["C"]

      elem = []
      tsv.through do |k,v| elem << k end
      assert_equal 1, elem.size

      tsv = TSV.open filename
      tsv.filter
      tsv.add_filter "field:ValueA", ["A"], Rbbt.tmp.test.Filter.fieldValueA.find(:user)

      elem = []
      tsv.through do |k,v| elem << k end
      assert_equal 1, elem.size
    end
  end

  def test_delete
    content =<<-EOF
#ID ValueA ValueB Comment
row1 a b c
row2 A B C
    EOF

    TmpFile.with_file(content) do |filename|
      tsv = TSV.open(File.open(filename), :double, :sep => /\s/)
      assert_equal 2, tsv.keys.length
      tsv.delete "row2"
      assert_equal 1, tsv.keys.length

      tsv = TSV.open(File.open(filename), :double, :sep => /\s/)
      tsv.filter
      tsv.add_filter "field:ValueA", ["A"]

      assert_equal 1, tsv.keys.length
      assert_equal ["row2"], tsv.keys

      tsv.delete "row2"
      assert_equal 0, tsv.keys.length

      tsv.pop_filter
      assert_equal ["row1"], tsv.keys
    end
  end

  def _test_filename
    content1 =<<-EOF
#: :sep=/\\s+/#:case_insensitive=false
#Id    ValueA    ValueB
row1    a|aa|aaa    b
row2    A    B
    EOF

    TmpFile.with_file(content1) do |filename|
      tsv = TSV.open filename
      tsv.filter
      tsv.add_filter "field:ValueA", ["A"]
      assert tsv.filename =~ /ValueA/
      tsv.pop_filter
      assert tsv.filename !~ /ValueA/
    end
 
  end
end