webrtc_m130/net/dcsctp/tx/mock_send_queue.h
Victor Boivie bd9031bf22 dcsctp: Add OnTotalBufferedAmountLow in Send Queue
This is similar to Change-Id: I12a16f44f775da3711f3aa52a68a0bf24f70d2f8
but with the entire send buffer as scope, not a single stream.

This can be used by clients to take alternate action (such as delaying
transmission or using other buffering) if the send buffer ever becomes
full, as they can now be notified when the send buffer is no longer
full.

Bug: webrtc:12794
Change-Id: Icf3be3b118888ffb5ced955fd7ba4826a37140f9
Reviewed-on: https://webrtc-review.googlesource.com/c/src/+/220360
Commit-Queue: Victor Boivie <boivie@webrtc.org>
Reviewed-by: Harald Alvestrand <hta@webrtc.org>
Cr-Commit-Position: refs/heads/master@{#34143}
2021-05-27 14:38:18 +00:00

61 lines
2.0 KiB
C++

/*
* Copyright (c) 2021 The WebRTC project authors. All Rights Reserved.
*
* Use of this source code is governed by a BSD-style license
* that can be found in the LICENSE file in the root of the source
* tree. An additional intellectual property rights grant can be found
* in the file PATENTS. All contributing project authors may
* be found in the AUTHORS file in the root of the source tree.
*/
#ifndef NET_DCSCTP_TX_MOCK_SEND_QUEUE_H_
#define NET_DCSCTP_TX_MOCK_SEND_QUEUE_H_
#include <cstdint>
#include "absl/types/optional.h"
#include "api/array_view.h"
#include "net/dcsctp/tx/send_queue.h"
#include "test/gmock.h"
namespace dcsctp {
class MockSendQueue : public SendQueue {
public:
MockSendQueue() {
ON_CALL(*this, Produce).WillByDefault([](TimeMs now, size_t max_size) {
return absl::nullopt;
});
}
MOCK_METHOD(absl::optional<SendQueue::DataToSend>,
Produce,
(TimeMs now, size_t max_size),
(override));
MOCK_METHOD(bool,
Discard,
(IsUnordered unordered, StreamID stream_id, MID message_id),
(override));
MOCK_METHOD(void,
PrepareResetStreams,
(rtc::ArrayView<const StreamID> streams),
(override));
MOCK_METHOD(bool, CanResetStreams, (), (const, override));
MOCK_METHOD(void, CommitResetStreams, (), (override));
MOCK_METHOD(void, RollbackResetStreams, (), (override));
MOCK_METHOD(void, Reset, (), (override));
MOCK_METHOD(size_t, buffered_amount, (StreamID stream_id), (const, override));
MOCK_METHOD(size_t, total_buffered_amount, (), (const, override));
MOCK_METHOD(size_t,
buffered_amount_low_threshold,
(StreamID stream_id),
(const, override));
MOCK_METHOD(void,
SetBufferedAmountLowThreshold,
(StreamID stream_id, size_t bytes),
(override));
};
} // namespace dcsctp
#endif // NET_DCSCTP_TX_MOCK_SEND_QUEUE_H_