Use FrameIsAd to decide whether to isolate a frame in TopDocumentIsolation mode.

content/public/test/test_utils.cc

 // Copyright (c) 2012 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/public/test/test_utils.h"
 
+#include <string>
 #include <utility>
 
+#include "base/base_switches.h"
 #include "base/bind.h"
 #include "base/command_line.h"
 #include "base/location.h"
 #include "base/macros.h"
 #include "base/message_loop/message_loop.h"
 #include "base/run_loop.h"
 #include "base/single_thread_task_runner.h"
 #include "base/strings/utf_string_conversions.h"
 #include "base/task_scheduler/task_scheduler.h"
 #include "base/threading/sequenced_worker_pool.h"
 #include "base/threading/thread_task_runner_handle.h"
 #include "base/values.h"
 #include "build/build_config.h"
+#include "components/variations/variations_switches.h"
 #include "content/common/site_isolation_policy.h"
 #include "content/common/url_schemes.h"
 #include "content/public/browser/browser_child_process_host_iterator.h"
 #include "content/public/browser/notification_service.h"
 #include "content/public/browser/render_frame_host.h"
 #include "content/public/browser/render_process_host.h"
 #include "content/public/browser/web_contents.h"
+#include "content/public/common/content_features.h"
 #include "content/public/common/content_switches.h"
 #include "content/public/common/process_type.h"
 #include "content/public/test/test_launcher.h"
 #include "content/public/test/test_service_manager_context.h"
 #include "testing/gtest/include/gtest/gtest.h"
 #include "url/url_util.h"
 
 #if defined(OS_ANDROID)
 #include "content/browser/android/browser_jni_registrar.h"
 #include "mojo/android/system/mojo_jni_registrar.h"
@@ skipping 160 matching lines @@
 
 bool AreAllSitesIsolatedForTesting() {
   return base::CommandLine::ForCurrentProcess()->HasSwitch(
       switches::kSitePerProcess);
 }
 
 void IsolateAllSitesForTesting(base::CommandLine* command_line) {
   command_line->AppendSwitch(switches::kSitePerProcess);
 }
 
+void EnableTopDocumentIsolationForTesting(base::CommandLine* command_line) {
+  // Enable top-document-isolation feature (features::kTopDocumentIsolation)
+  // and associate it with a fake "TopDocumentIsolation" trial.
+  command_line->AppendSwitchASCII(
+      switches::kEnableFeatures,
+      std::string(features::kTopDocumentIsolation.name) +
+          "<TopDocumentIsolation");
+
+  // Register a fake TopDocumentIsolation/BrowserTests study group.
+  command_line->AppendSwitchASCII(switches::kForceFieldTrials,
+                                  "TopDocumentIsolation/BrowserTests");

[Łukasz Anforowicz, 2017/06/30 16:50:20]

I think it is okay to hardcode "TopDocumentIsolation" and "BrowserTests"
strings, because they are local to this function.

I *think* it is okay if we use a string other than "TopDocumentIsolation" 1)
here, 2) in about_flags.cc (as the last argument of the
FEATURE_WITH_PARAMS_VALUE_TYPE macro), and 3) in Finch configs (in google3
repo).

OTOH, I am open to declaring and using a kTopDocumentIsolationTrialName =
"TopDocumentIsolation" constant.

[Charlie Reis, 2017/06/30 23:28:49]

I think this may be more of a question for the Features folks-- I'm not sure
what the right practice is here.

[Łukasz Anforowicz, 2017/07/01 00:10:53]

Ok.

+
+  // Set kTopDocumentIsolationModeParam to TopDocumentIsolationMode::Xsite.
+  // This is not really needed for content_browsertests (which will go through
+  // ShellContentBrowserClient::ShouldIsolateFrameFromMainContent), but is
+  // needed for browser_tests and other tests that use
+  // ChromeContentBrowserClient::ShouldIsolateFrameFromMainContent.
+  command_line->AppendSwitchASCII(
+      variations::switches::kForceFieldTrialParams,
+      "TopDocumentIsolation.BrowserTests:" +
+          std::string(features::kTopDocumentIsolationModeParam) + "/" +
+          std::to_string(
+              static_cast<int>(features::TopDocumentIsolationMode::Xsite)));

[Łukasz Anforowicz, 2017/06/30 16:50:20]

In the latest patchset, I am not hardcoding a "1" above.

+}
+
 void ResetSchemesAndOriginsWhitelist() {
   url::Shutdown();
   RegisterContentSchemes(false);
   url::Initialize();
 }
 
 #if defined(OS_ANDROID)
 // Registers content/browser and mojo JNI bindings necessary for some types of
 // tests.
 bool RegisterJniForTesting(JNIEnv* env) {
@@ skipping 182 matching lines @@
 
 void WebContentsDestroyedWatcher::Wait() {
   run_loop_.Run();
 }
 
 void WebContentsDestroyedWatcher::WebContentsDestroyed() {
   run_loop_.Quit();
 }
 
 }  // namespace content