f5dfd5df8cba2e5c69d58035f9b56142832bf44e
[WebKit-https.git] / Tools / TestWebKitAPI / cocoa / TestNavigationDelegate.mm
1 /*
2  * Copyright (C) 2016 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 #import "TestNavigationDelegate.h"
28
29 #import "Utilities.h"
30 #import <wtf/RetainPtr.h>
31
32 #if WK_API_ENABLED
33
34 @implementation TestNavigationDelegate
35
36 - (void)webView:(WKWebView *)webView didFailProvisionalNavigation:(WKNavigation *)navigation withError:(NSError *)error
37 {
38     if (_didFailProvisionalNavigation)
39         _didFailProvisionalNavigation(webView, navigation, error);
40 }
41
42 - (void)webView:(WKWebView *)webView didFinishNavigation:(WKNavigation *)navigation
43 {
44     if (_didFinishNavigation)
45         _didFinishNavigation(webView, navigation);
46 }
47
48 - (void)webViewWebContentProcessDidTerminate:(WKWebView *)webView
49 {
50     if (_webContentProcessDidTerminate)
51         _webContentProcessDidTerminate(webView);
52 }
53
54 - (void)_webView:(WKWebView *)webView renderingProgressDidChange:(_WKRenderingProgressEvents)progressEvents
55 {
56     if (_renderingProgressDidChange)
57         _renderingProgressDidChange(webView, progressEvents);
58 }
59
60 - (void)waitForDidFinishNavigation
61 {
62     EXPECT_FALSE(self.didFinishNavigation);
63
64     __block bool finished = false;
65     self.didFinishNavigation = ^(WKWebView *, WKNavigation *) {
66         finished = true;
67     };
68
69     TestWebKitAPI::Util::run(&finished);
70
71     self.didFinishNavigation = nil;
72 }
73
74 @end
75
76 @implementation WKWebView (TestWebKitAPIExtras)
77
78 - (void)_test_waitForDidFinishNavigation
79 {
80     EXPECT_FALSE(self.navigationDelegate);
81
82     auto navigationDelegate = adoptNS([[TestNavigationDelegate alloc] init]);
83     self.navigationDelegate = navigationDelegate.get();
84     [navigationDelegate waitForDidFinishNavigation];
85
86     self.navigationDelegate = nil;
87
88 #if PLATFORM(IOS)
89     __block bool presentationUpdateHappened = false;
90     [self _doAfterNextPresentationUpdate:^{
91         presentationUpdateHappened = true;
92     }];
93     TestWebKitAPI::Util::run(&presentationUpdateHappened);
94 #endif
95 }
96
97 @end
98
99 #endif