OLD | NEW |
| (Empty) |
1 /* | |
2 * Copyright (C) 2012 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 #import "config.h" | |
27 #import "Test.h" | |
28 | |
29 #import "PlatformUtilities.h" | |
30 #import "TestBrowsingContextLoadDelegate.h" | |
31 #import <Foundation/Foundation.h> | |
32 #import <WebKit2/WebKit2.h> | |
33 | |
34 static NSString *testScheme = @"test"; | |
35 static NSString *testHost = @"test"; | |
36 static bool testFinished = false; | |
37 | |
38 @interface TestProtocol : NSURLProtocol { | |
39 } | |
40 @end | |
41 | |
42 @implementation TestProtocol | |
43 | |
44 + (BOOL)canInitWithRequest:(NSURLRequest *)request | |
45 { | |
46 return [[[request URL] scheme] caseInsensitiveCompare:testScheme] == NSOrder
edSame; | |
47 } | |
48 | |
49 + (NSURLRequest *)canonicalRequestForRequest:(NSURLRequest *)request | |
50 { | |
51 return request; | |
52 } | |
53 | |
54 + (BOOL)requestIsCacheEquivalent:(NSURLRequest *)a toRequest:(NSURLRequest *)b | |
55 { | |
56 return NO; | |
57 } | |
58 | |
59 - (void)startLoading | |
60 { | |
61 EXPECT_TRUE([[[[self request] URL] scheme] isEqualToString:testScheme]); | |
62 EXPECT_TRUE([[[[self request] URL] host] isEqualToString:testHost]); | |
63 | |
64 NSData *data = [@"<body>PASS</body>" dataUsingEncoding:NSASCIIStringEncoding
]; | |
65 NSURLResponse *response = [[NSURLResponse alloc] initWithURL:[[self request]
URL] MIMEType:@"text/html" expectedContentLength:[data length] textEncodingName
:nil]; | |
66 [[self client] URLProtocol:self didReceiveResponse:response cacheStoragePoli
cy:NSURLCacheStorageNotAllowed]; | |
67 [[self client] URLProtocol:self didLoadData:data]; | |
68 [[self client] URLProtocolDidFinishLoading:self]; | |
69 [response release]; | |
70 } | |
71 | |
72 - (void)stopLoading | |
73 { | |
74 } | |
75 | |
76 @end | |
77 | |
78 TEST(WebKit2CustomProtocolsTest, CustomProtocolUsed) | |
79 { | |
80 [NSURLProtocol registerClass:[TestProtocol class]]; | |
81 [WKBrowsingContextController registerSchemeForCustomProtocol:testScheme]; | |
82 | |
83 WKProcessGroup *processGroup = [[WKProcessGroup alloc] init]; | |
84 WKBrowsingContextGroup *browsingContextGroup = [[WKBrowsingContextGroup allo
c] initWithIdentifier:@"TestIdentifier"]; | |
85 WKView *wkView = [[WKView alloc] initWithFrame:NSMakeRect(0, 0, 800, 600) pr
ocessGroup:processGroup browsingContextGroup:browsingContextGroup]; | |
86 wkView.browsingContextController.loadDelegate = [[TestBrowsingContextLoadDel
egate alloc] initWithBlockToRunOnLoad:^(WKBrowsingContextController *sender) { | |
87 testFinished = true; | |
88 }]; | |
89 [wkView.browsingContextController loadRequest:[NSURLRequest requestWithURL:[
NSURL URLWithString:[NSString stringWithFormat:@"%@://%@", testScheme, testHost]
]]]; | |
90 | |
91 TestWebKitAPI::Util::run(&testFinished); | |
92 } | |
OLD | NEW |