Make it possible for apps that use both WK1 and WK2 to use MessagePorts.
[WebKit-https.git] / Tools / TestWebKitAPI / Tests / WebKitCocoa / MessagePortProviders.mm
1 /*
2  * Copyright (C) 2018 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 "PlatformUtilities.h"
31 #import "TestWKWebView.h"
32 #import <WebKit/WKWebViewPrivate.h>
33 #import <WebKit/WebFrame.h>
34 #import <wtf/RetainPtr.h>
35
36 static bool didFinishLoad;
37
38 @interface MessagePortFrameLoadDelegate : NSObject <WebFrameLoadDelegate> {
39 }
40 @end
41 @implementation MessagePortFrameLoadDelegate
42
43 - (void)webView:(WebView *)sender didFinishLoadForFrame:(WebFrame *)frame
44 {
45     didFinishLoad = true;
46 }
47
48 @end
49
50 namespace TestWebKitAPI {
51
52 TEST(MessagePort, Providers)
53 {
54     // Loading a WebView that uses message ports guarantees that the default MessagePortChannelProviderImpl is set.
55     auto wk1View = adoptNS([[WebView alloc] initWithFrame:NSMakeRect(0, 0, 400, 400) frameName:nil groupName:nil]);
56     auto delegate = adoptNS([[MessagePortFrameLoadDelegate alloc] init]);
57     [wk1View.get() setFrameLoadDelegate:delegate.get()];
58     [[wk1View mainFrame] loadHTMLString:@"<script>new MessageChannel;</script>" baseURL:nil];
59
60     Util::run(&didFinishLoad);
61
62     // Now using a WKWebView to load content that uses message ports will use the WK2-style message ports.
63     // This should not conflict with WK1-style message ports.
64     // The conflict is caught by a RELEASE_ASSERT so, if this doesn't crash, it passes.
65     auto webView = adoptNS([[TestWKWebView alloc] initWithFrame:NSMakeRect(0, 0, 400, 400)]);
66     [webView synchronouslyLoadHTMLString:@"<script>new MessageChannel;</script>"];
67 }
68
69 } // namespace TestWebKitAPI
70
71 #endif // WK_API_ENABLED