dffac9b6eca840ddfbf969f077e212058db87f53
[WebKit-https.git] / Tools / TestWebKitAPI / Tests / WebKit2 / CanHandleRequest.cpp
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
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
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
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