| Index: content/browser/frame_host/navigation_controller_impl_unittest.cc
|
| diff --git a/content/browser/frame_host/navigation_controller_impl_unittest.cc b/content/browser/frame_host/navigation_controller_impl_unittest.cc
|
| index 1520c031a50a64ac69c377addf9f26fb664ee273..6303b067c417822260621d26496c8d881f0237ed 100644
|
| --- a/content/browser/frame_host/navigation_controller_impl_unittest.cc
|
| +++ b/content/browser/frame_host/navigation_controller_impl_unittest.cc
|
| @@ -32,6 +32,7 @@
|
| #include "content/public/common/content_switches.h"
|
| #include "content/public/common/page_state.h"
|
| #include "content/public/common/page_type.h"
|
| +#include "content/public/common/site_isolation_policy.h"
|
| #include "content/public/common/url_constants.h"
|
| #include "content/public/test/mock_render_process_host.h"
|
| #include "content/public/test/test_notification_tracker.h"
|
| @@ -2083,8 +2084,7 @@ TEST_F(NavigationControllerTest, NewSubframe) {
|
| EXPECT_EQ(params.page_id, entry->GetPageID());
|
|
|
| // Verify subframe entries if we're in --site-per-process mode.
|
| - if (base::CommandLine::ForCurrentProcess()->HasSwitch(
|
| - switches::kSitePerProcess)) {
|
| + if (SiteIsolationPolicy::AreAllSitesIsolatedForTesting()) {
|
| // The entry should have a subframe FrameNavigationEntry.
|
| ASSERT_EQ(1U, entry->root_node()->children.size());
|
| EXPECT_EQ(url2, entry->root_node()->children[0]->frame_entry->url());
|
| @@ -2141,8 +2141,7 @@ TEST_F(NavigationControllerTest, AutoSubframe) {
|
| EXPECT_EQ(url1, root_entry->url());
|
|
|
| // Verify subframe entries if we're in --site-per-process mode.
|
| - if (base::CommandLine::ForCurrentProcess()->HasSwitch(
|
| - switches::kSitePerProcess)) {
|
| + if (SiteIsolationPolicy::AreAllSitesIsolatedForTesting()) {
|
| // The entry should now have a subframe FrameNavigationEntry.
|
| ASSERT_EQ(1U, entry->root_node()->children.size());
|
| FrameNavigationEntry* frame_entry =
|
| @@ -2187,8 +2186,7 @@ TEST_F(NavigationControllerTest, AutoSubframe) {
|
| EXPECT_EQ(url1, root_entry->url());
|
|
|
| // Verify subframe entries if we're in --site-per-process mode.
|
| - if (base::CommandLine::ForCurrentProcess()->HasSwitch(
|
| - switches::kSitePerProcess)) {
|
| + if (SiteIsolationPolicy::AreAllSitesIsolatedForTesting()) {
|
| // The entry should now have 2 subframe FrameNavigationEntries.
|
| ASSERT_EQ(2U, entry->root_node()->children.size());
|
| FrameNavigationEntry* new_frame_entry =
|
| @@ -2237,8 +2235,7 @@ TEST_F(NavigationControllerTest, AutoSubframe) {
|
| EXPECT_EQ(url1, root_entry->url());
|
|
|
| // Verify subframe entries if we're in --site-per-process mode.
|
| - if (base::CommandLine::ForCurrentProcess()->HasSwitch(
|
| - switches::kSitePerProcess)) {
|
| + if (SiteIsolationPolicy::AreAllSitesIsolatedForTesting()) {
|
| // The entry should now have a nested FrameNavigationEntry.
|
| EXPECT_EQ(2U, entry->root_node()->children.size());
|
| ASSERT_EQ(1U, entry->root_node()->children[0]->children.size());
|
| @@ -2311,8 +2308,7 @@ TEST_F(NavigationControllerTest, BackSubframe) {
|
| EXPECT_EQ(2, controller.GetEntryCount());
|
|
|
| // Verify subframe entries if we're in --site-per-process mode.
|
| - if (base::CommandLine::ForCurrentProcess()->HasSwitch(
|
| - switches::kSitePerProcess)) {
|
| + if (SiteIsolationPolicy::AreAllSitesIsolatedForTesting()) {
|
| // The entry should have a subframe FrameNavigationEntry.
|
| ASSERT_EQ(1U, entry2->root_node()->children.size());
|
| EXPECT_EQ(url2, entry2->root_node()->children[0]->frame_entry->url());
|
| @@ -2336,8 +2332,7 @@ TEST_F(NavigationControllerTest, BackSubframe) {
|
| EXPECT_EQ(2, controller.GetCurrentEntryIndex());
|
|
|
| // Verify subframe entries if we're in --site-per-process mode.
|
| - if (base::CommandLine::ForCurrentProcess()->HasSwitch(
|
| - switches::kSitePerProcess)) {
|
| + if (SiteIsolationPolicy::AreAllSitesIsolatedForTesting()) {
|
| // The entry should have a subframe FrameNavigationEntry.
|
| ASSERT_EQ(1U, entry3->root_node()->children.size());
|
| EXPECT_EQ(url3, entry3->root_node()->children[0]->frame_entry->url());
|
|
|