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

 /*
  * Copyright (C) 2020 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 "DeprecatedGlobalValues.h"
 #import "PlatformUtilities.h"
 #import "TestWKWebView.h"
 #import <WebKit/WKPreferencesPrivate.h>
 #import <WebKit/WKProcessPoolPrivate.h>
 #import <WebKit/WKUIDelegatePrivate.h>
 #import <WebKit/WKWebView.h>
 #import <WebKit/WKWebViewConfigurationPrivate.h>
 #import <wtf/RetainPtr.h>

 static bool shouldGrantPermissionRequest = true;
 static bool permissionRequested = false;
 static bool captureStateDidChange;
 static bool isCapturing;
 static RetainPtr<WKWebView> createdWebView;

 @interface SpeechRecognitionUIDelegate : NSObject<WKUIDelegatePrivate>
 - (void)_webView:(WKWebView *)webView requestSpeechRecognitionPermissionForOrigin:(WKSecurityOrigin *)origin decisionHandler:(void (^)(BOOL))decisionHandler;
 - (void)webView:(WKWebView *)webView requestMediaCapturePermissionForOrigin:(WKSecurityOrigin *)origin initiatedByFrame:(WKFrameInfo *)frame type:(WKMediaCaptureType)type decisionHandler:(void (^)(WKPermissionDecision decision))decisionHandler;
 - (WKWebView *)webView:(WKWebView *)webView createWebViewWithConfiguration:(WKWebViewConfiguration *)configuration forNavigationAction:(WKNavigationAction *)navigationAction windowFeatures:(WKWindowFeatures *)windowFeatures;
 - (void)_webView:(WKWebView *)webView mediaCaptureStateDidChange:(_WKMediaCaptureStateDeprecated)state;
 @end

 @implementation SpeechRecognitionUIDelegate
 - (void)_webView:(WKWebView *)webView requestSpeechRecognitionPermissionForOrigin:(WKSecurityOrigin *)origin decisionHandler:(void (^)(BOOL))decisionHandler
 {
     permissionRequested = true;
     decisionHandler(shouldGrantPermissionRequest);
 }

 - (void)webView:(WKWebView *)webView requestMediaCapturePermissionForOrigin:(WKSecurityOrigin *)origin initiatedByFrame:(WKFrameInfo *)frame type:(WKMediaCaptureType)type decisionHandler:(void (^)(WKPermissionDecision decision))decisionHandler
 {
     permissionRequested = true;
     decisionHandler(shouldGrantPermissionRequest ? WKPermissionDecisionGrant : WKPermissionDecisionDeny);
 }

 - (void)_webView:(WKWebView *)webView requestMediaCaptureAuthorization: (_WKCaptureDevices)devices decisionHandler:(void (^)(BOOL))decisionHandler
 {
     permissionRequested = true;
     decisionHandler(shouldGrantPermissionRequest);
 }

 - (WKWebView *)webView:(WKWebView *)webView createWebViewWithConfiguration:(WKWebViewConfiguration *)configuration forNavigationAction:(WKNavigationAction *)navigationAction windowFeatures:(WKWindowFeatures *)windowFeatures
 {
     createdWebView = adoptNS([[WKWebView alloc] initWithFrame:CGRectMake(0, 0, 800, 600) configuration:configuration]);
     return createdWebView.get();
 }

 - (void)_webView:(WKWebView *)webView mediaCaptureStateDidChange:(_WKMediaCaptureStateDeprecated)state
 {
     isCapturing = state == _WKMediaCaptureStateDeprecatedActiveMicrophone;
     captureStateDidChange = true;
 }
 @end

 @interface SpeechRecognitionMessageHandler : NSObject <WKScriptMessageHandler>
 @end

 @implementation SpeechRecognitionMessageHandler
 - (void)userContentController:(WKUserContentController *)userContentController didReceiveScriptMessage:(WKScriptMessage *)message
 {
     receivedScriptMessage = true;
     lastScriptMessage = message;
 }
 @end

 @interface SpeechRecognitionNavigationDelegate : NSObject <WKNavigationDelegate>
 - (void)webView:(WKWebView *)webView decidePolicyForNavigationAction:(WKNavigationAction *)navigationAction preferences:(WKWebpagePreferences *)preferences decisionHandler:(void (^)(WKNavigationActionPolicy, WKWebpagePreferences *))decisionHandler;
 - (void)webView:(WKWebView *)webView didFinishNavigation:(WKNavigation *)navigation;
 @end

 @implementation SpeechRecognitionNavigationDelegate
 - (void)webView:(WKWebView *)webView decidePolicyForNavigationAction:(WKNavigationAction *)navigationAction preferences:(WKWebpagePreferences *)preferences decisionHandler:(void (^)(WKNavigationActionPolicy, WKWebpagePreferences *))decisionHandler
 {
     decisionHandler(WKNavigationActionPolicyAllow, preferences);
 }

 - (void)webView:(WKWebView *)webView didFinishNavigation:(WKNavigation *)navigation
 {
     didFinishNavigationBoolean = true;
 }
 @end

 namespace TestWebKitAPI {

 TEST(WebKit2, SpeechRecognitionUserPermissionPersistence)
 {
     auto configuration = adoptNS([[WKWebViewConfiguration alloc] init]);
     auto handler = adoptNS([[SpeechRecognitionMessageHandler alloc] init]);
     [[configuration userContentController] addScriptMessageHandler:handler.get() name:@"testHandler"];
     auto preferences = [configuration preferences];
     preferences._mockCaptureDevicesEnabled = YES;
     preferences._speechRecognitionEnabled = YES;
     auto webView = adoptNS([[TestWKWebView alloc] initWithFrame:CGRectMake(0, 0, 800, 600) configuration:configuration.get()]);
     auto delegate = adoptNS([[SpeechRecognitionUIDelegate alloc] init]);
     [webView setUIDelegate:delegate.get()];

     shouldGrantPermissionRequest = false;
     permissionRequested = false;
     receivedScriptMessage = false;
     [webView loadTestPageNamed:@"speechrecognition-user-permission-persistence"];
     TestWebKitAPI::Util::run(&receivedScriptMessage);
     EXPECT_WK_STREQ(@"Error: not-allowed - User permission check has failed", [lastScriptMessage body]);
     EXPECT_TRUE(permissionRequested);

     // Permission result is remembered.
     permissionRequested = false;
     receivedScriptMessage = false;
     [webView stringByEvaluatingJavaScript:@"start()"];
     TestWebKitAPI::Util::run(&receivedScriptMessage);
     EXPECT_WK_STREQ(@"Error: not-allowed - User permission check has failed", [lastScriptMessage body]);
     EXPECT_FALSE(permissionRequested);

     // Permission result will be cleared after document changes.
     [webView synchronouslyLoadRequest:[NSURLRequest requestWithURL:[NSURL URLWithString:@"about:blank"]]];

     shouldGrantPermissionRequest = true;
     permissionRequested = false;
     receivedScriptMessage = false;
     [webView loadTestPageNamed:@"speechrecognition-user-permission-persistence"];
     TestWebKitAPI::Util::run(&permissionRequested);
     // Should not get error message as permission is granted.
     EXPECT_FALSE(receivedScriptMessage);
 }

 TEST(WebKit2, SpeechRecognitionErrorWhenStartingAudioCaptureOnDifferentPage)
 {
     shouldGrantPermissionRequest = true;

     auto configuration = adoptNS([[WKWebViewConfiguration alloc] init]);
     auto handler = adoptNS([[SpeechRecognitionMessageHandler alloc] init]);
     [[configuration userContentController] addScriptMessageHandler:handler.get() name:@"testHandler"];
     configuration.get()._mediaCaptureEnabled = YES;
     auto preferences = [configuration preferences];
     preferences._mockCaptureDevicesEnabled = YES;
     preferences._speechRecognitionEnabled = YES;
     preferences._mediaCaptureRequiresSecureConnection = NO;
     auto delegate = adoptNS([[SpeechRecognitionUIDelegate alloc] init]);
     auto firstWebView = adoptNS([[TestWKWebView alloc] initWithFrame:CGRectMake(0, 0, 100, 100) configuration:configuration.get()]);
     [firstWebView setUIDelegate:delegate.get()];
     auto secondWebView = adoptNS([[TestWKWebView alloc] initWithFrame:CGRectMake(100, 0, 100, 100) configuration:configuration.get()]);
     [secondWebView setUIDelegate:delegate.get()];

     // First page starts recognition successfully.
     receivedScriptMessage = false;
     [firstWebView synchronouslyLoadTestPageNamed:@"speechrecognition-basic"];
     [firstWebView stringByEvaluatingJavaScript:@"start()"];
     TestWebKitAPI::Util::run(&receivedScriptMessage);
     EXPECT_WK_STREQ(@"Start", [lastScriptMessage body]);

     // First page is muted when second page starts recognition.
     // Load html string instead of test page to make sure message only comes from one page.
     receivedScriptMessage = false;
     [secondWebView synchronouslyLoadHTMLString:@"<script>speechRecognition = new webkitSpeechRecognition(); speechRecognition.start();</script>" baseURL:[NSURL URLWithString:@"https://webkit.org"]];
     TestWebKitAPI::Util::run(&receivedScriptMessage);
     EXPECT_WK_STREQ(@"Error: audio-capture - Source is muted", [lastScriptMessage body]);

     // First page restarts recognition successfully.
     receivedScriptMessage = false;
     [firstWebView stringByEvaluatingJavaScript:@"start()"];
     TestWebKitAPI::Util::run(&receivedScriptMessage);
     EXPECT_WK_STREQ(@"Start", [lastScriptMessage body]);

     // First page is muted when second page starts media recorder.
     receivedScriptMessage = false;
     [secondWebView synchronouslyLoadTestPageNamed:@"speechrecognition-basic"];
     [secondWebView stringByEvaluatingJavaScript:@"startAudio()"];
     TestWebKitAPI::Util::run(&receivedScriptMessage);
     EXPECT_WK_STREQ(@"Error: audio-capture - Source is muted", [lastScriptMessage body]);

     // Second page is muted when first page starts recognition.
     receivedScriptMessage = false;
     [firstWebView synchronouslyLoadHTMLString:@"<script>speechRecognition = new webkitSpeechRecognition(); speechRecognition.start();</script>" baseURL:[NSURL URLWithString:@"https://webkit.org"]];
     TestWebKitAPI::Util::run(&receivedScriptMessage);
     EXPECT_WK_STREQ(@"Audio Mute", [lastScriptMessage body]);
 }

 // FIXME: test this on iOS when https://webkit.org/b/175204 is fixed.
 #if PLATFORM(MAC)

 TEST(WebKit2, SpeechRecognitionPageBecomesInvisible)
 {
     auto configuration = adoptNS([[WKWebViewConfiguration alloc] init]);
     auto handler = adoptNS([[SpeechRecognitionMessageHandler alloc] init]);
     [[configuration userContentController] addScriptMessageHandler:handler.get() name:@"testHandler"];
     auto preferences = [configuration preferences];
     preferences._mockCaptureDevicesEnabled = YES;
     preferences._speechRecognitionEnabled = YES;
     auto webView = adoptNS([[TestWKWebView alloc] initWithFrame:CGRectMake(0, 0, 800, 600) configuration:configuration.get()]);
     auto delegate = adoptNS([[SpeechRecognitionUIDelegate alloc] init]);
     [webView setUIDelegate:delegate.get()];

     // Page is visible.
     shouldGrantPermissionRequest = true;
     receivedScriptMessage = false;
     [webView synchronouslyLoadTestPageNamed:@"speechrecognition-basic"];
     [webView evaluateJavaScript:@"setShouldHandleEndEvent(true); start();" completionHandler:nil];
     TestWebKitAPI::Util::run(&receivedScriptMessage);
     EXPECT_WK_STREQ(@"Start", [lastScriptMessage body]);

     // Hide page.
     receivedScriptMessage = false;
     [webView.get().window setIsVisible:NO];
     Util::sleep(0.1);
     // Ongoing recognition does not stop automatically.
     EXPECT_FALSE(receivedScriptMessage);
     [webView stringByEvaluatingJavaScript:@"stop()"];
     TestWebKitAPI::Util::run(&receivedScriptMessage);
     EXPECT_WK_STREQ(@"End", [lastScriptMessage body]);

     // Page is invisible.
     receivedScriptMessage = false;
     [webView evaluateJavaScript:@"start()" completionHandler:nil];
     TestWebKitAPI::Util::run(&receivedScriptMessage);
     EXPECT_WK_STREQ(@"Error: not-allowed - Page is not visible to user", [lastScriptMessage body]);
 }

 TEST(WebKit2, SpeechRecognitionPageIsDestroyed)
 {
     auto configuration = adoptNS([[WKWebViewConfiguration alloc] init]);
     auto preferences = [configuration preferences];
     preferences._mockCaptureDevicesEnabled = YES;
     preferences._speechRecognitionEnabled = YES;
     preferences.javaScriptCanOpenWindowsAutomatically = YES;
     auto delegate = adoptNS([[SpeechRecognitionUIDelegate alloc] init]);
     auto navigationDelegate = adoptNS([[SpeechRecognitionNavigationDelegate alloc] init]);
     shouldGrantPermissionRequest = true;
     createdWebView = nullptr;

     @autoreleasepool {
         auto webView = adoptNS([[WKWebView alloc] initWithFrame:NSMakeRect(0, 0, 800, 600) configuration:configuration.get()]);
         [webView setUIDelegate:delegate.get()];
         [webView setNavigationDelegate:navigationDelegate.get()];

         didFinishNavigationBoolean = false;
         [webView loadHTMLString:@"<script>speechRecognition = new webkitSpeechRecognition(); speechRecognition.start(); speechRecognition = null;</script>" baseURL:[NSURL URLWithString:@"http://webkit.org"]];
         TestWebKitAPI::Util::run(&didFinishNavigationBoolean);
         [configuration.get().processPool _garbageCollectJavaScriptObjectsForTesting];

         bool finishedRunningScript = false;
         [webView evaluateJavaScript:@"open('http://webkit.org')" completionHandler: [&] (id result, NSError *error) {
             finishedRunningScript = true;
         }];
         TestWebKitAPI::Util::run(&finishedRunningScript);
     }

     TestWebKitAPI::Util::sleep(0.5);

     EXPECT_TRUE(!!createdWebView);
 }

 TEST(WebKit2, SpeechRecognitionMediaCaptureStateChange)
 {
     auto configuration = adoptNS([[WKWebViewConfiguration alloc] init]);
     auto handler = adoptNS([[SpeechRecognitionMessageHandler alloc] init]);
     [[configuration userContentController] addScriptMessageHandler:handler.get() name:@"testHandler"];
     auto preferences = [configuration preferences];
     preferences._mockCaptureDevicesEnabled = YES;
     preferences._speechRecognitionEnabled = YES;
     auto delegate = adoptNS([[SpeechRecognitionUIDelegate alloc] init]);
     auto webView = adoptNS([[TestWKWebView alloc] initWithFrame:CGRectMake(0, 0, 800, 600) configuration:configuration.get()]);
     [webView setUIDelegate:delegate.get()];
     shouldGrantPermissionRequest = true;

     captureStateDidChange = false;
     [webView synchronouslyLoadTestPageNamed:@"speechrecognition-basic"];
     [webView stringByEvaluatingJavaScript:@"start()"];
     TestWebKitAPI::Util::run(&captureStateDidChange);
     EXPECT_TRUE(isCapturing);

     captureStateDidChange = false;
     [webView stringByEvaluatingJavaScript:@"stop()"];
     TestWebKitAPI::Util::run(&captureStateDidChange);
     EXPECT_FALSE(isCapturing);
 }

 #endif

 TEST(WebKit2, SpeechRecognitionWebProcessCrash)
 {
     auto configuration = adoptNS([[WKWebViewConfiguration alloc] init]);
     auto handler = adoptNS([[SpeechRecognitionMessageHandler alloc] init]);
     [[configuration userContentController] addScriptMessageHandler:handler.get() name:@"testHandler"];
     auto preferences = [configuration preferences];
     preferences._mockCaptureDevicesEnabled = YES;
     preferences._speechRecognitionEnabled = YES;
     auto delegate = adoptNS([[SpeechRecognitionUIDelegate alloc] init]);
     shouldGrantPermissionRequest = true;

     @autoreleasepool {
         auto webView = adoptNS([[TestWKWebView alloc] initWithFrame:CGRectMake(0, 0, 800, 600) configuration:configuration.get()]);
         [webView setUIDelegate:delegate.get()];

         receivedScriptMessage = false;
         [webView synchronouslyLoadTestPageNamed:@"speechrecognition-basic"];
         [webView evaluateJavaScript:@"start();" completionHandler:nil];
         TestWebKitAPI::Util::run(&receivedScriptMessage);
         EXPECT_WK_STREQ(@"Start", [lastScriptMessage body]);

         [webView _killWebContentProcess];
     }

     TestWebKitAPI::Util::sleep(0.5);
 }

 } // namespace TestWebKitAPI
	/*
	* Copyright (C) 2020 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 "DeprecatedGlobalValues.h"
	#import "PlatformUtilities.h"
	#import "TestWKWebView.h"
	#import <WebKit/WKPreferencesPrivate.h>
	#import <WebKit/WKProcessPoolPrivate.h>
	#import <WebKit/WKUIDelegatePrivate.h>
	#import <WebKit/WKWebView.h>
	#import <WebKit/WKWebViewConfigurationPrivate.h>
	#import <wtf/RetainPtr.h>

	static bool shouldGrantPermissionRequest = true;
	static bool permissionRequested = false;
	static bool captureStateDidChange;
	static bool isCapturing;
	static RetainPtr<WKWebView> createdWebView;

	@interface SpeechRecognitionUIDelegate : NSObject<WKUIDelegatePrivate>
	- (void)_webView:(WKWebView )webView requestSpeechRecognitionPermissionForOrigin:(WKSecurityOrigin )origin decisionHandler:(void (^)(BOOL))decisionHandler;
	- (void)webView:(WKWebView )webView requestMediaCapturePermissionForOrigin:(WKSecurityOrigin )origin initiatedByFrame:(WKFrameInfo *)frame type:(WKMediaCaptureType)type decisionHandler:(void (^)(WKPermissionDecision decision))decisionHandler;
	- (WKWebView )webView:(WKWebView )webView createWebViewWithConfiguration:(WKWebViewConfiguration )configuration forNavigationAction:(WKNavigationAction )navigationAction windowFeatures:(WKWindowFeatures *)windowFeatures;
	- (void)_webView:(WKWebView *)webView mediaCaptureStateDidChange:(_WKMediaCaptureStateDeprecated)state;
	@end

	@implementation SpeechRecognitionUIDelegate
	- (void)_webView:(WKWebView )webView requestSpeechRecognitionPermissionForOrigin:(WKSecurityOrigin )origin decisionHandler:(void (^)(BOOL))decisionHandler
	{
	permissionRequested = true;
	decisionHandler(shouldGrantPermissionRequest);
	}

	- (void)webView:(WKWebView )webView requestMediaCapturePermissionForOrigin:(WKSecurityOrigin )origin initiatedByFrame:(WKFrameInfo *)frame type:(WKMediaCaptureType)type decisionHandler:(void (^)(WKPermissionDecision decision))decisionHandler
	{
	permissionRequested = true;
	decisionHandler(shouldGrantPermissionRequest ? WKPermissionDecisionGrant : WKPermissionDecisionDeny);
	}

	- (void)_webView:(WKWebView *)webView requestMediaCaptureAuthorization: (_WKCaptureDevices)devices decisionHandler:(void (^)(BOOL))decisionHandler
	{
	permissionRequested = true;
	decisionHandler(shouldGrantPermissionRequest);
	}

	- (WKWebView )webView:(WKWebView )webView createWebViewWithConfiguration:(WKWebViewConfiguration )configuration forNavigationAction:(WKNavigationAction )navigationAction windowFeatures:(WKWindowFeatures *)windowFeatures
	{
	createdWebView = adoptNS([[WKWebView alloc] initWithFrame:CGRectMake(0, 0, 800, 600) configuration:configuration]);
	return createdWebView.get();
	}

	- (void)_webView:(WKWebView *)webView mediaCaptureStateDidChange:(_WKMediaCaptureStateDeprecated)state
	{
	isCapturing = state == _WKMediaCaptureStateDeprecatedActiveMicrophone;
	captureStateDidChange = true;
	}
	@end

	@interface SpeechRecognitionMessageHandler : NSObject <WKScriptMessageHandler>
	@end

	@implementation SpeechRecognitionMessageHandler
	- (void)userContentController:(WKUserContentController )userContentController didReceiveScriptMessage:(WKScriptMessage )message
	{
	receivedScriptMessage = true;
	lastScriptMessage = message;
	}
	@end

	@interface SpeechRecognitionNavigationDelegate : NSObject <WKNavigationDelegate>
	- (void)webView:(WKWebView )webView decidePolicyForNavigationAction:(WKNavigationAction )navigationAction preferences:(WKWebpagePreferences )preferences decisionHandler:(void (^)(WKNavigationActionPolicy, WKWebpagePreferences ))decisionHandler;
	- (void)webView:(WKWebView )webView didFinishNavigation:(WKNavigation )navigation;
	@end

	@implementation SpeechRecognitionNavigationDelegate
	- (void)webView:(WKWebView )webView decidePolicyForNavigationAction:(WKNavigationAction )navigationAction preferences:(WKWebpagePreferences )preferences decisionHandler:(void (^)(WKNavigationActionPolicy, WKWebpagePreferences ))decisionHandler
	{
	decisionHandler(WKNavigationActionPolicyAllow, preferences);
	}

	- (void)webView:(WKWebView )webView didFinishNavigation:(WKNavigation )navigation
	{
	didFinishNavigationBoolean = true;
	}
	@end

	namespace TestWebKitAPI {

	TEST(WebKit2, SpeechRecognitionUserPermissionPersistence)
	{
	auto configuration = adoptNS([[WKWebViewConfiguration alloc] init]);
	auto handler = adoptNS([[SpeechRecognitionMessageHandler alloc] init]);
	[[configuration userContentController] addScriptMessageHandler:handler.get() name:@"testHandler"];
	auto preferences = [configuration preferences];
	preferences._mockCaptureDevicesEnabled = YES;
	preferences._speechRecognitionEnabled = YES;
	auto webView = adoptNS([[TestWKWebView alloc] initWithFrame:CGRectMake(0, 0, 800, 600) configuration:configuration.get()]);
	auto delegate = adoptNS([[SpeechRecognitionUIDelegate alloc] init]);
	[webView setUIDelegate:delegate.get()];

	shouldGrantPermissionRequest = false;
	permissionRequested = false;
	receivedScriptMessage = false;
	[webView loadTestPageNamed:@"speechrecognition-user-permission-persistence"];
	TestWebKitAPI::Util::run(&receivedScriptMessage);
	EXPECT_WK_STREQ(@"Error: not-allowed - User permission check has failed", [lastScriptMessage body]);
	EXPECT_TRUE(permissionRequested);

	// Permission result is remembered.
	permissionRequested = false;
	receivedScriptMessage = false;
	[webView stringByEvaluatingJavaScript:@"start()"];
	TestWebKitAPI::Util::run(&receivedScriptMessage);
	EXPECT_WK_STREQ(@"Error: not-allowed - User permission check has failed", [lastScriptMessage body]);
	EXPECT_FALSE(permissionRequested);

	// Permission result will be cleared after document changes.
	[webView synchronouslyLoadRequest:[NSURLRequest requestWithURL:[NSURL URLWithString:@"about:blank"]]];

	shouldGrantPermissionRequest = true;
	permissionRequested = false;
	receivedScriptMessage = false;
	[webView loadTestPageNamed:@"speechrecognition-user-permission-persistence"];
	TestWebKitAPI::Util::run(&permissionRequested);
	// Should not get error message as permission is granted.
	EXPECT_FALSE(receivedScriptMessage);
	}

	TEST(WebKit2, SpeechRecognitionErrorWhenStartingAudioCaptureOnDifferentPage)
	{
	shouldGrantPermissionRequest = true;

	auto configuration = adoptNS([[WKWebViewConfiguration alloc] init]);
	auto handler = adoptNS([[SpeechRecognitionMessageHandler alloc] init]);
	[[configuration userContentController] addScriptMessageHandler:handler.get() name:@"testHandler"];
	configuration.get()._mediaCaptureEnabled = YES;
	auto preferences = [configuration preferences];
	preferences._mockCaptureDevicesEnabled = YES;
	preferences._speechRecognitionEnabled = YES;
	preferences._mediaCaptureRequiresSecureConnection = NO;
	auto delegate = adoptNS([[SpeechRecognitionUIDelegate alloc] init]);
	auto firstWebView = adoptNS([[TestWKWebView alloc] initWithFrame:CGRectMake(0, 0, 100, 100) configuration:configuration.get()]);
	[firstWebView setUIDelegate:delegate.get()];
	auto secondWebView = adoptNS([[TestWKWebView alloc] initWithFrame:CGRectMake(100, 0, 100, 100) configuration:configuration.get()]);
	[secondWebView setUIDelegate:delegate.get()];

	// First page starts recognition successfully.
	receivedScriptMessage = false;
	[firstWebView synchronouslyLoadTestPageNamed:@"speechrecognition-basic"];
	[firstWebView stringByEvaluatingJavaScript:@"start()"];
	TestWebKitAPI::Util::run(&receivedScriptMessage);
	EXPECT_WK_STREQ(@"Start", [lastScriptMessage body]);

	// First page is muted when second page starts recognition.
	// Load html string instead of test page to make sure message only comes from one page.
	receivedScriptMessage = false;
	[secondWebView synchronouslyLoadHTMLString:@"<script>speechRecognition = new webkitSpeechRecognition(); speechRecognition.start();</script>" baseURL:[NSURL URLWithString:@"https://webkit.org"]];
	TestWebKitAPI::Util::run(&receivedScriptMessage);
	EXPECT_WK_STREQ(@"Error: audio-capture - Source is muted", [lastScriptMessage body]);

	// First page restarts recognition successfully.
	receivedScriptMessage = false;
	[firstWebView stringByEvaluatingJavaScript:@"start()"];
	TestWebKitAPI::Util::run(&receivedScriptMessage);
	EXPECT_WK_STREQ(@"Start", [lastScriptMessage body]);

	// First page is muted when second page starts media recorder.
	receivedScriptMessage = false;
	[secondWebView synchronouslyLoadTestPageNamed:@"speechrecognition-basic"];
	[secondWebView stringByEvaluatingJavaScript:@"startAudio()"];
	TestWebKitAPI::Util::run(&receivedScriptMessage);
	EXPECT_WK_STREQ(@"Error: audio-capture - Source is muted", [lastScriptMessage body]);

	// Second page is muted when first page starts recognition.
	receivedScriptMessage = false;
	[firstWebView synchronouslyLoadHTMLString:@"<script>speechRecognition = new webkitSpeechRecognition(); speechRecognition.start();</script>" baseURL:[NSURL URLWithString:@"https://webkit.org"]];
	TestWebKitAPI::Util::run(&receivedScriptMessage);
	EXPECT_WK_STREQ(@"Audio Mute", [lastScriptMessage body]);
	}

	// FIXME: test this on iOS when https://webkit.org/b/175204 is fixed.
	#if PLATFORM(MAC)

	TEST(WebKit2, SpeechRecognitionPageBecomesInvisible)
	{
	auto configuration = adoptNS([[WKWebViewConfiguration alloc] init]);
	auto handler = adoptNS([[SpeechRecognitionMessageHandler alloc] init]);
	[[configuration userContentController] addScriptMessageHandler:handler.get() name:@"testHandler"];
	auto preferences = [configuration preferences];
	preferences._mockCaptureDevicesEnabled = YES;
	preferences._speechRecognitionEnabled = YES;
	auto webView = adoptNS([[TestWKWebView alloc] initWithFrame:CGRectMake(0, 0, 800, 600) configuration:configuration.get()]);
	auto delegate = adoptNS([[SpeechRecognitionUIDelegate alloc] init]);
	[webView setUIDelegate:delegate.get()];

	// Page is visible.
	shouldGrantPermissionRequest = true;
	receivedScriptMessage = false;
	[webView synchronouslyLoadTestPageNamed:@"speechrecognition-basic"];
	[webView evaluateJavaScript:@"setShouldHandleEndEvent(true); start();" completionHandler:nil];
	TestWebKitAPI::Util::run(&receivedScriptMessage);
	EXPECT_WK_STREQ(@"Start", [lastScriptMessage body]);

	// Hide page.
	receivedScriptMessage = false;
	[webView.get().window setIsVisible:NO];
	Util::sleep(0.1);
	// Ongoing recognition does not stop automatically.
	EXPECT_FALSE(receivedScriptMessage);
	[webView stringByEvaluatingJavaScript:@"stop()"];
	TestWebKitAPI::Util::run(&receivedScriptMessage);
	EXPECT_WK_STREQ(@"End", [lastScriptMessage body]);

	// Page is invisible.
	receivedScriptMessage = false;
	[webView evaluateJavaScript:@"start()" completionHandler:nil];
	TestWebKitAPI::Util::run(&receivedScriptMessage);
	EXPECT_WK_STREQ(@"Error: not-allowed - Page is not visible to user", [lastScriptMessage body]);
	}

	TEST(WebKit2, SpeechRecognitionPageIsDestroyed)
	{
	auto configuration = adoptNS([[WKWebViewConfiguration alloc] init]);
	auto preferences = [configuration preferences];
	preferences._mockCaptureDevicesEnabled = YES;
	preferences._speechRecognitionEnabled = YES;
	preferences.javaScriptCanOpenWindowsAutomatically = YES;
	auto delegate = adoptNS([[SpeechRecognitionUIDelegate alloc] init]);
	auto navigationDelegate = adoptNS([[SpeechRecognitionNavigationDelegate alloc] init]);
	shouldGrantPermissionRequest = true;
	createdWebView = nullptr;

	@autoreleasepool {
	auto webView = adoptNS([[WKWebView alloc] initWithFrame:NSMakeRect(0, 0, 800, 600) configuration:configuration.get()]);
	[webView setUIDelegate:delegate.get()];
	[webView setNavigationDelegate:navigationDelegate.get()];

	didFinishNavigationBoolean = false;
	[webView loadHTMLString:@"<script>speechRecognition = new webkitSpeechRecognition(); speechRecognition.start(); speechRecognition = null;</script>" baseURL:[NSURL URLWithString:@"http://webkit.org"]];
	TestWebKitAPI::Util::run(&didFinishNavigationBoolean);
	[configuration.get().processPool _garbageCollectJavaScriptObjectsForTesting];

	bool finishedRunningScript = false;
	[webView evaluateJavaScript:@"open('http://webkit.org')" completionHandler: [&] (id result, NSError *error) {
	finishedRunningScript = true;
	}];
	TestWebKitAPI::Util::run(&finishedRunningScript);
	}

	TestWebKitAPI::Util::sleep(0.5);

	EXPECT_TRUE(!!createdWebView);
	}

	TEST(WebKit2, SpeechRecognitionMediaCaptureStateChange)
	{
	auto configuration = adoptNS([[WKWebViewConfiguration alloc] init]);
	auto handler = adoptNS([[SpeechRecognitionMessageHandler alloc] init]);
	[[configuration userContentController] addScriptMessageHandler:handler.get() name:@"testHandler"];
	auto preferences = [configuration preferences];
	preferences._mockCaptureDevicesEnabled = YES;
	preferences._speechRecognitionEnabled = YES;
	auto delegate = adoptNS([[SpeechRecognitionUIDelegate alloc] init]);
	auto webView = adoptNS([[TestWKWebView alloc] initWithFrame:CGRectMake(0, 0, 800, 600) configuration:configuration.get()]);
	[webView setUIDelegate:delegate.get()];
	shouldGrantPermissionRequest = true;

	captureStateDidChange = false;
	[webView synchronouslyLoadTestPageNamed:@"speechrecognition-basic"];
	[webView stringByEvaluatingJavaScript:@"start()"];
	TestWebKitAPI::Util::run(&captureStateDidChange);
	EXPECT_TRUE(isCapturing);

	captureStateDidChange = false;
	[webView stringByEvaluatingJavaScript:@"stop()"];
	TestWebKitAPI::Util::run(&captureStateDidChange);
	EXPECT_FALSE(isCapturing);
	}

	#endif

	TEST(WebKit2, SpeechRecognitionWebProcessCrash)
	{
	auto configuration = adoptNS([[WKWebViewConfiguration alloc] init]);
	auto handler = adoptNS([[SpeechRecognitionMessageHandler alloc] init]);
	[[configuration userContentController] addScriptMessageHandler:handler.get() name:@"testHandler"];
	auto preferences = [configuration preferences];
	preferences._mockCaptureDevicesEnabled = YES;
	preferences._speechRecognitionEnabled = YES;
	auto delegate = adoptNS([[SpeechRecognitionUIDelegate alloc] init]);
	shouldGrantPermissionRequest = true;

	@autoreleasepool {
	auto webView = adoptNS([[TestWKWebView alloc] initWithFrame:CGRectMake(0, 0, 800, 600) configuration:configuration.get()]);
	[webView setUIDelegate:delegate.get()];

	receivedScriptMessage = false;
	[webView synchronouslyLoadTestPageNamed:@"speechrecognition-basic"];
	[webView evaluateJavaScript:@"start();" completionHandler:nil];
	TestWebKitAPI::Util::run(&receivedScriptMessage);
	EXPECT_WK_STREQ(@"Start", [lastScriptMessage body]);

	[webView _killWebContentProcess];
	}

	TestWebKitAPI::Util::sleep(0.5);
	}

	} // namespace TestWebKitAPI