Make Document::isSecureContext() work for OOPIFs

third_party/WebKit/Source/core/dom/DocumentTest.cpp

Index: third_party/WebKit/Source/core/dom/DocumentTest.cpp
diff --git a/third_party/WebKit/Source/core/dom/DocumentTest.cpp b/third_party/WebKit/Source/core/dom/DocumentTest.cpp
index b83d53bc9b6e018e041bf0db41ab1f61f87dbb1e..148d0380e4fc9ae1a0dd849c4e1b89ff1a88ad8d 100644
--- a/third_party/WebKit/Source/core/dom/DocumentTest.cpp
+++ b/third_party/WebKit/Source/core/dom/DocumentTest.cpp
@@ -37,6 +37,7 @@
 #include "core/testing/DummyPageHolder.h"
 #include "platform/heap/Handle.h"
 #include "platform/weborigin/ReferrerPolicy.h"
+#include "platform/weborigin/SchemeRegistry.h"
 #include "platform/weborigin/SecurityOrigin.h"
 #include "testing/gmock/include/gmock/gmock.h"
 #include "testing/gtest/include/gtest/gtest.h"
@@ -353,4 +354,32 @@ TEST_F(DocumentTest, StyleVersion)
     EXPECT_NE(previousStyleVersion, document().styleVersion());
 }
 
+TEST_F(DocumentTest, EnforceSandboxFlags)
+{
+    RefPtr<SecurityOrigin> origin = SecurityOrigin::createFromString("http://example.test");
+    document().setSecurityOrigin(origin);
+    SandboxFlags mask = SandboxNavigation;
+    document().enforceSandboxFlags(mask);
+    EXPECT_EQ(origin, document().securityOrigin());
+    EXPECT_FALSE(document().securityOrigin()->isPotentiallyTrustworthy());
+
+    mask |= SandboxOrigin;
+    document().enforceSandboxFlags(mask);
+    EXPECT_TRUE(document().securityOrigin()->isUnique());
+    EXPECT_FALSE(document().securityOrigin()->isPotentiallyTrustworthy());
+
+    SchemeRegistry::registerURLSchemeAsSecure("very-special-scheme");
+    origin = SecurityOrigin::createFromString("very-special-scheme://example.test");
+    document().setSecurityOrigin(origin);
+    document().enforceSandboxFlags(mask);
+    EXPECT_TRUE(document().securityOrigin()->isUnique());
+    EXPECT_TRUE(document().securityOrigin()->isPotentiallyTrustworthy());
+
+    origin = SecurityOrigin::createFromString("https://example.test");
+    document().setSecurityOrigin(origin);
+    document().enforceSandboxFlags(mask);
+    EXPECT_TRUE(document().securityOrigin()->isUnique());
+    EXPECT_TRUE(document().securityOrigin()->isPotentiallyTrustworthy());
+}
+
 } // namespace blink