stream_capture.h

Go to the documentation of this file.

// Copyright 2013 The Flutter 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 FLUTTER_TESTING_STREAM_CAPTURE_H_
#define FLUTTER_TESTING_STREAM_CAPTURE_H_
 
#include <ostream>
#include <sstream>
#include <string>
 
namespace flutter::testing {
 
// Temporarily replaces the specified stream's output buffer to capture output.
//
// Example:
// StreamCapture captured_stdout(&std::cout);
// ... code that writest to std::cout ...
// std::string output = captured_stdout.GetCapturedOutput();
class StreamCapture {
 public:
  // Begins capturing output to the specified stream.
  explicit StreamCapture(std::ostream* ostream);
 
  // Stops capturing output to the specified stream, and restores the original
  // output buffer, if |Stop| has not already been called.
  ~StreamCapture();
 
  // Stops capturing output to the specified stream, and restores the original
  // output buffer.
  void Stop();
 
  // Returns any output written to the captured stream between construction and
  // the first call to |Stop|, if any, or now.
  std::string GetOutput() const;
 
 private:
  std::ostream* ostream_;
  std::stringstream buffer_;
  std::streambuf* old_buffer_;
};
 
}  // namespace flutter::testing
 
#endif  // FLUTTER_TESTING_STREAM_CAPTURE_H_