| /* |
| * Copyright (C) 2014 Apple Inc. All rights reserved. |
| * |
| * Redistribution and use in source and binary forms, with or without |
| * modification, are permitted provided that the following conditions |
| * are met: |
| * 1. Redistributions of source code must retain the above copyright |
| * notice, this list of conditions and the following disclaimer. |
| * 2. Redistributions in binary form must reproduce the above copyright |
| * notice, this list of conditions and the following disclaimer in the |
| * documentation and/or other materials provided with the distribution. |
| * |
| * THIS SOFTWARE IS PROVIDED BY APPLE INC. AND ITS CONTRIBUTORS ``AS IS'' |
| * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, |
| * THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR |
| * PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL APPLE INC. OR ITS CONTRIBUTORS |
| * BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR |
| * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF |
| * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS |
| * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN |
| * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) |
| * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF |
| * THE POSSIBILITY OF SUCH DAMAGE. |
| */ |
| |
| #import "config.h" |
| #import <WebKit/WKFoundation.h> |
| |
| #if WK_API_ENABLED |
| #if PLATFORM(MAC) // No downloading on iOS |
| |
| #import "PlatformUtilities.h" |
| #import "Test.h" |
| #import "TestProtocol.h" |
| #import "TestWKWebView.h" |
| #import <WebCore/FileSystem.h> |
| #import <WebKit/_WKDownload.h> |
| #import <WebKit/_WKDownloadDelegate.h> |
| #import <WebKit/WKNavigationDelegatePrivate.h> |
| #import <WebKit/WKProcessPoolPrivate.h> |
| #import <WebKit/WKUIDelegatePrivate.h> |
| #import <WebKit/WKWebView.h> |
| #import <WebKit/WKWebViewConfiguration.h> |
| #import <wtf/RetainPtr.h> |
| #import <wtf/mac/AppKitCompatibilityDeclarations.h> |
| #import <wtf/text/WTFString.h> |
| |
| static bool isDone; |
| static unsigned redirectCount = 0; |
| static bool hasReceivedResponse; |
| static NSURL *sourceURL = [[NSBundle mainBundle] URLForResource:@"simple" withExtension:@"html" subdirectory:@"TestWebKitAPI.resources"]; |
| static WKWebView* expectedOriginatingWebView; |
| static bool expectedUserInitiatedState = false; |
| |
| @interface DownloadDelegate : NSObject <_WKDownloadDelegate> |
| @end |
| |
| @implementation DownloadDelegate { |
| RetainPtr<_WKDownload> _download; |
| String _destinationPath; |
| long long _expectedContentLength; |
| uint64_t _receivedContentLength; |
| } |
| |
| - (void)_downloadDidStart:(_WKDownload *)download |
| { |
| EXPECT_NULL(_download); |
| EXPECT_NOT_NULL(download); |
| EXPECT_TRUE([[[[download request] URL] path] isEqualToString:[sourceURL path]]); |
| EXPECT_EQ(expectedUserInitiatedState, download.wasUserInitiated); |
| _download = download; |
| } |
| |
| - (void)_download:(_WKDownload *)download didReceiveResponse:(NSURLResponse *)response |
| { |
| hasReceivedResponse = true; |
| EXPECT_EQ(_download, download); |
| EXPECT_TRUE(_expectedContentLength == 0); |
| EXPECT_TRUE(_receivedContentLength == 0); |
| EXPECT_TRUE([[[response URL] path] isEqualToString:[sourceURL path]]); |
| _expectedContentLength = [response expectedContentLength]; |
| } |
| |
| - (void)_download:(_WKDownload *)download didReceiveData:(uint64_t)length |
| { |
| EXPECT_EQ(_download, download); |
| _receivedContentLength += length; |
| } |
| |
| - (NSString *)_download:(_WKDownload *)download decideDestinationWithSuggestedFilename:(NSString *)filename allowOverwrite:(BOOL *)allowOverwrite |
| { |
| EXPECT_TRUE(hasReceivedResponse); |
| EXPECT_EQ(_download, download); |
| |
| WebCore::FileSystem::PlatformFileHandle fileHandle; |
| _destinationPath = WebCore::FileSystem::openTemporaryFile("TestWebKitAPI", fileHandle); |
| EXPECT_TRUE(fileHandle != WebCore::FileSystem::invalidPlatformFileHandle); |
| WebCore::FileSystem::closeFile(fileHandle); |
| |
| *allowOverwrite = YES; |
| return _destinationPath; |
| } |
| |
| - (void)_downloadDidFinish:(_WKDownload *)download |
| { |
| EXPECT_EQ(_download, download); |
| EXPECT_EQ(expectedUserInitiatedState, download.wasUserInitiated); |
| EXPECT_TRUE(_expectedContentLength == NSURLResponseUnknownLength || static_cast<uint64_t>(_expectedContentLength) == _receivedContentLength); |
| EXPECT_TRUE([[NSFileManager defaultManager] contentsEqualAtPath:_destinationPath andPath:[sourceURL path]]); |
| WebCore::FileSystem::deleteFile(_destinationPath); |
| isDone = true; |
| } |
| |
| @end |
| |
| TEST(_WKDownload, DownloadDelegate) |
| { |
| RetainPtr<WKProcessPool> processPool = adoptNS([[WKProcessPool alloc] init]); |
| DownloadDelegate *downloadDelegate = [[DownloadDelegate alloc] init]; |
| [processPool _setDownloadDelegate:downloadDelegate]; |
| |
| @autoreleasepool { |
| EXPECT_EQ(downloadDelegate, [processPool _downloadDelegate]); |
| } |
| |
| [downloadDelegate release]; |
| EXPECT_NULL([processPool _downloadDelegate]); |
| } |
| |
| static void runTest(id <WKNavigationDelegate> navigationDelegate, id <_WKDownloadDelegate> downloadDelegate, NSURL *url) |
| { |
| RetainPtr<WKWebView> webView = adoptNS([[WKWebView alloc] initWithFrame:NSMakeRect(0, 0, 800, 600)]); |
| [webView setNavigationDelegate:navigationDelegate]; |
| [[[webView configuration] processPool] _setDownloadDelegate:downloadDelegate]; |
| |
| isDone = false; |
| hasReceivedResponse = false; |
| expectedUserInitiatedState = false; |
| [webView loadRequest:[NSURLRequest requestWithURL:url]]; |
| TestWebKitAPI::Util::run(&isDone); |
| } |
| |
| @interface DownloadNavigationDelegate : NSObject <WKNavigationDelegate> |
| @end |
| |
| @implementation DownloadNavigationDelegate |
| - (void)webView:(WKWebView *)webView decidePolicyForNavigationAction:(WKNavigationAction *)navigationAction decisionHandler:(void (^)(WKNavigationActionPolicy))decisionHandler |
| { |
| decisionHandler(_WKNavigationActionPolicyDownload); |
| } |
| @end |
| |
| TEST(_WKDownload, DownloadRequest) |
| { |
| runTest(adoptNS([[DownloadNavigationDelegate alloc] init]).get(), adoptNS([[DownloadDelegate alloc] init]).get(), sourceURL); |
| } |
| |
| @interface ConvertResponseToDownloadNavigationDelegate : NSObject <WKNavigationDelegate> |
| @end |
| |
| @implementation ConvertResponseToDownloadNavigationDelegate |
| - (void)webView:(WKWebView *)webView decidePolicyForNavigationResponse:(WKNavigationResponse *)navigationResponse decisionHandler:(void (^)(WKNavigationResponsePolicy))decisionHandler |
| { |
| decisionHandler(_WKNavigationResponsePolicyBecomeDownload); |
| } |
| @end |
| |
| TEST(_WKDownload, ConvertResponseToDownload) |
| { |
| runTest(adoptNS([[ConvertResponseToDownloadNavigationDelegate alloc] init]).get(), adoptNS([[DownloadDelegate alloc] init]).get(), sourceURL); |
| } |
| |
| @interface FailingDownloadDelegate : NSObject <_WKDownloadDelegate> |
| @end |
| |
| @implementation FailingDownloadDelegate |
| |
| - (void)_downloadDidFinish:(_WKDownload *)download |
| { |
| EXPECT_TRUE(false); |
| isDone = true; |
| } |
| |
| - (void)_download:(_WKDownload *)download didFailWithError:(NSError *)error |
| { |
| isDone = true; |
| } |
| |
| - (void)_downloadDidCancel:(_WKDownload *)download |
| { |
| EXPECT_TRUE(false); |
| isDone = true; |
| } |
| |
| @end |
| |
| TEST(_WKDownload, DownloadMissingResource) |
| { |
| runTest(adoptNS([[DownloadNavigationDelegate alloc] init]).get(), adoptNS([[FailingDownloadDelegate alloc] init]).get(), [NSURL URLWithString:@"non-existant-scheme://"]); |
| } |
| |
| @interface CancelledDownloadDelegate : NSObject <_WKDownloadDelegate> |
| @end |
| |
| @implementation CancelledDownloadDelegate |
| |
| - (void)_downloadDidStart:(_WKDownload *)download |
| { |
| [download cancel]; |
| } |
| |
| - (void)_downloadDidFinish:(_WKDownload *)download |
| { |
| EXPECT_TRUE(false); |
| isDone = true; |
| } |
| |
| - (void)_download:(_WKDownload *)download didFailWithError:(NSError *)error |
| { |
| EXPECT_TRUE(false); |
| isDone = true; |
| } |
| |
| - (void)_downloadDidCancel:(_WKDownload *)download |
| { |
| isDone = true; |
| } |
| |
| @end |
| |
| TEST(_WKDownload, CancelDownload) |
| { |
| runTest(adoptNS([[DownloadNavigationDelegate alloc] init]).get(), adoptNS([[CancelledDownloadDelegate alloc] init]).get(), sourceURL); |
| } |
| |
| @interface OriginatingWebViewDownloadDelegate : NSObject <_WKDownloadDelegate> |
| - (instancetype)initWithWebView:(WKWebView *)webView; |
| @end |
| |
| @implementation OriginatingWebViewDownloadDelegate { |
| RetainPtr<WKWebView> _webView; |
| } |
| |
| - (instancetype)initWithWebView:(WKWebView *)webView |
| { |
| if (!(self = [super init])) |
| return nil; |
| |
| _webView = webView; |
| return self; |
| } |
| |
| - (void)_downloadDidStart:(_WKDownload *)download |
| { |
| @autoreleasepool { |
| EXPECT_EQ([download originatingWebView], _webView); |
| } |
| |
| _webView = nullptr; |
| EXPECT_NULL([download originatingWebView]); |
| isDone = true; |
| } |
| |
| @end |
| |
| TEST(_WKDownload, OriginatingWebView) |
| { |
| RetainPtr<DownloadNavigationDelegate> navigationDelegate = adoptNS([[DownloadNavigationDelegate alloc] init]); |
| RetainPtr<OriginatingWebViewDownloadDelegate> downloadDelegate; |
| @autoreleasepool { |
| RetainPtr<WKWebView> webView = adoptNS([[WKWebView alloc] initWithFrame:NSMakeRect(0, 0, 800, 600)]); |
| [webView setNavigationDelegate:navigationDelegate.get()]; |
| downloadDelegate = adoptNS([[OriginatingWebViewDownloadDelegate alloc] initWithWebView:webView.get()]); |
| [[[webView configuration] processPool] _setDownloadDelegate:downloadDelegate.get()]; |
| [webView loadRequest:[NSURLRequest requestWithURL:sourceURL]]; |
| } |
| |
| isDone = false; |
| TestWebKitAPI::Util::run(&isDone); |
| } |
| |
| @interface DownloadRequestOriginalURLDelegate : NSObject <_WKDownloadDelegate> |
| - (instancetype)initWithExpectedOriginalURL:(NSURL *)expectOriginalURL; |
| @end |
| |
| @implementation DownloadRequestOriginalURLDelegate { |
| NSURL *_expectedOriginalURL; |
| } |
| |
| - (instancetype)initWithExpectedOriginalURL:(NSURL *)expectedOriginalURL |
| { |
| if (!(self = [super init])) |
| return nil; |
| |
| _expectedOriginalURL = expectedOriginalURL; |
| return self; |
| } |
| |
| - (void)_downloadDidStart:(_WKDownload *)download |
| { |
| if ([_expectedOriginalURL isEqual:sourceURL]) |
| EXPECT_TRUE(!download.request.mainDocumentURL); |
| else |
| EXPECT_TRUE([_expectedOriginalURL isEqual:download.request.mainDocumentURL]); |
| isDone = true; |
| } |
| |
| @end |
| |
| @interface DownloadRequestOriginalURLNavigationDelegate : NSObject <WKNavigationDelegate> |
| @end |
| |
| @implementation DownloadRequestOriginalURLNavigationDelegate |
| - (void)webView:(WKWebView *)webView decidePolicyForNavigationAction:(WKNavigationAction *)navigationAction decisionHandler:(void (^)(WKNavigationActionPolicy))decisionHandler |
| { |
| if ([navigationAction.request.URL isEqual:sourceURL]) |
| decisionHandler(_WKNavigationActionPolicyDownload); |
| else |
| decisionHandler(WKNavigationActionPolicyAllow); |
| } |
| @end |
| |
| TEST(_WKDownload, DownloadRequestOriginalURL) |
| { |
| NSURL *originalURL = [[NSBundle mainBundle] URLForResource:@"DownloadRequestOriginalURL" withExtension:@"html" subdirectory:@"TestWebKitAPI.resources"]; |
| runTest(adoptNS([[DownloadRequestOriginalURLNavigationDelegate alloc] init]).get(), adoptNS([[DownloadRequestOriginalURLDelegate alloc] initWithExpectedOriginalURL:originalURL]).get(), originalURL); |
| } |
| |
| TEST(_WKDownload, DownloadRequestOriginalURLFrame) |
| { |
| NSURL *originalURL = [[NSBundle mainBundle] URLForResource:@"DownloadRequestOriginalURL2" withExtension:@"html" subdirectory:@"TestWebKitAPI.resources"]; |
| runTest(adoptNS([[DownloadRequestOriginalURLNavigationDelegate alloc] init]).get(), adoptNS([[DownloadRequestOriginalURLDelegate alloc] initWithExpectedOriginalURL:originalURL]).get(), originalURL); |
| } |
| |
| TEST(_WKDownload, DownloadRequestOriginalURLDirectDownload) |
| { |
| runTest(adoptNS([[DownloadRequestOriginalURLNavigationDelegate alloc] init]).get(), adoptNS([[DownloadRequestOriginalURLDelegate alloc] initWithExpectedOriginalURL:sourceURL]).get(), sourceURL); |
| } |
| |
| TEST(_WKDownload, DownloadRequestOriginalURLDirectDownloadWithLoadedContent) |
| { |
| auto webView = adoptNS([[WKWebView alloc] initWithFrame:NSMakeRect(0, 0, 800, 600)]); |
| [webView setNavigationDelegate:[[DownloadRequestOriginalURLNavigationDelegate alloc] init]]; |
| [[[webView configuration] processPool] _setDownloadDelegate:[[DownloadRequestOriginalURLDelegate alloc] initWithExpectedOriginalURL:sourceURL]]; |
| |
| expectedUserInitiatedState = false; |
| NSURL *contentURL = [[NSBundle mainBundle] URLForResource:@"simple2" withExtension:@"html" subdirectory:@"TestWebKitAPI.resources"]; |
| // Here is to test if the original URL can be set correctly when the current document |
| // is completely unrelated to the download. |
| [webView loadRequest:[NSURLRequest requestWithURL:contentURL]]; |
| [webView loadRequest:[NSURLRequest requestWithURL:sourceURL]]; |
| isDone = false; |
| TestWebKitAPI::Util::run(&isDone); |
| } |
| |
| @interface BlobDownloadDelegate : NSObject <_WKDownloadDelegate> |
| @end |
| |
| @implementation BlobDownloadDelegate { |
| RetainPtr<_WKDownload> _download; |
| String _destinationPath; |
| long long _expectedContentLength; |
| uint64_t _receivedContentLength; |
| } |
| |
| - (void)_downloadDidStart:(_WKDownload *)download |
| { |
| EXPECT_NULL(_download); |
| EXPECT_NOT_NULL(download); |
| EXPECT_TRUE([[[[download request] URL] scheme] isEqualToString:@"blob"]); |
| EXPECT_EQ(expectedUserInitiatedState, download.wasUserInitiated); |
| _download = download; |
| } |
| |
| - (void)_download:(_WKDownload *)download didReceiveResponse:(NSURLResponse *)response |
| { |
| hasReceivedResponse = true; |
| EXPECT_EQ(_download, download); |
| EXPECT_EQ(_expectedContentLength, 0U); |
| EXPECT_EQ(_receivedContentLength, 0U); |
| EXPECT_TRUE([[[response URL] scheme] isEqualToString:@"blob"]); |
| _expectedContentLength = [response expectedContentLength]; |
| } |
| |
| - (void)_download:(_WKDownload *)download didReceiveData:(uint64_t)length |
| { |
| EXPECT_EQ(_download, download); |
| _receivedContentLength += length; |
| } |
| |
| - (NSString *)_download:(_WKDownload *)download decideDestinationWithSuggestedFilename:(NSString *)filename allowOverwrite:(BOOL *)allowOverwrite |
| { |
| EXPECT_TRUE(hasReceivedResponse); |
| EXPECT_EQ(_download, download); |
| |
| WebCore::FileSystem::PlatformFileHandle fileHandle; |
| _destinationPath = WebCore::FileSystem::openTemporaryFile("TestWebKitAPI", fileHandle); |
| EXPECT_TRUE(fileHandle != WebCore::FileSystem::invalidPlatformFileHandle); |
| WebCore::FileSystem::closeFile(fileHandle); |
| |
| *allowOverwrite = YES; |
| return _destinationPath; |
| } |
| |
| - (void)_downloadDidFinish:(_WKDownload *)download |
| { |
| EXPECT_EQ(_download, download); |
| EXPECT_EQ(expectedUserInitiatedState, download.wasUserInitiated); |
| EXPECT_TRUE(_expectedContentLength == NSURLResponseUnknownLength || static_cast<uint64_t>(_expectedContentLength) == _receivedContentLength); |
| NSString* expectedContent = @"{\"x\":42,\"s\":\"hello, world\"}"; |
| NSData* expectedData = [expectedContent dataUsingEncoding:NSUTF8StringEncoding]; |
| EXPECT_TRUE([[[NSFileManager defaultManager] contentsAtPath:_destinationPath] isEqualToData:expectedData]); |
| WebCore::FileSystem::deleteFile(_destinationPath); |
| isDone = true; |
| } |
| |
| @end |
| |
| @interface DownloadBlobURLNavigationDelegate : NSObject <WKNavigationDelegate> |
| @end |
| |
| @implementation DownloadBlobURLNavigationDelegate |
| - (void)webView:(WKWebView *)webView decidePolicyForNavigationAction:(WKNavigationAction *)navigationAction decisionHandler:(void (^)(WKNavigationActionPolicy))decisionHandler |
| { |
| if ([navigationAction.request.URL.scheme isEqualToString:@"blob"]) |
| decisionHandler(_WKNavigationActionPolicyDownload); |
| else |
| decisionHandler(WKNavigationActionPolicyAllow); |
| } |
| @end |
| |
| TEST(_WKDownload, DownloadRequestBlobURL) |
| { |
| NSURL *originalURL = [[NSBundle mainBundle] URLForResource:@"DownloadRequestBlobURL" withExtension:@"html" subdirectory:@"TestWebKitAPI.resources"]; |
| runTest(adoptNS([[DownloadBlobURLNavigationDelegate alloc] init]).get(), adoptNS([[BlobDownloadDelegate alloc] init]).get(), originalURL); |
| } |
| |
| @interface UIDownloadAsFileTestDelegate : NSObject <WKUIDelegatePrivate> |
| @end |
| |
| @implementation UIDownloadAsFileTestDelegate |
| |
| - (NSMenu *)_webView:(WKWebView *)webView contextMenu:(NSMenu *)menu forElement:(_WKContextMenuElementInfo *)element |
| { |
| static const long downloadLinkedFileTag = 2; |
| auto index = [menu indexOfItemWithTag:downloadLinkedFileTag]; |
| [menu performActionForItemAtIndex:index]; |
| return nil; |
| } |
| |
| @end |
| |
| @interface RedirectedDownloadDelegate : NSObject <_WKDownloadDelegate> |
| @end |
| |
| @implementation RedirectedDownloadDelegate { |
| String _destinationPath; |
| } |
| |
| - (void)_downloadDidStart:(_WKDownload *)download |
| { |
| EXPECT_NOT_NULL(download); |
| EXPECT_EQ(expectedOriginatingWebView, download.originatingWebView); |
| EXPECT_EQ(expectedUserInitiatedState, download.wasUserInitiated); |
| } |
| |
| - (NSString *)_download:(_WKDownload *)download decideDestinationWithSuggestedFilename:(NSString *)filename allowOverwrite:(BOOL *)allowOverwrite |
| { |
| WebCore::FileSystem::PlatformFileHandle fileHandle; |
| _destinationPath = WebCore::FileSystem::openTemporaryFile("TestWebKitAPI", fileHandle); |
| EXPECT_TRUE(fileHandle != WebCore::FileSystem::invalidPlatformFileHandle); |
| WebCore::FileSystem::closeFile(fileHandle); |
| *allowOverwrite = YES; |
| return _destinationPath; |
| } |
| |
| - (void)_download:(_WKDownload *)download didReceiveServerRedirectToURL:(NSURL *)url |
| { |
| if (!redirectCount) |
| EXPECT_STREQ("http://redirect/?pass", [url.absoluteString UTF8String]); |
| else |
| EXPECT_STREQ("http://pass/", [url.absoluteString UTF8String]); |
| ++redirectCount = true; |
| } |
| |
| - (void)_downloadDidFinish:(_WKDownload *)download |
| { |
| EXPECT_EQ(expectedUserInitiatedState, download.wasUserInitiated); |
| |
| NSArray<NSURL *> *redirectChain = download.redirectChain; |
| EXPECT_EQ(3U, redirectChain.count); |
| if (redirectChain.count > 0) |
| EXPECT_STREQ("http://redirect/?redirect/?pass", [redirectChain[0].absoluteString UTF8String]); |
| if (redirectChain.count > 1) |
| EXPECT_STREQ("http://redirect/?pass", [redirectChain[1].absoluteString UTF8String]); |
| if (redirectChain.count > 2) |
| EXPECT_STREQ("http://pass/", [redirectChain[2].absoluteString UTF8String]); |
| |
| WebCore::FileSystem::deleteFile(_destinationPath); |
| isDone = true; |
| } |
| |
| @end |
| |
| TEST(_WKDownload, RedirectedDownload) |
| { |
| [TestProtocol registerWithScheme:@"http"]; |
| |
| redirectCount = 0; |
| isDone = false; |
| |
| auto delegate = adoptNS([[UIDownloadAsFileTestDelegate alloc] init]); |
| auto configuration = adoptNS([[WKWebViewConfiguration alloc] init]); |
| auto webView = adoptNS([[TestWKWebView alloc] initWithFrame:CGRectMake(0, 0, 800, 600) configuration:configuration.get()]); |
| [webView setUIDelegate:delegate.get()]; |
| auto downloadDelegate = adoptNS([[RedirectedDownloadDelegate alloc] init]); |
| [[[webView configuration] processPool] _setDownloadDelegate:downloadDelegate.get()]; |
| |
| auto window = adoptNS([[NSWindow alloc] initWithContentRect:[webView frame] styleMask:NSWindowStyleMaskBorderless backing:NSBackingStoreBuffered defer:YES]); |
| [[window contentView] addSubview:webView.get()]; |
| |
| // Do 2 loads in the same view to make sure the redirect chain is properly cleared between loads. |
| [webView synchronouslyLoadHTMLString:@"<div>First load</div>"]; |
| [webView synchronouslyLoadHTMLString:@"<a style='display: block; height: 100%; width: 100%' href='http://redirect/?redirect/?pass'>test</a>"]; |
| |
| expectedOriginatingWebView = webView.get(); |
| expectedUserInitiatedState = true; |
| NSPoint clickPoint = NSMakePoint(100, 100); |
| [[webView hitTest:clickPoint] mouseDown:[NSEvent mouseEventWithType:NSEventTypeRightMouseDown location:clickPoint modifierFlags:0 timestamp:0 windowNumber:[window windowNumber] context:nil eventNumber:0 clickCount:1 pressure:1]]; |
| [[webView hitTest:clickPoint] mouseUp:[NSEvent mouseEventWithType:NSEventTypeRightMouseUp location:clickPoint modifierFlags:0 timestamp:0 windowNumber:[window windowNumber] context:nil eventNumber:0 clickCount:1 pressure:1]]; |
| |
| isDone = false; |
| TestWebKitAPI::Util::run(&isDone); |
| EXPECT_EQ(1U, redirectCount); |
| |
| [TestProtocol unregister]; |
| } |
| |
| TEST(_WKDownload, RedirectedLoadConvertedToDownload) |
| { |
| [TestProtocol registerWithScheme:@"http"]; |
| |
| auto navigationDelegate = adoptNS([[ConvertResponseToDownloadNavigationDelegate alloc] init]); |
| auto downloadDelegate = adoptNS([[RedirectedDownloadDelegate alloc] init]); |
| |
| auto webView = adoptNS([[WKWebView alloc] initWithFrame:NSMakeRect(0, 0, 800, 600)]); |
| [webView setNavigationDelegate:navigationDelegate.get()]; |
| [[[webView configuration] processPool] _setDownloadDelegate:downloadDelegate.get()]; |
| |
| expectedOriginatingWebView = webView.get(); |
| expectedUserInitiatedState = false; |
| isDone = false; |
| redirectCount = 0; |
| hasReceivedResponse = false; |
| [webView loadRequest:[NSURLRequest requestWithURL:[NSURL URLWithString:@"http://redirect/?redirect/?pass"]]]; |
| TestWebKitAPI::Util::run(&isDone); |
| EXPECT_EQ(0U, redirectCount); |
| |
| [TestProtocol unregister]; |
| } |
| |
| TEST(_WKDownload, RedirectedSubframeLoadConvertedToDownload) |
| { |
| [TestProtocol registerWithScheme:@"http"]; |
| |
| auto navigationDelegate = adoptNS([[ConvertResponseToDownloadNavigationDelegate alloc] init]); |
| auto downloadDelegate = adoptNS([[RedirectedDownloadDelegate alloc] init]); |
| |
| auto webView = adoptNS([[WKWebView alloc] initWithFrame:NSMakeRect(0, 0, 800, 600)]); |
| [webView setNavigationDelegate:navigationDelegate.get()]; |
| [[[webView configuration] processPool] _setDownloadDelegate:downloadDelegate.get()]; |
| |
| expectedOriginatingWebView = webView.get(); |
| expectedUserInitiatedState = false; |
| isDone = false; |
| redirectCount = 0; |
| hasReceivedResponse = false; |
| [webView loadHTMLString:@"<body><iframe src='http://redirect/?redirect/?pass'></iframe></body>" baseURL:nil]; |
| TestWebKitAPI::Util::run(&isDone); |
| EXPECT_EQ(0U, redirectCount); |
| |
| [TestProtocol unregister]; |
| } |
| |
| #endif |
| #endif |