RTCPeerConnection.addTrack and removeTrack added (behind 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>
@@ skipping 233 matching lines @@
  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();
   }
 
+  void SetTrack(std::unique_ptr<blink::WebMediaStreamTrack> track) {
+    track_ = std::move(track);
+  }
+
  private:
   uintptr_t id_;
   std::unique_ptr<blink::WebMediaStreamTrack> track_;
 };
 
 class MockWebRTCRtpContributingSource
     : public blink::WebRTCRtpContributingSource {
  public:
   MockWebRTCRtpContributingSource(
       blink::WebRTCRtpContributingSourceType source_type,
@@ skipping 367 matching lines @@
   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;
+  // Senders of tracks in |local_streams_| (from |Add/RemoveStream|).
   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))));
     }
   }
+  // Senders of tracks in |tracks_| (from |Add/RemoveTrack|).
+  for (const auto& pair : tracks_) {
+    const auto& track = pair.second;
+    bool has_sender_for_track = false;
+    for (const auto& sender : senders) {
+      if (sender->Track()->Id() == track.Id()) {
+        has_sender_for_track = true;
+        break;
+      }
+    }
+    if (has_sender_for_track)
+      continue;
+    senders.push_back(base::MakeUnique<MockWebRTCRtpSender>(
+        GetIDByTrack(track.Id().Utf8(), &id_by_track_),
+        base::MakeUnique<WebMediaStreamTrack>(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() {
@@ skipping 17 matching lines @@
     }
   }
   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;
 }
 
+std::unique_ptr<blink::WebRTCRtpSender>
+MockWebRTCPeerConnectionHandler::AddTrack(
+    const blink::WebMediaStreamTrack& web_track,
+    const blink::WebVector<blink::WebMediaStream>& web_streams) {
+  for (const auto& sender : GetSenders()) {
+    if (sender->Track() && sender->Track()->Id() == web_track.Id()) {
+      return nullptr;
+    }
+  }
+  tracks_[web_track.Id().Utf8()] = web_track;
+  client_->NegotiationNeeded();
+  return base::MakeUnique<MockWebRTCRtpSender>(
+      GetIDByTrack(web_track.Id().Utf8(), &id_by_track_),
+      base::MakeUnique<blink::WebMediaStreamTrack>(web_track));
+}
+
+bool MockWebRTCPeerConnectionHandler::RemoveTrack(
+    blink::WebRTCRtpSender* web_sender) {
+  if (!tracks_.erase(web_sender->Track()->Id().Utf8()))
+    return false;
+  MockWebRTCRtpSender* sender = static_cast<MockWebRTCRtpSender*>(web_sender);
+  sender->SetTrack(nullptr);
+  client_->NegotiationNeeded();
+  return true;
+}
+
 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