missive/client/mock_report_queue.h

// Copyright 2021 The Chromium OS 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 MISSIVE_CLIENT_MOCK_REPORT_QUEUE_H_
#define MISSIVE_CLIENT_MOCK_REPORT_QUEUE_H_

#include <memory>

#include <base/callback.h>
#include <gmock/gmock.h>
#include <google/protobuf/message_lite.h>

#include "missive/client/report_queue.h"
#include "missive/proto/record.pb.h"
#include "missive/proto/record_constants.pb.h"
#include "missive/util/status.h"

namespace reporting {

// A mock of ReportQueue for use in testing.
class MockReportQueue : public ReportQueue {
 public:
  MockReportQueue();
  ~MockReportQueue() override;

  MOCK_METHOD(void,
              AddRecord,
              (base::StringPiece, Priority, ReportQueue::EnqueueCallback),
              (const override));

  MOCK_METHOD(void, Flush, (Priority, ReportQueue::FlushCallback), (override));

  MOCK_METHOD(
      (base::OnceCallback<void(StatusOr<std::unique_ptr<ReportQueue>>)>),
      PrepareToAttachActualQueue,
      (),
      (const override));
};

}  // namespace reporting

#endif  // MISSIVE_CLIENT_MOCK_REPORT_QUEUE_H_