| Index: webrtc/api/rtcstatscollector_unittest.cc
|
| diff --git a/webrtc/api/rtcstatscollector_unittest.cc b/webrtc/api/rtcstatscollector_unittest.cc
|
| index b973044c94751582d187dbc112f02703b1ff06c8..2681a6995929c31463b6cac6f0dd6d071833448c 100644
|
| --- a/webrtc/api/rtcstatscollector_unittest.cc
|
| +++ b/webrtc/api/rtcstatscollector_unittest.cc
|
| @@ -311,10 +311,10 @@ class RTCStatsCollectorTestHelper : public SetSessionDescriptionObserver {
|
| ReturnRef(data_channels_));
|
| EXPECT_CALL(session_, video_channel()).WillRepeatedly(ReturnNull());
|
| EXPECT_CALL(session_, voice_channel()).WillRepeatedly(ReturnNull());
|
| - EXPECT_CALL(session_, GetTransportStats(_)).WillRepeatedly(Return(false));
|
| - EXPECT_CALL(session_, GetLocalCertificate(_, _)).WillRepeatedly(
|
| + EXPECT_CALL(session_, GetSessionStats_n(_)).WillRepeatedly(ReturnNull());
|
| + EXPECT_CALL(session_, GetLocalCertificate_n(_, _)).WillRepeatedly(
|
| Return(false));
|
| - EXPECT_CALL(session_, GetRemoteSSLCertificate_ReturnsRawPointer(_))
|
| + EXPECT_CALL(session_, GetRemoteSSLCertificate_n_ReturnsRawPointer(_))
|
| .WillRepeatedly(Return(nullptr));
|
| }
|
|
|
| @@ -626,12 +626,13 @@ TEST_F(RTCStatsCollectorTest, CollectRTCCertificateStatsSingle) {
|
| std::vector<std::string>({ "(remote) single certificate" }));
|
|
|
| // Mock the session to return the local and remote certificates.
|
| - EXPECT_CALL(test_->session(), GetTransportStats(_)).WillRepeatedly(Invoke(
|
| - [this](SessionStats* stats) {
|
| + EXPECT_CALL(test_->session(), GetSessionStats_n(_)).WillRepeatedly(Invoke(
|
| + [this](const ChannelNamePairs&) {
|
| + std::unique_ptr<SessionStats> stats(new SessionStats());
|
| stats->transport_stats["transport"].transport_name = "transport";
|
| - return true;
|
| + return stats;
|
| }));
|
| - EXPECT_CALL(test_->session(), GetLocalCertificate(_, _)).WillRepeatedly(
|
| + EXPECT_CALL(test_->session(), GetLocalCertificate_n(_, _)).WillRepeatedly(
|
| Invoke([this, &local_certinfo](const std::string& transport_name,
|
| rtc::scoped_refptr<rtc::RTCCertificate>* certificate) {
|
| if (transport_name == "transport") {
|
| @@ -641,7 +642,7 @@ TEST_F(RTCStatsCollectorTest, CollectRTCCertificateStatsSingle) {
|
| return false;
|
| }));
|
| EXPECT_CALL(test_->session(),
|
| - GetRemoteSSLCertificate_ReturnsRawPointer(_)).WillRepeatedly(Invoke(
|
| + GetRemoteSSLCertificate_n_ReturnsRawPointer(_)).WillRepeatedly(Invoke(
|
| [this, &remote_certinfo](const std::string& transport_name) {
|
| if (transport_name == "transport")
|
| return remote_certinfo->certificate->ssl_certificate().GetReference();
|
| @@ -710,8 +711,10 @@ TEST_F(RTCStatsCollectorTest, CollectRTCCodecStats) {
|
| session_stats.transport_stats["TransportName"].transport_name =
|
| "TransportName";
|
|
|
| - EXPECT_CALL(test_->session(), GetTransportStats(_))
|
| - .WillRepeatedly(DoAll(SetArgPointee<0>(session_stats), Return(true)));
|
| + EXPECT_CALL(test_->session(), GetSessionStats_n(_)).WillRepeatedly(Invoke(
|
| + [&session_stats](const ChannelNamePairs&) {
|
| + return std::unique_ptr<SessionStats>(new SessionStats(session_stats));
|
| + }));
|
| EXPECT_CALL(test_->session(), voice_channel())
|
| .WillRepeatedly(Return(&voice_channel));
|
| EXPECT_CALL(test_->session(), video_channel())
|
| @@ -788,13 +791,14 @@ TEST_F(RTCStatsCollectorTest, CollectRTCCertificateStatsMultiple) {
|
| video_remote_certinfo->ders);
|
|
|
| // Mock the session to return the local and remote certificates.
|
| - EXPECT_CALL(test_->session(), GetTransportStats(_)).WillRepeatedly(Invoke(
|
| - [this](SessionStats* stats) {
|
| + EXPECT_CALL(test_->session(), GetSessionStats_n(_)).WillRepeatedly(Invoke(
|
| + [this](const ChannelNamePairs&) {
|
| + std::unique_ptr<SessionStats> stats(new SessionStats());
|
| stats->transport_stats["audio"].transport_name = "audio";
|
| stats->transport_stats["video"].transport_name = "video";
|
| - return true;
|
| + return stats;
|
| }));
|
| - EXPECT_CALL(test_->session(), GetLocalCertificate(_, _)).WillRepeatedly(
|
| + EXPECT_CALL(test_->session(), GetLocalCertificate_n(_, _)).WillRepeatedly(
|
| Invoke([this, &audio_local_certinfo, &video_local_certinfo](
|
| const std::string& transport_name,
|
| rtc::scoped_refptr<rtc::RTCCertificate>* certificate) {
|
| @@ -809,7 +813,7 @@ TEST_F(RTCStatsCollectorTest, CollectRTCCertificateStatsMultiple) {
|
| return false;
|
| }));
|
| EXPECT_CALL(test_->session(),
|
| - GetRemoteSSLCertificate_ReturnsRawPointer(_)).WillRepeatedly(Invoke(
|
| + GetRemoteSSLCertificate_n_ReturnsRawPointer(_)).WillRepeatedly(Invoke(
|
| [this, &audio_remote_certinfo, &video_remote_certinfo](
|
| const std::string& transport_name) {
|
| if (transport_name == "audio") {
|
| @@ -847,12 +851,13 @@ TEST_F(RTCStatsCollectorTest, CollectRTCCertificateStatsChain) {
|
| CreateFakeCertificateAndInfoFromDers(remote_ders);
|
|
|
| // Mock the session to return the local and remote certificates.
|
| - EXPECT_CALL(test_->session(), GetTransportStats(_)).WillRepeatedly(Invoke(
|
| - [this](SessionStats* stats) {
|
| + EXPECT_CALL(test_->session(), GetSessionStats_n(_)).WillRepeatedly(Invoke(
|
| + [this](const ChannelNamePairs&) {
|
| + std::unique_ptr<SessionStats> stats(new SessionStats());
|
| stats->transport_stats["transport"].transport_name = "transport";
|
| - return true;
|
| + return stats;
|
| }));
|
| - EXPECT_CALL(test_->session(), GetLocalCertificate(_, _)).WillRepeatedly(
|
| + EXPECT_CALL(test_->session(), GetLocalCertificate_n(_, _)).WillRepeatedly(
|
| Invoke([this, &local_certinfo](const std::string& transport_name,
|
| rtc::scoped_refptr<rtc::RTCCertificate>* certificate) {
|
| if (transport_name == "transport") {
|
| @@ -862,7 +867,7 @@ TEST_F(RTCStatsCollectorTest, CollectRTCCertificateStatsChain) {
|
| return false;
|
| }));
|
| EXPECT_CALL(test_->session(),
|
| - GetRemoteSSLCertificate_ReturnsRawPointer(_)).WillRepeatedly(Invoke(
|
| + GetRemoteSSLCertificate_n_ReturnsRawPointer(_)).WillRepeatedly(Invoke(
|
| [this, &remote_certinfo](const std::string& transport_name) {
|
| if (transport_name == "transport")
|
| return remote_certinfo->certificate->ssl_certificate().GetReference();
|
| @@ -975,10 +980,9 @@ TEST_F(RTCStatsCollectorTest, CollectRTCIceCandidateStats) {
|
| b_transport_channel_stats);
|
|
|
| // Mock the session to return the desired candidates.
|
| - EXPECT_CALL(test_->session(), GetTransportStats(_)).WillRepeatedly(Invoke(
|
| - [this, &session_stats](SessionStats* stats) {
|
| - *stats = session_stats;
|
| - return true;
|
| + EXPECT_CALL(test_->session(), GetSessionStats_n(_)).WillRepeatedly(Invoke(
|
| + [&session_stats](const ChannelNamePairs&) {
|
| + return std::unique_ptr<SessionStats>(new SessionStats(session_stats));
|
| }));
|
|
|
| rtc::scoped_refptr<const RTCStatsReport> report = GetStatsReport();
|
| @@ -1019,10 +1023,9 @@ TEST_F(RTCStatsCollectorTest, CollectRTCIceCandidatePairStats) {
|
| transport_channel_stats);
|
|
|
| // Mock the session to return the desired candidates.
|
| - EXPECT_CALL(test_->session(), GetTransportStats(_)).WillRepeatedly(Invoke(
|
| - [this, &session_stats](SessionStats* stats) {
|
| - *stats = session_stats;
|
| - return true;
|
| + EXPECT_CALL(test_->session(), GetSessionStats_n(_)).WillRepeatedly(Invoke(
|
| + [&session_stats](const ChannelNamePairs&) {
|
| + return std::unique_ptr<SessionStats>(new SessionStats(session_stats));
|
| }));
|
|
|
| rtc::scoped_refptr<const RTCStatsReport> report = GetStatsReport();
|
| @@ -1343,8 +1346,10 @@ TEST_F(RTCStatsCollectorTest, CollectRTCInboundRTPStreamStats_Audio) {
|
| session_stats.transport_stats["TransportName"].channel_stats.push_back(
|
| channel_stats);
|
|
|
| - EXPECT_CALL(test_->session(), GetTransportStats(_))
|
| - .WillRepeatedly(DoAll(SetArgPointee<0>(session_stats), Return(true)));
|
| + EXPECT_CALL(test_->session(), GetSessionStats_n(_)).WillRepeatedly(Invoke(
|
| + [&session_stats](const ChannelNamePairs&) {
|
| + return std::unique_ptr<SessionStats>(new SessionStats(session_stats));
|
| + }));
|
| EXPECT_CALL(test_->session(), voice_channel())
|
| .WillRepeatedly(Return(&voice_channel));
|
|
|
| @@ -1415,8 +1420,10 @@ TEST_F(RTCStatsCollectorTest, CollectRTCInboundRTPStreamStats_Video) {
|
| session_stats.transport_stats["TransportName"].channel_stats.push_back(
|
| channel_stats);
|
|
|
| - EXPECT_CALL(test_->session(), GetTransportStats(_))
|
| - .WillRepeatedly(DoAll(SetArgPointee<0>(session_stats), Return(true)));
|
| + EXPECT_CALL(test_->session(), GetSessionStats_n(_)).WillRepeatedly(Invoke(
|
| + [&session_stats](const ChannelNamePairs&) {
|
| + return std::unique_ptr<SessionStats>(new SessionStats(session_stats));
|
| + }));
|
| EXPECT_CALL(test_->session(), video_channel())
|
| .WillRepeatedly(Return(&video_channel));
|
|
|
| @@ -1484,8 +1491,10 @@ TEST_F(RTCStatsCollectorTest, CollectRTCOutboundRTPStreamStats_Audio) {
|
| session_stats.transport_stats["TransportName"].channel_stats.push_back(
|
| channel_stats);
|
|
|
| - EXPECT_CALL(test_->session(), GetTransportStats(_))
|
| - .WillRepeatedly(DoAll(SetArgPointee<0>(session_stats), Return(true)));
|
| + EXPECT_CALL(test_->session(), GetSessionStats_n(_)).WillRepeatedly(Invoke(
|
| + [&session_stats](const ChannelNamePairs&) {
|
| + return std::unique_ptr<SessionStats>(new SessionStats(session_stats));
|
| + }));
|
| EXPECT_CALL(test_->session(), voice_channel())
|
| .WillRepeatedly(Return(&voice_channel));
|
|
|
| @@ -1552,8 +1561,10 @@ TEST_F(RTCStatsCollectorTest, CollectRTCOutboundRTPStreamStats_Video) {
|
| session_stats.transport_stats["TransportName"].channel_stats.push_back(
|
| channel_stats);
|
|
|
| - EXPECT_CALL(test_->session(), GetTransportStats(_))
|
| - .WillRepeatedly(DoAll(SetArgPointee<0>(session_stats), Return(true)));
|
| + EXPECT_CALL(test_->session(), GetSessionStats_n(_)).WillRepeatedly(Invoke(
|
| + [&session_stats](const ChannelNamePairs&) {
|
| + return std::unique_ptr<SessionStats>(new SessionStats(session_stats));
|
| + }));
|
| EXPECT_CALL(test_->session(), video_channel())
|
| .WillRepeatedly(Return(&video_channel));
|
|
|
| @@ -1613,10 +1624,9 @@ TEST_F(RTCStatsCollectorTest, CollectRTCTransportStats) {
|
|
|
|
|
| // Mock the session to return the desired candidates.
|
| - EXPECT_CALL(test_->session(), GetTransportStats(_)).WillRepeatedly(Invoke(
|
| - [this, &session_stats](SessionStats* stats) {
|
| - *stats = session_stats;
|
| - return true;
|
| + EXPECT_CALL(test_->session(), GetSessionStats_n(_)).WillRepeatedly(Invoke(
|
| + [&session_stats](const ChannelNamePairs&) {
|
| + return std::unique_ptr<SessionStats>(new SessionStats(session_stats));
|
| }));
|
|
|
| // Get stats without RTCP, an active connection or certificates.
|
| @@ -1701,7 +1711,7 @@ TEST_F(RTCStatsCollectorTest, CollectRTCTransportStats) {
|
| std::unique_ptr<CertificateInfo> remote_certinfo =
|
| CreateFakeCertificateAndInfoFromDers(
|
| std::vector<std::string>({ "(remote) local", "(remote) chain" }));
|
| - EXPECT_CALL(test_->session(), GetLocalCertificate(_, _)).WillRepeatedly(
|
| + EXPECT_CALL(test_->session(), GetLocalCertificate_n(_, _)).WillRepeatedly(
|
| Invoke([this, &local_certinfo](const std::string& transport_name,
|
| rtc::scoped_refptr<rtc::RTCCertificate>* certificate) {
|
| if (transport_name == "transport") {
|
| @@ -1711,7 +1721,7 @@ TEST_F(RTCStatsCollectorTest, CollectRTCTransportStats) {
|
| return false;
|
| }));
|
| EXPECT_CALL(test_->session(),
|
| - GetRemoteSSLCertificate_ReturnsRawPointer(_)).WillRepeatedly(Invoke(
|
| + GetRemoteSSLCertificate_n_ReturnsRawPointer(_)).WillRepeatedly(Invoke(
|
| [this, &remote_certinfo](const std::string& transport_name) {
|
| if (transport_name == "transport")
|
| return remote_certinfo->certificate->ssl_certificate().GetReference();
|
|
|