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 = Util::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(), Util::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