1
0
Fork 0
mirror of https://github.com/ruby/ruby.git synced 2022-11-09 12:17:21 -05:00
ruby--ruby/test/csv/write/test_general.rb

259 lines
7.1 KiB
Ruby
Raw Normal View History

# -*- coding: utf-8 -*-
# frozen_string_literal: false
require_relative "../helper"
module TestCSVWriteGeneral
def test_tab
assert_equal("\t#{$INPUT_RECORD_SEPARATOR}",
generate_line(["\t"]))
end
def test_quote_character
assert_equal(%Q[foo,"""",baz#{$INPUT_RECORD_SEPARATOR}],
generate_line(["foo", %Q["], "baz"]))
end
def test_quote_character_double
assert_equal(%Q[foo,"""""",baz#{$INPUT_RECORD_SEPARATOR}],
generate_line(["foo", %Q[""], "baz"]))
end
def test_quote
assert_equal(%Q[foo,"""bar""",baz#{$INPUT_RECORD_SEPARATOR}],
generate_line(["foo", %Q["bar"], "baz"]))
end
def test_quote_lf
assert_equal(%Q["""\n","""\n"#{$INPUT_RECORD_SEPARATOR}],
generate_line([%Q["\n], %Q["\n]]))
end
def test_quote_cr
assert_equal(%Q["""\r","""\r"#{$INPUT_RECORD_SEPARATOR}],
generate_line([%Q["\r], %Q["\r]]))
end
def test_quote_last
assert_equal(%Q[foo,"bar"""#{$INPUT_RECORD_SEPARATOR}],
generate_line(["foo", %Q[bar"]]))
end
def test_quote_lf_last
assert_equal(%Q[foo,"\nbar"""#{$INPUT_RECORD_SEPARATOR}],
generate_line(["foo", %Q[\nbar"]]))
end
def test_quote_lf_value_lf
assert_equal(%Q[foo,"""\nbar\n"""#{$INPUT_RECORD_SEPARATOR}],
generate_line(["foo", %Q["\nbar\n"]]))
end
def test_quote_lf_value_lf_nil
assert_equal(%Q[foo,"""\nbar\n""",#{$INPUT_RECORD_SEPARATOR}],
generate_line(["foo", %Q["\nbar\n"], nil]))
end
def test_cr
assert_equal(%Q[foo,"\r",baz#{$INPUT_RECORD_SEPARATOR}],
generate_line(["foo", "\r", "baz"]))
end
def test_lf
assert_equal(%Q[foo,"\n",baz#{$INPUT_RECORD_SEPARATOR}],
generate_line(["foo", "\n", "baz"]))
end
def test_cr_lf
assert_equal(%Q[foo,"\r\n",baz#{$INPUT_RECORD_SEPARATOR}],
generate_line(["foo", "\r\n", "baz"]))
end
def test_cr_dot_lf
assert_equal(%Q[foo,"\r.\n",baz#{$INPUT_RECORD_SEPARATOR}],
generate_line(["foo", "\r.\n", "baz"]))
end
def test_cr_lf_cr
assert_equal(%Q[foo,"\r\n\r",baz#{$INPUT_RECORD_SEPARATOR}],
generate_line(["foo", "\r\n\r", "baz"]))
end
def test_cr_lf_lf
assert_equal(%Q[foo,"\r\n\n",baz#{$INPUT_RECORD_SEPARATOR}],
generate_line(["foo", "\r\n\n", "baz"]))
end
def test_cr_lf_comma
assert_equal(%Q["\r\n,"#{$INPUT_RECORD_SEPARATOR}],
generate_line(["\r\n,"]))
end
def test_cr_lf_comma_nil
assert_equal(%Q["\r\n,",#{$INPUT_RECORD_SEPARATOR}],
generate_line(["\r\n,", nil]))
end
def test_comma
assert_equal(%Q[","#{$INPUT_RECORD_SEPARATOR}],
generate_line([","]))
end
def test_comma_double
assert_equal(%Q[",",","#{$INPUT_RECORD_SEPARATOR}],
generate_line([",", ","]))
end
def test_comma_and_value
assert_equal(%Q[foo,"foo,bar",baz#{$INPUT_RECORD_SEPARATOR}],
generate_line(["foo", "foo,bar", "baz"]))
end
def test_one_element
assert_equal(%Q[foo#{$INPUT_RECORD_SEPARATOR}],
generate_line(["foo"]))
end
def test_nil_values_only
assert_equal(%Q[,,#{$INPUT_RECORD_SEPARATOR}],
generate_line([nil, nil, nil]))
end
def test_nil_double_only
assert_equal(%Q[,#{$INPUT_RECORD_SEPARATOR}],
generate_line([nil, nil]))
end
def test_nil_values
assert_equal(%Q[foo,,,#{$INPUT_RECORD_SEPARATOR}],
generate_line(["foo", nil, nil, nil]))
end
def test_nil_value_first
assert_equal(%Q[,foo,baz#{$INPUT_RECORD_SEPARATOR}],
generate_line([nil, "foo", "baz"]))
end
def test_nil_value_middle
assert_equal(%Q[foo,,baz#{$INPUT_RECORD_SEPARATOR}],
generate_line(["foo", nil, "baz"]))
end
def test_nil_value_last
assert_equal(%Q[foo,baz,#{$INPUT_RECORD_SEPARATOR}],
generate_line(["foo", "baz", nil]))
end
def test_nil_empty
assert_equal(%Q[,""#{$INPUT_RECORD_SEPARATOR}],
generate_line([nil, ""]))
end
def test_nil_cr
assert_equal(%Q[,"\r"#{$INPUT_RECORD_SEPARATOR}],
generate_line([nil, "\r"]))
end
def test_values
assert_equal(%Q[foo,bar#{$INPUT_RECORD_SEPARATOR}],
generate_line(["foo", "bar"]))
end
def test_semi_colon
assert_equal(%Q[;#{$INPUT_RECORD_SEPARATOR}],
generate_line([";"]))
end
def test_semi_colon_values
assert_equal(%Q[;,;#{$INPUT_RECORD_SEPARATOR}],
generate_line([";", ";"]))
end
def test_tab_values
assert_equal(%Q[\t,\t#{$INPUT_RECORD_SEPARATOR}],
generate_line(["\t", "\t"]))
end
def test_col_sep
assert_equal(%Q[a;b;;c#{$INPUT_RECORD_SEPARATOR}],
generate_line(["a", "b", nil, "c"],
col_sep: ";"))
assert_equal(%Q[a\tb\t\tc#{$INPUT_RECORD_SEPARATOR}],
generate_line(["a", "b", nil, "c"],
col_sep: "\t"))
end
def test_row_sep
assert_equal(%Q[a,b,,c\r\n],
generate_line(["a", "b", nil, "c"],
row_sep: "\r\n"))
end
def test_force_quotes
assert_equal(%Q["1","b","","already ""quoted"""#{$INPUT_RECORD_SEPARATOR}],
generate_line([1, "b", nil, %Q{already "quoted"}],
force_quotes: true))
end
def test_encoding_utf8
assert_equal(%Q[あ,い,う#{$INPUT_RECORD_SEPARATOR}],
generate_line(["" , "", ""]))
end
def test_encoding_euc_jp
row = ["", "", ""].collect {|field| field.encode("EUC-JP")}
assert_equal(%Q[あ,い,う#{$INPUT_RECORD_SEPARATOR}].encode("EUC-JP"),
generate_line(row))
end
2019-10-12 01:03:21 -04:00
def test_encoding_with_default_internal
with_default_internal(Encoding::UTF_8) do
row = ["", "", ""].collect {|field| field.encode("EUC-JP")}
assert_equal(%Q[あ,い,う#{$INPUT_RECORD_SEPARATOR}].encode("EUC-JP"),
generate_line(row, encoding: Encoding::EUC_JP))
end
end
def test_with_default_internal
with_default_internal(Encoding::UTF_8) do
row = ["", "", ""].collect {|field| field.encode("EUC-JP")}
assert_equal(%Q[あ,い,う#{$INPUT_RECORD_SEPARATOR}].encode("EUC-JP"),
generate_line(row))
end
end
def with_default_internal(encoding)
original = Encoding.default_internal
begin
$VERBOSE, verbose_bak = nil, $VERBOSE
2019-10-12 01:03:21 -04:00
Encoding.default_internal = encoding
$VERBOSE = verbose_bak
2019-10-12 01:03:21 -04:00
yield
ensure
$VERBOSE, verbose_bak = nil, $VERBOSE
2019-10-12 01:03:21 -04:00
Encoding.default_internal = original
$VERBOSE = verbose_bak
2019-10-12 01:03:21 -04:00
end
end
end
class TestCSVWriteGeneralGenerateLine < Test::Unit::TestCase
include TestCSVWriteGeneral
extend DifferentOFS
def generate_line(row, **kwargs)
CSV.generate_line(row, **kwargs)
end
end
class TestCSVWriteGeneralGenerate < Test::Unit::TestCase
include TestCSVWriteGeneral
extend DifferentOFS
def generate_line(row, **kwargs)
CSV.generate(**kwargs) do |csv|
csv << row
end
end
end