Add UIDelegatePrivate SPI corresponding to WKPageUIClient.showPage
[WebKit-https.git] / Tools / TestWebKitAPI / Tests / WebKit2Cocoa / UIDelegate.mm
1 /*
2  * Copyright (C) 2017 Apple Inc. All rights reserved.
3  *
4  * Redistribution and use in source and binary forms, with or without
5  * modification, are permitted provided that the following conditions
6  * are met:
7  * 1. Redistributions of source code must retain the above copyright
8  *    notice, this list of conditions and the following disclaimer.
9  * 2. Redistributions in binary form must reproduce the above copyright
10  *    notice, this list of conditions and the following disclaimer in the
11  *    documentation and/or other materials provided with the distribution.
12  *
13  * THIS SOFTWARE IS PROVIDED BY APPLE INC. AND ITS CONTRIBUTORS ``AS IS''
14  * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO,
15  * THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
16  * PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL APPLE INC. OR ITS CONTRIBUTORS
17  * BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
18  * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
19  * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
20  * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
21  * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
22  * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF
23  * THE POSSIBILITY OF SUCH DAMAGE.
24  */
25
26 #import "config.h"
27
28 #if WK_API_ENABLED
29
30 #import "TestWKWebView.h"
31 #import "Utilities.h"
32 #import <WebKit/WKUIDelegatePrivate.h>
33 #import <WebKit/WKWebView.h>
34 #import <wtf/RetainPtr.h>
35
36 @class UITestDelegate;
37
38 static RetainPtr<WKWebView> webViewFromDelegateCallback;
39 static RetainPtr<WKWebView> createdWebView;
40 static RetainPtr<UITestDelegate> delegate;
41 static bool done;
42
43 @interface UITestDelegate : NSObject <WKUIDelegatePrivate, WKURLSchemeHandler>
44 @end
45
46 @implementation UITestDelegate
47
48 - (nullable WKWebView *)webView:(WKWebView *)webView createWebViewWithConfiguration:(WKWebViewConfiguration *)configuration forNavigationAction:(WKNavigationAction *)navigationAction windowFeatures:(WKWindowFeatures *)windowFeatures
49 {
50     createdWebView = adoptNS([[WKWebView alloc] initWithFrame:CGRectMake(0, 0, 800, 600) configuration:configuration]);
51     [createdWebView setUIDelegate:delegate.get()];
52     return createdWebView.get();
53 }
54
55 - (void)_showPage:(WKWebView *)webView
56 {
57     webViewFromDelegateCallback = webView;
58     done = true;
59 }
60
61 - (void)webView:(WKWebView *)webView startURLSchemeTask:(id <WKURLSchemeTask>)urlSchemeTask
62 {
63     NSString *data = @"<script>window.open('other.html');</script>";
64     [urlSchemeTask didReceiveResponse:[[[NSURLResponse alloc] initWithURL:urlSchemeTask.request.URL MIMEType:@"text/html" expectedContentLength:data.length textEncodingName:nil] autorelease]];
65     [urlSchemeTask didReceiveData:[data dataUsingEncoding:NSUTF8StringEncoding]];
66     [urlSchemeTask didFinish];
67 }
68
69 - (void)webView:(WKWebView *)webView stopURLSchemeTask:(id <WKURLSchemeTask>)urlSchemeTask
70 {
71 }
72
73 @end
74
75 TEST(WebKit2, ShowPage)
76 {
77     delegate = adoptNS([[UITestDelegate alloc] init]);
78     auto configuration = adoptNS([[WKWebViewConfiguration alloc] init]);
79     [configuration setURLSchemeHandler:delegate.get() forURLScheme:@"test"];
80     auto webView = adoptNS([[WKWebView alloc] initWithFrame:CGRectMake(0, 0, 800, 600) configuration:configuration.get()]);
81     [webView setUIDelegate:delegate.get()];
82     [webView loadRequest:[NSURLRequest requestWithURL:[NSURL URLWithString:@"test:///first"]]];
83     TestWebKitAPI::Util::run(&done);
84     
85     ASSERT_EQ(webViewFromDelegateCallback, createdWebView);
86 }
87
88 #endif // WK_API_ENABLED