summaryrefslogtreecommitdiff
path: root/test/csv/parse/test_rewind.rb
blob: 73a69e9ccdd98a52719cee98cf1b295206e034d7 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
# -*- coding: utf-8 -*-
# frozen_string_literal: false

require_relative "../helper"

class TestCSVParseRewind < Test::Unit::TestCase
  extend DifferentOFS

  def parse(data, options={})
    csv = CSV.new(data, options)
    records = csv.to_a
    csv.rewind
    [records, csv.to_a]
  end

  def test_default
    data = <<-CSV
Ruby,2.6.0,script
    CSV
    assert_equal([
                   [["Ruby", "2.6.0", "script"]],
                   [["Ruby", "2.6.0", "script"]],
                 ],
                 parse(data))
  end

  def test_have_headers
    data = <<-CSV
Language,Version,Type
Ruby,2.6.0,script
    CSV
    assert_equal([
                   [CSV::Row.new(["Language", "Version", "Type"],
                                 ["Ruby", "2.6.0", "script"])],
                   [CSV::Row.new(["Language", "Version", "Type"],
                                 ["Ruby", "2.6.0", "script"])],
                 ],
                 parse(data, headers: true))
  end
end