Store/restore RTP state for audio streams with same SSRC within a call

webrtc/test/mock_voice_engine.h

 /*
  *  Copyright (c) 2015 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 WEBRTC_AUDIO_MOCK_VOICE_ENGINE_H_
 #define WEBRTC_AUDIO_MOCK_VOICE_ENGINE_H_
 
 #include <memory>
 
 #include "webrtc/modules/audio_device/include/mock_audio_device.h"
 #include "webrtc/modules/audio_device/include/mock_audio_transport.h"
 #include "webrtc/modules/audio_processing/include/mock_audio_processing.h"
+#include "webrtc/modules/rtp_rtcp/mocks/mock_rtp_rtcp.h"
 #include "webrtc/test/gmock.h"
 #include "webrtc/test/mock_voe_channel_proxy.h"
 #include "webrtc/voice_engine/voice_engine_impl.h"
 
 namespace webrtc {
 namespace voe {
 class TransmitMixer;
 }  // namespace voe
 
 namespace test {
@@ skipping 21 matching lines @@
         .WillByDefault(testing::Invoke([this](int channel_id) {
           auto* proxy =
               new testing::NiceMock<webrtc::test::MockVoEChannelProxy>();
           EXPECT_CALL(*proxy, GetAudioDecoderFactory())
               .WillRepeatedly(testing::ReturnRef(decoder_factory_));
           EXPECT_CALL(*proxy, SetReceiveCodecs(testing::_))
               .WillRepeatedly(testing::Invoke(
                   [](const std::map<int, SdpAudioFormat>& codecs) {
                     EXPECT_THAT(codecs, testing::IsEmpty());
                   }));
+          EXPECT_CALL(*proxy, GetRtpRtcp(testing::_, testing::_))
+              .WillRepeatedly(
+                  testing::SetArgPointee<0>(GetMockRtpRtcp(channel_id)));
           return proxy;
         }));
 
     ON_CALL(mock_audio_device_, TimeUntilNextProcess())
         .WillByDefault(testing::Return(1000));
     ON_CALL(*this, audio_device_module())
         .WillByDefault(testing::Return(&mock_audio_device_));
     ON_CALL(*this, audio_processing())
         .WillByDefault(testing::Return(&mock_audio_processing_));
     ON_CALL(*this, audio_transport())
         .WillByDefault(testing::Return(&mock_audio_transport_));
   }
   virtual ~MockVoiceEngine() /* override */ {
     // Decrease ref count before base class d-tor is called; otherwise it will
     // trigger an assertion.
     --_ref_count;
   }
+
+  // These need to be the same each call to channel_id and must not leak.
+  MockRtpRtcp* GetMockRtpRtcp(int channel_id) {
+    if (mock_rtp_rtcps_.find(channel_id) == mock_rtp_rtcps_.end()) {
+      mock_rtp_rtcps_[channel_id].reset(new ::testing::NiceMock<MockRtpRtcp>);
+    }
+    return mock_rtp_rtcps_[channel_id].get();
+  }
+
   // Allows injecting a ChannelProxy factory.
   MOCK_METHOD1(ChannelProxyFactory, voe::ChannelProxy*(int channel_id));
 
   // VoiceEngineImpl
   virtual std::unique_ptr<voe::ChannelProxy> GetChannelProxy(
       int channel_id) /* override */ {
     return std::unique_ptr<voe::ChannelProxy>(ChannelProxyFactory(channel_id));
   }
 
   // VoEBase
@@ skipping 147 matching lines @@
 
  private:
   // TODO(ossu): I'm not particularly happy about keeping the decoder factory
   // here, but due to how gmock is implemented, I cannot just keep it in the
   // functor implementing the default version of ChannelProxyFactory, above.
   // GMock creates an unfortunate copy of the functor, which would cause us to
   // return a dangling reference. Fortunately, this should go away once
   // voe::Channel does.
   rtc::scoped_refptr<AudioDecoderFactory> decoder_factory_;
 
+  std::map<int, std::unique_ptr<MockRtpRtcp>> mock_rtp_rtcps_;
+
   MockAudioDeviceModule mock_audio_device_;
   MockAudioProcessing mock_audio_processing_;
   MockAudioTransport mock_audio_transport_;
 };
 }  // namespace test
 }  // namespace webrtc
 
 #endif  // WEBRTC_AUDIO_MOCK_VOICE_ENGINE_H_