InjectedBundleBasic.cpp 3.0 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778
  1. /*
  2. * Copyright (C) 2010 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. #include "config.h"
  26. #include "PlatformUtilities.h"
  27. #include "PlatformWebView.h"
  28. #include <WebKit2/WKRetainPtr.h>
  29. namespace TestWebKitAPI {
  30. static bool done;
  31. static bool loadDone;
  32. static bool messageReceived;
  33. static void didReceiveMessageFromInjectedBundle(WKContextRef context, WKStringRef messageName, WKTypeRef messageBody, const void* clientInfo)
  34. {
  35. messageReceived = true;
  36. if (loadDone)
  37. done = true;
  38. }
  39. static void didFinishLoadForFrame(WKPageRef page, WKFrameRef frame, WKTypeRef userData, const void* clientInfo)
  40. {
  41. loadDone = true;
  42. if (messageReceived)
  43. done = true;
  44. }
  45. TEST(WebKit2, InjectedBundleBasic)
  46. {
  47. WKRetainPtr<WKContextRef> context(AdoptWK, Util::createContextForInjectedBundleTest("InjectedBundleBasicTest"));
  48. WKContextInjectedBundleClient injectedBundleClient;
  49. memset(&injectedBundleClient, 0, sizeof(injectedBundleClient));
  50. injectedBundleClient.version = 0;
  51. injectedBundleClient.clientInfo = 0;
  52. injectedBundleClient.didReceiveMessageFromInjectedBundle = didReceiveMessageFromInjectedBundle;
  53. WKContextSetInjectedBundleClient(context.get(), &injectedBundleClient);
  54. PlatformWebView webView(context.get());
  55. WKPageLoaderClient loaderClient;
  56. memset(&loaderClient, 0, sizeof(loaderClient));
  57. loaderClient.version = 0;
  58. loaderClient.clientInfo = 0;
  59. loaderClient.didFinishLoadForFrame = didFinishLoadForFrame;
  60. WKPageSetPageLoaderClient(webView.page(), &loaderClient);
  61. WKRetainPtr<WKURLRef> url(AdoptWK, Util::createURLForResource("simple", "html"));
  62. WKPageLoadURL(webView.page(), url.get());
  63. Util::run(&done);
  64. }
  65. } // namespace TestWebKitAPI