Interface RTCRtpReceiver and RTCPeerConnection.getReceivers() 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/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 {
 
 namespace {
 
+uintptr_t GetReceiverIDByTrack(
+    const std::string& track_id,
+    std::map<std::string, uintptr_t>* receiver_id_by_track) {
+  const auto& it = receiver_id_by_track->find(track_id);
+  if (it == receiver_id_by_track->end()) {
+    uintptr_t id = static_cast<uintptr_t>(receiver_id_by_track->size()) + 1;
+    receiver_id_by_track->insert(std::make_pair(track_id, id));
+    return id;
+  }
+  return it->second;
+}
+
 class MockWebRTCLegacyStats : public blink::WebRTCLegacyStats {
  public:
   class MemberIterator : public blink::WebRTCLegacyStatsMemberIterator {
    public:
     MemberIterator(
         const std::vector<std::pair<std::string, std::string>>* values)
         : values_(values) {}
 
     // blink::WebRTCLegacyStatsMemberIterator
     bool isEnd() const override { return i >= values_->size(); }
@@ skipping 170 matching lines @@
       return nullptr;
     return std::unique_ptr<blink::WebRTCStats>(
         new MockWebRTCStats(stats_[i_++]));
   }
 
  private:
   std::vector<MockWebRTCStats> stats_;
   size_t i_;
 };
 
+class MockWebRTCRtpReceiver : public blink::WebRTCRtpReceiver {
+ public:
+  MockWebRTCRtpReceiver(uintptr_t id, const blink::WebMediaStreamTrack& track)
+      : id_(id), track_(track) {}
+  ~MockWebRTCRtpReceiver() override {}
+
+  uintptr_t id() const override { return id_; }
+  const blink::WebMediaStreamTrack& track() const override { return track_; }
+
+ private:
+  uintptr_t id_;
+  blink::WebMediaStreamTrack track_;
+};
+
 }  // namespace
 
 MockWebRTCPeerConnectionHandler::MockWebRTCPeerConnectionHandler()
     : weak_factory_(this) {}
 
 MockWebRTCPeerConnectionHandler::~MockWebRTCPeerConnectionHandler() {}
 
 MockWebRTCPeerConnectionHandler::MockWebRTCPeerConnectionHandler(
     WebRTCPeerConnectionHandlerClient* client,
     TestInterfaces* interfaces)
@@ skipping 278 matching lines @@
   stats.addMember("sequenceUint32", blink::WebRTCStatsMemberTypeSequenceUint32);
   stats.addMember("sequenceInt64", blink::WebRTCStatsMemberTypeSequenceInt64);
   stats.addMember("sequenceUint64", blink::WebRTCStatsMemberTypeSequenceUint64);
   stats.addMember("sequenceDouble", blink::WebRTCStatsMemberTypeSequenceDouble);
   stats.addMember("sequenceString", blink::WebRTCStatsMemberTypeSequenceString);
   report->AddStats(stats);
   callback->OnStatsDelivered(
       std::unique_ptr<blink::WebRTCStatsReport>(report.release()));
 }
 
+blink::WebVector<std::unique_ptr<blink::WebRTCRtpReceiver>>
+MockWebRTCPeerConnectionHandler::getReceivers() {
+  std::vector<std::unique_ptr<blink::WebRTCRtpReceiver>> receivers;
+  for (const auto& pair : remote_streams_) {
+    const auto& remote_stream = pair.second;
+    blink::WebVector<blink::WebMediaStreamTrack> remote_tracks;
+    remote_stream.audioTracks(remote_tracks);
+    for (const auto& remote_track : remote_tracks) {
+      receivers.push_back(
+          std::unique_ptr<blink::WebRTCRtpReceiver>(new MockWebRTCRtpReceiver(
+              GetReceiverIDByTrack(remote_track.id().utf8(),
+                                   &receiver_id_by_track_),
+              remote_track)));
+    }
+    remote_stream.videoTracks(remote_tracks);
+    for (const auto& remote_track : remote_tracks) {
+      receivers.push_back(
+          std::unique_ptr<blink::WebRTCRtpReceiver>(new MockWebRTCRtpReceiver(
+              GetReceiverIDByTrack(remote_track.id().utf8(),
+                                   &receiver_id_by_track_),
+              remote_track)));
+    }
+  }
+  blink::WebVector<std::unique_ptr<blink::WebRTCRtpReceiver>> web_vector(
+      receivers.size());
+  for (size_t i = 0; i < receivers.size(); ++i) {
+    web_vector[i] = std::move(receivers[i]);
+  }
+  return web_vector;
+}
+
 void MockWebRTCPeerConnectionHandler::ReportCreationOfDataChannel() {
   WebRTCDataChannelInit init;
   WebRTCDataChannelHandler* remote_data_channel =
       new MockWebRTCDataChannelHandler("MockRemoteDataChannel", init,
                                        interfaces_->GetDelegate());
   client_->didAddRemoteDataChannel(remote_data_channel);
 }
 
 WebRTCDataChannelHandler* MockWebRTCPeerConnectionHandler::createDataChannel(
     const WebString& label,
@@ skipping 12 matching lines @@
     const WebMediaStreamTrack& track) {
   return new MockWebRTCDTMFSenderHandler(track, interfaces_->GetDelegate());
 }
 
 void MockWebRTCPeerConnectionHandler::stop() {
   stopped_ = true;
   weak_factory_.InvalidateWeakPtrs();
 }
 
 }  // namespace test_runner