70 lines
1.6 KiB
C++
70 lines
1.6 KiB
C++
|
#include "unittest.h"
|
||
|
|
#include "rapidjson/filestream.h"
|
||
|
|
#include "rapidjson/filereadstream.h"
|
||
|
|
|
||
|
|
using namespace rapidjson;
|
||
|
|
|
||
|
|
class FileStreamTest : public ::testing::Test {
|
||
|
|
virtual void SetUp() {
|
||
|
|
FILE *fp = fopen(filename_ = "data/sample.json", "rb");
|
||
|
|
if (!fp)
|
||
|
|
fp = fopen(filename_ = "../../bin/data/sample.json", "rb");
|
||
|
|
ASSERT_TRUE(fp != 0);
|
||
|
|
|
||
|
|
fseek(fp, 0, SEEK_END);
|
||
|
|
length_ = (size_t)ftell(fp);
|
||
|
|
fseek(fp, 0, SEEK_SET);
|
||
|
|
json_ = (char*)malloc(length_ + 1);
|
||
|
|
fread(json_, 1, length_, fp);
|
||
|
|
json_[length_] = '\0';
|
||
|
|
length_++; // include the null terminator
|
||
|
|
fclose(fp);
|
||
|
|
}
|
||
|
|
|
||
|
|
virtual void TearDown() {
|
||
|
|
free(json_);
|
||
|
|
}
|
||
|
|
|
||
|
|
protected:
|
||
|
|
const char* filename_;
|
||
|
|
char *json_;
|
||
|
|
size_t length_;
|
||
|
|
};
|
||
|
|
|
||
|
|
TEST_F(FileStreamTest, Read) {
|
||
|
|
FILE *fp = fopen(filename_, "rb");
|
||
|
|
ASSERT_TRUE(fp != 0);
|
||
|
|
FileStream s(fp);
|
||
|
|
|
||
|
|
for (size_t i = 0; i < length_; i++) {
|
||
|
|
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());
|
||
|
|
EXPECT_EQ('\0', s.Peek());
|
||
|
|
|
||
|
|
fclose(fp);
|
||
|
|
}
|
||
|
|
|
||
|
|
TEST_F(FileStreamTest, BufferedRead) {
|
||
|
|
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(length_, s.Tell());
|
||
|
|
EXPECT_EQ('\0', s.Peek());
|
||
|
|
|
||
|
|
fclose(fp);
|
||
|
|
}
|