RTCRtpSender with track behind RuntimeEnabled flag

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 "base/memory/ptr_util.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/WebRTCRtpSender.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));
+uintptr_t GetIDByTrack(const std::string& track_id,
+                       std::map<std::string, uintptr_t>* id_by_track) {
+  const auto& it = id_by_track->find(track_id);
+  if (it == id_by_track->end()) {
+    uintptr_t id = static_cast<uintptr_t>(id_by_track->size()) + 1;
+    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(
@@ skipping 176 matching lines @@
     return std::unique_ptr<blink::WebRTCStats>(
         new MockWebRTCStats(stats_[i_++]));
   }
   size_t Size() const override { return stats_.size(); }
 
  private:
   std::vector<MockWebRTCStats> stats_;
   size_t i_;
 };
 
+class MockWebRTCRtpSender : public blink::WebRTCRtpSender {
+ public:
+  MockWebRTCRtpSender(uintptr_t id,
+                      std::unique_ptr<blink::WebMediaStreamTrack> track)
+      : id_(id), track_(std::move(track)) {}
+
+  uintptr_t Id() const override { return id_; }
+  const blink::WebMediaStreamTrack* Track() const override {
+    return track_.get();
+  }
+
+ private:
+  uintptr_t id_;
+  std::unique_ptr<blink::WebMediaStreamTrack> track_;
+};
+
 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) {}
@@ skipping 359 matching lines @@
                   blink::kWebRTCStatsMemberTypeSequenceUint64);
   stats.addMember("sequenceDouble",
                   blink::kWebRTCStatsMemberTypeSequenceDouble);
   stats.addMember("sequenceString",
                   blink::kWebRTCStatsMemberTypeSequenceString);
   report->AddStats(stats);
   callback->OnStatsDelivered(
       std::unique_ptr<blink::WebRTCStatsReport>(report.release()));
 }
 
+blink::WebVector<std::unique_ptr<blink::WebRTCRtpSender>>
+MockWebRTCPeerConnectionHandler::GetSenders() {
+  std::vector<std::unique_ptr<blink::WebRTCRtpSender>> senders;
+  for (const auto& pair : local_streams_) {
+    const auto& local_stream = pair.second;
+    blink::WebVector<blink::WebMediaStreamTrack> local_tracks;
+    local_stream.AudioTracks(local_tracks);
+    for (const auto& local_track : local_tracks) {
+      senders.push_back(
+          std::unique_ptr<blink::WebRTCRtpSender>(new MockWebRTCRtpSender(
+              GetIDByTrack(local_track.Id().Utf8(), &id_by_track_),
+              base::MakeUnique<WebMediaStreamTrack>(local_track))));
+    }
+    local_stream.VideoTracks(local_tracks);
+    for (const auto& local_track : local_tracks) {
+      senders.push_back(
+          std::unique_ptr<blink::WebRTCRtpSender>(new MockWebRTCRtpSender(
+              GetIDByTrack(local_track.Id().Utf8(), &id_by_track_),
+              base::MakeUnique<WebMediaStreamTrack>(local_track))));
+    }
+  }
+  blink::WebVector<std::unique_ptr<blink::WebRTCRtpSender>> web_vector(
+      senders.size());
+  for (size_t i = 0; i < senders.size(); ++i) {
+    web_vector[i] = std::move(senders[i]);
+  }
+  return web_vector;
+}
+
 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_),
+              GetIDByTrack(remote_track.Id().Utf8(), &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_),
+              GetIDByTrack(remote_track.Id().Utf8(), &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;
 }
@@ skipping 23 matching lines @@
     const WebMediaStreamTrack& track) {
   return new MockWebRTCDTMFSenderHandler(track, interfaces_->GetDelegate());
 }
 
 void MockWebRTCPeerConnectionHandler::Stop() {
   stopped_ = true;
   weak_factory_.InvalidateWeakPtrs();
 }
 
 }  // namespace test_runner