2 files changed, 62 insertions, 0 deletions
diff --git a/testing/string_write_stream.cpp b/testing/string_write_stream.cpp
new file mode 100644
index 0000000000..47474389c5
--- /dev/null
+++ b/testing/string_write_stream.cpp
@@ -0,0 +1,32 @@
+// Copyright 2018 PDFium Authors. All rights reserved.
+// Use of this source code is governed by a BSD-style license that can be
+// found in the LICENSE file.
+
+#include "testing/string_write_stream.h"
+#include "core/fxcrt/bytestring.h"
+#include "core/fxcrt/widestring.h"
+
+StringWriteStream::StringWriteStream() = default;
+
+StringWriteStream::~StringWriteStream() = default;
+
+FX_FILESIZE StringWriteStream::GetSize() {
+  return stream_.tellp();
+}
+
+bool StringWriteStream::Flush() {
+  return true;
+}
+
+bool StringWriteStream::WriteBlock(const void* pData,
+                                   FX_FILESIZE offset,
+                                   size_t size) {
+  ASSERT(offset == 0);
+  stream_.write(static_cast<const char*>(pData), size);
+  return true;
+}
+
+bool StringWriteStream::WriteString(const ByteStringView& str) {
+  stream_.write(str.unterminated_c_str(), str.GetLength());
+  return true;
+}
diff --git a/testing/string_write_stream.h b/testing/string_write_stream.h
new file mode 100644
index 0000000000..7d28a8267d
--- /dev/null
+++ b/testing/string_write_stream.h
@@ -0,0 +1,30 @@
+// Copyright 2018 PDFium Authors. All rights reserved.
+// Use of this source code is governed by a BSD-style license that can be
+// found in the LICENSE file.
+
+#ifndef TESTING_STRING_WRITE_STREAM_H_
+#define TESTING_STRING_WRITE_STREAM_H_
+
+#include <sstream>
+#include <string>
+
+#include "core/fxcrt/fx_stream.h"
+
+class StringWriteStream : public IFX_SeekableWriteStream {
+ public:
+  StringWriteStream();
+  ~StringWriteStream() override;
+
+  // IFX_SeekableWriteStream
+  FX_FILESIZE GetSize() override;
+  bool Flush() override;
+  bool WriteBlock(const void* pData, FX_FILESIZE offset, size_t size) override;
+  bool WriteString(const ByteStringView& str) override;
+
+  std::string ToString() const { return stream_.str(); }
+
+ private:
+  std::ostringstream stream_;
+};
+
+#endif  // TESTING_STRING_WRITE_STREAM_H_