RTCRtpReceiver.getContributingSources() added.

content/shell/test_runner/mock_webrtc_peer_connection_handler.cc

 // Copyright 2014 The Chromium Authors. All rights reserved.
 // Use of this source code is governed by a BSD-style license that can be
 // found in the LICENSE file.
 
 #include "content/shell/test_runner/mock_webrtc_peer_connection_handler.h"
 
 #include <stddef.h>
 
 #include <string>
 #include <utility>
 #include <vector>
 
 #include "base/bind.h"
 #include "base/bind_helpers.h"
 #include "content/shell/test_runner/mock_webrtc_data_channel_handler.h"
 #include "content/shell/test_runner/mock_webrtc_dtmf_sender_handler.h"
 #include "content/shell/test_runner/test_interfaces.h"
 #include "content/shell/test_runner/web_test_delegate.h"
 #include "third_party/WebKit/public/platform/WebMediaStream.h"
 #include "third_party/WebKit/public/platform/WebMediaStreamSource.h"
 #include "third_party/WebKit/public/platform/WebMediaStreamTrack.h"
 #include "third_party/WebKit/public/platform/WebRTCAnswerOptions.h"
 #include "third_party/WebKit/public/platform/WebRTCDataChannelInit.h"
 #include "third_party/WebKit/public/platform/WebRTCOfferOptions.h"
 #include "third_party/WebKit/public/platform/WebRTCPeerConnectionHandlerClient.h"
+#include "third_party/WebKit/public/platform/WebRTCRtpContributingSource.h"
 #include "third_party/WebKit/public/platform/WebRTCRtpReceiver.h"
 #include "third_party/WebKit/public/platform/WebRTCStatsResponse.h"
 #include "third_party/WebKit/public/platform/WebRTCVoidRequest.h"
 #include "third_party/WebKit/public/platform/WebString.h"
 #include "third_party/WebKit/public/platform/WebVector.h"
 
 using namespace blink;
 
 namespace test_runner {
 
@@ skipping 194 matching lines @@
       return nullptr;
     return std::unique_ptr<blink::WebRTCStats>(
         new MockWebRTCStats(stats_[i_++]));
   }
 
  private:
   std::vector<MockWebRTCStats> stats_;
   size_t i_;
 };
 
+class MockWebRTCRtpContributingSource
+    : public blink::WebRTCRtpContributingSource {
+ public:
+  MockWebRTCRtpContributingSource(
+      blink::WebRTCRtpContributingSourceType source_type,
+      double timestamp_ms,
+      uint32_t source)
+      : source_type_(source_type),
+        timestamp_ms_(timestamp_ms),
+        source_(source) {}
+  ~MockWebRTCRtpContributingSource() override {}
+
+  blink::WebRTCRtpContributingSourceType sourceType() const override {
+    return source_type_;
+  }
+  double timestampMs() const override { return timestamp_ms_; }
+  uint32_t source() const override { return source_; }
+
+ private:
+  blink::WebRTCRtpContributingSourceType source_type_;
+  double timestamp_ms_;
+  uint32_t source_;
+};
+
 class MockWebRTCRtpReceiver : public blink::WebRTCRtpReceiver {
  public:
   MockWebRTCRtpReceiver(uintptr_t id, const blink::WebMediaStreamTrack& track)
-      : id_(id), track_(track) {}
+      : id_(id), track_(track), num_packets_(0) {}
   ~MockWebRTCRtpReceiver() override {}
 
   uintptr_t id() const override { return id_; }
   const blink::WebMediaStreamTrack& track() const override { return track_; }
 
+  // Every time called, mocks that a new packet has arrived updating the i-th
+  // CSRC such that the |kNumCSRCsActive| latest updated CSRCs are returned. "i"
+  // is the sequence number modulo number of CSRCs. Also returns an SSRC with
+  // the latest timestamp. For example, if 2 out of 3 CSRCs should be active,
+  // this will return the following "(type, source, timestamp)":
+  // - 1st call: { (CSRC, 0, 0), (SSRC, 0, 0) }
+  // - 2nd call: { (CSRC, 0, 0000), (CSRC, 1, 5000), (SSRC, 0, 5000) }
+  // - 3rd call: { (CSRC, 1, 5000), (CSRC, 2, 10000), (SSRC, 0, 10000) }
+  // - 4th call: { (CSRC, 2, 10000), (CSRC, 0, 15000), (SSRC, 0, 15000) }
+  // RTCPeerConnection-getReceivers.html depends on this behavior.
+  blink::WebVector<std::unique_ptr<blink::WebRTCRtpContributingSource>>
+  getSources() override {
+    ++num_packets_;
+    size_t num_csrcs = std::min(kNumCSRCsActive, num_packets_);
+    blink::WebVector<std::unique_ptr<blink::WebRTCRtpContributingSource>>
+        contributing_sources(num_csrcs + 1);
+    for (size_t i = 0; i < num_csrcs; ++i) {
+      size_t sequence_number = num_packets_ - num_csrcs + i;
+      contributing_sources[i].reset(new MockWebRTCRtpContributingSource(

[Guido Urdaneta, 2017/04/06 12:54:56]

use MakeUnique

[hbos_chromium, 2017/04/06 14:49:37]

Done.

+          blink::WebRTCRtpContributingSourceType::CSRC,
+          // Return value should include timestamps for the last 10 seconds, we
+          // pretend |10000.0 / kNumCSRCsActive| milliseconds have passed
+          // per packet in the sequence, starting from 0. This is not relative
+          // to any real clock.
+          sequence_number * 10000.0 / kNumCSRCsActive,
+          sequence_number % kNumCSRCs));
+    }
+    contributing_sources[num_csrcs].reset(new MockWebRTCRtpContributingSource(

[Guido Urdaneta, 2017/04/06 12:54:56]

use MakeUnique

[hbos_chromium, 2017/04/06 14:49:37]

Done.

+        blink::WebRTCRtpContributingSourceType::SSRC,
+        contributing_sources[num_csrcs - 1]->timestampMs(), 0));
+    return contributing_sources;
+  }
+
  private:
+  const size_t kNumCSRCs = 3;
+  const size_t kNumCSRCsActive = 2;
+
   uintptr_t id_;
   blink::WebMediaStreamTrack track_;
+  size_t num_packets_;
 };
 
 }  // namespace
 
 MockWebRTCPeerConnectionHandler::MockWebRTCPeerConnectionHandler()
     : weak_factory_(this) {}
 
 MockWebRTCPeerConnectionHandler::~MockWebRTCPeerConnectionHandler() {}
 
 MockWebRTCPeerConnectionHandler::MockWebRTCPeerConnectionHandler(
@@ skipping 343 matching lines @@
     const WebMediaStreamTrack& track) {
   return new MockWebRTCDTMFSenderHandler(track, interfaces_->GetDelegate());
 }
 
 void MockWebRTCPeerConnectionHandler::stop() {
   stopped_ = true;
   weak_factory_.InvalidateWeakPtrs();
 }
 
 }  // namespace test_runner