1 /*
2  *  Copyright (c) 2016 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_DECODER_FACTORY_H_
12 #define TEST_MOCK_AUDIO_DECODER_FACTORY_H_
13 
14 #include <memory>
15 #include <vector>
16 
17 #include "api/audio_codecs/audio_decoder_factory.h"
18 #include "api/audio_codecs/builtin_audio_decoder_factory.h"
19 #include "api/scoped_refptr.h"
20 #include "rtc_base/ref_counted_object.h"
21 #include "test/gmock.h"
22 
23 namespace webrtc {
24 
25 class MockAudioDecoderFactory : public AudioDecoderFactory {
26  public:
27   MOCK_METHOD0(GetSupportedDecoders, std::vector<AudioCodecSpec>());
28   MOCK_METHOD1(IsSupportedDecoder, bool(const SdpAudioFormat&));
MakeAudioDecoder(const SdpAudioFormat & format,absl::optional<AudioCodecPairId> codec_pair_id)29   std::unique_ptr<AudioDecoder> MakeAudioDecoder(
30       const SdpAudioFormat& format,
31       absl::optional<AudioCodecPairId> codec_pair_id) {
32     std::unique_ptr<AudioDecoder> return_value;
33     MakeAudioDecoderMock(format, codec_pair_id, &return_value);
34     return return_value;
35   }
36   MOCK_METHOD3(MakeAudioDecoderMock,
37                void(const SdpAudioFormat& format,
38                     absl::optional<AudioCodecPairId> codec_pair_id,
39                     std::unique_ptr<AudioDecoder>* return_value));
40 
41   // Creates a MockAudioDecoderFactory with no formats and that may not be
42   // invoked to create a codec - useful for initializing a voice engine, for
43   // example.
44   static rtc::scoped_refptr<webrtc::MockAudioDecoderFactory>
CreateUnusedFactory()45   CreateUnusedFactory() {
46     using ::testing::_;
47     using ::testing::AnyNumber;
48     using ::testing::Return;
49 
50     rtc::scoped_refptr<webrtc::MockAudioDecoderFactory> factory =
51         new rtc::RefCountedObject<webrtc::MockAudioDecoderFactory>;
52     ON_CALL(*factory.get(), GetSupportedDecoders())
53         .WillByDefault(Return(std::vector<webrtc::AudioCodecSpec>()));
54     EXPECT_CALL(*factory.get(), GetSupportedDecoders()).Times(AnyNumber());
55     ON_CALL(*factory, IsSupportedDecoder(_)).WillByDefault(Return(false));
56     EXPECT_CALL(*factory, IsSupportedDecoder(_)).Times(AnyNumber());
57     EXPECT_CALL(*factory.get(), MakeAudioDecoderMock(_, _, _)).Times(0);
58     return factory;
59   }
60 
61   // Creates a MockAudioDecoderFactory with no formats that may be invoked to
62   // create a codec any number of times. It will, though, return nullptr on each
63   // call, since it supports no codecs.
64   static rtc::scoped_refptr<webrtc::MockAudioDecoderFactory>
CreateEmptyFactory()65   CreateEmptyFactory() {
66     using ::testing::_;
67     using ::testing::AnyNumber;
68     using ::testing::Return;
69     using ::testing::SetArgPointee;
70 
71     rtc::scoped_refptr<webrtc::MockAudioDecoderFactory> factory =
72         new rtc::RefCountedObject<webrtc::MockAudioDecoderFactory>;
73     ON_CALL(*factory.get(), GetSupportedDecoders())
74         .WillByDefault(Return(std::vector<webrtc::AudioCodecSpec>()));
75     EXPECT_CALL(*factory.get(), GetSupportedDecoders()).Times(AnyNumber());
76     ON_CALL(*factory, IsSupportedDecoder(_)).WillByDefault(Return(false));
77     EXPECT_CALL(*factory, IsSupportedDecoder(_)).Times(AnyNumber());
78     ON_CALL(*factory.get(), MakeAudioDecoderMock(_, _, _))
79         .WillByDefault(SetArgPointee<2>(nullptr));
80     EXPECT_CALL(*factory.get(), MakeAudioDecoderMock(_, _, _))
81         .Times(AnyNumber());
82     return factory;
83   }
84 };
85 
86 }  // namespace webrtc
87 
88 #endif  // TEST_MOCK_AUDIO_DECODER_FACTORY_H_
89