1 /*
2 * Copyright (C) 2011 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 "Test.h"
27
28 #include "PlatformUtilities.h"
29 #include "PlatformWebView.h"
30 #include <WebKit2/WKContextPrivate.h>
31 #include <WebKit2/WKNumber.h>
32
33 namespace TestWebKitAPI {
34
35 static bool didReceiveMessage;
36 static bool canHandleRequest;
37
didReceiveMessageFromInjectedBundle(WKContextRef,WKStringRef messageName,WKTypeRef body,const void *)38 static void didReceiveMessageFromInjectedBundle(WKContextRef, WKStringRef messageName, WKTypeRef body, const void*)
39 {
40 didReceiveMessage = true;
41
42 TEST_ASSERT(WKStringIsEqualToUTF8CString(messageName, "DidCheckCanHandleRequest"));
43 TEST_ASSERT(WKGetTypeID(body) == WKBooleanGetTypeID());
44 canHandleRequest = WKBooleanGetValue(static_cast<WKBooleanRef>(body));
45 }
46
setInjectedBundleClient(WKContextRef context)47 static void setInjectedBundleClient(WKContextRef context)
48 {
49 WKContextInjectedBundleClient injectedBundleClient;
50 memset(&injectedBundleClient, 0, sizeof(injectedBundleClient));
51 injectedBundleClient.didReceiveMessageFromInjectedBundle = didReceiveMessageFromInjectedBundle;
52
53 WKContextSetInjectedBundleClient(context, &injectedBundleClient);
54 }
55
TEST(WebKit2,CanHandleRequest)56 TEST(WebKit2, CanHandleRequest)
57 {
58 WKRetainPtr<WKContextRef> context = adoptWK(Util::createContextForInjectedBundleTest("CanHandleRequestTest"));
59 setInjectedBundleClient(context.get());
60
61 _WKContextRegisterURLSchemeAsEmptyDocument(context.get(), Util::toWK("emptyscheme").get());
62
63 PlatformWebView webView(context.get());
64
65 WKPageLoadURL(webView.page(), adoptWK(Util::createURLForResource("simple", "html")).get());
66
67 WKContextPostMessageToInjectedBundle(context.get(), Util::toWK("CheckCanHandleRequest").get(), 0);
68 Util::run(&didReceiveMessage);
69 TEST_ASSERT(canHandleRequest);
70 }
71
72 } // namespace TestWebKitAPI
73