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

 /*
  * Copyright (C) 2019 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.
  */

 #import "config.h"
 #import <WebKit/WebKit.h>

 #import "PlatformUtilities.h"
 #import "Test.h"
 #import "TestWKWebView.h"
 #import <WebKit/WKPreferencesPrivate.h>
 #import <WebKit/WKProcessPoolPrivate.h>
 #import <WebKit/WKURLSchemeHandler.h>
 #import <WebKit/WKURLSchemeTaskPrivate.h>
 #import <WebKit/WKWebViewConfigurationPrivate.h>
 #import <WebKit/WKWebsiteDataStorePrivate.h>
 #import <WebKit/WKWebsiteDataStoreRef.h>
 #import <WebKit/_WKWebsiteDataStoreConfiguration.h>
 #import <wtf/BlockPtr.h>
 #import <wtf/HashMap.h>
 #import <wtf/RetainPtr.h>
 #import <wtf/Vector.h>
 #import <wtf/text/StringHash.h>
 #import <wtf/text/WTFString.h>

 using namespace TestWebKitAPI;

 @interface QuotaDelegate : NSObject <WKUIDelegate>
 -(bool)quotaDelegateCalled;
 -(void)grantQuota;
 -(void)denyQuota;
 @end

 static bool receivedQuotaDelegateCalled;

 @implementation QuotaDelegate {
     bool _quotaDelegateCalled;
     unsigned long long _currentQuota;
     unsigned long long _expectedUsage;
     BlockPtr<void(unsigned long long newQuota)> _decisionHandler;
 }

 - (instancetype)init
 {
     if (!(self = [super init]))
         return nil;

     _quotaDelegateCalled = false;
     _expectedUsage = 0;
     _currentQuota = 0;

     return self;
 }

 - (void)_webView:(WKWebView *)webView decideDatabaseQuotaForSecurityOrigin:(WKSecurityOrigin *)securityOrigin currentQuota:(unsigned long long)currentQuota currentOriginUsage:(unsigned long long)currentOriginUsage currentDatabaseUsage:(unsigned long long)currentUsage expectedUsage:(unsigned long long)expectedUsage decisionHandler:(void (^)(unsigned long long newQuota))decisionHandler
 {
     receivedQuotaDelegateCalled = true;
     _quotaDelegateCalled = true;
     _currentQuota = currentQuota;
     _expectedUsage = expectedUsage;
     _decisionHandler = decisionHandler;
 }

 -(bool)quotaDelegateCalled {
     return _quotaDelegateCalled;
 }

 -(void)grantQuota {
     if (_quotaDelegateCalled)
         _decisionHandler(_expectedUsage);
     _quotaDelegateCalled = false;
 }

 -(void)denyQuota {
     if (_quotaDelegateCalled)
         _decisionHandler(_currentQuota);
     _quotaDelegateCalled = false;
 }

 @end

 struct ResourceInfo {
     RetainPtr<NSString> mimeType;
     const char* data;
 };

 @interface StorageSchemes : NSObject <WKURLSchemeHandler> {
 @public
     HashMap<String, ResourceInfo> resources;
 }
 @end

 @implementation StorageSchemes {
 }

 - (void)webView:(WKWebView *)webView startURLSchemeTask:(id <WKURLSchemeTask>)task
 {
     auto entry = resources.find([task.request.URL absoluteString]);
     if (entry == resources.end()) {
         NSLog(@"Did not find resource entry for URL %@", task.request.URL);
         return;
     }

     RetainPtr<NSURLResponse> response = adoptNS([[NSURLResponse alloc] initWithURL:task.request.URL MIMEType:entry->value.mimeType.get() expectedContentLength:1 textEncodingName:nil]);
     [task didReceiveResponse:response.get()];

     [task didReceiveData:[NSData dataWithBytesNoCopy:(void*)entry->value.data length:strlen(entry->value.data) freeWhenDone:NO]];
     [task didFinish];
 }

 - (void)webView:(WKWebView *)webView stopURLSchemeTask:(id <WKURLSchemeTask>)task
 {
 }

 @end

 static bool receivedMessage;

 @interface QuotaMessageHandler : NSObject <WKScriptMessageHandler>
 -(void)setExpectedMessage:(NSString *)message;
 @end

 @implementation QuotaMessageHandler {
     NSString *_expectedMessage;
 }

 - (void)userContentController:(WKUserContentController *)userContentController didReceiveScriptMessage:(WKScriptMessage *)message
 {
     if (_expectedMessage) {
         EXPECT_TRUE([[message body] isEqualToString:_expectedMessage]);
         _expectedMessage = nil;
     }
     receivedMessage = true;
 }

 -(void)setExpectedMessage:(NSString *)message {
     _expectedMessage = message;
 }
 @end

 static const char* TestBytes = R"SWRESOURCE(
 <script>

 async function doTest()
 {
     const cache = await window.caches.open("mycache");
     const promise = cache.put("http://example.org/test", new Response(new ArrayBuffer(1024 * 500)));
     window.webkit.messageHandlers.qt.postMessage("start");
     promise.then(() => {
         window.webkit.messageHandlers.qt.postMessage("pass");
     }, () => {
         window.webkit.messageHandlers.qt.postMessage("fail");
     });
 }
 doTest();

 function doTestAgain()
 {
     doTest();
 }
 </script>
 )SWRESOURCE";

 static bool done;

 static inline void setVisible(TestWKWebView *webView)
 {
 #if PLATFORM(MAC)
     [webView.window setIsVisible:YES];
 #else
     webView.window.hidden = NO;
 #endif
 }

 TEST(WebKit, QuotaDelegate)
 {
     done = false;
     [[WKWebsiteDataStore defaultDataStore] removeDataOfTypes:[WKWebsiteDataStore allWebsiteDataTypes] modifiedSince:[NSDate distantPast] completionHandler:^() {
         done = true;
     }];
     TestWebKitAPI::Util::run(&done);

     auto configuration = adoptNS([[WKWebViewConfiguration alloc] init]);

     [[configuration websiteDataStore] _setPerOriginStorageQuota: 1024 * 400];

     auto messageHandler = adoptNS([[QuotaMessageHandler alloc] init]);
     [[configuration userContentController] addScriptMessageHandler:messageHandler.get() name:@"qt"];

     auto handler1 = adoptNS([[StorageSchemes alloc] init]);
     handler1->resources.set("qt1://test1.html", ResourceInfo { @"text/html", TestBytes });
     [configuration setURLSchemeHandler:handler1.get() forURLScheme:@"QT1"];
     [configuration.get().processPool _registerURLSchemeServiceWorkersCanHandle:@"qt1"];

     auto handler2 = adoptNS([[StorageSchemes alloc] init]);
     handler2->resources.set("qt2://test2.html", ResourceInfo { @"text/html", TestBytes });
     [configuration setURLSchemeHandler:handler2.get() forURLScheme:@"QT2"];
     [configuration.get().processPool _registerURLSchemeServiceWorkersCanHandle:@"qt2"];

     auto webView1 = adoptNS([[TestWKWebView alloc] initWithFrame:NSMakeRect(0, 0, 800, 600) configuration:configuration.get() addToWindow:YES]);
     auto delegate1 = adoptNS([[QuotaDelegate alloc] init]);
     [webView1 setUIDelegate:delegate1.get()];
     setVisible(webView1.get());

     receivedQuotaDelegateCalled = false;
     [webView1 loadRequest:[NSURLRequest requestWithURL:[NSURL URLWithString:@"qt1://test1.html"]]];
     Util::run(&receivedQuotaDelegateCalled);

     auto webView2 = adoptNS([[TestWKWebView alloc] initWithFrame:NSMakeRect(0, 0, 800, 600) configuration:configuration.get() addToWindow:YES]);
     auto delegate2 = adoptNS([[QuotaDelegate alloc] init]);
     [webView2 setUIDelegate:delegate2.get()];
     setVisible(webView2.get());

     receivedMessage = false;
     [webView2 loadRequest:[NSURLRequest requestWithURL:[NSURL URLWithString:@"qt2://test2.html"]]];
     [messageHandler setExpectedMessage: @"start"];
     Util::run(&receivedMessage);

     EXPECT_FALSE(delegate2.get().quotaDelegateCalled);
     [delegate1 grantQuota];

     [messageHandler setExpectedMessage: @"pass"];
     receivedMessage = false;
     Util::run(&receivedMessage);

     while (!delegate2.get().quotaDelegateCalled)
         TestWebKitAPI::Util::sleep(0.1);

     [delegate2 denyQuota];

     [messageHandler setExpectedMessage: @"fail"];
     receivedMessage = false;
     Util::run(&receivedMessage);
 }

 TEST(WebKit, QuotaDelegateReload)
 {
     done = false;
     [[WKWebsiteDataStore defaultDataStore] removeDataOfTypes:[WKWebsiteDataStore allWebsiteDataTypes] modifiedSince:[NSDate distantPast] completionHandler:^() {
         done = true;
     }];
     TestWebKitAPI::Util::run(&done);

     auto configuration = adoptNS([[WKWebViewConfiguration alloc] init]);

     [[configuration websiteDataStore] _setPerOriginStorageQuota: 1024 * 400];

     auto messageHandler = adoptNS([[QuotaMessageHandler alloc] init]);
     [[configuration userContentController] addScriptMessageHandler:messageHandler.get() name:@"qt"];

     auto handler = adoptNS([[StorageSchemes alloc] init]);
     handler->resources.set("qt://test1.html", ResourceInfo { @"text/html", TestBytes });
     [configuration setURLSchemeHandler:handler.get() forURLScheme:@"QT"];
     [configuration.get().processPool _registerURLSchemeServiceWorkersCanHandle:@"qt"];

     auto webView = adoptNS([[TestWKWebView alloc] initWithFrame:NSMakeRect(0, 0, 800, 600) configuration:configuration.get() addToWindow:YES]);
     auto delegate = adoptNS([[QuotaDelegate alloc] init]);
     [webView setUIDelegate:delegate.get()];
     setVisible(webView.get());

     receivedQuotaDelegateCalled = false;
     [webView loadRequest:[NSURLRequest requestWithURL:[NSURL URLWithString:@"qt://test1.html"]]];
     Util::run(&receivedQuotaDelegateCalled);

     [delegate denyQuota];

     [messageHandler setExpectedMessage: @"fail"];
     receivedMessage = false;
     Util::run(&receivedMessage);

     receivedQuotaDelegateCalled = false;
     [webView reload];
     Util::run(&receivedQuotaDelegateCalled);

     [delegate grantQuota];

     [messageHandler setExpectedMessage: @"pass"];
     receivedMessage = false;
     Util::run(&receivedMessage);
 }

 TEST(WebKit, QuotaDelegateNavigateFragment)
 {
     done = false;
     [[WKWebsiteDataStore defaultDataStore] removeDataOfTypes:[WKWebsiteDataStore allWebsiteDataTypes] modifiedSince:[NSDate distantPast] completionHandler:^() {
         done = true;
     }];
     TestWebKitAPI::Util::run(&done);

     auto configuration = adoptNS([[WKWebViewConfiguration alloc] init]);

     [[configuration websiteDataStore] _setPerOriginStorageQuota: 1024 * 400];

     auto messageHandler = adoptNS([[QuotaMessageHandler alloc] init]);
     [[configuration userContentController] addScriptMessageHandler:messageHandler.get() name:@"qt"];

     auto handler = adoptNS([[StorageSchemes alloc] init]);
     handler->resources.set("qt://test1.html", ResourceInfo { @"text/html", TestBytes });
     [configuration setURLSchemeHandler:handler.get() forURLScheme:@"QT"];
     [configuration.get().processPool _registerURLSchemeServiceWorkersCanHandle:@"qt"];

     auto webView = adoptNS([[TestWKWebView alloc] initWithFrame:NSMakeRect(0, 0, 800, 600) configuration:configuration.get() addToWindow:YES]);
     auto delegate = adoptNS([[QuotaDelegate alloc] init]);
     [webView setUIDelegate:delegate.get()];
     setVisible(webView.get());

     receivedQuotaDelegateCalled = false;
     [webView loadRequest:[NSURLRequest requestWithURL:[NSURL URLWithString:@"qt://test1.html"]]];
     Util::run(&receivedQuotaDelegateCalled);

     [delegate denyQuota];

     [messageHandler setExpectedMessage: @"fail"];
     receivedMessage = false;
     Util::run(&receivedMessage);

     receivedQuotaDelegateCalled = false;
     [webView loadRequest:[NSURLRequest requestWithURL:[NSURL URLWithString:@"qt://test1.html#fragment"]]];
     [webView stringByEvaluatingJavaScript:@"doTestAgain()"];

     [messageHandler setExpectedMessage: @"start"];
     receivedMessage = false;
     Util::run(&receivedMessage);

     [messageHandler setExpectedMessage: @"fail"];
     receivedMessage = false;
     Util::run(&receivedMessage);

     EXPECT_FALSE(receivedQuotaDelegateCalled);
 }
	/*
	* Copyright (C) 2019 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.
	*/

	#import "config.h"
	#import <WebKit/WebKit.h>

	#import "PlatformUtilities.h"
	#import "Test.h"
	#import "TestWKWebView.h"
	#import <WebKit/WKPreferencesPrivate.h>
	#import <WebKit/WKProcessPoolPrivate.h>
	#import <WebKit/WKURLSchemeHandler.h>
	#import <WebKit/WKURLSchemeTaskPrivate.h>
	#import <WebKit/WKWebViewConfigurationPrivate.h>
	#import <WebKit/WKWebsiteDataStorePrivate.h>
	#import <WebKit/WKWebsiteDataStoreRef.h>
	#import <WebKit/_WKWebsiteDataStoreConfiguration.h>
	#import <wtf/BlockPtr.h>
	#import <wtf/HashMap.h>
	#import <wtf/RetainPtr.h>
	#import <wtf/Vector.h>
	#import <wtf/text/StringHash.h>
	#import <wtf/text/WTFString.h>

	using namespace TestWebKitAPI;

	@interface QuotaDelegate : NSObject <WKUIDelegate>
	-(bool)quotaDelegateCalled;
	-(void)grantQuota;
	-(void)denyQuota;
	@end

	static bool receivedQuotaDelegateCalled;

	@implementation QuotaDelegate {
	bool _quotaDelegateCalled;
	unsigned long long _currentQuota;
	unsigned long long _expectedUsage;
	BlockPtr<void(unsigned long long newQuota)> _decisionHandler;
	}

	- (instancetype)init
	{
	if (!(self = [super init]))
	return nil;

	_quotaDelegateCalled = false;
	_expectedUsage = 0;
	_currentQuota = 0;

	return self;
	}

	- (void)_webView:(WKWebView )webView decideDatabaseQuotaForSecurityOrigin:(WKSecurityOrigin )securityOrigin currentQuota:(unsigned long long)currentQuota currentOriginUsage:(unsigned long long)currentOriginUsage currentDatabaseUsage:(unsigned long long)currentUsage expectedUsage:(unsigned long long)expectedUsage decisionHandler:(void (^)(unsigned long long newQuota))decisionHandler
	{
	receivedQuotaDelegateCalled = true;
	_quotaDelegateCalled = true;
	_currentQuota = currentQuota;
	_expectedUsage = expectedUsage;
	_decisionHandler = decisionHandler;
	}

	-(bool)quotaDelegateCalled {
	return _quotaDelegateCalled;
	}

	-(void)grantQuota {
	if (_quotaDelegateCalled)
	_decisionHandler(_expectedUsage);
	_quotaDelegateCalled = false;
	}

	-(void)denyQuota {
	if (_quotaDelegateCalled)
	_decisionHandler(_currentQuota);
	_quotaDelegateCalled = false;
	}

	@end

	struct ResourceInfo {
	RetainPtr<NSString> mimeType;
	const char* data;
	};

	@interface StorageSchemes : NSObject <WKURLSchemeHandler> {
	@public
	HashMap<String, ResourceInfo> resources;
	}
	@end

	@implementation StorageSchemes {
	}

	- (void)webView:(WKWebView *)webView startURLSchemeTask:(id <WKURLSchemeTask>)task
	{
	auto entry = resources.find([task.request.URL absoluteString]);
	if (entry == resources.end()) {
	NSLog(@"Did not find resource entry for URL %@", task.request.URL);
	return;
	}

	RetainPtr<NSURLResponse> response = adoptNS([[NSURLResponse alloc] initWithURL:task.request.URL MIMEType:entry->value.mimeType.get() expectedContentLength:1 textEncodingName:nil]);
	[task didReceiveResponse:response.get()];

	[task didReceiveData:[NSData dataWithBytesNoCopy:(void*)entry->value.data length:strlen(entry->value.data) freeWhenDone:NO]];
	[task didFinish];
	}

	- (void)webView:(WKWebView *)webView stopURLSchemeTask:(id <WKURLSchemeTask>)task
	{
	}

	@end

	static bool receivedMessage;

	@interface QuotaMessageHandler : NSObject <WKScriptMessageHandler>
	-(void)setExpectedMessage:(NSString *)message;
	@end

	@implementation QuotaMessageHandler {
	NSString *_expectedMessage;
	}

	- (void)userContentController:(WKUserContentController )userContentController didReceiveScriptMessage:(WKScriptMessage )message
	{
	if (_expectedMessage) {
	EXPECT_TRUE([[message body] isEqualToString:_expectedMessage]);
	_expectedMessage = nil;
	}
	receivedMessage = true;
	}

	-(void)setExpectedMessage:(NSString *)message {
	_expectedMessage = message;
	}
	@end

	static const char* TestBytes = R"SWRESOURCE(
	<script>

	async function doTest()
	{
	const cache = await window.caches.open("mycache");
	const promise = cache.put("http://example.org/test", new Response(new ArrayBuffer(1024 * 500)));
	window.webkit.messageHandlers.qt.postMessage("start");
	promise.then(() => {
	window.webkit.messageHandlers.qt.postMessage("pass");
	}, () => {
	window.webkit.messageHandlers.qt.postMessage("fail");
	});
	}
	doTest();

	function doTestAgain()
	{
	doTest();
	}
	</script>
	)SWRESOURCE";

	static bool done;

	static inline void setVisible(TestWKWebView *webView)
	{
	#if PLATFORM(MAC)
	[webView.window setIsVisible:YES];
	#else
	webView.window.hidden = NO;
	#endif
	}

	TEST(WebKit, QuotaDelegate)
	{
	done = false;
	[[WKWebsiteDataStore defaultDataStore] removeDataOfTypes:[WKWebsiteDataStore allWebsiteDataTypes] modifiedSince:[NSDate distantPast] completionHandler:^() {
	done = true;
	}];
	TestWebKitAPI::Util::run(&done);

	auto configuration = adoptNS([[WKWebViewConfiguration alloc] init]);

	[[configuration websiteDataStore] _setPerOriginStorageQuota: 1024 * 400];

	auto messageHandler = adoptNS([[QuotaMessageHandler alloc] init]);
	[[configuration userContentController] addScriptMessageHandler:messageHandler.get() name:@"qt"];

	auto handler1 = adoptNS([[StorageSchemes alloc] init]);
	handler1->resources.set("qt1://test1.html", ResourceInfo { @"text/html", TestBytes });
	[configuration setURLSchemeHandler:handler1.get() forURLScheme:@"QT1"];
	[configuration.get().processPool _registerURLSchemeServiceWorkersCanHandle:@"qt1"];

	auto handler2 = adoptNS([[StorageSchemes alloc] init]);
	handler2->resources.set("qt2://test2.html", ResourceInfo { @"text/html", TestBytes });
	[configuration setURLSchemeHandler:handler2.get() forURLScheme:@"QT2"];
	[configuration.get().processPool _registerURLSchemeServiceWorkersCanHandle:@"qt2"];

	auto webView1 = adoptNS([[TestWKWebView alloc] initWithFrame:NSMakeRect(0, 0, 800, 600) configuration:configuration.get() addToWindow:YES]);
	auto delegate1 = adoptNS([[QuotaDelegate alloc] init]);
	[webView1 setUIDelegate:delegate1.get()];
	setVisible(webView1.get());

	receivedQuotaDelegateCalled = false;
	[webView1 loadRequest:[NSURLRequest requestWithURL:[NSURL URLWithString:@"qt1://test1.html"]]];
	Util::run(&receivedQuotaDelegateCalled);

	auto webView2 = adoptNS([[TestWKWebView alloc] initWithFrame:NSMakeRect(0, 0, 800, 600) configuration:configuration.get() addToWindow:YES]);
	auto delegate2 = adoptNS([[QuotaDelegate alloc] init]);
	[webView2 setUIDelegate:delegate2.get()];
	setVisible(webView2.get());

	receivedMessage = false;
	[webView2 loadRequest:[NSURLRequest requestWithURL:[NSURL URLWithString:@"qt2://test2.html"]]];
	[messageHandler setExpectedMessage: @"start"];
	Util::run(&receivedMessage);

	EXPECT_FALSE(delegate2.get().quotaDelegateCalled);
	[delegate1 grantQuota];

	[messageHandler setExpectedMessage: @"pass"];
	receivedMessage = false;
	Util::run(&receivedMessage);

	while (!delegate2.get().quotaDelegateCalled)
	TestWebKitAPI::Util::sleep(0.1);

	[delegate2 denyQuota];

	[messageHandler setExpectedMessage: @"fail"];
	receivedMessage = false;
	Util::run(&receivedMessage);
	}

	TEST(WebKit, QuotaDelegateReload)
	{
	done = false;
	[[WKWebsiteDataStore defaultDataStore] removeDataOfTypes:[WKWebsiteDataStore allWebsiteDataTypes] modifiedSince:[NSDate distantPast] completionHandler:^() {
	done = true;
	}];
	TestWebKitAPI::Util::run(&done);

	auto configuration = adoptNS([[WKWebViewConfiguration alloc] init]);

	[[configuration websiteDataStore] _setPerOriginStorageQuota: 1024 * 400];

	auto messageHandler = adoptNS([[QuotaMessageHandler alloc] init]);
	[[configuration userContentController] addScriptMessageHandler:messageHandler.get() name:@"qt"];

	auto handler = adoptNS([[StorageSchemes alloc] init]);
	handler->resources.set("qt://test1.html", ResourceInfo { @"text/html", TestBytes });
	[configuration setURLSchemeHandler:handler.get() forURLScheme:@"QT"];
	[configuration.get().processPool _registerURLSchemeServiceWorkersCanHandle:@"qt"];

	auto webView = adoptNS([[TestWKWebView alloc] initWithFrame:NSMakeRect(0, 0, 800, 600) configuration:configuration.get() addToWindow:YES]);
	auto delegate = adoptNS([[QuotaDelegate alloc] init]);
	[webView setUIDelegate:delegate.get()];
	setVisible(webView.get());

	receivedQuotaDelegateCalled = false;
	[webView loadRequest:[NSURLRequest requestWithURL:[NSURL URLWithString:@"qt://test1.html"]]];
	Util::run(&receivedQuotaDelegateCalled);

	[delegate denyQuota];

	[messageHandler setExpectedMessage: @"fail"];
	receivedMessage = false;
	Util::run(&receivedMessage);

	receivedQuotaDelegateCalled = false;
	[webView reload];
	Util::run(&receivedQuotaDelegateCalled);

	[delegate grantQuota];

	[messageHandler setExpectedMessage: @"pass"];
	receivedMessage = false;
	Util::run(&receivedMessage);
	}

	TEST(WebKit, QuotaDelegateNavigateFragment)
	{
	done = false;
	[[WKWebsiteDataStore defaultDataStore] removeDataOfTypes:[WKWebsiteDataStore allWebsiteDataTypes] modifiedSince:[NSDate distantPast] completionHandler:^() {
	done = true;
	}];
	TestWebKitAPI::Util::run(&done);

	auto configuration = adoptNS([[WKWebViewConfiguration alloc] init]);

	[[configuration websiteDataStore] _setPerOriginStorageQuota: 1024 * 400];

	auto messageHandler = adoptNS([[QuotaMessageHandler alloc] init]);
	[[configuration userContentController] addScriptMessageHandler:messageHandler.get() name:@"qt"];

	auto handler = adoptNS([[StorageSchemes alloc] init]);
	handler->resources.set("qt://test1.html", ResourceInfo { @"text/html", TestBytes });
	[configuration setURLSchemeHandler:handler.get() forURLScheme:@"QT"];
	[configuration.get().processPool _registerURLSchemeServiceWorkersCanHandle:@"qt"];

	auto webView = adoptNS([[TestWKWebView alloc] initWithFrame:NSMakeRect(0, 0, 800, 600) configuration:configuration.get() addToWindow:YES]);
	auto delegate = adoptNS([[QuotaDelegate alloc] init]);
	[webView setUIDelegate:delegate.get()];
	setVisible(webView.get());

	receivedQuotaDelegateCalled = false;
	[webView loadRequest:[NSURLRequest requestWithURL:[NSURL URLWithString:@"qt://test1.html"]]];
	Util::run(&receivedQuotaDelegateCalled);

	[delegate denyQuota];

	[messageHandler setExpectedMessage: @"fail"];
	receivedMessage = false;
	Util::run(&receivedMessage);

	receivedQuotaDelegateCalled = false;
	[webView loadRequest:[NSURLRequest requestWithURL:[NSURL URLWithString:@"qt://test1.html#fragment"]]];
	[webView stringByEvaluatingJavaScript:@"doTestAgain()"];

	[messageHandler setExpectedMessage: @"start"];
	receivedMessage = false;
	Util::run(&receivedMessage);

	[messageHandler setExpectedMessage: @"fail"];
	receivedMessage = false;
	Util::run(&receivedMessage);

	EXPECT_FALSE(receivedQuotaDelegateCalled);
	}