Use appropriate style for serialized strings

When psych parses YAML with string in the folded style as:

    a: >
      some
      inline
      content

and serializes it back to YAML,
then it renders the string in the literal style instead:

    a: |
      some inline content

This patch fixes this behaviour. When a string doesn't contain newlines
(blank line at the end is ignored) and it's not necessary to be quoted,
then it will use plain style or folded style according to its length and
the `line_width` option.

    # line_width = 100
    a: some inline content

    # line_width = 11
    a: >
      some inline
      content

This change was originally proposed by @Iazel.

1 files changed, 24 insertions, 0 deletions

diff --git a/test/psych/test_string.rb b/test/psych/test_string.rb
index 26a4e20..3188b5c 100644
--- a/test/psych/test_string.rb
+++ b/test/psych/test_string.rb
@@ -34,6 +34,30 @@ module Psych
       assert_match(/---\s*"/, yaml)
     end
 
+    def test_plain_when_shorten_than_line_width
+      str = "Lorem ipsum"
+      yaml = Psych.dump str, line_width: 12
+      assert_match /---\s*[^>|]+\n/, yaml
+    end
+
+    def test_folded_when_longer_than_line_width_and_no_newlines
+      str = "Lorem ipsum dolor sit amet, consectetur"
+      yaml = Psych.dump str, line_width: 12
+      assert_match /---\s*>\n(.*\n){3}\Z/, yaml
+    end
+
+    def test_folded_when_longer_than_line_width_and_trailing_newline
+      str = "Lorem ipsum dolor sit\n"
+      yaml = Psych.dump str, line_width: 12
+      assert_match /---\s*>\n(.*\n){2}\Z/, yaml
+    end
+
+    def test_literal_when_inner_newline
+      str = "Lorem ipsum\ndolor\n"
+      yaml = Psych.dump str, line_width: 12
+      assert_match /---\s*|\n(.*\n){2}\Z/, yaml
+    end
+
     def test_cycle_x
       str = X.new 'abc'
       assert_cycle str