#!/usr/local/bin/ruby -w

# tc_data_converters.rb
#
#  Created by James Edward Gray II on 2005-12-30.
#  Copyright 2005 Gray Productions. All rights reserved.

require "test/unit"

require "faster_csv"

class TestDataConverters < Test::Unit::TestCase
  def setup
    @data   = "Numbers,:integer,1,:float,3.015"
    @parser = FasterCSV.new(@data)
    
    @custom = lambda { |field| field =~ /\A:(\S.*?)\s*\Z/ ? $1.to_sym : field }
  end
  
  def test_builtin_integer_converter
    # does convert
    [-5, 1, 10000000000].each do |n|
      assert_equal(n, FasterCSV::Converters[:integer][n.to_s])
    end
    
    # does not convert
    (%w{junk 1.0} + [""]).each do |str|
      assert_equal(str, FasterCSV::Converters[:integer][str])
    end
  end

  def test_builtin_float_converter
    # does convert
    [-5.1234, 0, 2.3e-11].each do |n|
      assert_equal(n, FasterCSV::Converters[:float][n.to_s])
    end
    
    # does not convert
    (%w{junk 1..0 .015F} + [""]).each do |str|
      assert_equal(str, FasterCSV::Converters[:float][str])
    end
  end

  def test_builtin_date_converter
    # does convert
    assert_instance_of(Date, FasterCSV::Converters[:date][Time.now.to_s])

    # does not convert
    assert_instance_of(String, FasterCSV::Converters[:date]["junk"])
  end

  def test_builtin_date_time_converter
    # does convert
    assert_instance_of( DateTime,
                        FasterCSV::Converters[:date_time][Time.now.to_s] )

    # does not convert
    assert_instance_of(String, FasterCSV::Converters[:date_time]["junk"])
  end
  
  def test_convert_with_builtin
    # setup parser...
    assert(@parser.respond_to?(:convert))
    assert_nothing_raised(Exception) { @parser.convert(:integer) }
    
    # and use
    assert_equal(["Numbers", ":integer", 1, ":float", "3.015"], @parser.shift)
    
    setup  # reset
    
    # setup parser...
    assert_nothing_raised(Exception) { @parser.convert(:float) }
    
    # and use
    assert_equal(["Numbers", ":integer", 1.0, ":float", 3.015], @parser.shift)
  end
  
  def test_convert_order
    # floats first, then integers...
    assert_nothing_raised(Exception) do 
      @parser.convert(:float)
      @parser.convert(:integer)
    end
    
    # gets us nothing but floats
    assert_equal( [String, String, Float, String, Float],
                  @parser.shift.map { |field| field.class } )
    
    setup  # reset

    # integers have precendance...
    assert_nothing_raised(Exception) do 
      @parser.convert(:integer)
      @parser.convert(:float)
    end
    
    # gives us proper number conversion
    assert_equal( [String, String, Fixnum, String, Float],
                  @parser.shift.map { |field| field.class } )
  end

  def test_builtin_numeric_combo_converter
    # setup parser...
    assert_nothing_raised(Exception) { @parser.convert(:numeric) }
    
    # and use
    assert_equal( [String, String, Fixnum, String, Float],
                  @parser.shift.map { |field| field.class } )
  end

  def test_builtin_all_nested_combo_converter
    # setup parser...
    @data   << ",#{Time.now}"        # add a DateTime field
    @parser =  FasterCSV.new(@data)  # reset parser
    assert_nothing_raised(Exception) { @parser.convert(:all) }
    
    # and use
    assert_equal( [String, String, Fixnum, String, Float, DateTime],
                  @parser.shift.map { |field| field.class } )
  end
  
  def test_convert_with_custom_code
    # define custom converter...
    assert_nothing_raised(Exception) do
      @parser.convert { |field| field =~ /\A:(\S.*?)\s*\Z/ ? $1.to_sym : field }
    end
    
    # and use
    assert_equal(["Numbers", :integer, "1", :float, "3.015"], @parser.shift)

    setup  # reset

    # mix built-in and custom...
    assert_nothing_raised(Exception) { @parser.convert(:numeric) }
    assert_nothing_raised(Exception) { @parser.convert(&@custom) }
    
    # and use
    assert_equal(["Numbers", :integer, 1, :float, 3.015], @parser.shift)
  end
  
  def test_convert_with_custom_code_using_field_info
    # define custom converter that uses field information...
    assert_nothing_raised(Exception) do
      @parser.convert do |field, info|
        assert_equal(1, info.line)
        info.index == 4 ? Float(field).floor : field
      end
    end
    
    # and use
    assert_equal(["Numbers", ":integer", "1", ":float", 3], @parser.shift)
  end
  
  def test_shortcut_interface
    assert_equal( ["Numbers", ":integer", 1, ":float", 3.015],
                  FasterCSV.parse_line(@data, :converters => :numeric) )

    assert_equal( ["Numbers", ":integer", 1, ":float", 3.015],
                  FasterCSV.parse_line( @data, :converters => [ :integer,
                                                                :float ] ) )

    assert_equal( ["Numbers", :integer, 1, :float, 3.015],
                  FasterCSV.parse_line( @data, :converters => [ :numeric,
                                                                @custom ] ) )
  end
end