Tools/TestWebKitAPI/Tests/WebKitCocoa/AsyncFunction.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 "PlatformUtilities.h"
 #import "Test.h"
 #import "TestUIDelegate.h"
 #import "TestWKWebView.h"
 #import <WebKit/WKContentWorld.h>
 #import <WebKit/WKFrameInfo.h>
 #import <WebKit/WKProcessPoolPrivate.h>
 #import <WebKit/WKWebViewPrivate.h>

 namespace TestWebKitAPI {

 TEST(AsyncFunction, Basic)
 {
     auto webView = adoptNS([[TestWKWebView alloc] initWithFrame:NSMakeRect(0, 0, 800, 600)]);
     NSError *error;

     // Function with no return value.
     // Returns JavaScript undefined which translates to nil.
     id result = [webView objectByCallingAsyncFunction:@"1" withArguments:nil error:&error];
     EXPECT_NULL(error);
     EXPECT_EQ(result, nil);

     // Function returns explicit null.
     // Returns JavaScript null which translates to NSNull.
     result = [webView objectByCallingAsyncFunction:@"return null" withArguments:nil error:&error];
     EXPECT_NULL(error);
     EXPECT_TRUE([result isKindOfClass:[NSNull class]]);

     // Function returns a number.
     result = [webView objectByCallingAsyncFunction:@"return 1" withArguments:nil error:&error];
     EXPECT_NULL(error);
     EXPECT_TRUE([result isKindOfClass:[NSNumber class]]);
     EXPECT_TRUE([result isEqualToNumber:@1]);

     // Function returns a string.
     result = [webView objectByCallingAsyncFunction:@"return '1'" withArguments:nil error:&error];
     EXPECT_NULL(error);
     EXPECT_TRUE([result isKindOfClass:[NSString class]]);
     EXPECT_TRUE([result isEqualToString:@"1"]);

     // Takes multiple arguments.
     result = [webView objectByCallingAsyncFunction:@"return a + b" withArguments:@{ @"a" : @40, @"b" : @2 } error:&error];
     EXPECT_NULL(error);
     EXPECT_TRUE([result isKindOfClass:[NSNumber class]]);
     EXPECT_TRUE([result isEqualToNumber:@42]);

     // Takes multiple arguments of different types, follows javascripts "type appending" rules.
     result = [webView objectByCallingAsyncFunction:@"return foo + bar" withArguments:@{ @"foo" : @"foo", @"bar" : @42 } error:&error];
     EXPECT_NULL(error);
     EXPECT_TRUE([result isKindOfClass:[NSString class]]);
     EXPECT_TRUE([result isEqualToString:@"foo42"]);

     // Invalid JavaScript, should return an error.
     result = [webView objectByCallingAsyncFunction:@"retunr null" withArguments:nil error:&error];
     EXPECT_FALSE(error == nil);
     EXPECT_NULL(result);
 }

 TEST(AsyncFunction, InvalidArguments)
 {
     auto webView = adoptNS([[TestWKWebView alloc] initWithFrame:NSMakeRect(0, 0, 800, 600)]);
     NSError *error;

     // Values can only be NSString, NSNumber, NSDate, NSNull, NS(Mutable)Array, NS(Mutable)Dictionary, NSNull, and may only contain those 6 types.
     id result = [webView objectByCallingAsyncFunction:@"return 1" withArguments:@{ @"a" : [NSData data] } error:&error];
     EXPECT_NULL(result);
     EXPECT_FALSE(error == nil);

     result = [webView objectByCallingAsyncFunction:@"return 1" withArguments:@{ @"a" : @[ @1, [NSData data] ] } error:&error];
     EXPECT_NULL(result);
     EXPECT_FALSE(error == nil);

     // References an argument that was not provided.
     result = [webView objectByCallingAsyncFunction:@"return a + b" withArguments:@{ @"a" : @40 } error:&error];
     EXPECT_NULL(result);
     EXPECT_FALSE(error == nil);
 }

 TEST(AsyncFunction, RoundTrip)
 {
     auto webView = adoptNS([[TestWKWebView alloc] initWithFrame:NSMakeRect(0, 0, 800, 600)]);
     NSError *error;

     // Tests round tripping a whole bunch of argument inputs and verifying the result.

     id value = @42;
     id arguments = @{ @"a" : value };
     id result = [webView objectByCallingAsyncFunction:@"return a" withArguments:arguments error:&error];
     EXPECT_NULL(error);
     EXPECT_TRUE([value isEqual:result]);

     value = [NSNull null];
     arguments = @{ @"a" : value };
     result = [webView objectByCallingAsyncFunction:@"return a" withArguments:arguments error:&error];
     EXPECT_NULL(error);
     EXPECT_TRUE([value isEqual:result]);

     value = @"Foo";
     arguments = @{ @"a" : value };
     result = [webView objectByCallingAsyncFunction:@"return a" withArguments:arguments error:&error];
     EXPECT_NULL(error);
     EXPECT_TRUE([value isEqual:result]);

     value = [NSDate dateWithTimeIntervalSinceReferenceDate:0];
     arguments = @{ @"a" : value };
     result = [webView objectByCallingAsyncFunction:@"return a" withArguments:arguments error:&error];
     EXPECT_NULL(error);
     EXPECT_TRUE([value isEqual:result]);

     value = @[ @1, @2 ];
     arguments = @{ @"a" : value };
     result = [webView objectByCallingAsyncFunction:@"return a" withArguments:arguments error:&error];
     EXPECT_NULL(error);
     EXPECT_TRUE([value isEqual:result]);

     auto mutableArray = adoptNS([[NSMutableArray alloc] init]);
     [mutableArray addObject:value];
     arguments = @{ @"a" : mutableArray.get() };
     result = [webView objectByCallingAsyncFunction:@"return a" withArguments:arguments error:&error];
     EXPECT_NULL(error);
     EXPECT_TRUE([mutableArray isEqual:result]);

     value = @[ @"foo", [NSDate dateWithTimeIntervalSinceReferenceDate:0] ];
     arguments = @{ @"a" : value };
     result = [webView objectByCallingAsyncFunction:@"return a" withArguments:arguments error:&error];
     EXPECT_NULL(error);
     EXPECT_TRUE([value isEqual:result]);

     value = @{ @"a" : @1, @"b" : @2 };
     arguments = @{ @"a" : value };
     result = [webView objectByCallingAsyncFunction:@"return a" withArguments:arguments error:&error];
     EXPECT_NULL(error);
     EXPECT_TRUE([value isEqual:result]);

     auto mutableDictionary = adoptNS((NSMutableDictionary<NSString *, id> *)[[NSMutableDictionary alloc] init]);
     mutableDictionary.get()[@"foo"] = value;
     arguments = @{ @"a" : mutableDictionary.get() };
     result = [webView objectByCallingAsyncFunction:@"return a" withArguments:arguments error:&error];
     EXPECT_NULL(error);
     EXPECT_TRUE([mutableDictionary isEqual:result]);

     value = @{ @"a" : @"foo", @"b" : [NSDate dateWithTimeIntervalSinceReferenceDate:0] };
     arguments = @{ @"a" : value };
     result = [webView objectByCallingAsyncFunction:@"return a" withArguments:arguments error:&error];
     EXPECT_NULL(error);
     EXPECT_TRUE([value isEqual:result]);

     value = @{ @"a" : @{ @"a" : @1 } };
     arguments = @{ @"a" : value };
     result = [webView objectByCallingAsyncFunction:@"return a" withArguments:arguments error:&error];
     EXPECT_NULL(error);
     EXPECT_TRUE([value isEqual:result]);
 }

 TEST(AsyncFunction, Promise)
 {
     auto webView = adoptNS([[TestWKWebView alloc] initWithFrame:NSMakeRect(0, 0, 800, 600)]);

     [webView callAsyncJavaScript:@"shouldn't crash" arguments:@{ @"invalidparameter" : webView.get() } inFrame:nil inContentWorld:WKContentWorld.pageWorld completionHandler:nil];

     NSString *functionBody = @"return new Promise(function(resolve, reject) { setTimeout(function(){ resolve(42) }, 0); })";

     bool done = false;
     [webView callAsyncJavaScript:functionBody arguments:nil inFrame:nil inContentWorld:WKContentWorld.pageWorld completionHandler:[&] (id result, NSError *error) {
         EXPECT_NULL(error);
         EXPECT_TRUE([result isKindOfClass:[NSNumber class]]);
         EXPECT_TRUE([result isEqualToNumber:@42]);
         done = true;
     }];
     TestWebKitAPI::Util::run(&done);
     done = false;

     functionBody = @"return new Promise(function(resolve, reject) { setTimeout(function(){ reject('Rejected!') }, 0); })";

     done = false;
     [webView callAsyncJavaScript:functionBody arguments:nil inFrame:nil inContentWorld:WKContentWorld.pageWorld completionHandler:[&] (id result, NSError *error) {
         EXPECT_NULL(result);
         EXPECT_TRUE(error != nil);
         EXPECT_TRUE([[error description] containsString:@"Rejected!"]);
         done = true;
     }];
     TestWebKitAPI::Util::run(&done);

     functionBody = @"let p = new Proxy(function(resolve, reject) { setTimeout(function() { resolve(42); }, 0); }, { }); return { then: p };";

     done = false;
     [webView callAsyncJavaScript:functionBody arguments:nil inFrame:nil inContentWorld:WKContentWorld.pageWorld completionHandler:[&] (id result, NSError *error) {
         EXPECT_NULL(error);
         EXPECT_TRUE([result isKindOfClass:[NSNumber class]]);
         EXPECT_TRUE([result isEqualToNumber:@42]);
         done = true;
     }];
     TestWebKitAPI::Util::run(&done);

     functionBody = @"let p = new Proxy(function(resolve, reject) { setTimeout(function() { reject('Rejected!'); }, 0); }, { }); return { then: p };";

     done = false;
     [webView callAsyncJavaScript:functionBody arguments:nil inFrame:nil inContentWorld:WKContentWorld.pageWorld completionHandler:[&] (id result, NSError *error) {
         EXPECT_NULL(result);
         EXPECT_TRUE(error != nil);
         EXPECT_TRUE([[error description] containsString:@"Rejected!"]);
         done = true;
     }];
     TestWebKitAPI::Util::run(&done);

     // Verify we can await for a promise to be resolved before returning.
     functionBody = @"var r = 0; var p = new Promise(function(fulfill, reject) { setTimeout(function(){ r = 42; fulfill(); }, 5);}); await p; return r;";

     done = false;
     [webView callAsyncJavaScript:functionBody arguments:nil inFrame:nil inContentWorld:WKContentWorld.pageWorld completionHandler:[&] (id result, NSError *error) {
         EXPECT_NULL(error);
         EXPECT_TRUE([result isKindOfClass:[NSNumber class]]);
         EXPECT_TRUE([result isEqualToNumber:@42]);
         done = true;
     }];
     TestWebKitAPI::Util::run(&done);

     // Returning an already resolved promise gives the value it was resolved with.
     functionBody = @"var p = new Promise(function(fulfill, reject) { setTimeout(function(){ fulfill('Fulfilled!') }, 5);}); await p; return p;";

     done = false;
     [webView callAsyncJavaScript:functionBody arguments:nil inFrame:nil inContentWorld:WKContentWorld.pageWorld completionHandler:[&] (id result, NSError *error) {
         EXPECT_NULL(error);
         EXPECT_TRUE([result isKindOfClass:[NSString class]]);
         EXPECT_TRUE([result isEqualToString:@"Fulfilled!"]);
         done = true;
     }];
     TestWebKitAPI::Util::run(&done);

     // Chaining thenables should work.
     functionBody = @"var p = new Promise(function (r) { r(new Promise(function (r) { r(42); })); }); await p; return 'Done';";

     done = false;
     [webView callAsyncJavaScript:functionBody arguments:nil inFrame:nil inContentWorld:WKContentWorld.pageWorld completionHandler:[&] (id result, NSError *error) {
         EXPECT_NULL(error);
         EXPECT_TRUE([result isKindOfClass:[NSString class]]);
         EXPECT_TRUE([result isEqualToString:@"Done"]);
         done = true;
     }];
     TestWebKitAPI::Util::run(&done);

     // Promises known to become unreachable (e.g. via garbage collection) should call back with an error.
     done = false;
     functionBody = @"return new Promise(function(resolve, reject) { })";
     for (int i = 0; i < 500; ++i) {
         [webView callAsyncJavaScript:functionBody arguments:nil inFrame:nil inContentWorld:WKContentWorld.pageWorld completionHandler:[&] (id result, NSError *error) {
             EXPECT_NULL(result);
             EXPECT_TRUE(error != nil);
             EXPECT_TRUE([[error description] containsString:@"no longer reachable"]);
             done = true;
         }];
     }

     [webView.get().configuration.processPool _garbageCollectJavaScriptObjectsForTesting];
     TestWebKitAPI::Util::run(&done);
 }

 TEST(AsyncFunction, PromiseDetachedFrame)
 {
     auto webView = adoptNS([[TestWKWebView alloc] initWithFrame:NSMakeRect(0, 0, 800, 600)]);
     auto uiDelegate = adoptNS([[TestUIDelegate alloc] init]);
     [webView setUIDelegate:uiDelegate.get()];

     __block RetainPtr<WKFrameInfo> subframe;
     __block bool doneGettingSubframe = false;
     uiDelegate.get().runJavaScriptAlertPanelWithMessage = ^(WKWebView *, NSString *, WKFrameInfo * frame, void (^completionHandler)(void)) {
         subframe = frame;
         doneGettingSubframe = true;
         completionHandler();
     };

     [webView synchronouslyLoadHTMLString:@"<iframe src='javascript:alert()'></iframe>"];

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

     ASSERT_TRUE(!!subframe);
     EXPECT_FALSE([subframe isMainFrame]);

     auto pid = [webView _webProcessIdentifier];
     EXPECT_NE(pid, 0);

     NSString *functionBody = @"return new Promise(function(resolve, reject) { setTimeout(function(){ top.setTimeout(function() { resolve(42); }, 100); top.document.querySelector('iframe').remove(); }, 0); })";

     bool done = false;
     [webView callAsyncJavaScript:functionBody arguments:nil inFrame:subframe.get() inContentWorld:WKContentWorld.pageWorld completionHandler:[&] (id result, NSError *error) {
         EXPECT_NULL(error);
         EXPECT_TRUE([result isKindOfClass:[NSNumber class]]);
         EXPECT_TRUE([result isEqualToNumber:@42]);
         done = true;
     }];
     TestWebKitAPI::Util::run(&done);

     EXPECT_EQ([webView _webProcessIdentifier], pid);
 }

 }
	/*
	* 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 "PlatformUtilities.h"
	#import "Test.h"
	#import "TestUIDelegate.h"
	#import "TestWKWebView.h"
	#import <WebKit/WKContentWorld.h>
	#import <WebKit/WKFrameInfo.h>
	#import <WebKit/WKProcessPoolPrivate.h>
	#import <WebKit/WKWebViewPrivate.h>

	namespace TestWebKitAPI {

	TEST(AsyncFunction, Basic)
	{
	auto webView = adoptNS([[TestWKWebView alloc] initWithFrame:NSMakeRect(0, 0, 800, 600)]);
	NSError *error;

	// Function with no return value.
	// Returns JavaScript undefined which translates to nil.
	id result = [webView objectByCallingAsyncFunction:@"1" withArguments:nil error:&error];
	EXPECT_NULL(error);
	EXPECT_EQ(result, nil);

	// Function returns explicit null.
	// Returns JavaScript null which translates to NSNull.
	result = [webView objectByCallingAsyncFunction:@"return null" withArguments:nil error:&error];
	EXPECT_NULL(error);
	EXPECT_TRUE([result isKindOfClass:[NSNull class]]);

	// Function returns a number.
	result = [webView objectByCallingAsyncFunction:@"return 1" withArguments:nil error:&error];
	EXPECT_NULL(error);
	EXPECT_TRUE([result isKindOfClass:[NSNumber class]]);
	EXPECT_TRUE([result isEqualToNumber:@1]);

	// Function returns a string.
	result = [webView objectByCallingAsyncFunction:@"return '1'" withArguments:nil error:&error];
	EXPECT_NULL(error);
	EXPECT_TRUE([result isKindOfClass:[NSString class]]);
	EXPECT_TRUE([result isEqualToString:@"1"]);

	// Takes multiple arguments.
	result = [webView objectByCallingAsyncFunction:@"return a + b" withArguments:@{ @"a" : @40, @"b" : @2 } error:&error];
	EXPECT_NULL(error);
	EXPECT_TRUE([result isKindOfClass:[NSNumber class]]);
	EXPECT_TRUE([result isEqualToNumber:@42]);

	// Takes multiple arguments of different types, follows javascripts "type appending" rules.
	result = [webView objectByCallingAsyncFunction:@"return foo + bar" withArguments:@{ @"foo" : @"foo", @"bar" : @42 } error:&error];
	EXPECT_NULL(error);
	EXPECT_TRUE([result isKindOfClass:[NSString class]]);
	EXPECT_TRUE([result isEqualToString:@"foo42"]);

	// Invalid JavaScript, should return an error.
	result = [webView objectByCallingAsyncFunction:@"retunr null" withArguments:nil error:&error];
	EXPECT_FALSE(error == nil);
	EXPECT_NULL(result);
	}

	TEST(AsyncFunction, InvalidArguments)
	{
	auto webView = adoptNS([[TestWKWebView alloc] initWithFrame:NSMakeRect(0, 0, 800, 600)]);
	NSError *error;

	// Values can only be NSString, NSNumber, NSDate, NSNull, NS(Mutable)Array, NS(Mutable)Dictionary, NSNull, and may only contain those 6 types.
	id result = [webView objectByCallingAsyncFunction:@"return 1" withArguments:@{ @"a" : [NSData data] } error:&error];
	EXPECT_NULL(result);
	EXPECT_FALSE(error == nil);

	result = [webView objectByCallingAsyncFunction:@"return 1" withArguments:@{ @"a" : @[ @1, [NSData data] ] } error:&error];
	EXPECT_NULL(result);
	EXPECT_FALSE(error == nil);

	// References an argument that was not provided.
	result = [webView objectByCallingAsyncFunction:@"return a + b" withArguments:@{ @"a" : @40 } error:&error];
	EXPECT_NULL(result);
	EXPECT_FALSE(error == nil);
	}

	TEST(AsyncFunction, RoundTrip)
	{
	auto webView = adoptNS([[TestWKWebView alloc] initWithFrame:NSMakeRect(0, 0, 800, 600)]);
	NSError *error;

	// Tests round tripping a whole bunch of argument inputs and verifying the result.

	id value = @42;
	id arguments = @{ @"a" : value };
	id result = [webView objectByCallingAsyncFunction:@"return a" withArguments:arguments error:&error];
	EXPECT_NULL(error);
	EXPECT_TRUE([value isEqual:result]);

	value = [NSNull null];
	arguments = @{ @"a" : value };
	result = [webView objectByCallingAsyncFunction:@"return a" withArguments:arguments error:&error];
	EXPECT_NULL(error);
	EXPECT_TRUE([value isEqual:result]);

	value = @"Foo";
	arguments = @{ @"a" : value };
	result = [webView objectByCallingAsyncFunction:@"return a" withArguments:arguments error:&error];
	EXPECT_NULL(error);
	EXPECT_TRUE([value isEqual:result]);

	value = [NSDate dateWithTimeIntervalSinceReferenceDate:0];
	arguments = @{ @"a" : value };
	result = [webView objectByCallingAsyncFunction:@"return a" withArguments:arguments error:&error];
	EXPECT_NULL(error);
	EXPECT_TRUE([value isEqual:result]);

	value = @[ @1, @2 ];
	arguments = @{ @"a" : value };
	result = [webView objectByCallingAsyncFunction:@"return a" withArguments:arguments error:&error];
	EXPECT_NULL(error);
	EXPECT_TRUE([value isEqual:result]);

	auto mutableArray = adoptNS([[NSMutableArray alloc] init]);
	[mutableArray addObject:value];
	arguments = @{ @"a" : mutableArray.get() };
	result = [webView objectByCallingAsyncFunction:@"return a" withArguments:arguments error:&error];
	EXPECT_NULL(error);
	EXPECT_TRUE([mutableArray isEqual:result]);

	value = @[ @"foo", [NSDate dateWithTimeIntervalSinceReferenceDate:0] ];
	arguments = @{ @"a" : value };
	result = [webView objectByCallingAsyncFunction:@"return a" withArguments:arguments error:&error];
	EXPECT_NULL(error);
	EXPECT_TRUE([value isEqual:result]);

	value = @{ @"a" : @1, @"b" : @2 };
	arguments = @{ @"a" : value };
	result = [webView objectByCallingAsyncFunction:@"return a" withArguments:arguments error:&error];
	EXPECT_NULL(error);
	EXPECT_TRUE([value isEqual:result]);

	auto mutableDictionary = adoptNS((NSMutableDictionary<NSString , id> )[[NSMutableDictionary alloc] init]);
	mutableDictionary.get()[@"foo"] = value;
	arguments = @{ @"a" : mutableDictionary.get() };
	result = [webView objectByCallingAsyncFunction:@"return a" withArguments:arguments error:&error];
	EXPECT_NULL(error);
	EXPECT_TRUE([mutableDictionary isEqual:result]);

	value = @{ @"a" : @"foo", @"b" : [NSDate dateWithTimeIntervalSinceReferenceDate:0] };
	arguments = @{ @"a" : value };
	result = [webView objectByCallingAsyncFunction:@"return a" withArguments:arguments error:&error];
	EXPECT_NULL(error);
	EXPECT_TRUE([value isEqual:result]);

	value = @{ @"a" : @{ @"a" : @1 } };
	arguments = @{ @"a" : value };
	result = [webView objectByCallingAsyncFunction:@"return a" withArguments:arguments error:&error];
	EXPECT_NULL(error);
	EXPECT_TRUE([value isEqual:result]);
	}

	TEST(AsyncFunction, Promise)
	{
	auto webView = adoptNS([[TestWKWebView alloc] initWithFrame:NSMakeRect(0, 0, 800, 600)]);

	[webView callAsyncJavaScript:@"shouldn't crash" arguments:@{ @"invalidparameter" : webView.get() } inFrame:nil inContentWorld:WKContentWorld.pageWorld completionHandler:nil];

	NSString *functionBody = @"return new Promise(function(resolve, reject) { setTimeout(function(){ resolve(42) }, 0); })";

	bool done = false;
	[webView callAsyncJavaScript:functionBody arguments:nil inFrame:nil inContentWorld:WKContentWorld.pageWorld completionHandler:[&] (id result, NSError *error) {
	EXPECT_NULL(error);
	EXPECT_TRUE([result isKindOfClass:[NSNumber class]]);
	EXPECT_TRUE([result isEqualToNumber:@42]);
	done = true;
	}];
	TestWebKitAPI::Util::run(&done);
	done = false;

	functionBody = @"return new Promise(function(resolve, reject) { setTimeout(function(){ reject('Rejected!') }, 0); })";

	done = false;
	[webView callAsyncJavaScript:functionBody arguments:nil inFrame:nil inContentWorld:WKContentWorld.pageWorld completionHandler:[&] (id result, NSError *error) {
	EXPECT_NULL(result);
	EXPECT_TRUE(error != nil);
	EXPECT_TRUE([[error description] containsString:@"Rejected!"]);
	done = true;
	}];
	TestWebKitAPI::Util::run(&done);

	functionBody = @"let p = new Proxy(function(resolve, reject) { setTimeout(function() { resolve(42); }, 0); }, { }); return { then: p };";

	done = false;
	[webView callAsyncJavaScript:functionBody arguments:nil inFrame:nil inContentWorld:WKContentWorld.pageWorld completionHandler:[&] (id result, NSError *error) {
	EXPECT_NULL(error);
	EXPECT_TRUE([result isKindOfClass:[NSNumber class]]);
	EXPECT_TRUE([result isEqualToNumber:@42]);
	done = true;
	}];
	TestWebKitAPI::Util::run(&done);

	functionBody = @"let p = new Proxy(function(resolve, reject) { setTimeout(function() { reject('Rejected!'); }, 0); }, { }); return { then: p };";

	done = false;
	[webView callAsyncJavaScript:functionBody arguments:nil inFrame:nil inContentWorld:WKContentWorld.pageWorld completionHandler:[&] (id result, NSError *error) {
	EXPECT_NULL(result);
	EXPECT_TRUE(error != nil);
	EXPECT_TRUE([[error description] containsString:@"Rejected!"]);
	done = true;
	}];
	TestWebKitAPI::Util::run(&done);

	// Verify we can await for a promise to be resolved before returning.
	functionBody = @"var r = 0; var p = new Promise(function(fulfill, reject) { setTimeout(function(){ r = 42; fulfill(); }, 5);}); await p; return r;";

	done = false;
	[webView callAsyncJavaScript:functionBody arguments:nil inFrame:nil inContentWorld:WKContentWorld.pageWorld completionHandler:[&] (id result, NSError *error) {
	EXPECT_NULL(error);
	EXPECT_TRUE([result isKindOfClass:[NSNumber class]]);
	EXPECT_TRUE([result isEqualToNumber:@42]);
	done = true;
	}];
	TestWebKitAPI::Util::run(&done);

	// Returning an already resolved promise gives the value it was resolved with.
	functionBody = @"var p = new Promise(function(fulfill, reject) { setTimeout(function(){ fulfill('Fulfilled!') }, 5);}); await p; return p;";

	done = false;
	[webView callAsyncJavaScript:functionBody arguments:nil inFrame:nil inContentWorld:WKContentWorld.pageWorld completionHandler:[&] (id result, NSError *error) {
	EXPECT_NULL(error);
	EXPECT_TRUE([result isKindOfClass:[NSString class]]);
	EXPECT_TRUE([result isEqualToString:@"Fulfilled!"]);
	done = true;
	}];
	TestWebKitAPI::Util::run(&done);

	// Chaining thenables should work.
	functionBody = @"var p = new Promise(function (r) { r(new Promise(function (r) { r(42); })); }); await p; return 'Done';";

	done = false;
	[webView callAsyncJavaScript:functionBody arguments:nil inFrame:nil inContentWorld:WKContentWorld.pageWorld completionHandler:[&] (id result, NSError *error) {
	EXPECT_NULL(error);
	EXPECT_TRUE([result isKindOfClass:[NSString class]]);
	EXPECT_TRUE([result isEqualToString:@"Done"]);
	done = true;
	}];
	TestWebKitAPI::Util::run(&done);

	// Promises known to become unreachable (e.g. via garbage collection) should call back with an error.
	done = false;
	functionBody = @"return new Promise(function(resolve, reject) { })";
	for (int i = 0; i < 500; ++i) {
	[webView callAsyncJavaScript:functionBody arguments:nil inFrame:nil inContentWorld:WKContentWorld.pageWorld completionHandler:[&] (id result, NSError *error) {
	EXPECT_NULL(result);
	EXPECT_TRUE(error != nil);
	EXPECT_TRUE([[error description] containsString:@"no longer reachable"]);
	done = true;
	}];
	}

	[webView.get().configuration.processPool _garbageCollectJavaScriptObjectsForTesting];
	TestWebKitAPI::Util::run(&done);
	}

	TEST(AsyncFunction, PromiseDetachedFrame)
	{
	auto webView = adoptNS([[TestWKWebView alloc] initWithFrame:NSMakeRect(0, 0, 800, 600)]);
	auto uiDelegate = adoptNS([[TestUIDelegate alloc] init]);
	[webView setUIDelegate:uiDelegate.get()];

	__block RetainPtr<WKFrameInfo> subframe;
	__block bool doneGettingSubframe = false;
	uiDelegate.get().runJavaScriptAlertPanelWithMessage = ^(WKWebView , NSString , WKFrameInfo * frame, void (^completionHandler)(void)) {
	subframe = frame;
	doneGettingSubframe = true;
	completionHandler();
	};

	[webView synchronouslyLoadHTMLString:@"<iframe src='javascript:alert()'></iframe>"];

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

	ASSERT_TRUE(!!subframe);
	EXPECT_FALSE([subframe isMainFrame]);

	auto pid = [webView _webProcessIdentifier];
	EXPECT_NE(pid, 0);

	NSString *functionBody = @"return new Promise(function(resolve, reject) { setTimeout(function(){ top.setTimeout(function() { resolve(42); }, 100); top.document.querySelector('iframe').remove(); }, 0); })";

	bool done = false;
	[webView callAsyncJavaScript:functionBody arguments:nil inFrame:subframe.get() inContentWorld:WKContentWorld.pageWorld completionHandler:[&] (id result, NSError *error) {
	EXPECT_NULL(error);
	EXPECT_TRUE([result isKindOfClass:[NSNumber class]]);
	EXPECT_TRUE([result isEqualToNumber:@42]);
	done = true;
	}];
	TestWebKitAPI::Util::run(&done);

	EXPECT_EQ([webView _webProcessIdentifier], pid);
	}

	}