| /* |
| * Copyright 2022 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 PC_TEST_MOCK_PEER_CONNECTION_INTERNAL_H_ |
| #define PC_TEST_MOCK_PEER_CONNECTION_INTERNAL_H_ |
| |
| #include <map> |
| #include <memory> |
| #include <set> |
| #include <string> |
| #include <vector> |
| |
| #include "pc/peer_connection_internal.h" |
| #include "test/gmock.h" |
| |
| namespace webrtc { |
| |
| class MockPeerConnectionInternal : public PeerConnectionInternal { |
| public: |
| MockPeerConnectionInternal() {} |
| ~MockPeerConnectionInternal() = default; |
| // PeerConnectionInterface |
| MOCK_METHOD(rtc::scoped_refptr<StreamCollectionInterface>, |
| local_streams, |
| (), |
| (override)); |
| MOCK_METHOD(rtc::scoped_refptr<StreamCollectionInterface>, |
| remote_streams, |
| (), |
| (override)); |
| MOCK_METHOD(bool, AddStream, (MediaStreamInterface*), (override)); |
| MOCK_METHOD(void, RemoveStream, (MediaStreamInterface*), (override)); |
| MOCK_METHOD(RTCErrorOr<rtc::scoped_refptr<RtpSenderInterface>>, |
| AddTrack, |
| (rtc::scoped_refptr<MediaStreamTrackInterface>, |
| const std::vector<std::string>&), |
| (override)); |
| MOCK_METHOD(RTCError, |
| RemoveTrackOrError, |
| (rtc::scoped_refptr<RtpSenderInterface>), |
| (override)); |
| MOCK_METHOD(RTCErrorOr<rtc::scoped_refptr<RtpTransceiverInterface>>, |
| AddTransceiver, |
| (rtc::scoped_refptr<MediaStreamTrackInterface>), |
| (override)); |
| MOCK_METHOD(RTCErrorOr<rtc::scoped_refptr<RtpTransceiverInterface>>, |
| AddTransceiver, |
| (rtc::scoped_refptr<MediaStreamTrackInterface>, |
| const RtpTransceiverInit&), |
| (override)); |
| MOCK_METHOD(RTCErrorOr<rtc::scoped_refptr<RtpTransceiverInterface>>, |
| AddTransceiver, |
| (cricket::MediaType), |
| (override)); |
| MOCK_METHOD(RTCErrorOr<rtc::scoped_refptr<RtpTransceiverInterface>>, |
| AddTransceiver, |
| (cricket::MediaType, const RtpTransceiverInit&), |
| (override)); |
| MOCK_METHOD(rtc::scoped_refptr<RtpSenderInterface>, |
| CreateSender, |
| (const std::string&, const std::string&), |
| (override)); |
| MOCK_METHOD(std::vector<rtc::scoped_refptr<RtpSenderInterface>>, |
| GetSenders, |
| (), |
| (const, override)); |
| MOCK_METHOD(std::vector<rtc::scoped_refptr<RtpReceiverInterface>>, |
| GetReceivers, |
| (), |
| (const, override)); |
| MOCK_METHOD(std::vector<rtc::scoped_refptr<RtpTransceiverInterface>>, |
| GetTransceivers, |
| (), |
| (const, override)); |
| MOCK_METHOD(bool, |
| GetStats, |
| (StatsObserver*, MediaStreamTrackInterface*, StatsOutputLevel), |
| (override)); |
| MOCK_METHOD(void, GetStats, (RTCStatsCollectorCallback*), (override)); |
| MOCK_METHOD(void, |
| GetStats, |
| (rtc::scoped_refptr<RtpSenderInterface>, |
| rtc::scoped_refptr<RTCStatsCollectorCallback>), |
| (override)); |
| MOCK_METHOD(void, |
| GetStats, |
| (rtc::scoped_refptr<RtpReceiverInterface>, |
| rtc::scoped_refptr<RTCStatsCollectorCallback>), |
| (override)); |
| MOCK_METHOD(void, ClearStatsCache, (), (override)); |
| MOCK_METHOD(RTCErrorOr<rtc::scoped_refptr<DataChannelInterface>>, |
| CreateDataChannelOrError, |
| (const std::string&, const DataChannelInit*), |
| (override)); |
| MOCK_METHOD(SessionDescriptionInterface*, |
| local_description, |
| (), |
| (const, override)); |
| MOCK_METHOD(SessionDescriptionInterface*, |
| remote_description, |
| (), |
| (const, override)); |
| MOCK_METHOD(SessionDescriptionInterface*, |
| current_local_description, |
| (), |
| (const, override)); |
| MOCK_METHOD(SessionDescriptionInterface*, |
| current_remote_description, |
| (), |
| (const, override)); |
| MOCK_METHOD(SessionDescriptionInterface*, |
| pending_local_description, |
| (), |
| (const, override)); |
| MOCK_METHOD(SessionDescriptionInterface*, |
| pending_remote_description, |
| (), |
| (const, override)); |
| MOCK_METHOD(void, RestartIce, (), (override)); |
| MOCK_METHOD(void, |
| CreateOffer, |
| (CreateSessionDescriptionObserver*, const RTCOfferAnswerOptions&), |
| (override)); |
| MOCK_METHOD(void, |
| CreateAnswer, |
| (CreateSessionDescriptionObserver*, const RTCOfferAnswerOptions&), |
| (override)); |
| |
| MOCK_METHOD(void, |
| SetLocalDescription, |
| (SetSessionDescriptionObserver*, SessionDescriptionInterface*), |
| (override)); |
| MOCK_METHOD(void, |
| SetRemoteDescription, |
| (SetSessionDescriptionObserver*, SessionDescriptionInterface*), |
| (override)); |
| MOCK_METHOD(void, |
| SetRemoteDescription, |
| (std::unique_ptr<SessionDescriptionInterface>, |
| rtc::scoped_refptr<SetRemoteDescriptionObserverInterface>), |
| (override)); |
| MOCK_METHOD(PeerConnectionInterface::RTCConfiguration, |
| GetConfiguration, |
| (), |
| (override)); |
| MOCK_METHOD(RTCError, |
| SetConfiguration, |
| (const PeerConnectionInterface::RTCConfiguration&), |
| (override)); |
| MOCK_METHOD(bool, |
| AddIceCandidate, |
| (const IceCandidateInterface*), |
| (override)); |
| MOCK_METHOD(bool, |
| RemoveIceCandidates, |
| (const std::vector<cricket::Candidate>&), |
| (override)); |
| MOCK_METHOD(RTCError, SetBitrate, (const BitrateSettings&), (override)); |
| MOCK_METHOD(void, SetAudioPlayout, (bool), (override)); |
| MOCK_METHOD(void, SetAudioRecording, (bool), (override)); |
| MOCK_METHOD(rtc::scoped_refptr<DtlsTransportInterface>, |
| LookupDtlsTransportByMid, |
| (const std::string&), |
| (override)); |
| MOCK_METHOD(rtc::scoped_refptr<SctpTransportInterface>, |
| GetSctpTransport, |
| (), |
| (const, override)); |
| MOCK_METHOD(SignalingState, signaling_state, (), (override)); |
| MOCK_METHOD(IceConnectionState, ice_connection_state, (), (override)); |
| MOCK_METHOD(IceConnectionState, |
| standardized_ice_connection_state, |
| (), |
| (override)); |
| MOCK_METHOD(PeerConnectionState, peer_connection_state, (), (override)); |
| MOCK_METHOD(IceGatheringState, ice_gathering_state, (), (override)); |
| MOCK_METHOD(absl::optional<bool>, can_trickle_ice_candidates, (), (override)); |
| MOCK_METHOD(bool, |
| StartRtcEventLog, |
| (std::unique_ptr<RtcEventLogOutput>, int64_t), |
| (override)); |
| MOCK_METHOD(bool, |
| StartRtcEventLog, |
| (std::unique_ptr<RtcEventLogOutput>), |
| (override)); |
| MOCK_METHOD(void, StopRtcEventLog, (), (override)); |
| MOCK_METHOD(void, Close, (), (override)); |
| MOCK_METHOD(rtc::Thread*, signaling_thread, (), (const, override)); |
| |
| // PeerConnectionSdpMethods |
| MOCK_METHOD(std::string, session_id, (), (const, override)); |
| MOCK_METHOD(bool, NeedsIceRestart, (const std::string&), (const, override)); |
| MOCK_METHOD(absl::optional<std::string>, sctp_mid, (), (const, override)); |
| MOCK_METHOD(PeerConnectionInterface::RTCConfiguration*, |
| configuration, |
| (), |
| (const, override)); |
| MOCK_METHOD(void, |
| ReportSdpFormatReceived, |
| (const SessionDescriptionInterface&), |
| (override)); |
| MOCK_METHOD(void, |
| ReportSdpBundleUsage, |
| (const SessionDescriptionInterface&), |
| (override)); |
| MOCK_METHOD(PeerConnectionMessageHandler*, message_handler, (), (override)); |
| MOCK_METHOD(RtpTransmissionManager*, rtp_manager, (), (override)); |
| MOCK_METHOD(const RtpTransmissionManager*, |
| rtp_manager, |
| (), |
| (const, override)); |
| MOCK_METHOD(bool, dtls_enabled, (), (const, override)); |
| MOCK_METHOD(const PeerConnectionFactoryInterface::Options*, |
| options, |
| (), |
| (const, override)); |
| MOCK_METHOD(CryptoOptions, GetCryptoOptions, (), (override)); |
| MOCK_METHOD(JsepTransportController*, transport_controller_s, (), (override)); |
| MOCK_METHOD(JsepTransportController*, transport_controller_n, (), (override)); |
| MOCK_METHOD(DataChannelController*, data_channel_controller, (), (override)); |
| MOCK_METHOD(cricket::PortAllocator*, port_allocator, (), (override)); |
| MOCK_METHOD(LegacyStatsCollector*, legacy_stats, (), (override)); |
| MOCK_METHOD(PeerConnectionObserver*, Observer, (), (const, override)); |
| MOCK_METHOD(bool, GetSctpSslRole, (rtc::SSLRole*), (override)); |
| MOCK_METHOD(PeerConnectionInterface::IceConnectionState, |
| ice_connection_state_internal, |
| (), |
| (override)); |
| MOCK_METHOD(void, |
| SetIceConnectionState, |
| (PeerConnectionInterface::IceConnectionState), |
| (override)); |
| MOCK_METHOD(void, NoteUsageEvent, (UsageEvent), (override)); |
| MOCK_METHOD(bool, IsClosed, (), (const, override)); |
| MOCK_METHOD(bool, IsUnifiedPlan, (), (const, override)); |
| MOCK_METHOD(bool, |
| ValidateBundleSettings, |
| (const cricket::SessionDescription*, |
| (const std::map<std::string, const cricket::ContentGroup*>&)), |
| (override)); |
| MOCK_METHOD(absl::optional<std::string>, GetDataMid, (), (const, override)); |
| MOCK_METHOD(RTCErrorOr<rtc::scoped_refptr<RtpTransceiverInterface>>, |
| AddTransceiver, |
| (cricket::MediaType, |
| rtc::scoped_refptr<MediaStreamTrackInterface>, |
| const RtpTransceiverInit&, |
| bool), |
| (override)); |
| MOCK_METHOD(void, StartSctpTransport, (int, int, int), (override)); |
| MOCK_METHOD(void, |
| AddRemoteCandidate, |
| (const std::string&, const cricket::Candidate&), |
| (override)); |
| MOCK_METHOD(Call*, call_ptr, (), (override)); |
| MOCK_METHOD(bool, SrtpRequired, (), (const, override)); |
| MOCK_METHOD(bool, |
| SetupDataChannelTransport_n, |
| (const std::string&), |
| (override)); |
| MOCK_METHOD(void, TeardownDataChannelTransport_n, (), (override)); |
| MOCK_METHOD(void, SetSctpDataMid, (const std::string&), (override)); |
| MOCK_METHOD(void, ResetSctpDataMid, (), (override)); |
| MOCK_METHOD(const FieldTrialsView&, trials, (), (const, override)); |
| |
| // PeerConnectionInternal |
| MOCK_METHOD(rtc::Thread*, network_thread, (), (const, override)); |
| MOCK_METHOD(rtc::Thread*, worker_thread, (), (const, override)); |
| MOCK_METHOD(bool, initial_offerer, (), (const, override)); |
| MOCK_METHOD( |
| std::vector< |
| rtc::scoped_refptr<RtpTransceiverProxyWithInternal<RtpTransceiver>>>, |
| GetTransceiversInternal, |
| (), |
| (const, override)); |
| MOCK_METHOD(sigslot::signal1<SctpDataChannel*>&, |
| SignalSctpDataChannelCreated, |
| (), |
| (override)); |
| MOCK_METHOD(std::vector<DataChannelStats>, |
| GetDataChannelStats, |
| (), |
| (const, override)); |
| MOCK_METHOD(absl::optional<std::string>, |
| sctp_transport_name, |
| (), |
| (const, override)); |
| MOCK_METHOD(cricket::CandidateStatsList, |
| GetPooledCandidateStats, |
| (), |
| (const, override)); |
| MOCK_METHOD((std::map<std::string, cricket::TransportStats>), |
| GetTransportStatsByNames, |
| (const std::set<std::string>&), |
| (override)); |
| MOCK_METHOD(Call::Stats, GetCallStats, (), (override)); |
| MOCK_METHOD(bool, |
| GetLocalCertificate, |
| (const std::string&, rtc::scoped_refptr<rtc::RTCCertificate>*), |
| (override)); |
| MOCK_METHOD(std::unique_ptr<rtc::SSLCertChain>, |
| GetRemoteSSLCertChain, |
| (const std::string&), |
| (override)); |
| MOCK_METHOD(bool, IceRestartPending, (const std::string&), (const, override)); |
| MOCK_METHOD(bool, |
| GetSslRole, |
| (const std::string&, rtc::SSLRole*), |
| (override)); |
| MOCK_METHOD(void, NoteDataAddedEvent, (), (override)); |
| MOCK_METHOD(void, |
| OnSctpDataChannelClosed, |
| (DataChannelInterface*), |
| (override)); |
| }; |
| |
| } // namespace webrtc |
| |
| #endif // PC_TEST_MOCK_PEER_CONNECTION_INTERNAL_H_ |