ruby--ruby/test/csv/write/test_general.rb

# -*- 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

  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
      Encoding.default_internal = encoding
      $VERBOSE = verbose_bak
      yield
    ensure
      $VERBOSE, verbose_bak = nil, $VERBOSE
      Encoding.default_internal = original
      $VERBOSE = verbose_bak
    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