79 lines
2.2 KiB
Ruby
79 lines
2.2 KiB
Ruby
# frozen_string_literal: false
|
|
|
|
require_relative "../helper"
|
|
|
|
module TestCSVWriteForceQuotes
|
|
def test_default
|
|
assert_equal(%Q[1,2,3#{$INPUT_RECORD_SEPARATOR}],
|
|
generate_line(["1", "2", "3"]))
|
|
end
|
|
|
|
def test_true
|
|
assert_equal(%Q["1","2","3"#{$INPUT_RECORD_SEPARATOR}],
|
|
generate_line(["1", "2", "3"],
|
|
force_quotes: true))
|
|
end
|
|
|
|
def test_false
|
|
assert_equal(%Q[1,2,3#{$INPUT_RECORD_SEPARATOR}],
|
|
generate_line(["1", "2", "3"],
|
|
force_quotes: false))
|
|
end
|
|
|
|
def test_field_name
|
|
assert_equal(%Q["1",2,"3"#{$INPUT_RECORD_SEPARATOR}],
|
|
generate_line(["1", "2", "3"],
|
|
headers: ["a", "b", "c"],
|
|
force_quotes: ["a", :c]))
|
|
end
|
|
|
|
def test_field_name_without_headers
|
|
force_quotes = ["a", "c"]
|
|
error = assert_raise(ArgumentError) do
|
|
generate_line(["1", "2", "3"],
|
|
force_quotes: force_quotes)
|
|
end
|
|
assert_equal(":headers is required when you use field name " +
|
|
"in :force_quotes: " +
|
|
"#{force_quotes.first.inspect}: #{force_quotes.inspect}",
|
|
error.message)
|
|
end
|
|
|
|
def test_field_index
|
|
assert_equal(%Q["1",2,"3"#{$INPUT_RECORD_SEPARATOR}],
|
|
generate_line(["1", "2", "3"],
|
|
force_quotes: [0, 2]))
|
|
end
|
|
|
|
def test_field_unknown
|
|
force_quotes = [1.1]
|
|
error = assert_raise(ArgumentError) do
|
|
generate_line(["1", "2", "3"],
|
|
force_quotes: force_quotes)
|
|
end
|
|
assert_equal(":force_quotes element must be field index or field name: " +
|
|
"#{force_quotes.first.inspect}: #{force_quotes.inspect}",
|
|
error.message)
|
|
end
|
|
end
|
|
|
|
class TestCSVWriteForceQuotesGenerateLine < Test::Unit::TestCase
|
|
include TestCSVWriteForceQuotes
|
|
extend DifferentOFS
|
|
|
|
def generate_line(row, **kwargs)
|
|
CSV.generate_line(row, **kwargs)
|
|
end
|
|
end
|
|
|
|
class TestCSVWriteForceQuotesGenerate < Test::Unit::TestCase
|
|
include TestCSVWriteForceQuotes
|
|
extend DifferentOFS
|
|
|
|
def generate_line(row, **kwargs)
|
|
CSV.generate(**kwargs) do |csv|
|
|
csv << row
|
|
end
|
|
end
|
|
end
|