operations_mock.hpp

#pragma once
 
#include <userver/dump/operations.hpp>
 
USERVER_NAMESPACE_BEGIN
 
namespace dump {
 
/// A `Writer` that appends to a string buffer (used in tests)
class MockWriter final : public Writer {
 public:
  /// Creates a `MockWriter` with an empty buffer
  MockWriter();
 
  void Finish() override;
 
  /// Extracts the built string, leaving the Writer in a valid but unspecified
  /// state
  std::string Extract() &&;
 
 private:
  void WriteRaw(std::string_view data) override;
 
  std::string data_;
};
 
/// A `Reader` that reads from a string buffer (used in tests)
class MockReader final : public Reader {
 public:
  /// Creates a `PipeReader` that references the given buffer
  explicit MockReader(std::string data);
 
  void Finish() override;
 
 private:
  std::string_view ReadRaw(std::size_t max_size) override;
 
  std::string data_;
  std::string_view unread_data_;
};
 
}  // namespace dump
 
USERVER_NAMESPACE_END