components/allocation_recorder/testing/mock_objects.h

// Copyright 2023 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

#ifndef COMPONENTS_ALLOCATION_RECORDER_TESTING_MOCK_OBJECTS_H_
#define COMPONENTS_ALLOCATION_RECORDER_TESTING_MOCK_OBJECTS_H_

#include <memory>

#include "base/strings/string_piece.h"
#include "components/allocation_recorder/crash_handler/stream_data_source_factory.h"
#include "testing/gmock/include/gmock/gmock.h"
#include "third_party/crashpad/crashpad/minidump/minidump_user_extension_stream_data_source.h"

namespace allocation_recorder::testing::crash_handler {

struct StreamDataSourceFactoryMock
    : public allocation_recorder::crash_handler::StreamDataSourceFactory {
  StreamDataSourceFactoryMock();
  ~StreamDataSourceFactoryMock() override;

#if BUILDFLAG(ENABLE_ALLOCATION_STACK_TRACE_RECORDER)
  MOCK_METHOD(
      std::unique_ptr<::crashpad::MinidumpUserExtensionStreamDataSource>,
      CreateReportStream,
      (),
      (const override));
#endif

  MOCK_METHOD(
      std::unique_ptr<::crashpad::MinidumpUserExtensionStreamDataSource>,
      CreateErrorMessage,
      (base::StringPiece error_message),
      (const override));
};

}  // namespace allocation_recorder::testing::crash_handler

#endif