1 /*
2  *  Copyright (c) 2017 The WebRTC project authors. All Rights Reserved.
3  *
4  *  Use of this source code is governed by a BSD-style license
5  *  that can be found in the LICENSE file in the root of the source
6  *  tree. An additional intellectual property rights grant can be found
7  *  in the file PATENTS.  All contributing project authors may
8  *  be found in the AUTHORS file in the root of the source tree.
9  */
10 
11 #ifndef TEST_MOCK_AUDIO_ENCODER_FACTORY_H_
12 #define TEST_MOCK_AUDIO_ENCODER_FACTORY_H_
13 
14 #include <memory>
15 #include <vector>
16 
17 #include "api/audio_codecs/audio_encoder_factory.h"
18 #include "api/scoped_refptr.h"
19 #include "rtc_base/ref_counted_object.h"
20 #include "test/gmock.h"
21 
22 namespace webrtc {
23 
24 class MockAudioEncoderFactory
25     : public ::testing::NiceMock<AudioEncoderFactory> {
26  public:
27   MOCK_METHOD(std::vector<AudioCodecSpec>,
28               GetSupportedEncoders,
29               (),
30               (override));
31   MOCK_METHOD(absl::optional<AudioCodecInfo>,
32               QueryAudioEncoder,
33               (const SdpAudioFormat& format),
34               (override));
35 
MakeAudioEncoder(int payload_type,const SdpAudioFormat & format,absl::optional<AudioCodecPairId> codec_pair_id)36   std::unique_ptr<AudioEncoder> MakeAudioEncoder(
37       int payload_type,
38       const SdpAudioFormat& format,
39       absl::optional<AudioCodecPairId> codec_pair_id) override {
40     std::unique_ptr<AudioEncoder> return_value;
41     MakeAudioEncoderMock(payload_type, format, codec_pair_id, &return_value);
42     return return_value;
43   }
44   MOCK_METHOD(void,
45               MakeAudioEncoderMock,
46               (int payload_type,
47                const SdpAudioFormat& format,
48                absl::optional<AudioCodecPairId> codec_pair_id,
49                std::unique_ptr<AudioEncoder>*));
50 
51   // Creates a MockAudioEncoderFactory with no formats and that may not be
52   // invoked to create a codec - useful for initializing a voice engine, for
53   // example.
54   static rtc::scoped_refptr<webrtc::MockAudioEncoderFactory>
CreateUnusedFactory()55   CreateUnusedFactory() {
56     using ::testing::_;
57     using ::testing::AnyNumber;
58     using ::testing::Return;
59 
60     rtc::scoped_refptr<webrtc::MockAudioEncoderFactory> factory =
61         new rtc::RefCountedObject<webrtc::MockAudioEncoderFactory>;
62     ON_CALL(*factory.get(), GetSupportedEncoders())
63         .WillByDefault(Return(std::vector<webrtc::AudioCodecSpec>()));
64     ON_CALL(*factory.get(), QueryAudioEncoder(_))
65         .WillByDefault(Return(absl::nullopt));
66 
67     EXPECT_CALL(*factory.get(), GetSupportedEncoders()).Times(AnyNumber());
68     EXPECT_CALL(*factory.get(), QueryAudioEncoder(_)).Times(AnyNumber());
69     EXPECT_CALL(*factory.get(), MakeAudioEncoderMock(_, _, _, _)).Times(0);
70     return factory;
71   }
72 
73   // Creates a MockAudioEncoderFactory with no formats that may be invoked to
74   // create a codec any number of times. It will, though, return nullptr on each
75   // call, since it supports no codecs.
76   static rtc::scoped_refptr<webrtc::MockAudioEncoderFactory>
CreateEmptyFactory()77   CreateEmptyFactory() {
78     using ::testing::_;
79     using ::testing::AnyNumber;
80     using ::testing::Return;
81     using ::testing::SetArgPointee;
82 
83     rtc::scoped_refptr<webrtc::MockAudioEncoderFactory> factory =
84         new rtc::RefCountedObject<webrtc::MockAudioEncoderFactory>;
85     ON_CALL(*factory.get(), GetSupportedEncoders())
86         .WillByDefault(Return(std::vector<webrtc::AudioCodecSpec>()));
87     ON_CALL(*factory.get(), QueryAudioEncoder(_))
88         .WillByDefault(Return(absl::nullopt));
89     ON_CALL(*factory.get(), MakeAudioEncoderMock(_, _, _, _))
90         .WillByDefault(SetArgPointee<3>(nullptr));
91 
92     EXPECT_CALL(*factory.get(), GetSupportedEncoders()).Times(AnyNumber());
93     EXPECT_CALL(*factory.get(), QueryAudioEncoder(_)).Times(AnyNumber());
94     EXPECT_CALL(*factory.get(), MakeAudioEncoderMock(_, _, _, _))
95         .Times(AnyNumber());
96     return factory;
97   }
98 };
99 
100 }  // namespace webrtc
101 
102 #endif  // TEST_MOCK_AUDIO_ENCODER_FACTORY_H_
103