RTCPeerConnection.getStats: Whitelist of stats in unittest.

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 "base/lazy_instance.h"
 #include "base/macros.h"
@@ skipping 102 matching lines @@
     message_loop_runner_->Quit();
   }
 
   PermissionRequestManager* request_manager_;
   bool request_shown_;
   scoped_refptr<content::MessageLoopRunner> message_loop_runner_;
 
   DISALLOW_COPY_AND_ASSIGN(PermissionRequestObserver);
 };
 
+std::vector<std::string> SemicolonListToVector(
+    const std::string str, size_t start_index = 0) {
+  std::vector<std::string> vector;
+  for (size_t begin = start_index, end = begin; begin <= str.length(); ++end) {
+    if (end >= str.length() || str[end] == ';') {
+      vector.push_back(str.substr(begin, end - begin));
+      begin = end + 1;
+    }
+  }
+  return vector;
+}
+
 }  // namespace
 
 WebRtcTestBase::WebRtcTestBase(): detect_errors_in_javascript_(false) {
   // The handler gets set for each test method, but that's fine since this
   // set operation is idempotent.
   logging::SetLogMessageHandler(&JavascriptErrorDetectingLogHandler);
   hit_javascript_errors_.Get() = false;
 
   EnablePixelOutput();
 }
@@ skipping 320 matching lines @@
   std::string javascript = base::StringPrintf(
       "generateAndCloneCertificate(%s)", keygen_algorithm.c_str());
   EXPECT_EQ("ok-generated-and-cloned", ExecuteJavascript(javascript, tab));
 }
 
 void WebRtcTestBase::VerifyStatsGeneratedCallback(
     content::WebContents* tab) const {
   EXPECT_EQ("ok-got-stats", ExecuteJavascript("verifyStatsGenerated()", tab));
 }
 
-void WebRtcTestBase::VerifyStatsGeneratedPromise(
+std::vector<std::string> WebRtcTestBase::VerifyStatsGeneratedPromise(
     content::WebContents* tab) const {
-  EXPECT_EQ("ok-got-stats",
-            ExecuteJavascript("verifyStatsGeneratedPromise()", tab));
+  std::string result = ExecuteJavascript("verifyStatsGeneratedPromise()", tab);
+  EXPECT_TRUE(base::StartsWith(result, "ok-", base::CompareCase::SENSITIVE));
+  return SemicolonListToVector(result, 3);
+}
+
+std::vector<std::string> WebRtcTestBase::GetWhitelistedStatsTypes(
+    content::WebContents* tab) const {
+  return SemicolonListToVector(
+      ExecuteJavascript("getWhitelistedStatsTypes()", tab));
 }
 
 void WebRtcTestBase::SetDefaultVideoCodec(
     content::WebContents* tab,
     const std::string& video_codec) const {
   EXPECT_EQ("ok-forced",
             ExecuteJavascript("forceVideoCodec('" + video_codec + "')", tab));
 }
 
 void WebRtcTestBase::EnableOpusDtx(content::WebContents* tab) const {
   EXPECT_EQ("ok-forced", ExecuteJavascript("forceOpusDtx()", tab));
 }