From 178649e6dcb679f4b42cbf723d4f34f81a591304 Mon Sep 17 00:00:00 2001
From: Sutou Kouhei <kou@clear-code.com>
Date: Thu, 16 Jul 2020 06:10:38 +0900
Subject: [ruby/csv] force_quotes: add support for specifying the target
 indexes or names

GitHub: fix GH-153

Reported by Aleksandr. Thanks!!!

https://github.com/ruby/csv/commit/8812c58a26
---
 test/csv/write/test_force_quotes.rb | 78 +++++++++++++++++++++++++++++++++++++
 1 file changed, 78 insertions(+)
 create mode 100644 test/csv/write/test_force_quotes.rb

(limited to 'test/csv/write/test_force_quotes.rb')

diff --git a/test/csv/write/test_force_quotes.rb b/test/csv/write/test_force_quotes.rb
new file mode 100644
index 0000000000..622dcb021b
--- /dev/null
+++ b/test/csv/write/test_force_quotes.rb
@@ -0,0 +1,78 @@
+# 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
-- 
cgit v1.2.3