Fix API test after r241728
[WebKit-https.git] / Tools / TestWebKitAPI / Tests / WebKitCocoa / JavaScriptDuringNavigation.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 #include "config.h"
27
28 #import "PlatformUtilities.h"
29 #import "Test.h"
30 #import <WebKit/WKWebView.h>
31 #import <wtf/RetainPtr.h>
32
33 #if WK_API_ENABLED
34
35 static bool navigationComplete;
36 static size_t alerts;
37 static bool receivedBothAlerts;
38 static RetainPtr<NSURL> firstURL;
39 static RetainPtr<NSURL> secondURL;
40
41 @interface JSNavigationDelegate : NSObject <WKNavigationDelegate, WKUIDelegate>
42 @end
43
44 @implementation JSNavigationDelegate
45
46 - (void)webView:(WKWebView *)webView didFinishNavigation:(null_unspecified WKNavigation *)navigation
47 {
48     navigationComplete = true;
49 }
50
51 - (void)webView:(WKWebView *)webView decidePolicyForNavigationAction:(WKNavigationAction *)navigationAction decisionHandler:(void (^)(WKNavigationActionPolicy))decisionHandler
52 {
53     if ([navigationAction.request.URL.absoluteString isEqualToString:[secondURL absoluteString]]) {
54         [webView evaluateJavaScript:@"alert(document.location);" completionHandler:^(id, NSError *) {
55             decisionHandler(WKNavigationActionPolicyAllow);
56         }];
57     } else
58         decisionHandler(WKNavigationActionPolicyAllow);
59 }
60
61 - (void)webView:(WKWebView *)webView decidePolicyForNavigationResponse:(WKNavigationResponse *)navigationResponse decisionHandler:(void (^)(WKNavigationResponsePolicy))decisionHandler
62 {
63     if ([navigationResponse.response.URL.absoluteString isEqualToString:[secondURL absoluteString]]) {
64         [webView evaluateJavaScript:@"alert(document.location);" completionHandler:^(id, NSError *) {
65             decisionHandler(WKNavigationResponsePolicyAllow);
66         }];
67     } else
68         decisionHandler(WKNavigationResponsePolicyAllow);
69 }
70
71 - (void)webView:(WKWebView *)webView runJavaScriptAlertPanelWithMessage:(NSString *)message initiatedByFrame:(WKFrameInfo *)frame completionHandler:(void (^)(void))completionHandler
72 {
73     EXPECT_STREQ(message.UTF8String, [[firstURL absoluteString] UTF8String]);
74     if (++alerts == 2)
75         receivedBothAlerts = true;
76     completionHandler();
77 }
78
79 @end
80
81 TEST(WebKit, JavaScriptDuringNavigation)
82 {
83     firstURL = [[NSBundle mainBundle] URLForResource:@"simple" withExtension:@"html" subdirectory:@"TestWebKitAPI.resources"];
84     secondURL = [[NSBundle mainBundle] URLForResource:@"simple2" withExtension:@"html" subdirectory:@"TestWebKitAPI.resources"];
85     
86     auto webView = adoptNS([[WKWebView alloc] init]);
87     auto delegate = adoptNS([[JSNavigationDelegate alloc] init]);
88     [webView setNavigationDelegate:delegate.get()];
89     [webView setUIDelegate:delegate.get()];
90
91     [webView loadRequest:[NSURLRequest requestWithURL:firstURL.get()]];
92     TestWebKitAPI::Util::run(&navigationComplete);
93
94     [webView loadRequest:[NSURLRequest requestWithURL:secondURL.get()]];
95     TestWebKitAPI::Util::run(&receivedBothAlerts);
96 }
97
98 #endif