| Index: webrtc/api/statscollector_unittest.cc
|
| diff --git a/webrtc/api/statscollector_unittest.cc b/webrtc/api/statscollector_unittest.cc
|
| index bf60d0ed959f9f6395599a5dcbe779e57d10f0ef..50ae51d6b0b9b8b0b375f35119a680f07f848957 100644
|
| --- a/webrtc/api/statscollector_unittest.cc
|
| +++ b/webrtc/api/statscollector_unittest.cc
|
| @@ -41,6 +41,7 @@
|
| using testing::_;
|
| using testing::DoAll;
|
| using testing::Field;
|
| +using testing::Invoke;
|
| using testing::Return;
|
| using testing::ReturnNull;
|
| using testing::ReturnRef;
|
| @@ -509,7 +510,7 @@ class StatsCollectorTest : public testing::Test {
|
| &event_log_)),
|
| session_(media_controller_.get()) {
|
| // By default, we ignore session GetStats calls.
|
| - EXPECT_CALL(session_, GetTransportStats(_)).WillRepeatedly(Return(false));
|
| + EXPECT_CALL(session_, GetStats(_)).WillRepeatedly(ReturnNull());
|
| // Add default returns for mock classes.
|
| EXPECT_CALL(session_, video_channel()).WillRepeatedly(ReturnNull());
|
| EXPECT_CALL(session_, voice_channel()).WillRepeatedly(ReturnNull());
|
| @@ -612,9 +613,11 @@ class StatsCollectorTest : public testing::Test {
|
|
|
| // Instruct the session to return stats containing the transport channel.
|
| InitSessionStats(vc_name);
|
| - EXPECT_CALL(session_, GetTransportStats(_))
|
| - .WillRepeatedly(DoAll(SetArgPointee<0>(session_stats_),
|
| - Return(true)));
|
| + EXPECT_CALL(session_, GetStats(_)).WillRepeatedly(Invoke(
|
| + [this](const ChannelNamePairs&) {
|
| + return std::unique_ptr<SessionStats>(
|
| + new SessionStats(session_stats_));
|
| + }));
|
|
|
| // Constructs an ssrc stats update.
|
| if (voice_sender_info)
|
| @@ -712,9 +715,11 @@ class StatsCollectorTest : public testing::Test {
|
| EXPECT_CALL(session_, GetRemoteSSLCertificate_ReturnsRawPointer(
|
| transport_stats.transport_name))
|
| .WillOnce(Return(remote_cert.release()));
|
| - EXPECT_CALL(session_, GetTransportStats(_))
|
| - .WillOnce(DoAll(SetArgPointee<0>(session_stats),
|
| - Return(true)));
|
| + EXPECT_CALL(session_, GetStats(_)).WillOnce(Invoke(
|
| + [&session_stats](const ChannelNamePairs&) {
|
| + return std::unique_ptr<SessionStats>(
|
| + new SessionStats(session_stats));
|
| + }));
|
|
|
| stats.UpdateStats(PeerConnectionInterface::kStatsOutputLevelStandard);
|
|
|
| @@ -853,9 +858,11 @@ TEST_F(StatsCollectorTest, BytesCounterHandles64Bits) {
|
| const char kVideoChannelName[] = "video";
|
|
|
| InitSessionStats(kVideoChannelName);
|
| - EXPECT_CALL(session_, GetTransportStats(_))
|
| - .WillRepeatedly(DoAll(SetArgPointee<0>(session_stats_),
|
| - Return(true)));
|
| + EXPECT_CALL(session_, GetStats(_)).WillRepeatedly(Invoke(
|
| + [this](const ChannelNamePairs&) {
|
| + return std::unique_ptr<SessionStats>(
|
| + new SessionStats(session_stats_));
|
| + }));
|
|
|
| MockVideoMediaChannel* media_channel = new MockVideoMediaChannel();
|
| cricket::VideoChannel video_channel(
|
| @@ -900,9 +907,11 @@ TEST_F(StatsCollectorTest, BandwidthEstimationInfoIsReported) {
|
| const char kVideoChannelName[] = "video";
|
|
|
| InitSessionStats(kVideoChannelName);
|
| - EXPECT_CALL(session_, GetTransportStats(_))
|
| - .WillRepeatedly(DoAll(SetArgPointee<0>(session_stats_),
|
| - Return(true)));
|
| + EXPECT_CALL(session_, GetStats(_)).WillRepeatedly(Invoke(
|
| + [this](const ChannelNamePairs&) {
|
| + return std::unique_ptr<SessionStats>(
|
| + new SessionStats(session_stats_));
|
| + }));
|
|
|
| MockVideoMediaChannel* media_channel = new MockVideoMediaChannel();
|
| cricket::VideoChannel video_channel(
|
| @@ -1013,9 +1022,11 @@ TEST_F(StatsCollectorTest, TrackAndSsrcObjectExistAfterUpdateSsrcStats) {
|
|
|
| const char kVideoChannelName[] = "video";
|
| InitSessionStats(kVideoChannelName);
|
| - EXPECT_CALL(session_, GetTransportStats(_))
|
| - .WillRepeatedly(DoAll(SetArgPointee<0>(session_stats_),
|
| - Return(true)));
|
| + EXPECT_CALL(session_, GetStats(_)).WillRepeatedly(Invoke(
|
| + [this](const ChannelNamePairs&) {
|
| + return std::unique_ptr<SessionStats>(
|
| + new SessionStats(session_stats_));
|
| + }));
|
|
|
| MockVideoMediaChannel* media_channel = new MockVideoMediaChannel();
|
| cricket::VideoChannel video_channel(
|
| @@ -1110,9 +1121,11 @@ TEST_F(StatsCollectorTest, TransportObjectLinkedFromSsrcObject) {
|
| Return(true)));
|
|
|
| InitSessionStats(kVcName);
|
| - EXPECT_CALL(session_, GetTransportStats(_))
|
| - .WillRepeatedly(DoAll(SetArgPointee<0>(session_stats_),
|
| - Return(true)));
|
| + EXPECT_CALL(session_, GetStats(_)).WillRepeatedly(Invoke(
|
| + [this](const ChannelNamePairs&) {
|
| + return std::unique_ptr<SessionStats>(
|
| + new SessionStats(session_stats_));
|
| + }));
|
|
|
| stats.UpdateStats(PeerConnectionInterface::kStatsOutputLevelStandard);
|
| StatsReports reports;
|
| @@ -1181,9 +1194,11 @@ TEST_F(StatsCollectorTest, RemoteSsrcInfoIsPresent) {
|
|
|
| // Instruct the session to return stats containing the transport channel.
|
| InitSessionStats(kVcName);
|
| - EXPECT_CALL(session_, GetTransportStats(_))
|
| - .WillRepeatedly(DoAll(SetArgPointee<0>(session_stats_),
|
| - Return(true)));
|
| + EXPECT_CALL(session_, GetStats(_)).WillRepeatedly(Invoke(
|
| + [this](const ChannelNamePairs&) {
|
| + return std::unique_ptr<SessionStats>(
|
| + new SessionStats(session_stats_));
|
| + }));
|
|
|
| // Constructs an ssrc stats update.
|
| cricket::VideoMediaInfo stats_read;
|
| @@ -1224,9 +1239,11 @@ TEST_F(StatsCollectorTest, ReportsFromRemoteTrack) {
|
|
|
| const char kVideoChannelName[] = "video";
|
| InitSessionStats(kVideoChannelName);
|
| - EXPECT_CALL(session_, GetTransportStats(_))
|
| - .WillRepeatedly(DoAll(SetArgPointee<0>(session_stats_),
|
| - Return(true)));
|
| + EXPECT_CALL(session_, GetStats(_)).WillRepeatedly(Invoke(
|
| + [this](const ChannelNamePairs&) {
|
| + return std::unique_ptr<SessionStats>(
|
| + new SessionStats(session_stats_));
|
| + }));
|
|
|
| MockVideoMediaChannel* media_channel = new MockVideoMediaChannel();
|
| cricket::VideoChannel video_channel(
|
| @@ -1430,9 +1447,11 @@ TEST_F(StatsCollectorTest, NoTransport) {
|
| transport_stats;
|
|
|
| // Configure MockWebRtcSession
|
| - EXPECT_CALL(session_, GetTransportStats(_))
|
| - .WillOnce(DoAll(SetArgPointee<0>(session_stats),
|
| - Return(true)));
|
| + EXPECT_CALL(session_, GetStats(_)).WillRepeatedly(Invoke(
|
| + [&session_stats](const ChannelNamePairs&) {
|
| + return std::unique_ptr<SessionStats>(
|
| + new SessionStats(session_stats));
|
| + }));
|
|
|
| stats.UpdateStats(PeerConnectionInterface::kStatsOutputLevelStandard);
|
| stats.GetStats(NULL, &reports);
|
| @@ -1487,9 +1506,11 @@ TEST_F(StatsCollectorTest, NoCertificates) {
|
| transport_stats;
|
|
|
| // Configure MockWebRtcSession
|
| - EXPECT_CALL(session_, GetTransportStats(_))
|
| - .WillOnce(DoAll(SetArgPointee<0>(session_stats),
|
| - Return(true)));
|
| + EXPECT_CALL(session_, GetStats(_)).WillRepeatedly(Invoke(
|
| + [&session_stats](const ChannelNamePairs&) {
|
| + return std::unique_ptr<SessionStats>(
|
| + new SessionStats(session_stats));
|
| + }));
|
| stats.UpdateStats(PeerConnectionInterface::kStatsOutputLevelStandard);
|
| stats.GetStats(NULL, &reports);
|
|
|
| @@ -1566,8 +1587,11 @@ TEST_F(StatsCollectorTest, FilterOutNegativeInitialValues) {
|
|
|
| // Instruct the session to return stats containing the transport channel.
|
| InitSessionStats(kVcName);
|
| - EXPECT_CALL(session_, GetTransportStats(_))
|
| - .WillRepeatedly(DoAll(SetArgPointee<0>(session_stats_), Return(true)));
|
| + EXPECT_CALL(session_, GetStats(_)).WillRepeatedly(Invoke(
|
| + [this](const ChannelNamePairs&) {
|
| + return std::unique_ptr<SessionStats>(
|
| + new SessionStats(session_stats_));
|
| + }));
|
|
|
| cricket::VoiceSenderInfo voice_sender_info;
|
| voice_sender_info.add_ssrc(kSsrcOfTrack);
|
| @@ -1720,9 +1744,11 @@ TEST_F(StatsCollectorTest, GetStatsAfterRemoveAudioStream) {
|
|
|
| // Instruct the session to return stats containing the transport channel.
|
| InitSessionStats(kVcName);
|
| - EXPECT_CALL(session_, GetTransportStats(_))
|
| - .WillRepeatedly(DoAll(SetArgPointee<0>(session_stats_),
|
| - Return(true)));
|
| + EXPECT_CALL(session_, GetStats(_)).WillRepeatedly(Invoke(
|
| + [this](const ChannelNamePairs&) {
|
| + return std::unique_ptr<SessionStats>(
|
| + new SessionStats(session_stats_));
|
| + }));
|
|
|
| stats.RemoveLocalAudioTrack(audio_track_.get(), kSsrcOfTrack);
|
| cricket::VoiceSenderInfo voice_sender_info;
|
| @@ -1794,9 +1820,11 @@ TEST_F(StatsCollectorTest, LocalAndRemoteTracksWithSameSsrc) {
|
|
|
| // Instruct the session to return stats containing the transport channel.
|
| InitSessionStats(kVcName);
|
| - EXPECT_CALL(session_, GetTransportStats(_))
|
| - .WillRepeatedly(DoAll(SetArgPointee<0>(session_stats_),
|
| - Return(true)));
|
| + EXPECT_CALL(session_, GetStats(_)).WillRepeatedly(Invoke(
|
| + [this](const ChannelNamePairs&) {
|
| + return std::unique_ptr<SessionStats>(
|
| + new SessionStats(session_stats_));
|
| + }));
|
|
|
| cricket::VoiceSenderInfo voice_sender_info;
|
| InitVoiceSenderInfo(&voice_sender_info);
|
| @@ -1914,8 +1942,11 @@ TEST_F(StatsCollectorTest, VerifyVideoSendSsrcStats) {
|
| const char kVideoChannelName[] = "video";
|
|
|
| InitSessionStats(kVideoChannelName);
|
| - EXPECT_CALL(session_, GetTransportStats(_))
|
| - .WillRepeatedly(DoAll(SetArgPointee<0>(session_stats_), Return(true)));
|
| + EXPECT_CALL(session_, GetStats(_)).WillRepeatedly(Invoke(
|
| + [this](const ChannelNamePairs&) {
|
| + return std::unique_ptr<SessionStats>(
|
| + new SessionStats(session_stats_));
|
| + }));
|
|
|
| MockVideoMediaChannel* media_channel = new MockVideoMediaChannel();
|
| cricket::VideoChannel video_channel(
|
| @@ -1959,8 +1990,11 @@ TEST_F(StatsCollectorTest, VerifyVideoReceiveSsrcStats) {
|
| const char kVideoChannelName[] = "video";
|
|
|
| InitSessionStats(kVideoChannelName);
|
| - EXPECT_CALL(session_, GetTransportStats(_))
|
| - .WillRepeatedly(DoAll(SetArgPointee<0>(session_stats_), Return(true)));
|
| + EXPECT_CALL(session_, GetStats(_)).WillRepeatedly(Invoke(
|
| + [this](const ChannelNamePairs&) {
|
| + return std::unique_ptr<SessionStats>(
|
| + new SessionStats(session_stats_));
|
| + }));
|
|
|
| MockVideoMediaChannel* media_channel = new MockVideoMediaChannel();
|
| cricket::VideoChannel video_channel(
|
|
|