blob: 1cdf75337dc633991b4191688883b4bb4a87c06f [file] [log] [blame]
/*
* Copyright (C) 2015 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/WKFoundation.h>
#import "PlatformUtilities.h"
#import "Test.h"
#import <WebKit/WKWebViewPrivate.h>
#import <wtf/RetainPtr.h>
#import <wtf/cocoa/NSURLExtras.h>
static bool isDone;
static int provisionalLoadCount;
@interface LoadAlternateHTMLStringFromProvisionalLoadErrorController : NSObject <WKNavigationDelegate>
@end
@implementation LoadAlternateHTMLStringFromProvisionalLoadErrorController
- (void)webView:(WKWebView *)webView didFailProvisionalNavigation:(WKNavigation *)navigation withError:(NSError *)error
{
if (error.code != NSURLErrorCancelled)
[webView _loadAlternateHTMLString:@"error page" baseURL:nil forUnreachableURL:error.userInfo[@"NSErrorFailingURLKey"]];
}
- (void)webView:(WKWebView *)webView didFinishNavigation:(WKNavigation *)navigation
{
isDone = true;
}
- (void)webView:(WKWebView *)webView didStartProvisionalNavigation:(null_unspecified WKNavigation *)navigation
{
provisionalLoadCount++;
}
@end
static NSString *unloadableURL = @"data:";
static NSString *loadableURL = @"data:text/html,no%20error";
TEST(WKWebView, LoadAlternateHTMLStringFromProvisionalLoadError)
{
auto webView = adoptNS([[WKWebView alloc] init]);
auto controller = adoptNS([[LoadAlternateHTMLStringFromProvisionalLoadErrorController alloc] init]);
[webView setNavigationDelegate:controller.get()];
[webView loadRequest:[NSURLRequest requestWithURL:[NSURL URLWithString:unloadableURL]]];
TestWebKitAPI::Util::run(&isDone);
isDone = false;
[webView loadRequest:[NSURLRequest requestWithURL:[NSURL URLWithString:loadableURL]]];
TestWebKitAPI::Util::run(&isDone);
isDone = false;
[webView goBack];
TestWebKitAPI::Util::run(&isDone);
isDone = false;
WKBackForwardList *list = [webView backForwardList];
EXPECT_EQ((NSUInteger)0, list.backList.count);
EXPECT_EQ((NSUInteger)1, list.forwardList.count);
EXPECT_STREQ([[list.forwardList.firstObject URL] absoluteString].UTF8String, loadableURL.UTF8String);
EXPECT_STREQ([[list.currentItem URL] absoluteString].UTF8String, unloadableURL.UTF8String);
EXPECT_TRUE([webView canGoForward]);
if (![webView canGoForward])
return;
[webView goForward];
TestWebKitAPI::Util::run(&isDone);
isDone = false;
EXPECT_EQ((NSUInteger)1, list.backList.count);
EXPECT_EQ((NSUInteger)0, list.forwardList.count);
EXPECT_STREQ([[list.backList.firstObject URL] absoluteString].UTF8String, unloadableURL.UTF8String);
EXPECT_STREQ([[list.currentItem URL] absoluteString].UTF8String, loadableURL.UTF8String);
}
TEST(WKWebView, LoadAlternateHTMLStringFromProvisionalLoadErrorBackToBack)
{
@autoreleasepool {
RetainPtr<WKWebView> webView = adoptNS([[WKWebView alloc] initWithFrame:NSMakeRect(0, 0, 800, 600)]);
RetainPtr<LoadAlternateHTMLStringFromProvisionalLoadErrorController> delegate = adoptNS([[LoadAlternateHTMLStringFromProvisionalLoadErrorController alloc] init]);
[webView setNavigationDelegate:delegate.get()];
char literal[] = "https://www.example.com<>/";
NSURL* targetURL = WTF::URLWithData([NSData dataWithBytes:literal length:strlen(literal)], nil);
[webView loadRequest:[NSURLRequest requestWithURL:targetURL]];
[webView loadRequest:[NSURLRequest requestWithURL:targetURL]];
isDone = false;
TestWebKitAPI::Util::run(&isDone);
// In success, we should only start 2 provisional loads: 1 for the second loadRequest, and 1 for the _loadAlternateHTMLString.
// The second loadRequest cancels the first one before its provisional load starts.
EXPECT_EQ(2, provisionalLoadCount);
}
}
TEST(WKWebView, LoadAlternateHTMLStringNoFileSystemPath)
{
auto webView = adoptNS([[WKWebView alloc] init]);
[webView loadHTMLString:@"<html>hi</html>" baseURL:[NSURL URLWithString:@"file:///.file/id="]];
}
TEST(WKWebView, LoadAlternateHTMLStringFromProvisionalLoadErrorReload)
{
auto webView = adoptNS([[WKWebView alloc] initWithFrame:NSMakeRect(0, 0, 800, 600)]);
auto controller = adoptNS([[LoadAlternateHTMLStringFromProvisionalLoadErrorController alloc] init]);
[webView setNavigationDelegate:controller.get()];
NSURL *invalidURL = [NSURL URLWithString:@"https://www.example.com%3C%3E/"];
[webView loadRequest:[NSURLRequest requestWithURL:invalidURL]];
TestWebKitAPI::Util::run(&isDone);
isDone = false;
[webView reload];
TestWebKitAPI::Util::run(&isDone);
isDone = false;
[webView reloadFromOrigin];
TestWebKitAPI::Util::run(&isDone);
isDone = false;
[webView _reloadExpiredOnly];
TestWebKitAPI::Util::run(&isDone);
isDone = false;
WKBackForwardList *list = [webView backForwardList];
EXPECT_EQ((NSUInteger)0, list.backList.count);
}