SameSite: Correctly set requests' initiator for new tabs.

third_party/WebKit/Source/core/loader/FrameFetchContextTest.cpp

 /*
  * Copyright (c) 2015, Google Inc. All rights reserved.
  *
  * Redistribution and use in source and binary forms, with or without
  * modification, are permitted provided that the following conditions are
  * met:
  *
  *     * Redistributions of source code must retain the above copyright
  * notice, this list of conditions and the following disclaimer.
  *     * Redistributions in binary form must reproduce the above
@@ skipping 25 matching lines @@
 #include "core/frame/FrameOwner.h"
 #include "core/frame/FrameView.h"
 #include "core/frame/Settings.h"
 #include "core/html/HTMLDocument.h"
 #include "core/loader/DocumentLoader.h"
 #include "core/loader/EmptyClients.h"
 #include "core/page/Page.h"
 #include "core/testing/DummyPageHolder.h"
 #include "platform/network/ResourceRequest.h"
 #include "platform/weborigin/KURL.h"
+#include "platform/weborigin/SecurityOrigin.h"
 #include "public/platform/WebAddressSpace.h"
 #include "public/platform/WebCachePolicy.h"
 #include "public/platform/WebInsecureRequestPolicy.h"
 #include "testing/gmock/include/gmock/gmock-generated-function-mockers.h"
 #include "testing/gtest/include/gtest/gtest.h"
 #include <memory>
 
 namespace blink {
 
 class StubFrameLoaderClientWithParent final : public EmptyFrameLoaderClient {
@@ skipping 573 matching lines @@
         ResourceRequest mainRequest(test.url);
         fetchContext->addAdditionalRequestHeaders(mainRequest, FetchMainResource);
         EXPECT_EQ(mainRequest.isExternalRequest(), test.isExternalExpectation);
 
         ResourceRequest subRequest(test.url);
         fetchContext->addAdditionalRequestHeaders(subRequest, FetchSubresource);
         EXPECT_EQ(subRequest.isExternalRequest(), test.isExternalExpectation);
     }
 }
 
+TEST_F(FrameFetchContextTest, PopulateRequestorData)
+{
+    FrameFetchContext::provideDocumentToContext(*fetchContext, document.get());
+    KURL documentURL(ParsedURLString, "http://example.com/document.html");
+    document->setURL(documentURL);
+    document->setSecurityOrigin(SecurityOrigin::create(documentURL));
+
+    {
+        ResourceRequest request("http://not-example.com/");
+        request.setFrameType(WebURLRequest::FrameTypeNone);
+
+        fetchContext->populateRequestorDataIfNecessary(request);
+
+        EXPECT_TRUE(SecurityOrigin::create(document->url())->isSameSchemeHostPortAndSuborigin(request.requestorOrigin().get()));
+        EXPECT_EQ(document->firstPartyForCookies(), request.firstPartyForCookies());
+    }
+
+    {
+        ResourceRequest request("http://not-example.com/");
+        request.setFrameType(WebURLRequest::FrameTypeTopLevel);
+
+        fetchContext->populateRequestorDataIfNecessary(request);
+
+        EXPECT_TRUE(request.requestorOrigin()->isUnique());
+        EXPECT_EQ(KURL(), request.firstPartyForCookies());
+    }
+}
+
 } // namespace blink