commit d7b25588ce4b19a94885c21985521ad9ee122bed
author Andrey Somov <public.somov@gmail.com> Sun May 05 18:08:38 2013 +0200
committer Andrey Somov <public.somov@gmail.com> Sun May 05 18:08:38 2013 +0200
tree 071573855800d0613ce860aa565282bc4ff57de6
parent c5d5148e06ad9457c6bc64783bec4d380843c86d

Improve SingleQuoteTest for issue 176

src/test/java/org/yaml/snakeyaml/issues/issue176/SingleQuoteTest.java

diff --git a/src/test/java/org/yaml/snakeyaml/issues/issue176/SingleQuoteTest.java b/src/test/java/org/yaml/snakeyaml/issues/issue176/SingleQuoteTest.java
index 48a5a6c..be75d30 100644
--- a/src/test/java/org/yaml/snakeyaml/issues/issue176/SingleQuoteTest.java
+++ b/src/test/java/org/yaml/snakeyaml/issues/issue176/SingleQuoteTest.java
@@ -27,12 +27,21 @@
 
 public class SingleQuoteTest extends TestCase {
 
-    public void testSingleQuote() throws Exception {
+    public void testNoSingleQuoteForBlockStyle() throws Exception {
+        checkQuotes(true, "cows:\n    steak:cow: '11'");
+    }
+
+    public void testSingleQuoteForFlowStyle() throws Exception {
+        checkQuotes(false, "cows: {'steak:cow': '11'}");
+    }
+
+    private void checkQuotes(boolean isBlock, String expectation) {
         DumperOptions options = new DumperOptions();
         options.setIndent(4);
-        options.setDefaultFlowStyle(FlowStyle.BLOCK);
-        Representer representer = new org.yaml.snakeyaml.representer.Representer();
-        representer.setDefaultFlowStyle(FlowStyle.BLOCK);
+        if (isBlock) {
+            options.setDefaultFlowStyle(FlowStyle.BLOCK);
+        }
+        Representer representer = new Representer();
 
         Yaml yaml = new Yaml(new SafeConstructor(), representer, options);
 
@@ -41,6 +50,6 @@
         LinkedHashMap<String, Object> root = new LinkedHashMap<String, Object>();
         root.put("cows", lvl1);
         String output = yaml.dump(root);
-        assertEquals("cows:\n    steak:cow: '11'\n", output);
+        assertEquals(expectation + "\n", output);
     }
 }