Open backdoor in VoiceEngineImpl to get at the actual voe::Channel objects from an ID.

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 "testing/gmock/include/gmock/gmock.h"
+#include "webrtc/test/mock_voe_channel_proxy.h"
 #include "webrtc/voice_engine/voice_engine_impl.h"
 
 namespace webrtc {
 namespace test {
 
 // NOTE: This class inherits from VoiceEngineImpl so that its clients will be
 // able to get the various interfaces as usual, via T::GetInterface().
 class MockVoiceEngine : public VoiceEngineImpl {
  public:
   MockVoiceEngine() : VoiceEngineImpl(new Config(), true) {
     // Increase ref count so this object isn't automatically deleted whenever
     // interfaces are Release():d.
     ++_ref_count;
+    // We add this default behavior to make the mock easier to use in tests. It
+    // will return a NiceMock of a voe::ChannelProxy.
+    ON_CALL(*this, GetChannelProxy(testing::_))
+        .WillByDefault(
+            testing::Invoke(this, &MockVoiceEngine::CreateChannelProxy));

[kwiberg-webrtc, 2015/11/25 10:44:59]

Lambda here too?

[the sun, 2015/11/25 12:28:01]

Done.

   }
   ~MockVoiceEngine() override {
     // Decrease ref count before base class d-tor is called; otherwise it will
     // trigger an assertion.
     --_ref_count;
   }
 
+  // VoiceEngineImpl
+  MOCK_METHOD1(GetChannelProxy, voe::ChannelProxy*(int channel_id));
+
   // VoEAudioProcessing
   MOCK_METHOD2(SetNsStatus, int(bool enable, NsModes mode));
   MOCK_METHOD2(GetNsStatus, int(bool& enabled, NsModes& mode));
   MOCK_METHOD2(SetAgcStatus, int(bool enable, AgcModes mode));
   MOCK_METHOD2(GetAgcStatus, int(bool& enabled, AgcModes& mode));
   MOCK_METHOD1(SetAgcConfig, int(AgcConfig config));
   MOCK_METHOD1(GetAgcConfig, int(AgcConfig& config));
   MOCK_METHOD2(SetEcStatus, int(bool enable, EcModes mode));
   MOCK_METHOD2(GetEcStatus, int(bool& enabled, EcModes& mode));
   MOCK_METHOD1(EnableDriftCompensation, int(bool enable));
@@ skipping 263 matching lines @@
   MOCK_METHOD2(GetInputMute, int(int channel, bool& enabled));
   MOCK_METHOD1(GetSpeechInputLevel, int(unsigned int& level));
   MOCK_METHOD2(GetSpeechOutputLevel, int(int channel, unsigned int& level));
   MOCK_METHOD1(GetSpeechInputLevelFullRange, int(unsigned int& level));
   MOCK_METHOD2(GetSpeechOutputLevelFullRange,
                int(int channel, unsigned& level));
   MOCK_METHOD2(SetChannelOutputVolumeScaling, int(int channel, float scaling));
   MOCK_METHOD2(GetChannelOutputVolumeScaling, int(int channel, float& scaling));
   MOCK_METHOD3(SetOutputVolumePan, int(int channel, float left, float right));
   MOCK_METHOD3(GetOutputVolumePan, int(int channel, float& left, float& right));
+
+ private:
+  voe::ChannelProxy* CreateChannelProxy(int channel_id) {
+   return new testing::NiceMock<MockVoEChannelProxy>();
+  }
 };
 }  // namespace test
 }  // namespace webrtc
 
 #endif  // WEBRTC_AUDIO_MOCK_VOICE_ENGINE_H_