require_relative "base"
class TestCSV::Headers < TestCSV
extend DifferentOFS
def setup
super
@data = <<-END_CSV.gsub(/^\s+/, "")
first,second,third
A,B,C
1,2,3
END_CSV
end
def test_first_row
[:first_row, true].each do |setting| csv = nil
assert_nothing_raised(Exception) do
csv = CSV.parse(@data, headers: setting)
end
row = csv[0]
assert_not_nil(row)
assert_instance_of(CSV::Row, row)
assert_equal([%w{first A}, %w{second B}, %w{third C}], row.to_a)
row = csv[1]
assert_not_nil(row)
assert_instance_of(CSV::Row, row)
assert_equal([%w{first 1}, %w{second 2}, %w{third 3}], row.to_a)
assert_nil(csv[2])
end
end
def test_array_of_headers
csv = nil
assert_nothing_raised(Exception) do
csv = CSV.parse(@data, headers: [:my, :new, :headers])
end
row = csv[0]
assert_not_nil(row)
assert_instance_of(CSV::Row, row)
assert_equal( [[:my, "first"], [:new, "second"], [:headers, "third"]],
row.to_a )
row = csv[1]
assert_not_nil(row)
assert_instance_of(CSV::Row, row)
assert_equal([[:my, "A"], [:new, "B"], [:headers, "C"]], row.to_a)
row = csv[2]
assert_not_nil(row)
assert_instance_of(CSV::Row, row)
assert_equal([[:my, "1"], [:new, "2"], [:headers, "3"]], row.to_a)
assert_nil(csv[3])
assert_nothing_raised(Exception) do
csv = CSV.parse( @data, headers: [:my, :new, :headers],
return_headers: true,
header_converters: lambda { |h| h.to_s } )
end
row = csv[0]
assert_not_nil(row)
assert_instance_of(CSV::Row, row)
assert_equal([["my", :my], ["new", :new], ["headers", :headers]], row.to_a)
assert(row.header_row?)
assert(!row.field_row?)
end
def test_csv_header_string
csv = nil
assert_nothing_raised(Exception) do
csv = CSV.parse(@data, headers: "my,new,headers")
end
row = csv[0]
assert_not_nil(row)
assert_instance_of(CSV::Row, row)
assert_equal([%w{my first}, %w{new second}, %w{headers third}], row.to_a)
row = csv[1]
assert_not_nil(row)
assert_instance_of(CSV::Row, row)
assert_equal([%w{my A}, %w{new B}, %w{headers C}], row.to_a)
row = csv[2]
assert_not_nil(row)
assert_instance_of(CSV::Row, row)
assert_equal([%w{my 1}, %w{new 2}, %w{headers 3}], row.to_a)
assert_nil(csv[3])
assert_nothing_raised(Exception) do
csv = CSV.parse( @data, headers: "my,new,headers",
return_headers: true,
header_converters: :symbol )
end
row = csv[0]
assert_not_nil(row)
assert_instance_of(CSV::Row, row)
assert_equal([[:my, "my"], [:new, "new"], [:headers, "headers"]], row.to_a)
assert(row.header_row?)
assert(!row.field_row?)
end
def test_csv_header_string_inherits_separators
csv = nil
assert_nothing_raised(Exception) do
csv = CSV.parse( @data.tr(",", "|"), col_sep: "|",
headers: "my|new|headers" )
end
row = csv[0]
assert_not_nil(row)
assert_instance_of(CSV::Row, row)
assert_equal([%w{my first}, %w{new second}, %w{headers third}], row.to_a)
end
def test_return_headers
csv = nil
assert_nothing_raised(Exception) do
csv = CSV.parse(@data, headers: true, return_headers: true)
end
row = csv[0]
assert_not_nil(row)
assert_instance_of(CSV::Row, row)
assert_equal( [%w{first first}, %w{second second}, %w{third third}],
row.to_a )
assert(row.header_row?)
assert(!row.field_row?)
row = csv[1]
assert_not_nil(row)
assert_instance_of(CSV::Row, row)
assert_equal([%w{first A}, %w{second B}, %w{third C}], row.to_a)
assert(!row.header_row?)
assert(row.field_row?)
row = csv[2]
assert_not_nil(row)
assert_instance_of(CSV::Row, row)
assert_equal([%w{first 1}, %w{second 2}, %w{third 3}], row.to_a)
assert(!row.header_row?)
assert(row.field_row?)
assert_nil(csv[3])
end
def test_converters
data = <<-END_MATCHING_CSV.gsub(/^\s+/, "")
1,2,3
1,2,3
END_MATCHING_CSV
csv = CSV.parse( data, headers: true,
return_headers: true,
converters: :numeric )
assert_equal([%w{1 1}, %w{2 2}, %w{3 3}], csv[0].to_a)
assert_equal([["1", 1], ["2", 2], ["3", 3]], csv[1].to_a)
assert_nil(csv[2])
assert_nothing_raised(Exception) do
csv = CSV.parse( data, headers: true,
return_headers: true,
converters: :numeric,
header_converters: :symbol )
end
assert_equal([[:"1", "1"], [:"2", "2"], [:"3", "3"]], csv[0].to_a)
assert_equal([[:"1", 1], [:"2", 2], [:"3", 3]], csv[1].to_a)
assert_nil(csv[2])
end
def test_builtin_downcase_converter
csv = CSV.parse( "One,TWO Three", headers: true,
return_headers: true,
header_converters: :downcase )
assert_equal(%w{one two\ three}, csv.headers)
end
def test_builtin_symbol_converter
csv = CSV.parse( "One,TWO Three", headers: true,
return_headers: true,
header_converters: :symbol )
assert_equal([:one, :two_three], csv.headers)
end
def test_custom_converter
converter = lambda { |header| header.tr(" ", "_") }
csv = CSV.parse( "One,TWO Three",
headers: true,
return_headers: true,
header_converters: converter )
assert_equal(%w{One TWO_Three}, csv.headers)
end
def test_table_support
csv = nil
assert_nothing_raised(Exception) do
csv = CSV.parse(@data, headers: true)
end
assert_instance_of(CSV::Table, csv)
end
def test_skip_blanks
@data = <<-END_CSV.gsub(/^ +/, "")
A,B,C
1,2,3
END_CSV
expected = [%w[1 2 3]]
CSV.parse(@data, headers: true, skip_blanks: true) do |row|
assert_equal(expected.shift, row.fields)
end
expected = [%w[A B C], %w[1 2 3]]
CSV.parse( @data,
headers: true,
return_headers: true,
skip_blanks: true ) do |row|
assert_equal(expected.shift, row.fields)
end
end
def test_headers_reader
assert_nil(CSV.new(@data).headers)
csv = CSV.new(@data, headers: true)
assert_equal(true, csv.headers) csv.shift assert_equal(%w[first second third], csv.headers) end
def test_blank_row_bug_fix
@data += "\n#{@data}"
CSV.parse(@data, headers: true) do |row|
assert_instance_of(CSV::Row, row)
end
end
end