OLD | NEW |
| (Empty) |
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 | |
26 #include "config.h" | |
27 #include "InjectedBundleTest.h" | |
28 #include "PlatformUtilities.h" | |
29 #include "Test.h" | |
30 | |
31 #include <WebKit2/WKBundlePage.h> | |
32 #include <WebKit2/WKBundleFrame.h> | |
33 #include <WebKit2/WKRetainPtr.h> | |
34 | |
35 #include <wtf/Assertions.h> | |
36 | |
37 namespace TestWebKitAPI { | |
38 | |
39 class LoadCanceledNoServerRedirectCallbackTest : public InjectedBundleTest { | |
40 public: | |
41 LoadCanceledNoServerRedirectCallbackTest(const std::string& identifier) | |
42 : InjectedBundleTest(identifier) | |
43 { | |
44 } | |
45 | |
46 static WKURLRequestRef willSendRequestForFrame(WKBundlePageRef, WKBundleFram
eRef frame, uint64_t resourceIdentifier, WKURLRequestRef request, WKURLResponseR
ef redirectResponse, const void *clientInfo) | |
47 { | |
48 // Allow the loading of the main resource, but don't allow the loading o
f an iframe, return null from willSendRequest. | |
49 if (WKBundleFrameIsMainFrame(frame)) { | |
50 WKRetainPtr<WKURLRequestRef> newRequest = request; | |
51 return newRequest.leakRef(); | |
52 } | |
53 | |
54 return 0; | |
55 } | |
56 | |
57 virtual void didCreatePage(WKBundleRef bundle, WKBundlePageRef page) | |
58 { | |
59 WKBundlePageResourceLoadClient resourceLoadClient; | |
60 memset(&resourceLoadClient, 0, sizeof(resourceLoadClient)); | |
61 | |
62 resourceLoadClient.version = 0; | |
63 resourceLoadClient.willSendRequestForFrame = willSendRequestForFrame; | |
64 | |
65 WKBundlePageSetResourceLoadClient(page, &resourceLoadClient); | |
66 | |
67 } | |
68 }; | |
69 | |
70 static InjectedBundleTest::Register<LoadCanceledNoServerRedirectCallbackTest> re
gistrar("LoadCanceledNoServerRedirectCallbackTest"); | |
71 | |
72 } // namespace TestWebKitAPI | |
OLD | NEW |