diff options
Diffstat (limited to 'test/csv/tc_features.rb')
| -rw-r--r-- | test/csv/tc_features.rb | 177 |
1 files changed, 177 insertions, 0 deletions
diff --git a/test/csv/tc_features.rb b/test/csv/tc_features.rb new file mode 100644 index 0000000000..6766839e10 --- /dev/null +++ b/test/csv/tc_features.rb @@ -0,0 +1,177 @@ +#!/usr/local/bin/ruby -w + +# tc_features.rb +# +# Created by James Edward Gray II on 2005-10-31. +# Copyright 2005 James Edward Gray II. You can redistribute or modify this code +# under the terms of Ruby's license. + +require "test/unit" +require "zlib" + +require "csv" + +class TestCSVFeatures < Test::Unit::TestCase + TEST_CASES = [ [%Q{a,b}, ["a", "b"]], + [%Q{a,"""b"""}, ["a", "\"b\""]], + [%Q{a,"""b"}, ["a", "\"b"]], + [%Q{a,"b"""}, ["a", "b\""]], + [%Q{a,"\nb"""}, ["a", "\nb\""]], + [%Q{a,"""\nb"}, ["a", "\"\nb"]], + [%Q{a,"""\nb\n"""}, ["a", "\"\nb\n\""]], + [%Q{a,"""\nb\n""",\nc}, ["a", "\"\nb\n\"", nil]], + [%Q{a,,,}, ["a", nil, nil, nil]], + [%Q{,}, [nil, nil]], + [%Q{"",""}, ["", ""]], + [%Q{""""}, ["\""]], + [%Q{"""",""}, ["\"",""]], + [%Q{,""}, [nil,""]], + [%Q{,"\r"}, [nil,"\r"]], + [%Q{"\r\n,"}, ["\r\n,"]], + [%Q{"\r\n,",}, ["\r\n,", nil]] ] + + def setup + @sample_data = <<-END_DATA.gsub(/^ +/, "") + line,1,abc + line,2,"def\nghi" + + line,4,jkl + END_DATA + @csv = CSV.new(@sample_data) + end + + def test_col_sep + [";", "\t"].each do |sep| + TEST_CASES.each do |test_case| + assert_equal( test_case.last.map { |t| t.tr(",", sep) unless t.nil? }, + CSV.parse_line( test_case.first.tr(",", sep), + :col_sep => sep ) ) + end + end + assert_equal([",,,", nil], CSV.parse_line(",,,;", :col_sep => ";")) + end + + def test_row_sep + assert_raise(CSV::MalformedCSVError) do + CSV.parse_line("1,2,3\n,4,5\r\n", :row_sep => "\r\n") + end + assert_equal( ["1", "2", "3\n", "4", "5"], + CSV.parse_line(%Q{1,2,"3\n",4,5\r\n}, :row_sep => "\r\n")) + end + + def test_quote_char + TEST_CASES.each do |test_case| + assert_equal( test_case.last.map { |t| t.tr('"', "'") unless t.nil? }, + CSV.parse_line( test_case.first.tr('"', "'"), + :quote_char => "'" ) ) + end + end + + def test_row_sep_auto_discovery + ["\r\n", "\n", "\r"].each do |line_end| + data = "1,2,3#{line_end}4,5#{line_end}" + discovered = CSV.new(data).instance_eval { @row_sep } + assert_equal(line_end, discovered) + end + + assert_equal("\n", CSV.new("\n\r\n\r").instance_eval { @row_sep }) + + assert_equal($/, CSV.new("").instance_eval { @row_sep }) + + assert_equal($/, CSV.new(STDERR).instance_eval { @row_sep }) + end + + def test_lineno + assert_equal(5, @sample_data.to_a.size) + + 4.times do |line_count| + assert_equal(line_count, @csv.lineno) + assert_not_nil(@csv.shift) + assert_equal(line_count + 1, @csv.lineno) + end + assert_nil(@csv.shift) + end + + def test_readline + test_lineno + + @csv.rewind + + test_lineno + end + + def test_unknown_options + assert_raise(ArgumentError) { CSV.new(String.new, :unknown => :error) } + end + + def test_skip_blanks + assert_equal(4, @csv.to_a.size) + + @csv = CSV.new(@sample_data, :skip_blanks => true) + + count = 0 + @csv.each do |row| + count += 1 + assert_equal("line", row.first) + end + assert_equal(3, count) + end + + # reported by Kev Jackson + def test_failing_to_escape_col_sep_bug_fix + assert_nothing_raised(Exception) { CSV.new(String.new, :col_sep => "|") } + end + + # reported by Chris Roos + def test_failing_to_reset_headers_in_rewind_bug_fix + csv = CSV.new("forename,surname", :headers => true, :return_headers => true) + csv.each { |row| assert row.header_row? } + csv.rewind + csv.each { |row| assert row.header_row? } + end + + # reported by Dave Burt + def test_leading_empty_fields_with_multibyte_col_sep_bug_fix + data = <<-END_DATA.gsub(/^\s+/, "") + <=><=>A<=>B<=>C + 1<=>2<=>3 + END_DATA + parsed = CSV.parse(data, :col_sep => "<=>") + assert_equal([[nil, nil, "A", "B", "C"], ["1", "2", "3"]], parsed) + end + + def test_gzip_reader_bug_fix + zipped = nil + assert_nothing_raised(NoMethodError) do + zipped = CSV.new( + Zlib::GzipReader.open( + File.join(File.dirname(__FILE__), "line_endings.gz") + ) + ) + end + assert_equal("\r\n", zipped.instance_eval { @row_sep }) + end + + def test_gzip_writer_bug_fix + file = File.join(File.dirname(__FILE__), "temp.gz") + zipped = nil + assert_nothing_raised(NoMethodError) do + zipped = CSV.new(Zlib::GzipWriter.open(file)) + end + zipped << %w[one two three] + zipped << [1, 2, 3] + zipped.close + + assert( Zlib::GzipReader.open(file) { |f| f.read }. + include?($INPUT_RECORD_SEPARATOR), + "@row_sep did not default" ) + File.unlink(file) + end + + def test_version + assert_not_nil(CSV::VERSION) + assert_instance_of(String, CSV::VERSION) + assert(CSV::VERSION.frozen?) + assert_match(/\A\d\.\d\.\d\Z/, CSV::VERSION) + end +end |