Provide a TestWebKitAPI-wide WKNavigationDelegate to simplify tests
[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)_webView:(WKWebView *)webView renderingProgressDidChange:(_WKRenderingProgressEvents)progressEvents
49 {
50     if (_renderingProgressDidChange)
51         _renderingProgressDidChange(webView, progressEvents);
52 }
53
54 - (void)waitForDidFinishNavigation
55 {
56     EXPECT_FALSE(self.didFinishNavigation);
57
58     __block bool finished = false;
59     self.didFinishNavigation = ^(WKWebView *, WKNavigation *) {
60         finished = true;
61     };
62
63     TestWebKitAPI::Util::run(&finished);
64
65     self.didFinishNavigation = nil;
66 }
67
68 @end
69
70 @implementation WKWebView (TestWebKitAPIExtras)
71
72 - (void)_test_waitForDidFinishNavigation
73 {
74     EXPECT_FALSE(self.navigationDelegate);
75
76     auto navigationDelegate = adoptNS([[TestNavigationDelegate alloc] init]);
77     self.navigationDelegate = navigationDelegate.get();
78     [navigationDelegate waitForDidFinishNavigation];
79
80     self.navigationDelegate = nil;
81 }
82
83 @end
84
85 #endif