Tools/TestWebKitAPI/Tests/ios/ApplicationStateTracking.mm - WebKit - Git at Google

 /*
  * Copyright (C) 2022 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"

 #if PLATFORM(IOS_FAMILY)

 #import "ClassMethodSwizzler.h"
 #import "PlatformUtilities.h"
 #import "TestNavigationDelegate.h"
 #import "TestPDFHostViewController.h"
 #import "TestProtocol.h"
 #import "TestWKWebView.h"
 #import "WKWebViewConfigurationExtras.h"

 namespace TestWebKitAPI {

 TEST(ApplicationStateTracking, WindowDeallocDoesNotPermanentlyFreezeLayerTree)
 {
     auto configuration = adoptNS([WKWebViewConfiguration new]);

     RetainPtr<TestWKWebView> webView;
     RetainPtr<UIWindow> window;

     auto postWithScene = [&](NSNotificationName name) {
         [NSNotificationCenter.defaultCenter postNotificationName:name object:[window windowScene] userInfo:nil];
     };

     @autoreleasepool {
         webView = adoptNS([[TestWKWebView alloc] initWithFrame:CGRectMake(0, 0, 320, 568) configuration:configuration.get() addToWindow:NO]);
         window = adoptNS([[UIWindow alloc] initWithFrame:CGRectMake(0, 0, 320, 568)]);
         [window addSubview:webView.get()];
         [webView synchronouslyLoadHTMLString:@"<body>Foo</body>"];
         [webView waitForNextPresentationUpdate];
         postWithScene(UISceneDidEnterBackgroundNotification);
         window = nil;
     }

     window = adoptNS([[UIWindow alloc] initWithFrame:CGRectMake(0, 0, 320, 568)]);
     postWithScene(UISceneWillEnterForegroundNotification);
     [window addSubview:webView.get()];
     [webView synchronouslyLoadHTMLString:@"<body>Bar</body>"];
     [webView waitForNextPresentationUpdate];
 }

 #if HAVE(PDFKIT)

 TEST(ApplicationStateTracking, NavigatingFromPDFDoesNotLeaveWebViewInactive)
 {
     RetainPtr configuration = [WKWebViewConfiguration _test_configurationWithTestPlugInClassName:@"WebProcessPlugInWithInternals" configureJSCForTesting:YES];

     auto pdfHostViewControllerSwizzler = createPDFHostViewControllerSwizzler();
     auto webView = adoptNS([[TestWKWebView alloc] initWithFrame:CGRectMake(0, 0, 320, 568) configuration:configuration.get()]);
     auto delegate = adoptNS([TestNavigationDelegate new]);
     [webView setNavigationDelegate:delegate.get()];

     [webView loadSimulatedRequest:[NSURLRequest requestWithURL:[NSURL URLWithString:@"https://foo.com"]] responseHTMLString:@"<body>Hello world</body>"];
     [delegate waitForDidFinishNavigation];
     [webView waitForNextPresentationUpdate];

     RetainPtr fakeURL = [NSURL URLWithString:@"https://bar.com"];
     RetainPtr pdfData = [NSData dataWithContentsOfURL:[[NSBundle mainBundle] URLForResource:@"test" withExtension:@"pdf" subdirectory:@"TestWebKitAPI.resources"]];
     auto response = adoptNS([[NSURLResponse alloc] initWithURL:fakeURL.get() MIMEType:@"application/pdf" expectedContentLength:[pdfData length] textEncodingName:nil]);
     [webView loadSimulatedRequest:[NSURLRequest requestWithURL:fakeURL.get()] response:response.get() responseData:pdfData.get()];
     [delegate waitForDidFinishNavigation];

     [webView goBack];
     [delegate waitForDidFinishNavigation];
     [webView waitForNextPresentationUpdate];

     EXPECT_TRUE([[webView objectByEvaluatingJavaScript:@"internals.isPageActive()"] boolValue]);
 }

 #endif // HAVE(PDFKIT)

 } // namespace TestWebKitAPI

 #endif // PLATFORM(IOS_FAMILY)
	/*
	* Copyright (C) 2022 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"

	#if PLATFORM(IOS_FAMILY)

	#import "ClassMethodSwizzler.h"
	#import "PlatformUtilities.h"
	#import "TestNavigationDelegate.h"
	#import "TestPDFHostViewController.h"
	#import "TestProtocol.h"
	#import "TestWKWebView.h"
	#import "WKWebViewConfigurationExtras.h"

	namespace TestWebKitAPI {

	TEST(ApplicationStateTracking, WindowDeallocDoesNotPermanentlyFreezeLayerTree)
	{
	auto configuration = adoptNS([WKWebViewConfiguration new]);

	RetainPtr<TestWKWebView> webView;
	RetainPtr<UIWindow> window;

	auto postWithScene = [&](NSNotificationName name) {
	[NSNotificationCenter.defaultCenter postNotificationName:name object:[window windowScene] userInfo:nil];
	};

	@autoreleasepool {
	webView = adoptNS([[TestWKWebView alloc] initWithFrame:CGRectMake(0, 0, 320, 568) configuration:configuration.get() addToWindow:NO]);
	window = adoptNS([[UIWindow alloc] initWithFrame:CGRectMake(0, 0, 320, 568)]);
	[window addSubview:webView.get()];
	[webView synchronouslyLoadHTMLString:@"<body>Foo</body>"];
	[webView waitForNextPresentationUpdate];
	postWithScene(UISceneDidEnterBackgroundNotification);
	window = nil;
	}

	window = adoptNS([[UIWindow alloc] initWithFrame:CGRectMake(0, 0, 320, 568)]);
	postWithScene(UISceneWillEnterForegroundNotification);
	[window addSubview:webView.get()];
	[webView synchronouslyLoadHTMLString:@"<body>Bar</body>"];
	[webView waitForNextPresentationUpdate];
	}

	#if HAVE(PDFKIT)

	TEST(ApplicationStateTracking, NavigatingFromPDFDoesNotLeaveWebViewInactive)
	{
	RetainPtr configuration = [WKWebViewConfiguration _test_configurationWithTestPlugInClassName:@"WebProcessPlugInWithInternals" configureJSCForTesting:YES];

	auto pdfHostViewControllerSwizzler = createPDFHostViewControllerSwizzler();
	auto webView = adoptNS([[TestWKWebView alloc] initWithFrame:CGRectMake(0, 0, 320, 568) configuration:configuration.get()]);
	auto delegate = adoptNS([TestNavigationDelegate new]);
	[webView setNavigationDelegate:delegate.get()];

	[webView loadSimulatedRequest:[NSURLRequest requestWithURL:[NSURL URLWithString:@"https://foo.com"]] responseHTMLString:@"<body>Hello world</body>"];
	[delegate waitForDidFinishNavigation];
	[webView waitForNextPresentationUpdate];

	RetainPtr fakeURL = [NSURL URLWithString:@"https://bar.com"];
	RetainPtr pdfData = [NSData dataWithContentsOfURL:[[NSBundle mainBundle] URLForResource:@"test" withExtension:@"pdf" subdirectory:@"TestWebKitAPI.resources"]];
	auto response = adoptNS([[NSURLResponse alloc] initWithURL:fakeURL.get() MIMEType:@"application/pdf" expectedContentLength:[pdfData length] textEncodingName:nil]);
	[webView loadSimulatedRequest:[NSURLRequest requestWithURL:fakeURL.get()] response:response.get() responseData:pdfData.get()];
	[delegate waitForDidFinishNavigation];

	[webView goBack];
	[delegate waitForDidFinishNavigation];
	[webView waitForNextPresentationUpdate];

	EXPECT_TRUE([[webView objectByEvaluatingJavaScript:@"internals.isPageActive()"] boolValue]);
	}

	#endif // HAVE(PDFKIT)

	} // namespace TestWebKitAPI

	#endif // PLATFORM(IOS_FAMILY)