RTCPeerConnection.addTrack and removeTrack added (behind flag)

chrome/browser/media/webrtc/webrtc_browsertest_base.cc

 // Copyright 2013 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 "chrome/browser/media/webrtc/webrtc_browsertest_base.h"
 
 #include <stddef.h>
 
 #include <limits>
 
 #include "base/json/json_reader.h"
 #include "base/lazy_instance.h"
 #include "base/logging.h"
 #include "base/macros.h"
 #include "base/strings/string_number_conversions.h"
+#include "base/strings/string_split.h"
 #include "base/strings/string_util.h"
 #include "base/strings/stringprintf.h"
 #include "base/values.h"
 #include "build/build_config.h"
 #include "chrome/browser/media/webrtc/webrtc_browsertest_common.h"
 #include "chrome/browser/permissions/permission_request_manager.h"
 #include "chrome/browser/ui/browser.h"
 #include "chrome/browser/ui/browser_tabstrip.h"
 #include "chrome/browser/ui/tabs/tab_strip_model.h"
 #include "chrome/test/base/ui_test_utils.h"
@@ skipping 31 matching lines @@
     "failed-with-error-PermissionDismissedError";
 const char WebRtcTestBase::kAudioVideoCallConstraints360p[] =
    "{audio: true, video: {mandatory: {minWidth: 640, maxWidth: 640, "
    " minHeight: 360, maxHeight: 360}}}";
 const char WebRtcTestBase::kAudioVideoCallConstraints720p[] =
    "{audio: true, video: {mandatory: {minWidth: 1280, maxWidth: 1280, "
    " minHeight: 720, maxHeight: 720}}}";
 const char WebRtcTestBase::kUseDefaultCertKeygen[] = "null";
 const char WebRtcTestBase::kUseDefaultAudioCodec[] = "";
 const char WebRtcTestBase::kUseDefaultVideoCodec[] = "";
+const char WebRtcTestBase::kUndefined[] = "undefined";
 
 namespace {
 
 base::LazyInstance<bool>::DestructorAtExit hit_javascript_errors_ =
     LAZY_INSTANCE_INITIALIZER;
 
 // Intercepts all log messages. We always attach this handler but only look at
 // the results if the test requests so. Note that this will only work if the
 // WebrtcTestBase-inheriting test cases do not run in parallel (if they did they
 // would race to look at the log, which is global to all tests).
@@ skipping 511 matching lines @@
 
 void WebRtcTestBase::VerifyRtpReceivers(
     content::WebContents* tab,
     base::Optional<size_t> expected_num_tracks) const {
   std::string javascript =
       expected_num_tracks ? "verifyRtpReceivers(" +
                                 base::SizeTToString(*expected_num_tracks) + ")"
                           : "verifyRtpReceivers()";
   EXPECT_EQ("ok-receivers-verified", ExecuteJavascript(javascript, tab));
 }
+
+std::vector<std::string> WebRtcTestBase::CreateAndAddAudioAndVideoTrack(
+    content::WebContents* tab,
+    StreamArgumentType stream_argument_type) const {
+  const char* string_argument_type_str = nullptr;
+  switch (stream_argument_type) {
+    case StreamArgumentType::NO_STREAM:
+      string_argument_type_str = "'no-stream'";
+      break;
+    case StreamArgumentType::SHARED_STREAM:
+      string_argument_type_str = "'shared-stream'";
+      break;
+    case StreamArgumentType::INDIVIDUAL_STREAMS:
+      string_argument_type_str = "'individual-streams'";
+      break;
+  }
+  std::string result =
+      ExecuteJavascript(base::StringPrintf("createAndAddAudioAndVideoTrack(%s)",
+                                           string_argument_type_str),
+                        tab);
+  EXPECT_TRUE(base::StartsWith(result, "ok-", base::CompareCase::SENSITIVE));
+  std::vector<std::string> ids = base::SplitString(
+      result.substr(3), " ", base::KEEP_WHITESPACE, base::SPLIT_WANT_ALL);
+  EXPECT_EQ(4u, ids.size());
+  return ids;
+}
+
+void WebRtcTestBase::RemoveTrack(content::WebContents* tab,
+                                 const std::string& track_id) const {
+  EXPECT_EQ(
+      "ok-sender-removed",
+      ExecuteJavascript(
+          base::StringPrintf("removeTrack('%s')", track_id.c_str()), tab));
+}
+
+bool WebRtcTestBase::HasLocalStreamWithTrack(
+    content::WebContents* tab,
+    const std::string& stream_id,
+    const std::string& track_id) const {
+  return HasStreamWithTrack(tab, "hasLocalStreamWithTrack", stream_id,
+                            track_id);
+}
+
+bool WebRtcTestBase::HasRemoteStreamWithTrack(
+    content::WebContents* tab,
+    const std::string& stream_id,
+    const std::string& track_id) const {
+  return HasStreamWithTrack(tab, "hasRemoteStreamWithTrack", stream_id,
+                            track_id);
+}
+
+bool WebRtcTestBase::HasStreamWithTrack(content::WebContents* tab,
+                                        const char* function_name,
+                                        std::string stream_id,
+                                        std::string track_id) const {
+  if (stream_id != kUndefined)
+    stream_id = "'" + stream_id + "'";
+  std::string javascript = base::StringPrintf(
+      "%s(%s, '%s')", function_name, stream_id.c_str(), track_id.c_str());
+  std::string result = ExecuteJavascript(javascript, tab);
+  EXPECT_TRUE(result == "ok-stream-with-track-found" ||
+              result == "ok-stream-with-track-not-found");
+  return result == "ok-stream-with-track-found";
+}
+
+bool WebRtcTestBase::HasSenderWithTrack(content::WebContents* tab,
+                                        std::string track_id) const {
+  std::string javascript =
+      base::StringPrintf("hasSenderWithTrack('%s')", track_id.c_str());
+  std::string result = ExecuteJavascript(javascript, tab);
+  EXPECT_TRUE(result == "ok-sender-with-track-found" ||
+              result == "ok-sender-with-track-not-found");
+  return result == "ok-sender-with-track-found";
+}
+
+bool WebRtcTestBase::HasReceiverWithTrack(content::WebContents* tab,
+                                          std::string track_id) const {
+  std::string javascript =
+      base::StringPrintf("hasReceiverWithTrack('%s')", track_id.c_str());
+  std::string result = ExecuteJavascript(javascript, tab);
+  EXPECT_TRUE(result == "ok-receiver-with-track-found" ||
+              result == "ok-receiver-with-track-not-found");
+  return result == "ok-receiver-with-track-found";
+}
+
+size_t WebRtcTestBase::GetNegotiationNeededCount(
+    content::WebContents* tab) const {
+  std::string result = ExecuteJavascript("getNegotiationNeededCount()", tab);
+  EXPECT_TRUE(base::StartsWith(result, "ok-negotiation-count-is-",
+                               base::CompareCase::SENSITIVE));
+  size_t count = 0;
+  EXPECT_TRUE(base::StringToSizeT(result.substr(24), &count));
+  return count;
+}