// Copyright 2018 The Chromium 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 THIRD_PARTY_BLINK_RENDERER_MODULES_PEERCONNECTION_ADAPTERS_TEST_MOCK_P2P_QUIC_STREAM_H_
#define THIRD_PARTY_BLINK_RENDERER_MODULES_PEERCONNECTION_ADAPTERS_TEST_MOCK_P2P_QUIC_STREAM_H_

#include "testing/gmock/include/gmock/gmock.h"
#include "third_party/blink/renderer/modules/peerconnection/adapters/p2p_quic_stream.h"

namespace blink {

class MockP2PQuicStream : public testing::NiceMock<P2PQuicStream> {
 public:
  explicit MockP2PQuicStream(P2PQuicStream::Delegate** delegate_out = nullptr) {
    if (delegate_out) {
      // Ensure the caller has not left the delegate_out value floating.
      DCHECK_EQ(nullptr, *delegate_out);
      EXPECT_CALL(*this, SetDelegate(testing::_))
          .WillOnce(testing::Invoke(
              [delegate_out](P2PQuicStream::Delegate* delegate) {
                *delegate_out = delegate;
              }));
    }
  }

  // P2PQuicStream overrides.
  MOCK_METHOD0(Reset, void());
  MOCK_METHOD1(MarkReceivedDataConsumed, void(uint32_t));
  MOCK_METHOD2(WriteData, void(Vector<uint8_t>, bool));
  MOCK_METHOD1(SetDelegate, void(Delegate*));
};

}  // namespace blink

#endif  // THIRD_PARTY_BLINK_RENDERER_MODULES_PEERCONNECTION_ADAPTERS_TEST_MOCK_P2P_QUIC_STREAM_H_