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
33namespace TestWebKitAPI {
34
35static bool didReceiveMessage;
36static bool canHandleRequest;
37
38static 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
47static 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
56TEST(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