summaryrefslogtreecommitdiff
path: root/test/csv/write/test_converters.rb
blob: 0e0080b4c5cb73bcbfecb30b6958c6ac6c092f60 (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
41
42
43
44
45
46
47
48
49
50
51
52
53
# -*- coding: utf-8 -*-
# frozen_string_literal: false

require_relative "../helper"

module TestCSVWriteConverters
  def test_one
    assert_equal(%Q[=a,=b,=c\n],
                 generate_line(["a", "b", "c"],
                               write_converters: ->(value) {"=" + value}))
  end

  def test_multiple
    assert_equal(%Q[=a_,=b_,=c_\n],
                 generate_line(["a", "b", "c"],
                               write_converters: [
                                 ->(value) {"=" + value},
                                 ->(value) {value + "_"},
                               ]))
  end

  def test_nil_value
    assert_equal(%Q[a,NaN,29\n],
                 generate_line(["a", nil, 29],
                               write_nil_value: "NaN"))
  end

  def test_empty_value
    assert_equal(%Q[a,,29\n],
                 generate_line(["a", "", 29],
                               write_empty_value: nil))
  end
end

class TestCSVWriteConvertersGenerateLine < Test::Unit::TestCase
  include TestCSVWriteConverters
  extend DifferentOFS

  def generate_line(row, **kwargs)
    CSV.generate_line(row, **kwargs)
  end
end

class TestCSVWriteConvertersGenerate < Test::Unit::TestCase
  include TestCSVWriteConverters
  extend DifferentOFS

  def generate_line(row, **kwargs)
    CSV.generate(**kwargs) do |csv|
      csv << row
    end
  end
end