diff options
Diffstat (limited to 'test/csv/parse/test_skip_lines.rb')
-rw-r--r-- | test/csv/parse/test_skip_lines.rb | 105 |
1 files changed, 105 insertions, 0 deletions
diff --git a/test/csv/parse/test_skip_lines.rb b/test/csv/parse/test_skip_lines.rb new file mode 100644 index 0000000000..196858f1b0 --- /dev/null +++ b/test/csv/parse/test_skip_lines.rb @@ -0,0 +1,105 @@ +# frozen_string_literal: false + +require_relative "../helper" + +class TestCSVParseSkipLines < Test::Unit::TestCase + extend DifferentOFS + include Helper + + def test_default + csv = CSV.new("a,b,c\n") + assert_nil(csv.skip_lines) + end + + def test_regexp + csv = <<-CSV +1 +#2 + #3 +4 + CSV + assert_equal([ + ["1"], + ["4"], + ], + CSV.parse(csv, :skip_lines => /\A\s*#/)) + end + + def test_regexp_quoted + csv = <<-CSV +1 +#2 +"#3" +4 + CSV + assert_equal([ + ["1"], + ["#3"], + ["4"], + ], + CSV.parse(csv, :skip_lines => /\A\s*#/)) + end + + def test_string + csv = <<-CSV +1 +.2 +3. +4 + CSV + assert_equal([ + ["1"], + ["4"], + ], + CSV.parse(csv, :skip_lines => ".")) + end + + class RegexStub + end + + def test_not_matchable + regex_stub = RegexStub.new + csv = CSV.new("1\n", :skip_lines => regex_stub) + error = assert_raise(ArgumentError) do + csv.shift + end + assert_equal(":skip_lines has to respond to #match: #{regex_stub.inspect}", + error.message) + end + + class Matchable + def initialize(pattern) + @pattern = pattern + end + + def match(line) + @pattern.match(line) + end + end + + def test_matchable + csv = <<-CSV +1 +# 2 +3 +# 4 + CSV + assert_equal([ + ["1"], + ["3"], + ], + CSV.parse(csv, :skip_lines => Matchable.new(/\A#/))) + end + + def test_multibyte_data + # U+3042 HIRAGANA LETTER A + # U+3044 HIRAGANA LETTER I + # U+3046 HIRAGANA LETTER U + value = "\u3042\u3044\u3046" + with_chunk_size("5") do + assert_equal([[value], [value]], + CSV.parse("#{value}\n#{value}\n", + :skip_lines => /\A#/)) + end + end +end |