Home | History | Annotate | Download | only in WebKit2 
      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
     73