Tools/TestWebKitAPI/Tests/WebKitCocoa/WKHTTPCookieStore.mm - WebKit - Git at Google

 /*
  * Copyright (C) 2017 Apple 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:
  * 1. Redistributions of source code must retain the above copyright
  *    notice, this list of conditions and the following disclaimer.
  * 2. Redistributions in binary form must reproduce the above copyright
  *    notice, this list of conditions and the following disclaimer in the
  *    documentation and/or other materials provided with the distribution.
  *
  * THIS SOFTWARE IS PROVIDED BY APPLE INC. AND ITS CONTRIBUTORS ``AS IS''
  * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO,
  * THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
  * PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL APPLE INC. OR ITS CONTRIBUTORS
  * BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
  * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
  * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
  * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
  * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
  * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF
  * THE POSSIBILITY OF SUCH DAMAGE.
  */

 #include "config.h"

 #import "PlatformUtilities.h"
 #import "TestNavigationDelegate.h"
 #import <WebKit/WKFoundation.h>
 #import <WebKit/WKHTTPCookieStore.h>
 #import <WebKit/WKProcessPoolPrivate.h>
 #import <WebKit/WKWebsiteDataStorePrivate.h>
 #import <WebKit/_WKWebsiteDataStoreConfiguration.h>
 #import <wtf/RetainPtr.h>

 #if WK_API_ENABLED

 static bool gotFlag;
 static uint64_t observerCallbacks;
 static RetainPtr<WKHTTPCookieStore> globalCookieStore;

 @interface CookieObserver : NSObject<WKHTTPCookieStoreObserver>
 - (void)cookiesDidChangeInCookieStore:(WKHTTPCookieStore *)cookieStore;
 @end

 @implementation CookieObserver

 - (void)cookiesDidChangeInCookieStore:(WKHTTPCookieStore *)cookieStore
 {
     ASSERT_EQ(cookieStore, globalCookieStore.get());
     ++observerCallbacks;
 }

 @end

 static void runTestWithWebsiteDataStore(WKWebsiteDataStore* dataStore)
 {
     observerCallbacks = 0;

     auto configuration = adoptNS([[WKWebViewConfiguration alloc] init]);
     configuration.get().websiteDataStore = dataStore;
     auto webView = adoptNS([[WKWebView alloc] initWithFrame:NSMakeRect(0, 0, 800, 600) configuration:configuration.get()]);

     [webView loadHTMLString:@"Oh hello" baseURL:[NSURL URLWithString:@"http://webkit.org"]];
     [webView _test_waitForDidFinishNavigation];

     [dataStore removeDataOfTypes:[WKWebsiteDataStore allWebsiteDataTypes] modifiedSince:[NSDate distantPast] completionHandler:[] {
         gotFlag = true;
     }];

     TestWebKitAPI::Util::run(&gotFlag);
     gotFlag = false;

     // Triggering removeData when we don't have plugin data to remove should not trigger the plugin process to launch.
     id pool = [WKProcessPool _sharedProcessPool];
     EXPECT_EQ([pool _pluginProcessCount], static_cast<size_t>(0));

     globalCookieStore = dataStore.httpCookieStore;
     RetainPtr<CookieObserver> observer1 = adoptNS([[CookieObserver alloc] init]);
     RetainPtr<CookieObserver> observer2 = adoptNS([[CookieObserver alloc] init]);
     [globalCookieStore addObserver:observer1.get()];
     [globalCookieStore addObserver:observer2.get()];

     NSArray<NSHTTPCookie *> *cookies = nil;
     [globalCookieStore getAllCookies:[cookiesPtr = &cookies](NSArray<NSHTTPCookie *> *nsCookies) {
         *cookiesPtr = [nsCookies retain];
         gotFlag = true;
     }];

     TestWebKitAPI::Util::run(&gotFlag);

     ASSERT_EQ(cookies.count, 0u);
     [cookies release];

     gotFlag = false;

     RetainPtr<NSHTTPCookie> cookie1 = [NSHTTPCookie cookieWithProperties:@{
         NSHTTPCookiePath: @"/",
         NSHTTPCookieName: @"CookieName",
         NSHTTPCookieValue: @"CookieValue",
         NSHTTPCookieDomain: @".www.webkit.org",
         NSHTTPCookieSecure: @"TRUE",
         NSHTTPCookieDiscard: @"TRUE",
         NSHTTPCookieMaximumAge: @"10000",
     }];

     RetainPtr<NSHTTPCookie> cookie2 = [NSHTTPCookie cookieWithProperties:@{
         NSHTTPCookiePath: @"/path",
         NSHTTPCookieName: @"OtherCookieName",
         NSHTTPCookieValue: @"OtherCookieValue",
         NSHTTPCookieDomain: @".www.w3c.org",
         NSHTTPCookieMaximumAge: @"10000",
     }];

     [globalCookieStore setCookie:cookie1.get() completionHandler:[](){
         gotFlag = true;
     }];

     TestWebKitAPI::Util::run(&gotFlag);
     gotFlag = false;

     [globalCookieStore setCookie:cookie2.get() completionHandler:[](){
         gotFlag = true;
     }];

     TestWebKitAPI::Util::run(&gotFlag);
     gotFlag = false;

     [globalCookieStore getAllCookies:[cookiesPtr = &cookies](NSArray<NSHTTPCookie *> *nsCookies) {
         *cookiesPtr = [nsCookies retain];
         gotFlag = true;
     }];

     TestWebKitAPI::Util::run(&gotFlag);
     gotFlag = false;

     ASSERT_EQ(cookies.count, 2u);
     ASSERT_EQ(observerCallbacks, 4u);

     for (NSHTTPCookie *cookie : cookies) {
         if ([cookie.name isEqual:@"CookieName"]) {
             ASSERT_TRUE([cookie1.get().path isEqualToString:cookie.path]);
             ASSERT_TRUE([cookie1.get().value isEqualToString:cookie.value]);
             ASSERT_TRUE([cookie1.get().domain isEqualToString:cookie.domain]);
             ASSERT_TRUE(cookie.secure);
             ASSERT_TRUE(cookie.sessionOnly);
         } else {
             ASSERT_TRUE([cookie2.get().path isEqualToString:cookie.path]);
             ASSERT_TRUE([cookie2.get().value isEqualToString:cookie.value]);
             ASSERT_TRUE([cookie2.get().name isEqualToString:cookie.name]);
             ASSERT_TRUE([cookie2.get().domain isEqualToString:cookie.domain]);
             ASSERT_FALSE(cookie.secure);
             ASSERT_FALSE(cookie.sessionOnly);
         }
     }
     [cookies release];

     [globalCookieStore deleteCookie:cookie2.get() completionHandler:[](){
         gotFlag = true;
     }];

     TestWebKitAPI::Util::run(&gotFlag);
     gotFlag = false;

     [globalCookieStore getAllCookies:[cookiesPtr = &cookies](NSArray<NSHTTPCookie *> *nsCookies) {
         *cookiesPtr = [nsCookies retain];
         gotFlag = true;
     }];

     TestWebKitAPI::Util::run(&gotFlag);
     gotFlag = false;

     ASSERT_EQ(cookies.count, 1u);
     ASSERT_EQ(observerCallbacks, 6u);

     for (NSHTTPCookie *cookie : cookies) {
         ASSERT_TRUE([cookie1.get().path isEqualToString:cookie.path]);
         ASSERT_TRUE([cookie1.get().value isEqualToString:cookie.value]);
         ASSERT_TRUE([cookie1.get().domain isEqualToString:cookie.domain]);
         ASSERT_TRUE(cookie.secure);
         ASSERT_TRUE(cookie.sessionOnly);
     }
     [cookies release];

     [globalCookieStore removeObserver:observer1.get()];
     [globalCookieStore removeObserver:observer2.get()];
 }

 TEST(WebKit, WKHTTPCookieStore)
 {
     runTestWithWebsiteDataStore([WKWebsiteDataStore defaultDataStore]);
 }

 TEST(WebKit, WKHTTPCookieStoreHttpOnly)
 {
     WKWebsiteDataStore* dataStore = [WKWebsiteDataStore defaultDataStore];

     auto configuration = adoptNS([[WKWebViewConfiguration alloc] init]);
     configuration.get().websiteDataStore = dataStore;
     auto webView = adoptNS([[WKWebView alloc] initWithFrame:NSMakeRect(0, 0, 800, 600) configuration:configuration.get()]);

     [webView loadHTMLString:@"WebKit Test" baseURL:[NSURL URLWithString:@"http://webkit.org"]];
     [webView _test_waitForDidFinishNavigation];

     [dataStore removeDataOfTypes:[WKWebsiteDataStore allWebsiteDataTypes] modifiedSince:[NSDate distantPast] completionHandler:[] {
         gotFlag = true;
     }];

     TestWebKitAPI::Util::run(&gotFlag);
     gotFlag = false;

     id pool = [WKProcessPool _sharedProcessPool];
     EXPECT_EQ([pool _pluginProcessCount], static_cast<size_t>(0));

     globalCookieStore = dataStore.httpCookieStore;

     NSArray<NSHTTPCookie *> *cookies = nil;
     [globalCookieStore getAllCookies:[cookiesPtr = &cookies](NSArray<NSHTTPCookie *> *nsCookies) {
         *cookiesPtr = [nsCookies retain];
         gotFlag = true;
     }];

     TestWebKitAPI::Util::run(&gotFlag);
     gotFlag = false;

     ASSERT_EQ(cookies.count, 0u);
     [cookies release];

     CFMutableDictionaryRef cookieProperties = CFDictionaryCreateMutable(NULL, 0, &kCFTypeDictionaryKeyCallBacks, &kCFTypeDictionaryValueCallBacks);
     CFDictionarySetValue(cookieProperties, CFSTR("Name"), CFSTR("httpCookie"));
     CFDictionarySetValue(cookieProperties, CFSTR("Value"), CFSTR("httpCookieValue"));
     CFDictionarySetValue(cookieProperties, CFSTR("Domain"), CFSTR(".www.webkit.org"));
     CFDictionarySetValue(cookieProperties, CFSTR("Path"), CFSTR("/httpPath"));
     CFDictionarySetValue(cookieProperties, CFSTR("HttpOnly"), kCFBooleanTrue);
     RetainPtr<NSHTTPCookie> httpOnlyCookie = [NSHTTPCookie cookieWithProperties:(NSDictionary*) cookieProperties];
     CFDictionaryRemoveValue(cookieProperties, CFSTR("HttpOnly"));
     RetainPtr<NSHTTPCookie> notHttpOnlyCookie = [NSHTTPCookie cookieWithProperties:(NSDictionary*) cookieProperties];
     EXPECT_TRUE(httpOnlyCookie.get().HTTPOnly);
     EXPECT_FALSE(notHttpOnlyCookie.get().HTTPOnly);

     [globalCookieStore setCookie:notHttpOnlyCookie.get() completionHandler:[]() {
         gotFlag = true;
     }];

     TestWebKitAPI::Util::run(&gotFlag);
     gotFlag = false;

     [globalCookieStore getAllCookies:[cookiesPtr = &cookies](NSArray<NSHTTPCookie *> *nsCookies) {
         *cookiesPtr = [nsCookies retain];
         gotFlag = true;
     }];

     TestWebKitAPI::Util::run(&gotFlag);
     gotFlag = false;

     ASSERT_EQ(cookies.count, 1u);
     [cookies release];

     [globalCookieStore deleteCookie:httpOnlyCookie.get() completionHandler:[]() {
         gotFlag = true;
     }];

     TestWebKitAPI::Util::run(&gotFlag);
     gotFlag = false;
     [globalCookieStore getAllCookies:[cookiesPtr = &cookies](NSArray<NSHTTPCookie *> *nsCookies) {
         *cookiesPtr = [nsCookies retain];
         gotFlag = true;
     }];

     TestWebKitAPI::Util::run(&gotFlag);
     gotFlag = false;

     // Delete httpOnlyCookie should fail because it is different from notHttpOnlyCookie.
     ASSERT_EQ(cookies.count, 1u);
     [cookies release];
 }

 // FIXME: This should be removed once <rdar://problem/35344202> is resolved and bots are updated.
 #if (PLATFORM(MAC) && __MAC_OS_X_VERSION_MAX_ALLOWED <= 101301) || (PLATFORM(IOS) && __IPHONE_OS_VERSION_MAX_ALLOWED <= 110102)
 TEST(WebKit, WKHTTPCookieStoreNonPersistent)
 {
     RetainPtr<WKWebsiteDataStore> nonPersistentDataStore;
     @autoreleasepool {
         nonPersistentDataStore = [WKWebsiteDataStore nonPersistentDataStore];
     }

     runTestWithWebsiteDataStore(nonPersistentDataStore.get());
 }

 TEST(WebKit, WKHTTPCookieStoreCustom)
 {
     NSURL *cookieStorageFile = [NSURL fileURLWithPath:[@"~/Library/WebKit/TestWebKitAPI/CustomWebsiteData/CookieStorage/Cookie.File" stringByExpandingTildeInPath] isDirectory:NO];
     NSURL *idbPath = [NSURL fileURLWithPath:[@"~/Library/WebKit/TestWebKitAPI/CustomWebsiteData/IndexedDB/" stringByExpandingTildeInPath] isDirectory:YES];

     [[NSFileManager defaultManager] removeItemAtURL:cookieStorageFile error:nil];
     [[NSFileManager defaultManager] removeItemAtURL:idbPath error:nil];

     EXPECT_FALSE([[NSFileManager defaultManager] fileExistsAtPath:cookieStorageFile.path]);
     EXPECT_FALSE([[NSFileManager defaultManager] fileExistsAtPath:idbPath.path]);

     auto websiteDataStoreConfiguration = adoptNS([[_WKWebsiteDataStoreConfiguration alloc] init]);
     websiteDataStoreConfiguration.get()._indexedDBDatabaseDirectory = idbPath;
     websiteDataStoreConfiguration.get()._cookieStorageFile = cookieStorageFile;

     auto customDataStore = adoptNS([[WKWebsiteDataStore alloc] _initWithConfiguration:websiteDataStoreConfiguration.get()]);
     runTestWithWebsiteDataStore(customDataStore.get());
 }
 #endif // (PLATFORM(MAC) && __MAC_OS_X_VERSION_MAX_ALLOWED <= 101301) || (PLATFORM(IOS) && __IPHONE_OS_VERSION_MAX_ALLOWED <= 110102)

 TEST(WebKit, CookieObserverCrash)
 {
     RetainPtr<WKWebsiteDataStore> nonPersistentDataStore;
     @autoreleasepool {
         nonPersistentDataStore = [WKWebsiteDataStore nonPersistentDataStore];
     }

     auto configuration = adoptNS([[WKWebViewConfiguration alloc] init]);
     configuration.get().websiteDataStore = nonPersistentDataStore.get();

     auto webView = adoptNS([[WKWebView alloc] initWithFrame:NSMakeRect(0, 0, 800, 600) configuration:configuration.get()]);

     [webView loadHTMLString:@"Oh hello" baseURL:[NSURL URLWithString:@"http://webkit.org"]];
     [webView _test_waitForDidFinishNavigation];

     globalCookieStore = nonPersistentDataStore.get().httpCookieStore;
     RetainPtr<CookieObserver> observer = adoptNS([[CookieObserver alloc] init]);
     [globalCookieStore addObserver:observer.get()];

     [globalCookieStore getAllCookies:[](NSArray<NSHTTPCookie *> *) {
         gotFlag = true;
     }];

     TestWebKitAPI::Util::run(&gotFlag);
 }

 static bool finished;

 @interface CookieUIDelegate : NSObject <WKUIDelegate>
 @end

 @implementation CookieUIDelegate
 - (void)webView:(WKWebView *)webView runJavaScriptAlertPanelWithMessage:(NSString *)message initiatedByFrame:(WKFrameInfo *)frame completionHandler:(void (^)(void))completionHandler
 {
     EXPECT_STREQ("cookie:cookiename=cookievalue", message.UTF8String);
     finished = true;
     completionHandler();
 }
 @end

 // FIXME: This should be removed once <rdar://problem/35344202> is resolved and bots are updated.
 #if (PLATFORM(MAC) && __MAC_OS_X_VERSION_MAX_ALLOWED <= 101301) || (PLATFORM(IOS) && __IPHONE_OS_VERSION_MAX_ALLOWED <= 110102)
 TEST(WebKit, WKHTTPCookieStoreWithoutProcessPool)
 {
     NSHTTPCookie *cookie = [NSHTTPCookie cookieWithProperties:[NSDictionary dictionaryWithObjectsAndKeys:@"127.0.0.1", NSHTTPCookieDomain, @"/", NSHTTPCookiePath, @"cookiename", NSHTTPCookieName, @"cookievalue", NSHTTPCookieValue, [NSDate distantFuture], NSHTTPCookieExpires, nil]];
     NSString *alertCookieHTML = @"<script>alert('cookie:'+document.cookie);</script>";

     finished = false;
     WKWebsiteDataStore *ephemeralStoreWithCookies = [WKWebsiteDataStore nonPersistentDataStore];
     [ephemeralStoreWithCookies.httpCookieStore setCookie:cookie completionHandler:^ {
         WKWebsiteDataStore *ephemeralStoreWithIndependentCookieStorage = [WKWebsiteDataStore nonPersistentDataStore];
         [ephemeralStoreWithIndependentCookieStorage.httpCookieStore getAllCookies:^(NSArray<NSHTTPCookie *> *cookies) {
             ASSERT_EQ(cookies.count, 0u);

             WKWebViewConfiguration *configuration = [[WKWebViewConfiguration alloc] init];
             configuration.websiteDataStore = ephemeralStoreWithCookies;
             WKWebView *view = [[WKWebView alloc] initWithFrame:NSMakeRect(0, 0, 800, 600) configuration:configuration];
             view.UIDelegate = [[CookieUIDelegate alloc] init];

             [view loadHTMLString:alertCookieHTML baseURL:[NSURL URLWithString:@"http://127.0.0.1/"]];
         }];
     }];
     TestWebKitAPI::Util::run(&finished);

     // FIXME: Get this to work on iOS. <rdar://problem/32260156>
 #if !PLATFORM(IOS)
     finished = false;
     WKWebsiteDataStore *defaultStore = [WKWebsiteDataStore defaultDataStore];
     [defaultStore.httpCookieStore setCookie:cookie completionHandler:^ {
         [defaultStore.httpCookieStore getAllCookies:^(NSArray<NSHTTPCookie *> *cookies) {
             ASSERT_EQ(cookies.count, 1u);

             WKWebViewConfiguration *configuration = [[WKWebViewConfiguration alloc] init];
             configuration.websiteDataStore = defaultStore;
             WKWebView *view = [[WKWebView alloc] initWithFrame:NSMakeRect(0, 0, 800, 600) configuration:configuration];
             view.UIDelegate = [[CookieUIDelegate alloc] init];

             [view loadHTMLString:alertCookieHTML baseURL:[NSURL URLWithString:@"http://127.0.0.1/"]];
         }];
     }];
     TestWebKitAPI::Util::run(&finished);

     [defaultStore.httpCookieStore deleteCookie:cookie completionHandler:^ {
         [defaultStore.httpCookieStore getAllCookies:^(NSArray<NSHTTPCookie *> *cookies) {
             ASSERT_EQ(cookies.count, 0u);
             finished = true;
         }];
     }];
     TestWebKitAPI::Util::run(&finished);
 #endif
 }
 #endif // (PLATFORM(MAC) && __MAC_OS_X_VERSION_MAX_ALLOWED <= 101301) || (PLATFORM(IOS) && __IPHONE_OS_VERSION_MAX_ALLOWED <= 110102)
 #endif
	/*
	* Copyright (C) 2017 Apple 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:
	* 1. Redistributions of source code must retain the above copyright
	* notice, this list of conditions and the following disclaimer.
	* 2. Redistributions in binary form must reproduce the above copyright
	* notice, this list of conditions and the following disclaimer in the
	* documentation and/or other materials provided with the distribution.
	*
	* THIS SOFTWARE IS PROVIDED BY APPLE INC. AND ITS CONTRIBUTORS ``AS IS''
	* AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO,
	* THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
	* PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL APPLE INC. OR ITS CONTRIBUTORS
	* BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
	* CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
	* SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
	* INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
	* CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
	* ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF
	* THE POSSIBILITY OF SUCH DAMAGE.
	*/

	#include "config.h"

	#import "PlatformUtilities.h"
	#import "TestNavigationDelegate.h"
	#import <WebKit/WKFoundation.h>
	#import <WebKit/WKHTTPCookieStore.h>
	#import <WebKit/WKProcessPoolPrivate.h>
	#import <WebKit/WKWebsiteDataStorePrivate.h>
	#import <WebKit/_WKWebsiteDataStoreConfiguration.h>
	#import <wtf/RetainPtr.h>

	#if WK_API_ENABLED

	static bool gotFlag;
	static uint64_t observerCallbacks;
	static RetainPtr<WKHTTPCookieStore> globalCookieStore;

	@interface CookieObserver : NSObject<WKHTTPCookieStoreObserver>
	- (void)cookiesDidChangeInCookieStore:(WKHTTPCookieStore *)cookieStore;
	@end

	@implementation CookieObserver

	- (void)cookiesDidChangeInCookieStore:(WKHTTPCookieStore *)cookieStore
	{
	ASSERT_EQ(cookieStore, globalCookieStore.get());
	++observerCallbacks;
	}

	@end

	static void runTestWithWebsiteDataStore(WKWebsiteDataStore* dataStore)
	{
	observerCallbacks = 0;

	auto configuration = adoptNS([[WKWebViewConfiguration alloc] init]);
	configuration.get().websiteDataStore = dataStore;
	auto webView = adoptNS([[WKWebView alloc] initWithFrame:NSMakeRect(0, 0, 800, 600) configuration:configuration.get()]);

	[webView loadHTMLString:@"Oh hello" baseURL:[NSURL URLWithString:@"http://webkit.org"]];
	[webView _test_waitForDidFinishNavigation];

	[dataStore removeDataOfTypes:[WKWebsiteDataStore allWebsiteDataTypes] modifiedSince:[NSDate distantPast] completionHandler:[] {
	gotFlag = true;
	}];

	TestWebKitAPI::Util::run(&gotFlag);
	gotFlag = false;

	// Triggering removeData when we don't have plugin data to remove should not trigger the plugin process to launch.
	id pool = [WKProcessPool _sharedProcessPool];
	EXPECT_EQ([pool _pluginProcessCount], static_cast<size_t>(0));

	globalCookieStore = dataStore.httpCookieStore;
	RetainPtr<CookieObserver> observer1 = adoptNS([[CookieObserver alloc] init]);
	RetainPtr<CookieObserver> observer2 = adoptNS([[CookieObserver alloc] init]);
	[globalCookieStore addObserver:observer1.get()];
	[globalCookieStore addObserver:observer2.get()];

	NSArray<NSHTTPCookie > cookies = nil;
	[globalCookieStore getAllCookies:[cookiesPtr = &cookies](NSArray<NSHTTPCookie > nsCookies) {
	*cookiesPtr = [nsCookies retain];
	gotFlag = true;
	}];

	TestWebKitAPI::Util::run(&gotFlag);

	ASSERT_EQ(cookies.count, 0u);
	[cookies release];

	gotFlag = false;

	RetainPtr<NSHTTPCookie> cookie1 = [NSHTTPCookie cookieWithProperties:@{
	NSHTTPCookiePath: @"/",
	NSHTTPCookieName: @"CookieName",
	NSHTTPCookieValue: @"CookieValue",
	NSHTTPCookieDomain: @".www.webkit.org",
	NSHTTPCookieSecure: @"TRUE",
	NSHTTPCookieDiscard: @"TRUE",
	NSHTTPCookieMaximumAge: @"10000",
	}];

	RetainPtr<NSHTTPCookie> cookie2 = [NSHTTPCookie cookieWithProperties:@{
	NSHTTPCookiePath: @"/path",
	NSHTTPCookieName: @"OtherCookieName",
	NSHTTPCookieValue: @"OtherCookieValue",
	NSHTTPCookieDomain: @".www.w3c.org",
	NSHTTPCookieMaximumAge: @"10000",
	}];

	[globalCookieStore setCookie:cookie1.get() completionHandler:[](){
	gotFlag = true;
	}];

	TestWebKitAPI::Util::run(&gotFlag);
	gotFlag = false;

	[globalCookieStore setCookie:cookie2.get() completionHandler:[](){
	gotFlag = true;
	}];

	TestWebKitAPI::Util::run(&gotFlag);
	gotFlag = false;

	[globalCookieStore getAllCookies:[cookiesPtr = &cookies](NSArray<NSHTTPCookie > nsCookies) {
	*cookiesPtr = [nsCookies retain];
	gotFlag = true;
	}];

	TestWebKitAPI::Util::run(&gotFlag);
	gotFlag = false;

	ASSERT_EQ(cookies.count, 2u);
	ASSERT_EQ(observerCallbacks, 4u);

	for (NSHTTPCookie *cookie : cookies) {
	if ([cookie.name isEqual:@"CookieName"]) {
	ASSERT_TRUE([cookie1.get().path isEqualToString:cookie.path]);
	ASSERT_TRUE([cookie1.get().value isEqualToString:cookie.value]);
	ASSERT_TRUE([cookie1.get().domain isEqualToString:cookie.domain]);
	ASSERT_TRUE(cookie.secure);
	ASSERT_TRUE(cookie.sessionOnly);
	} else {
	ASSERT_TRUE([cookie2.get().path isEqualToString:cookie.path]);
	ASSERT_TRUE([cookie2.get().value isEqualToString:cookie.value]);
	ASSERT_TRUE([cookie2.get().name isEqualToString:cookie.name]);
	ASSERT_TRUE([cookie2.get().domain isEqualToString:cookie.domain]);
	ASSERT_FALSE(cookie.secure);
	ASSERT_FALSE(cookie.sessionOnly);
	}
	}
	[cookies release];

	[globalCookieStore deleteCookie:cookie2.get() completionHandler:[](){
	gotFlag = true;
	}];

	TestWebKitAPI::Util::run(&gotFlag);
	gotFlag = false;

	[globalCookieStore getAllCookies:[cookiesPtr = &cookies](NSArray<NSHTTPCookie > nsCookies) {
	*cookiesPtr = [nsCookies retain];
	gotFlag = true;
	}];

	TestWebKitAPI::Util::run(&gotFlag);
	gotFlag = false;

	ASSERT_EQ(cookies.count, 1u);
	ASSERT_EQ(observerCallbacks, 6u);

	for (NSHTTPCookie *cookie : cookies) {
	ASSERT_TRUE([cookie1.get().path isEqualToString:cookie.path]);
	ASSERT_TRUE([cookie1.get().value isEqualToString:cookie.value]);
	ASSERT_TRUE([cookie1.get().domain isEqualToString:cookie.domain]);
	ASSERT_TRUE(cookie.secure);
	ASSERT_TRUE(cookie.sessionOnly);
	}
	[cookies release];

	[globalCookieStore removeObserver:observer1.get()];
	[globalCookieStore removeObserver:observer2.get()];
	}

	TEST(WebKit, WKHTTPCookieStore)
	{
	runTestWithWebsiteDataStore([WKWebsiteDataStore defaultDataStore]);
	}

	TEST(WebKit, WKHTTPCookieStoreHttpOnly)
	{
	WKWebsiteDataStore* dataStore = [WKWebsiteDataStore defaultDataStore];

	auto configuration = adoptNS([[WKWebViewConfiguration alloc] init]);
	configuration.get().websiteDataStore = dataStore;
	auto webView = adoptNS([[WKWebView alloc] initWithFrame:NSMakeRect(0, 0, 800, 600) configuration:configuration.get()]);

	[webView loadHTMLString:@"WebKit Test" baseURL:[NSURL URLWithString:@"http://webkit.org"]];
	[webView _test_waitForDidFinishNavigation];

	[dataStore removeDataOfTypes:[WKWebsiteDataStore allWebsiteDataTypes] modifiedSince:[NSDate distantPast] completionHandler:[] {
	gotFlag = true;
	}];

	TestWebKitAPI::Util::run(&gotFlag);
	gotFlag = false;

	id pool = [WKProcessPool _sharedProcessPool];
	EXPECT_EQ([pool _pluginProcessCount], static_cast<size_t>(0));

	globalCookieStore = dataStore.httpCookieStore;

	NSArray<NSHTTPCookie > cookies = nil;
	[globalCookieStore getAllCookies:[cookiesPtr = &cookies](NSArray<NSHTTPCookie > nsCookies) {
	*cookiesPtr = [nsCookies retain];
	gotFlag = true;
	}];

	TestWebKitAPI::Util::run(&gotFlag);
	gotFlag = false;

	ASSERT_EQ(cookies.count, 0u);
	[cookies release];

	CFMutableDictionaryRef cookieProperties = CFDictionaryCreateMutable(NULL, 0, &kCFTypeDictionaryKeyCallBacks, &kCFTypeDictionaryValueCallBacks);
	CFDictionarySetValue(cookieProperties, CFSTR("Name"), CFSTR("httpCookie"));
	CFDictionarySetValue(cookieProperties, CFSTR("Value"), CFSTR("httpCookieValue"));
	CFDictionarySetValue(cookieProperties, CFSTR("Domain"), CFSTR(".www.webkit.org"));
	CFDictionarySetValue(cookieProperties, CFSTR("Path"), CFSTR("/httpPath"));
	CFDictionarySetValue(cookieProperties, CFSTR("HttpOnly"), kCFBooleanTrue);
	RetainPtr<NSHTTPCookie> httpOnlyCookie = [NSHTTPCookie cookieWithProperties:(NSDictionary*) cookieProperties];
	CFDictionaryRemoveValue(cookieProperties, CFSTR("HttpOnly"));
	RetainPtr<NSHTTPCookie> notHttpOnlyCookie = [NSHTTPCookie cookieWithProperties:(NSDictionary*) cookieProperties];
	EXPECT_TRUE(httpOnlyCookie.get().HTTPOnly);
	EXPECT_FALSE(notHttpOnlyCookie.get().HTTPOnly);

	[globalCookieStore setCookie:notHttpOnlyCookie.get() completionHandler:[]() {
	gotFlag = true;
	}];

	TestWebKitAPI::Util::run(&gotFlag);
	gotFlag = false;

	[globalCookieStore getAllCookies:[cookiesPtr = &cookies](NSArray<NSHTTPCookie > nsCookies) {
	*cookiesPtr = [nsCookies retain];
	gotFlag = true;
	}];

	TestWebKitAPI::Util::run(&gotFlag);
	gotFlag = false;

	ASSERT_EQ(cookies.count, 1u);
	[cookies release];

	[globalCookieStore deleteCookie:httpOnlyCookie.get() completionHandler:[]() {
	gotFlag = true;
	}];

	TestWebKitAPI::Util::run(&gotFlag);
	gotFlag = false;
	[globalCookieStore getAllCookies:[cookiesPtr = &cookies](NSArray<NSHTTPCookie > nsCookies) {
	*cookiesPtr = [nsCookies retain];
	gotFlag = true;
	}];

	TestWebKitAPI::Util::run(&gotFlag);
	gotFlag = false;

	// Delete httpOnlyCookie should fail because it is different from notHttpOnlyCookie.
	ASSERT_EQ(cookies.count, 1u);
	[cookies release];
	}

	// FIXME: This should be removed once <rdar://problem/35344202> is resolved and bots are updated.
	#if (PLATFORM(MAC) && __MAC_OS_X_VERSION_MAX_ALLOWED <= 101301) \|\| (PLATFORM(IOS) && __IPHONE_OS_VERSION_MAX_ALLOWED <= 110102)
	TEST(WebKit, WKHTTPCookieStoreNonPersistent)
	{
	RetainPtr<WKWebsiteDataStore> nonPersistentDataStore;
	@autoreleasepool {
	nonPersistentDataStore = [WKWebsiteDataStore nonPersistentDataStore];
	}

	runTestWithWebsiteDataStore(nonPersistentDataStore.get());
	}

	TEST(WebKit, WKHTTPCookieStoreCustom)
	{
	NSURL *cookieStorageFile = [NSURL fileURLWithPath:[@"~/Library/WebKit/TestWebKitAPI/CustomWebsiteData/CookieStorage/Cookie.File" stringByExpandingTildeInPath] isDirectory:NO];
	NSURL *idbPath = [NSURL fileURLWithPath:[@"~/Library/WebKit/TestWebKitAPI/CustomWebsiteData/IndexedDB/" stringByExpandingTildeInPath] isDirectory:YES];

	[[NSFileManager defaultManager] removeItemAtURL:cookieStorageFile error:nil];
	[[NSFileManager defaultManager] removeItemAtURL:idbPath error:nil];

	EXPECT_FALSE([[NSFileManager defaultManager] fileExistsAtPath:cookieStorageFile.path]);
	EXPECT_FALSE([[NSFileManager defaultManager] fileExistsAtPath:idbPath.path]);

	auto websiteDataStoreConfiguration = adoptNS([[_WKWebsiteDataStoreConfiguration alloc] init]);
	websiteDataStoreConfiguration.get()._indexedDBDatabaseDirectory = idbPath;
	websiteDataStoreConfiguration.get()._cookieStorageFile = cookieStorageFile;

	auto customDataStore = adoptNS([[WKWebsiteDataStore alloc] _initWithConfiguration:websiteDataStoreConfiguration.get()]);
	runTestWithWebsiteDataStore(customDataStore.get());
	}
	#endif // (PLATFORM(MAC) && __MAC_OS_X_VERSION_MAX_ALLOWED <= 101301) \|\| (PLATFORM(IOS) && __IPHONE_OS_VERSION_MAX_ALLOWED <= 110102)

	TEST(WebKit, CookieObserverCrash)
	{
	RetainPtr<WKWebsiteDataStore> nonPersistentDataStore;
	@autoreleasepool {
	nonPersistentDataStore = [WKWebsiteDataStore nonPersistentDataStore];
	}

	auto configuration = adoptNS([[WKWebViewConfiguration alloc] init]);
	configuration.get().websiteDataStore = nonPersistentDataStore.get();

	auto webView = adoptNS([[WKWebView alloc] initWithFrame:NSMakeRect(0, 0, 800, 600) configuration:configuration.get()]);

	[webView loadHTMLString:@"Oh hello" baseURL:[NSURL URLWithString:@"http://webkit.org"]];
	[webView _test_waitForDidFinishNavigation];

	globalCookieStore = nonPersistentDataStore.get().httpCookieStore;
	RetainPtr<CookieObserver> observer = adoptNS([[CookieObserver alloc] init]);
	[globalCookieStore addObserver:observer.get()];

	[globalCookieStore getAllCookies:[](NSArray<NSHTTPCookie > ) {
	gotFlag = true;
	}];

	TestWebKitAPI::Util::run(&gotFlag);
	}

	static bool finished;

	@interface CookieUIDelegate : NSObject <WKUIDelegate>
	@end

	@implementation CookieUIDelegate
	- (void)webView:(WKWebView )webView runJavaScriptAlertPanelWithMessage:(NSString )message initiatedByFrame:(WKFrameInfo *)frame completionHandler:(void (^)(void))completionHandler
	{
	EXPECT_STREQ("cookie:cookiename=cookievalue", message.UTF8String);
	finished = true;
	completionHandler();
	}
	@end

	// FIXME: This should be removed once <rdar://problem/35344202> is resolved and bots are updated.
	#if (PLATFORM(MAC) && __MAC_OS_X_VERSION_MAX_ALLOWED <= 101301) \|\| (PLATFORM(IOS) && __IPHONE_OS_VERSION_MAX_ALLOWED <= 110102)
	TEST(WebKit, WKHTTPCookieStoreWithoutProcessPool)
	{
	NSHTTPCookie *cookie = [NSHTTPCookie cookieWithProperties:[NSDictionary dictionaryWithObjectsAndKeys:@"127.0.0.1", NSHTTPCookieDomain, @"/", NSHTTPCookiePath, @"cookiename", NSHTTPCookieName, @"cookievalue", NSHTTPCookieValue, [NSDate distantFuture], NSHTTPCookieExpires, nil]];
	NSString *alertCookieHTML = @"<script>alert('cookie:'+document.cookie);</script>";

	finished = false;
	WKWebsiteDataStore *ephemeralStoreWithCookies = [WKWebsiteDataStore nonPersistentDataStore];
	[ephemeralStoreWithCookies.httpCookieStore setCookie:cookie completionHandler:^ {
	WKWebsiteDataStore *ephemeralStoreWithIndependentCookieStorage = [WKWebsiteDataStore nonPersistentDataStore];
	[ephemeralStoreWithIndependentCookieStorage.httpCookieStore getAllCookies:^(NSArray<NSHTTPCookie > cookies) {
	ASSERT_EQ(cookies.count, 0u);

	WKWebViewConfiguration *configuration = [[WKWebViewConfiguration alloc] init];
	configuration.websiteDataStore = ephemeralStoreWithCookies;
	WKWebView *view = [[WKWebView alloc] initWithFrame:NSMakeRect(0, 0, 800, 600) configuration:configuration];
	view.UIDelegate = [[CookieUIDelegate alloc] init];

	[view loadHTMLString:alertCookieHTML baseURL:[NSURL URLWithString:@"http://127.0.0.1/"]];
	}];
	}];
	TestWebKitAPI::Util::run(&finished);

	// FIXME: Get this to work on iOS. <rdar://problem/32260156>
	#if !PLATFORM(IOS)
	finished = false;
	WKWebsiteDataStore *defaultStore = [WKWebsiteDataStore defaultDataStore];
	[defaultStore.httpCookieStore setCookie:cookie completionHandler:^ {
	[defaultStore.httpCookieStore getAllCookies:^(NSArray<NSHTTPCookie > cookies) {
	ASSERT_EQ(cookies.count, 1u);

	WKWebViewConfiguration *configuration = [[WKWebViewConfiguration alloc] init];
	configuration.websiteDataStore = defaultStore;
	WKWebView *view = [[WKWebView alloc] initWithFrame:NSMakeRect(0, 0, 800, 600) configuration:configuration];
	view.UIDelegate = [[CookieUIDelegate alloc] init];

	[view loadHTMLString:alertCookieHTML baseURL:[NSURL URLWithString:@"http://127.0.0.1/"]];
	}];
	}];
	TestWebKitAPI::Util::run(&finished);

	[defaultStore.httpCookieStore deleteCookie:cookie completionHandler:^ {
	[defaultStore.httpCookieStore getAllCookies:^(NSArray<NSHTTPCookie > cookies) {
	ASSERT_EQ(cookies.count, 0u);
	finished = true;
	}];
	}];
	TestWebKitAPI::Util::run(&finished);
	#endif
	}
	#endif // (PLATFORM(MAC) && __MAC_OS_X_VERSION_MAX_ALLOWED <= 101301) \|\| (PLATFORM(IOS) && __IPHONE_OS_VERSION_MAX_ALLOWED <= 110102)
	#endif