summaryrefslogtreecommitdiff
path: root/testing/string_write_stream.h
diff options
context:
space:
mode:
Diffstat (limited to 'testing/string_write_stream.h')
-rw-r--r--testing/string_write_stream.h30
1 files changed, 30 insertions, 0 deletions
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_