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