"add testing"

Author: dzhwinter

paddle/fluid/recordio/header_test.cc

@@ -32,14 +32,5 @@ TEST(Recordio, ChunkHead) {
   std::ostringstream oss2;
   hdr2.Write(oss2);
   EXPECT_STREQ(oss2.str().c_str(), oss.str().c_str());
-}
-
-TEST(Recordio, Stream) {
-  Header hdr(0, 1, static_cast<Compressor>(2), 3);
-  std::ostringstream oss1;
-  hdr.Write(oss1);
-
-  std::ostringstream oss2;
-  oss2 << hdr;
-  EXPECT_STREQ(oss2.str().c_str(), oss1.str().c_str());
+  EXPECT_EQ(hdr == hdr2);
 }

paddle/fluid/recordio/writer.cc

@@ -29,13 +29,14 @@ Writer::Writer(std::ostream& os, int maxChunkSize, int compressor)
   chunk_.reset(new Chunk);
 }
 
-size_t Writer::Write(const std::string& buf) {}
+size_t Writer::Write(const std::string& buf) { return Write(std::string(buf)); }
 
 size_t Writer::Write(const char* buf, size_t length) {
-  // std::string s(buf, length);
-  Write(std::string(buf, length));
+  return Write(std::string(buf, length));
 }
 
+size_t Writer::Write(std::string&& buf) {}
+
 void Writer::Close() {
   stream_.flush();
   stream_.setstate(std::ios::eofbit);