generic-library/rapidjson
1
0
Fork 0
mirror of https://github.com/Tencent/rapidjson.git synced 2025-10-27 02:53:13 +01:00
Code Issues Projects Releases Wiki Activity

Implemented FileWriteStream with buffering

Browse Source 
git-svn-id: https://rapidjson.googlecode.com/svn/trunk@23 c5894555-1306-4e8d-425f-1f6f381ee07c
This commit is contained in:
miloyip@gmail.com
2011-11-21 10:00:33 +00:00
parent bce34fbea8
commit 04515a639e
6 changed files with 120 additions and 20 deletions
Download Patch File Download Diff File Expand all files Collapse all files

40
test/unittest/filestreamtest.cpp
View File

@@ -1,6 +1,7 @@
#include "unittest.h"
#include "rapidjson/filestream.h"
#include "rapidjson/filereadstream.h"
#include "rapidjson/filewritestream.h"
using namespace rapidjson;
@@ -17,7 +18,6 @@ class FileStreamTest : public ::testing::Test {
json_ = (char*)malloc(length_ + 1);
fread(json_, 1, length_, fp);
json_[length_] = '\0';
length_++; // include the null terminator
fclose(fp);
}
@@ -31,7 +31,7 @@ protected:
size_t length_;
};
TEST_F(FileStreamTest, Read) {
TEST_F(FileStreamTest, FileStream_Read) {
FILE *fp = fopen(filename_, "rb");
ASSERT_TRUE(fp != 0);
FileStream s(fp);
@@ -48,18 +48,16 @@ TEST_F(FileStreamTest, Read) {
fclose(fp);
}
TEST_F(FileStreamTest, BufferedRead) {
TEST_F(FileStreamTest, FileReadStream) {
FILE *fp = fopen(filename_, "rb");
ASSERT_TRUE(fp != 0);
char buffer[65536];
FileReadStream s(fp, buffer, sizeof(buffer));
for (size_t i = 0; i < length_; i++) {
if (json_[i] != s.Peek())
__asm int 3;
ASSERT_EQ(json_[i], s.Peek());
ASSERT_EQ(json_[i], s.Peek()); // 2nd time should be the same
ASSERT_EQ(json_[i], s.Take());
EXPECT_EQ(json_[i], s.Peek());
EXPECT_EQ(json_[i], s.Peek()); // 2nd time should be the same
EXPECT_EQ(json_[i], s.Take());
}
EXPECT_EQ(length_, s.Tell());
@@ -67,3 +65,29 @@ TEST_F(FileStreamTest, BufferedRead) {
fclose(fp);
}
TEST_F(FileStreamTest, FileWriteStream) {
char filename[L_tmpnam];
tmpnam(filename);
FILE *fp = fopen(filename, "wb");
char buffer[65536];
FileWriteStream os(fp, buffer, sizeof(buffer));
for (size_t i = 0; i < length_; i++)
os.Put(json_[i]);
os.Flush();
fclose(fp);
// Read it back to verify
fp = fopen(filename, "rb");
FileReadStream is(fp, buffer, sizeof(buffer));
for (size_t i = 0; i < length_; i++)
EXPECT_EQ(json_[i], is.Take());
EXPECT_EQ(length_, is.Tell());
fclose(fp);
//std::cout << filename << std::endl;
remove(filename);
}