modules/congestion_controller/rtp/transport_feedback_demuxer_unittest.cc

/*
 *  Copyright (c) 2019 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.
 */
#include "modules/congestion_controller/rtp/transport_feedback_demuxer.h"

#include "modules/rtp_rtcp/source/rtcp_packet/transport_feedback.h"
#include "test/gmock.h"
#include "test/gtest.h"

namespace webrtc {
namespace {

using ::testing::_;
static constexpr uint32_t kSsrc = 8492;

class MockStreamFeedbackObserver : public webrtc::StreamFeedbackObserver {
 public:
  MOCK_METHOD1(OnPacketFeedbackVector,
               void(std::vector<StreamPacketInfo> packet_feedback_vector));
};

RtpPacketSendInfo CreatePacket(uint32_t ssrc,
                               int16_t rtp_sequence_number,
                               int64_t transport_sequence_number) {
  RtpPacketSendInfo res;
  res.ssrc = ssrc;
  res.transport_sequence_number = transport_sequence_number;
  res.rtp_sequence_number = rtp_sequence_number;
  return res;
}
}  // namespace
TEST(TransportFeedbackDemuxerTest, ObserverSanity) {
  TransportFeedbackDemuxer demuxer;
  MockStreamFeedbackObserver mock;
  demuxer.RegisterStreamFeedbackObserver({kSsrc}, &mock);

  demuxer.AddPacket(CreatePacket(kSsrc, 55, 1));
  demuxer.AddPacket(CreatePacket(kSsrc, 56, 2));
  demuxer.AddPacket(CreatePacket(kSsrc, 57, 3));

  rtcp::TransportFeedback feedback;
  feedback.SetBase(1, 1000);
  ASSERT_TRUE(feedback.AddReceivedPacket(1, 1000));
  ASSERT_TRUE(feedback.AddReceivedPacket(2, 2000));
  ASSERT_TRUE(feedback.AddReceivedPacket(3, 3000));

  EXPECT_CALL(mock, OnPacketFeedbackVector(_)).Times(1);
  demuxer.OnTransportFeedback(feedback);

  demuxer.DeRegisterStreamFeedbackObserver(&mock);

  demuxer.AddPacket(CreatePacket(kSsrc, 58, 4));
  rtcp::TransportFeedback second_feedback;
  second_feedback.SetBase(4, 4000);
  ASSERT_TRUE(second_feedback.AddReceivedPacket(4, 4000));

  EXPECT_CALL(mock, OnPacketFeedbackVector(_)).Times(0);
  demuxer.OnTransportFeedback(second_feedback);
}
}  // namespace webrtc