| // 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/filereadstream.h" | |
| #include "rapidjson/filewritestream.h" | |
| #include "rapidjson/encodedstream.h" | |
| #include "rapidjson/stringbuffer.h" | |
| #include "rapidjson/memorystream.h" | |
| #include "rapidjson/memorybuffer.h" | |
| using namespace rapidjson; | |
| class EncodedStreamTest : public ::testing::Test { | |
| public: | |
| EncodedStreamTest() : json_(), length_() {} | |
| virtual ~EncodedStreamTest(); | |
| virtual void SetUp() { | |
| json_ = ReadFile("utf8.json", true, &length_); | |
| } | |
| virtual void TearDown() { | |
| free(json_); | |
| json_ = 0; | |
| } | |
| private: | |
| EncodedStreamTest(const EncodedStreamTest&); | |
| EncodedStreamTest& operator=(const EncodedStreamTest&); | |
| protected: | |
| static FILE* Open(const char* filename) { | |
| const char *paths[] = { | |
| "encodings", | |
| "bin/encodings", | |
| "../bin/encodings", | |
| "../../bin/encodings", | |
| "../../../bin/encodings" | |
| }; | |
| char buffer[1024]; | |
| for (size_t i = 0; i < sizeof(paths) / sizeof(paths[0]); i++) { | |
| sprintf(buffer, "%s/%s", paths[i], filename); | |
| FILE *fp = fopen(buffer, "rb"); | |
| if (fp) | |
| return fp; | |
| } | |
| return 0; | |
| } | |
| static char *ReadFile(const char* filename, bool appendPath, size_t* outLength) { | |
| FILE *fp = appendPath ? Open(filename) : fopen(filename, "rb"); | |
| if (!fp) { | |
| *outLength = 0; | |
| return 0; | |
| } | |
| fseek(fp, 0, SEEK_END); | |
| *outLength = static_cast<size_t>(ftell(fp)); | |
| fseek(fp, 0, SEEK_SET); | |
| char* buffer = static_cast<char*>(malloc(*outLength + 1)); | |
| size_t readLength = fread(buffer, 1, *outLength, fp); | |
| buffer[readLength] = '\0'; | |
| fclose(fp); | |
| return buffer; | |
| } | |
| template <typename FileEncoding, typename MemoryEncoding> | |
| void TestEncodedInputStream(const char* filename) { | |
| // Test FileReadStream | |
| { | |
| char buffer[16]; | |
| FILE *fp = Open(filename); | |
| ASSERT_TRUE(fp != 0); | |
| FileReadStream fs(fp, buffer, sizeof(buffer)); | |
| EncodedInputStream<FileEncoding, FileReadStream> eis(fs); | |
| StringStream s(json_); | |
| while (eis.Peek() != '\0') { | |
| unsigned expected, actual; | |
| EXPECT_TRUE(UTF8<>::Decode(s, &expected)); | |
| EXPECT_TRUE(MemoryEncoding::Decode(eis, &actual)); | |
| EXPECT_EQ(expected, actual); | |
| } | |
| EXPECT_EQ('\0', s.Peek()); | |
| fclose(fp); | |
| } | |
| // Test MemoryStream | |
| { | |
| size_t size; | |
| char* data = ReadFile(filename, true, &size); | |
| MemoryStream ms(data, size); | |
| EncodedInputStream<FileEncoding, MemoryStream> eis(ms); | |
| StringStream s(json_); | |
| while (eis.Peek() != '\0') { | |
| unsigned expected, actual; | |
| EXPECT_TRUE(UTF8<>::Decode(s, &expected)); | |
| EXPECT_TRUE(MemoryEncoding::Decode(eis, &actual)); | |
| EXPECT_EQ(expected, actual); | |
| } | |
| EXPECT_EQ('\0', s.Peek()); | |
| free(data); | |
| EXPECT_EQ(size, eis.Tell()); | |
| } | |
| } | |
| void TestAutoUTFInputStream(const char *filename, bool expectHasBOM) { | |
| // Test FileReadStream | |
| { | |
| char buffer[16]; | |
| FILE *fp = Open(filename); | |
| ASSERT_TRUE(fp != 0); | |
| FileReadStream fs(fp, buffer, sizeof(buffer)); | |
| AutoUTFInputStream<unsigned, FileReadStream> eis(fs); | |
| EXPECT_EQ(expectHasBOM, eis.HasBOM()); | |
| StringStream s(json_); | |
| while (eis.Peek() != '\0') { | |
| unsigned expected, actual; | |
| EXPECT_TRUE(UTF8<>::Decode(s, &expected)); | |
| EXPECT_TRUE(AutoUTF<unsigned>::Decode(eis, &actual)); | |
| EXPECT_EQ(expected, actual); | |
| } | |
| EXPECT_EQ('\0', s.Peek()); | |
| fclose(fp); | |
| } | |
| // Test MemoryStream | |
| { | |
| size_t size; | |
| char* data = ReadFile(filename, true, &size); | |
| MemoryStream ms(data, size); | |
| AutoUTFInputStream<unsigned, MemoryStream> eis(ms); | |
| EXPECT_EQ(expectHasBOM, eis.HasBOM()); | |
| StringStream s(json_); | |
| while (eis.Peek() != '\0') { | |
| unsigned expected, actual; | |
| EXPECT_TRUE(UTF8<>::Decode(s, &expected)); | |
| EXPECT_TRUE(AutoUTF<unsigned>::Decode(eis, &actual)); | |
| EXPECT_EQ(expected, actual); | |
| } | |
| EXPECT_EQ('\0', s.Peek()); | |
| free(data); | |
| EXPECT_EQ(size, eis.Tell()); | |
| } | |
| } | |
| template <typename FileEncoding, typename MemoryEncoding> | |
| void TestEncodedOutputStream(const char* expectedFilename, bool putBOM) { | |
| // Test FileWriteStream | |
| { | |
| char filename[L_tmpnam]; | |
| FILE* fp = TempFile(filename); | |
| char buffer[16]; | |
| FileWriteStream os(fp, buffer, sizeof(buffer)); | |
| EncodedOutputStream<FileEncoding, FileWriteStream> eos(os, putBOM); | |
| StringStream s(json_); | |
| while (s.Peek() != '\0') { | |
| bool success = Transcoder<UTF8<>, MemoryEncoding>::Transcode(s, eos); | |
| EXPECT_TRUE(success); | |
| } | |
| eos.Flush(); | |
| fclose(fp); | |
| EXPECT_TRUE(CompareFile(filename, expectedFilename)); | |
| remove(filename); | |
| } | |
| // Test MemoryBuffer | |
| { | |
| MemoryBuffer mb; | |
| EncodedOutputStream<FileEncoding, MemoryBuffer> eos(mb, putBOM); | |
| StringStream s(json_); | |
| while (s.Peek() != '\0') { | |
| bool success = Transcoder<UTF8<>, MemoryEncoding>::Transcode(s, eos); | |
| EXPECT_TRUE(success); | |
| } | |
| eos.Flush(); | |
| EXPECT_TRUE(CompareBufferFile(mb.GetBuffer(), mb.GetSize(), expectedFilename)); | |
| } | |
| } | |
| void TestAutoUTFOutputStream(UTFType type, bool putBOM, const char *expectedFilename) { | |
| // Test FileWriteStream | |
| { | |
| char filename[L_tmpnam]; | |
| FILE* fp = TempFile(filename); | |
| char buffer[16]; | |
| FileWriteStream os(fp, buffer, sizeof(buffer)); | |
| AutoUTFOutputStream<unsigned, FileWriteStream> eos(os, type, putBOM); | |
| StringStream s(json_); | |
| while (s.Peek() != '\0') { | |
| bool success = Transcoder<UTF8<>, AutoUTF<unsigned> >::Transcode(s, eos); | |
| EXPECT_TRUE(success); | |
| } | |
| eos.Flush(); | |
| fclose(fp); | |
| EXPECT_TRUE(CompareFile(filename, expectedFilename)); | |
| remove(filename); | |
| } | |
| // Test MemoryBuffer | |
| { | |
| MemoryBuffer mb; | |
| AutoUTFOutputStream<unsigned, MemoryBuffer> eos(mb, type, putBOM); | |
| StringStream s(json_); | |
| while (s.Peek() != '\0') { | |
| bool success = Transcoder<UTF8<>, AutoUTF<unsigned> >::Transcode(s, eos); | |
| EXPECT_TRUE(success); | |
| } | |
| eos.Flush(); | |
| EXPECT_TRUE(CompareBufferFile(mb.GetBuffer(), mb.GetSize(), expectedFilename)); | |
| } | |
| } | |
| bool CompareFile(const char* filename, const char* expectedFilename) { | |
| size_t actualLength, expectedLength; | |
| char* actualBuffer = ReadFile(filename, false, &actualLength); | |
| char* expectedBuffer = ReadFile(expectedFilename, true, &expectedLength); | |
| bool ret = (expectedLength == actualLength) && memcmp(expectedBuffer, actualBuffer, actualLength) == 0; | |
| free(actualBuffer); | |
| free(expectedBuffer); | |
| return ret; | |
| } | |
| bool CompareBufferFile(const char* actualBuffer, size_t actualLength, const char* expectedFilename) { | |
| size_t expectedLength; | |
| char* expectedBuffer = ReadFile(expectedFilename, true, &expectedLength); | |
| bool ret = (expectedLength == actualLength) && memcmp(expectedBuffer, actualBuffer, actualLength) == 0; | |
| free(expectedBuffer); | |
| return ret; | |
| } | |
| char *json_; | |
| size_t length_; | |
| }; | |
| EncodedStreamTest::~EncodedStreamTest() {} | |
| TEST_F(EncodedStreamTest, EncodedInputStream) { | |
| TestEncodedInputStream<UTF8<>, UTF8<> >("utf8.json"); | |
| TestEncodedInputStream<UTF8<>, UTF8<> >("utf8bom.json"); | |
| TestEncodedInputStream<UTF16LE<>, UTF16<> >("utf16le.json"); | |
| TestEncodedInputStream<UTF16LE<>, UTF16<> >("utf16lebom.json"); | |
| TestEncodedInputStream<UTF16BE<>, UTF16<> >("utf16be.json"); | |
| TestEncodedInputStream<UTF16BE<>, UTF16<> >("utf16bebom.json"); | |
| TestEncodedInputStream<UTF32LE<>, UTF32<> >("utf32le.json"); | |
| TestEncodedInputStream<UTF32LE<>, UTF32<> >("utf32lebom.json"); | |
| TestEncodedInputStream<UTF32BE<>, UTF32<> >("utf32be.json"); | |
| TestEncodedInputStream<UTF32BE<>, UTF32<> >("utf32bebom.json"); | |
| } | |
| TEST_F(EncodedStreamTest, AutoUTFInputStream) { | |
| TestAutoUTFInputStream("utf8.json", false); | |
| TestAutoUTFInputStream("utf8bom.json", true); | |
| TestAutoUTFInputStream("utf16le.json", false); | |
| TestAutoUTFInputStream("utf16lebom.json",true); | |
| TestAutoUTFInputStream("utf16be.json", false); | |
| TestAutoUTFInputStream("utf16bebom.json",true); | |
| TestAutoUTFInputStream("utf32le.json", false); | |
| TestAutoUTFInputStream("utf32lebom.json",true); | |
| TestAutoUTFInputStream("utf32be.json", false); | |
| TestAutoUTFInputStream("utf32bebom.json", true); | |
| { | |
| // Auto detection fail, use user defined UTF type | |
| const char json[] = "{ }"; | |
| MemoryStream ms(json, sizeof(json)); | |
| AutoUTFInputStream<unsigned, MemoryStream> eis(ms, kUTF8); | |
| EXPECT_FALSE(eis.HasBOM()); | |
| EXPECT_EQ(kUTF8, eis.GetType()); | |
| } | |
| } | |
| TEST_F(EncodedStreamTest, EncodedOutputStream) { | |
| TestEncodedOutputStream<UTF8<>, UTF8<> >("utf8.json", false); | |
| TestEncodedOutputStream<UTF8<>, UTF8<> >("utf8bom.json", true); | |
| TestEncodedOutputStream<UTF16LE<>, UTF16<> >("utf16le.json", false); | |
| TestEncodedOutputStream<UTF16LE<>, UTF16<> >("utf16lebom.json",true); | |
| TestEncodedOutputStream<UTF16BE<>, UTF16<> >("utf16be.json", false); | |
| TestEncodedOutputStream<UTF16BE<>, UTF16<> >("utf16bebom.json",true); | |
| TestEncodedOutputStream<UTF32LE<>, UTF32<> >("utf32le.json", false); | |
| TestEncodedOutputStream<UTF32LE<>, UTF32<> >("utf32lebom.json",true); | |
| TestEncodedOutputStream<UTF32BE<>, UTF32<> >("utf32be.json", false); | |
| TestEncodedOutputStream<UTF32BE<>, UTF32<> >("utf32bebom.json",true); | |
| } | |
| TEST_F(EncodedStreamTest, AutoUTFOutputStream) { | |
| TestAutoUTFOutputStream(kUTF8, false, "utf8.json"); | |
| TestAutoUTFOutputStream(kUTF8, true, "utf8bom.json"); | |
| TestAutoUTFOutputStream(kUTF16LE, false, "utf16le.json"); | |
| TestAutoUTFOutputStream(kUTF16LE, true, "utf16lebom.json"); | |
| TestAutoUTFOutputStream(kUTF16BE, false, "utf16be.json"); | |
| TestAutoUTFOutputStream(kUTF16BE, true, "utf16bebom.json"); | |
| TestAutoUTFOutputStream(kUTF32LE, false, "utf32le.json"); | |
| TestAutoUTFOutputStream(kUTF32LE, true, "utf32lebom.json"); | |
| TestAutoUTFOutputStream(kUTF32BE, false, "utf32be.json"); | |
| TestAutoUTFOutputStream(kUTF32BE, true, "utf32bebom.json"); | |
| } |