| // Tencent is pleased to support the open source community by making RapidJSON available. |
| // |
| // Copyright (C) 2015 THL A29 Limited, a Tencent company, and Milo Yip. All rights reserved. |
| // |
| // Licensed under the MIT License (the "License"); you may not use this file except |
| // in compliance with the License. You may obtain a copy of the License at |
| // |
| // http://opensource.org/licenses/MIT |
| // |
| // Unless required by applicable law or agreed to in writing, software distributed |
| // under the License is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES OR |
| // CONDITIONS OF ANY KIND, either express or implied. See the License for the |
| // specific language governing permissions and limitations under the License. |
| |
| #include "unittest.h" |
| #include "rapidjson/reader.h" |
| #include "rapidjson/prettywriter.h" |
| #include "rapidjson/stringbuffer.h" |
| #include "rapidjson/filewritestream.h" |
| |
| #ifdef __clang__ |
| RAPIDJSON_DIAG_PUSH |
| RAPIDJSON_DIAG_OFF(c++98-compat) |
| #endif |
| |
| using namespace rapidjson; |
| |
| static const char kJson[] = "{\"hello\":\"world\",\"t\":true,\"f\":false,\"n\":null,\"i\":123,\"pi\":3.1416,\"a\":[1,2,3,-1],\"u64\":1234567890123456789,\"i64\":-1234567890123456789}"; |
| static const char kPrettyJson[] = |
| "{\n" |
| " \"hello\": \"world\",\n" |
| " \"t\": true,\n" |
| " \"f\": false,\n" |
| " \"n\": null,\n" |
| " \"i\": 123,\n" |
| " \"pi\": 3.1416,\n" |
| " \"a\": [\n" |
| " 1,\n" |
| " 2,\n" |
| " 3,\n" |
| " -1\n" |
| " ],\n" |
| " \"u64\": 1234567890123456789,\n" |
| " \"i64\": -1234567890123456789\n" |
| "}"; |
| |
| static const char kPrettyJson_FormatOptions_SLA[] = |
| "{\n" |
| " \"hello\": \"world\",\n" |
| " \"t\": true,\n" |
| " \"f\": false,\n" |
| " \"n\": null,\n" |
| " \"i\": 123,\n" |
| " \"pi\": 3.1416,\n" |
| " \"a\": [1, 2, 3, -1],\n" |
| " \"u64\": 1234567890123456789,\n" |
| " \"i64\": -1234567890123456789\n" |
| "}"; |
| |
| TEST(PrettyWriter, Basic) { |
| StringBuffer buffer; |
| PrettyWriter<StringBuffer> writer(buffer); |
| Reader reader; |
| StringStream s(kJson); |
| reader.Parse(s, writer); |
| EXPECT_STREQ(kPrettyJson, buffer.GetString()); |
| } |
| |
| TEST(PrettyWriter, FormatOptions) { |
| StringBuffer buffer; |
| PrettyWriter<StringBuffer> writer(buffer); |
| writer.SetFormatOptions(kFormatSingleLineArray); |
| Reader reader; |
| StringStream s(kJson); |
| reader.Parse(s, writer); |
| EXPECT_STREQ(kPrettyJson_FormatOptions_SLA, buffer.GetString()); |
| } |
| |
| TEST(PrettyWriter, SetIndent) { |
| StringBuffer buffer; |
| PrettyWriter<StringBuffer> writer(buffer); |
| writer.SetIndent('\t', 1); |
| Reader reader; |
| StringStream s(kJson); |
| reader.Parse(s, writer); |
| EXPECT_STREQ( |
| "{\n" |
| "\t\"hello\": \"world\",\n" |
| "\t\"t\": true,\n" |
| "\t\"f\": false,\n" |
| "\t\"n\": null,\n" |
| "\t\"i\": 123,\n" |
| "\t\"pi\": 3.1416,\n" |
| "\t\"a\": [\n" |
| "\t\t1,\n" |
| "\t\t2,\n" |
| "\t\t3,\n" |
| "\t\t-1\n" |
| "\t],\n" |
| "\t\"u64\": 1234567890123456789,\n" |
| "\t\"i64\": -1234567890123456789\n" |
| "}", |
| buffer.GetString()); |
| } |
| |
| TEST(PrettyWriter, String) { |
| StringBuffer buffer; |
| PrettyWriter<StringBuffer> writer(buffer); |
| EXPECT_TRUE(writer.StartArray()); |
| EXPECT_TRUE(writer.String("Hello\n")); |
| EXPECT_TRUE(writer.EndArray()); |
| EXPECT_STREQ("[\n \"Hello\\n\"\n]", 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 |
| |
| #include <sstream> |
| |
| class OStreamWrapper { |
| public: |
| typedef char Ch; |
| |
| OStreamWrapper(std::ostream& os) : os_(os) {} |
| |
| Ch Peek() const { assert(false); return '\0'; } |
| Ch Take() { assert(false); return '\0'; } |
| size_t Tell() const { return 0; } |
| |
| Ch* PutBegin() { assert(false); return 0; } |
| void Put(Ch c) { os_.put(c); } |
| void Flush() { os_.flush(); } |
| size_t PutEnd(Ch*) { assert(false); return 0; } |
| |
| private: |
| OStreamWrapper(const OStreamWrapper&); |
| OStreamWrapper& operator=(const OStreamWrapper&); |
| |
| std::ostream& os_; |
| }; |
| |
| // For covering PutN() generic version |
| TEST(PrettyWriter, OStreamWrapper) { |
| StringStream s(kJson); |
| |
| std::stringstream ss; |
| OStreamWrapper os(ss); |
| |
| PrettyWriter<OStreamWrapper> writer(os); |
| |
| Reader reader; |
| reader.Parse(s, writer); |
| |
| std::string actual = ss.str(); |
| EXPECT_STREQ(kPrettyJson, actual.c_str()); |
| } |
| |
| // For covering FileWriteStream::PutN() |
| TEST(PrettyWriter, FileWriteStream) { |
| char filename[L_tmpnam]; |
| FILE* fp = TempFile(filename); |
| ASSERT_TRUE(fp!=NULL); |
| char buffer[16]; |
| FileWriteStream os(fp, buffer, sizeof(buffer)); |
| PrettyWriter<FileWriteStream> writer(os); |
| Reader reader; |
| StringStream s(kJson); |
| reader.Parse(s, writer); |
| fclose(fp); |
| |
| fp = fopen(filename, "rb"); |
| fseek(fp, 0, SEEK_END); |
| size_t size = static_cast<size_t>(ftell(fp)); |
| fseek(fp, 0, SEEK_SET); |
| char* json = static_cast<char*>(malloc(size + 1)); |
| size_t readLength = fread(json, 1, size, fp); |
| json[readLength] = '\0'; |
| fclose(fp); |
| remove(filename); |
| EXPECT_STREQ(kPrettyJson, json); |
| free(json); |
| } |
| |
| TEST(PrettyWriter, RawValue) { |
| StringBuffer buffer; |
| PrettyWriter<StringBuffer> writer(buffer); |
| writer.StartObject(); |
| writer.Key("a"); |
| writer.Int(1); |
| writer.Key("raw"); |
| const char json[] = "[\"Hello\\nWorld\", 123.456]"; |
| writer.RawValue(json, strlen(json), kArrayType); |
| writer.EndObject(); |
| EXPECT_TRUE(writer.IsComplete()); |
| EXPECT_STREQ( |
| "{\n" |
| " \"a\": 1,\n" |
| " \"raw\": [\"Hello\\nWorld\", 123.456]\n" // no indentation within raw value |
| "}", |
| buffer.GetString()); |
| } |
| |
| TEST(PrettyWriter, InvalidEventSequence) { |
| // {] |
| { |
| StringBuffer buffer; |
| PrettyWriter<StringBuffer> writer(buffer); |
| writer.StartObject(); |
| EXPECT_THROW(writer.EndArray(), AssertException); |
| EXPECT_FALSE(writer.IsComplete()); |
| } |
| |
| // [} |
| { |
| StringBuffer buffer; |
| PrettyWriter<StringBuffer> writer(buffer); |
| writer.StartArray(); |
| EXPECT_THROW(writer.EndObject(), AssertException); |
| EXPECT_FALSE(writer.IsComplete()); |
| } |
| |
| // { 1: |
| { |
| StringBuffer buffer; |
| PrettyWriter<StringBuffer> writer(buffer); |
| writer.StartObject(); |
| EXPECT_THROW(writer.Int(1), AssertException); |
| EXPECT_FALSE(writer.IsComplete()); |
| } |
| |
| // { 'a' } |
| { |
| StringBuffer buffer; |
| PrettyWriter<StringBuffer> writer(buffer); |
| writer.StartObject(); |
| writer.Key("a"); |
| EXPECT_THROW(writer.EndObject(), AssertException); |
| EXPECT_FALSE(writer.IsComplete()); |
| } |
| |
| // { 'a':'b','c' } |
| { |
| StringBuffer buffer; |
| PrettyWriter<StringBuffer> writer(buffer); |
| writer.StartObject(); |
| writer.Key("a"); |
| writer.String("b"); |
| writer.Key("c"); |
| EXPECT_THROW(writer.EndObject(), AssertException); |
| EXPECT_FALSE(writer.IsComplete()); |
| } |
| } |
| |
| TEST(PrettyWriter, NaN) { |
| double nan = std::numeric_limits<double>::quiet_NaN(); |
| |
| EXPECT_TRUE(internal::Double(nan).IsNan()); |
| StringBuffer buffer; |
| { |
| PrettyWriter<StringBuffer> writer(buffer); |
| EXPECT_FALSE(writer.Double(nan)); |
| } |
| { |
| PrettyWriter<StringBuffer, UTF8<>, UTF8<>, CrtAllocator, kWriteNanAndInfFlag> writer(buffer); |
| EXPECT_TRUE(writer.Double(nan)); |
| EXPECT_STREQ("NaN", buffer.GetString()); |
| } |
| GenericStringBuffer<UTF16<> > buffer2; |
| PrettyWriter<GenericStringBuffer<UTF16<> > > writer2(buffer2); |
| EXPECT_FALSE(writer2.Double(nan)); |
| } |
| |
| TEST(PrettyWriter, Inf) { |
| double inf = std::numeric_limits<double>::infinity(); |
| |
| EXPECT_TRUE(internal::Double(inf).IsInf()); |
| StringBuffer buffer; |
| { |
| PrettyWriter<StringBuffer> writer(buffer); |
| EXPECT_FALSE(writer.Double(inf)); |
| } |
| { |
| PrettyWriter<StringBuffer> writer(buffer); |
| EXPECT_FALSE(writer.Double(-inf)); |
| } |
| { |
| PrettyWriter<StringBuffer, UTF8<>, UTF8<>, CrtAllocator, kWriteNanAndInfFlag> writer(buffer); |
| EXPECT_TRUE(writer.Double(inf)); |
| } |
| { |
| PrettyWriter<StringBuffer, UTF8<>, UTF8<>, CrtAllocator, kWriteNanAndInfFlag> writer(buffer); |
| EXPECT_TRUE(writer.Double(-inf)); |
| } |
| EXPECT_STREQ("Infinity-Infinity", buffer.GetString()); |
| } |
| |
| TEST(PrettyWriter, Issue_889) { |
| char buf[100] = "Hello"; |
| |
| StringBuffer buffer; |
| PrettyWriter<StringBuffer> writer(buffer); |
| writer.StartArray(); |
| writer.String(buf); |
| writer.EndArray(); |
| |
| EXPECT_STREQ("[\n \"Hello\"\n]", buffer.GetString()); |
| EXPECT_TRUE(writer.IsComplete()); \ |
| } |
| |
| |
| #if RAPIDJSON_HAS_CXX11_RVALUE_REFS |
| |
| static PrettyWriter<StringBuffer> WriterGen(StringBuffer &target) { |
| PrettyWriter<StringBuffer> writer(target); |
| writer.StartObject(); |
| writer.Key("a"); |
| writer.Int(1); |
| return writer; |
| } |
| |
| TEST(PrettyWriter, MoveCtor) { |
| StringBuffer buffer; |
| PrettyWriter<StringBuffer> writer(WriterGen(buffer)); |
| writer.EndObject(); |
| EXPECT_TRUE(writer.IsComplete()); |
| EXPECT_STREQ( |
| "{\n" |
| " \"a\": 1\n" |
| "}", |
| buffer.GetString()); |
| } |
| #endif |
| |
| TEST(PrettyWriter, Issue_1336) { |
| #define T(meth, val, expected) \ |
| { \ |
| StringBuffer buffer; \ |
| PrettyWriter<StringBuffer> writer(buffer); \ |
| writer.meth(val); \ |
| \ |
| EXPECT_STREQ(expected, buffer.GetString()); \ |
| EXPECT_TRUE(writer.IsComplete()); \ |
| } |
| |
| T(Bool, false, "false"); |
| T(Bool, true, "true"); |
| T(Int, 0, "0"); |
| T(Uint, 0, "0"); |
| T(Int64, 0, "0"); |
| T(Uint64, 0, "0"); |
| T(Double, 0, "0.0"); |
| T(String, "Hello", "\"Hello\""); |
| #undef T |
| |
| StringBuffer buffer; |
| PrettyWriter<StringBuffer> writer(buffer); |
| writer.Null(); |
| |
| EXPECT_STREQ("null", buffer.GetString()); |
| EXPECT_TRUE(writer.IsComplete()); |
| } |
| |
| #ifdef __clang__ |
| RAPIDJSON_DIAG_POP |
| #endif |