From 3c028685dfdb23ba3a8c28defa857640b9c93ed1 Mon Sep 17 00:00:00 2001
From: miloyip <miloyip@gmail.com>
Date: Mon, 13 Apr 2015 14:25:05 +0800
Subject: [PATCH] Add tests for Writer API for RAPIDJSON_HAS_STDSTRING

---
 include/rapidjson/prettywriter.h   |  2 +-
 include/rapidjson/writer.h         |  2 +-
 test/unittest/prettywritertest.cpp | 11 +++++++++++
 test/unittest/writertest.cpp       |  9 +++++++++
 4 files changed, 22 insertions(+), 2 deletions(-)

diff --git a/include/rapidjson/prettywriter.h b/include/rapidjson/prettywriter.h
index 90d1983..14b8477 100644
--- a/include/rapidjson/prettywriter.h
+++ b/include/rapidjson/prettywriter.h
@@ -78,7 +78,7 @@ public:
 
 #if RAPIDJSON_HAS_STDSTRING
     bool String(const std::basic_string<Ch>& str) {
-      return String(str.data(), SizeType(str.size()));
+        return String(str.data(), SizeType(str.size()));
     }
 #endif
 
diff --git a/include/rapidjson/writer.h b/include/rapidjson/writer.h
index 5a4d156..bf17f3a 100644
--- a/include/rapidjson/writer.h
+++ b/include/rapidjson/writer.h
@@ -127,7 +127,7 @@ public:
 
 #if RAPIDJSON_HAS_STDSTRING
     bool String(const std::basic_string<Ch>& str) {
-      return String(str.data(), SizeType(str.size()));
+        return String(str.data(), SizeType(str.size()));
     }
 #endif
 
diff --git a/test/unittest/prettywritertest.cpp b/test/unittest/prettywritertest.cpp
index d0abf33..fcb1121 100644
--- a/test/unittest/prettywritertest.cpp
+++ b/test/unittest/prettywritertest.cpp
@@ -79,3 +79,14 @@ TEST(PrettyWriter, SetIndent) {
         "}",
         buffer.GetString());
 }
+
+#if RAPIDJSON_HAS_STDSTRING
+TEST(PrettyWriter, String_STDSTRING) {
+    StringBuffer buffer;
+    PrettyWriter<StringBuffer> writer(buffer);
+    EXPECT_TRUE(writer.StartArray());
+    EXPECT_TRUE(writer.String(std::string("Hello\n")));
+    EXPECT_TRUE(writer.EndArray());
+    EXPECT_STREQ("[\n    \"Hello\\n\"\n]", buffer.GetString());
+}
+#endif
diff --git a/test/unittest/writertest.cpp b/test/unittest/writertest.cpp
index 2a0f484..7b9fa9a 100644
--- a/test/unittest/writertest.cpp
+++ b/test/unittest/writertest.cpp
@@ -89,6 +89,15 @@ TEST(Writer, String) {
     TEST_ROUNDTRIP("[\"Hello\"]");
     TEST_ROUNDTRIP("[\"Hello\\u0000World\"]");
     TEST_ROUNDTRIP("[\"\\\"\\\\/\\b\\f\\n\\r\\t\"]");
+
+#if RAPIDJSON_HAS_STDSTRING
+    {
+        StringBuffer buffer;
+        Writer<StringBuffer> writer(buffer);
+        writer.String(std::string("Hello\n"));
+        EXPECT_STREQ("\"Hello\\n\"", buffer.GetString());
+    }
+#endif
 }
 
 TEST(Writer, Double) {