/* * Copyright (C) 2018 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 "Test.h" #if ENABLE(DRAG_SUPPORT) && !PLATFORM(MACCATALYST) #import "DragAndDropSimulator.h" #import "PlatformUtilities.h" #import "WKWebViewConfigurationExtras.h" #import #import #import #if PLATFORM(IOS_FAMILY) #import #endif @implementation TestWKWebView (DragAndDropTests) - (void)selectElementWithID:(NSString *)elementID { [self evaluateJavaScript:[NSString stringWithFormat:@"getSelection().selectAllChildren(document.getElementById('%@'))", elementID] completionHandler:nil]; [self waitForNextPresentationUpdate]; } @end @implementation DragAndDropSimulator (DragAndDropTests) - (void)dragFromElementWithID:(NSString *)elementID to:(CGPoint)destination { NSArray *rectValues = [self.webView objectByEvaluatingJavaScript:[NSString stringWithFormat:@"(() => {" " const element = document.getElementById('%@');" " const bounds = element.getBoundingClientRect();" " return [bounds.left, bounds.top, bounds.width, bounds.height];" "})();", elementID]]; auto bounds = CGRectMake(rectValues[0].floatValue, rectValues[1].floatValue, rectValues[2].floatValue, rectValues[3].floatValue); auto midPoint = CGPointMake(CGRectGetMidX(bounds), CGRectGetMidY(bounds)); [self runFrom:midPoint to:destination]; } @end TEST(DragAndDropTests, ModernWebArchiveType) { NSData *markupData = [@"Hello world" dataUsingEncoding:NSUTF8StringEncoding]; auto mainResource = adoptNS([[WebResource alloc] initWithData:markupData URL:[NSURL URLWithString:@"foo.html"] MIMEType:@"text/html" textEncodingName:@"utf-8" frameName:nil]); auto archive = adoptNS([[WebArchive alloc] initWithMainResource:mainResource.get() subresources:@[ ] subframeArchives:@[ ]]); NSString *webArchiveType = (__bridge NSString *)kUTTypeWebArchive; auto simulator = adoptNS([[DragAndDropSimulator alloc] initWithWebViewFrame:CGRectMake(0, 0, 320, 500)]); auto webView = [simulator webView]; [webView synchronouslyLoadHTMLString:@""]; #if PLATFORM(MAC) NSPasteboard *pasteboard = [NSPasteboard pasteboardWithUniqueName]; [pasteboard declareTypes:@[webArchiveType, (__bridge NSString *)kUTTypeUTF8PlainText] owner:nil]; [pasteboard setData:[archive data] forType:webArchiveType]; [pasteboard setData:[@"Hello world" dataUsingEncoding:NSUTF8StringEncoding] forType:(__bridge NSString *)kUTTypeUTF8PlainText]; [simulator setExternalDragPasteboard:pasteboard]; #else auto item = adoptNS([[NSItemProvider alloc] init]); [item registerDataRepresentationForTypeIdentifier:webArchiveType visibility:NSItemProviderRepresentationVisibilityAll loadHandler:[&] (void (^completionHandler)(NSData *, NSError *)) -> NSProgress * { completionHandler([archive data], nil); return nil; }]; [item registerDataRepresentationForTypeIdentifier:(__bridge NSString *)kUTTypeUTF8PlainText visibility:NSItemProviderRepresentationVisibilityAll loadHandler:[&] (void (^completionHandler)(NSData *, NSError *)) -> NSProgress * { completionHandler([@"Hello world" dataUsingEncoding:NSUTF8StringEncoding], nil); return nil; }]; [simulator setExternalItemProviders:@[ item.get() ]]; #endif [simulator runFrom:CGPointMake(0, 0) to:CGPointMake(50, 50)]; [webView stringByEvaluatingJavaScript:@"document.body.focus(); getSelection().setBaseAndExtent(document.body, 0, document.body, 1)"]; EXPECT_WK_STREQ("Hello world", [webView stringByEvaluatingJavaScript:@"document.body.textContent"]); EXPECT_TRUE([webView stringByEvaluatingJavaScript:@"document.queryCommandState('bold')"].boolValue); EXPECT_TRUE([webView stringByEvaluatingJavaScript:@"document.queryCommandState('italic')"].boolValue); } TEST(DragAndDropTests, DragImageLocationForLinkInSubframe) { auto simulator = adoptNS([[DragAndDropSimulator alloc] initWithWebViewFrame:CGRectMake(0, 0, 400, 400)]); [[simulator webView] synchronouslyLoadTestPageNamed:@"link-in-iframe-and-input"]; [simulator runFrom:CGPointMake(200, 375) to:CGPointMake(200, 125)]; EXPECT_WK_STREQ("https://www.apple.com/", [[simulator webView] stringByEvaluatingJavaScript:@"document.querySelector('input').value"]); #if PLATFORM(MAC) EXPECT_TRUE(NSPointInRect([simulator initialDragImageLocationInView], NSMakeRect(0, 250, 400, 250))); #endif } TEST(DragAndDropTests, ExposeMultipleURLsInDataTransfer) { auto simulator = adoptNS([[DragAndDropSimulator alloc] initWithWebViewFrame:CGRectMake(0, 0, 320, 500)]); auto webView = [simulator webView]; WKPreferencesSetCustomPasteboardDataEnabled((__bridge WKPreferencesRef)[webView configuration].preferences, true); [webView synchronouslyLoadTestPageNamed:@"DataTransfer"]; NSString *stringData = @"Hello world"; NSURL *firstURL = [NSURL URLWithString:@"https://webkit.org/"]; NSURL *secondURL = [NSURL URLWithString:@"https://apple.com/"]; #if PLATFORM(MAC) NSPasteboard *pasteboard = [NSPasteboard pasteboardWithUniqueName]; [pasteboard writeObjects:@[ stringData, firstURL, secondURL ]]; [simulator setExternalDragPasteboard:pasteboard]; #else auto stringItem = adoptNS([[NSItemProvider alloc] initWithObject:stringData]); auto firstURLItem = adoptNS([[NSItemProvider alloc] initWithObject:firstURL]); auto secondURLItem = adoptNS([[NSItemProvider alloc] initWithObject:secondURL]); for (NSItemProvider *item in @[ stringItem.get(), firstURLItem.get(), secondURLItem.get() ]) item.preferredPresentationStyle = UIPreferredPresentationStyleInline; [simulator setExternalItemProviders:@[ stringItem.get(), firstURLItem.get(), secondURLItem.get() ]]; #endif [simulator runFrom:CGPointMake(0, 0) to:CGPointMake(100, 100)]; EXPECT_WK_STREQ("text/plain, text/uri-list", [webView stringByEvaluatingJavaScript:@"types.textContent"]); EXPECT_WK_STREQ("(STRING, text/plain), (STRING, text/uri-list)", [webView stringByEvaluatingJavaScript:@"items.textContent"]); EXPECT_WK_STREQ("Hello world", [webView stringByEvaluatingJavaScript:@"textData.textContent"]); EXPECT_WK_STREQ("https://webkit.org/\nhttps://apple.com/", [webView stringByEvaluatingJavaScript:@"urlData.textContent"]); } #if PLATFORM(MAC) TEST(DragAndDropTests, DragAndDropOnEmptyView) { auto simulator = adoptNS([[DragAndDropSimulator alloc] initWithWebViewFrame:CGRectMake(0, 0, 320, 500)]); simulator.get().dragDestinationAction = WKDragDestinationActionAny; auto webView = [simulator webView]; NSURL *url = [[NSBundle mainBundle] URLForResource:@"simple" withExtension:@"html" subdirectory:@"TestWebKitAPI.resources"]; NSPasteboard *pasteboard = [NSPasteboard pasteboardWithUniqueName]; [pasteboard writeObjects:@[ url ]]; [simulator setExternalDragPasteboard:pasteboard]; [simulator runFrom:CGPointMake(0, 0) to:CGPointMake(100, 100)]; __block bool finished = false; [webView performAfterLoading:^{ finished = true; }]; TestWebKitAPI::Util::run(&finished); EXPECT_WK_STREQ("Simple HTML file.", [webView stringByEvaluatingJavaScript:@"document.body.innerText"]); } #endif // PLATFORM(MAC) TEST(DragAndDropTests, PreventingMouseDownShouldPreventDragStart) { auto simulator = adoptNS([[DragAndDropSimulator alloc] initWithWebViewFrame:CGRectMake(0, 0, 320, 500)]); auto webView = [simulator webView]; WKPreferencesSetCustomPasteboardDataEnabled((__bridge WKPreferencesRef)[webView configuration].preferences, true); [webView synchronouslyLoadTestPageNamed:@"link-and-target-div"]; [simulator runFrom:CGPointMake(160, 100) to:CGPointMake(160, 300)]; EXPECT_WK_STREQ("PASS", [webView stringByEvaluatingJavaScript:@"target.textContent"]); EXPECT_WK_STREQ("dragstart dragend", [webView stringByEvaluatingJavaScript:@"output.textContent"]); // Now verify that preventing default on the 'mousedown' event cancels the drag altogether. [webView evaluateJavaScript:@"target.textContent = output.textContent = ''" completionHandler:nil]; [webView evaluateJavaScript:@"source.addEventListener('mousedown', event => { event.preventDefault(); window.observedMousedown = true; })" completionHandler:nil]; [simulator runFrom:CGPointMake(160, 100) to:CGPointMake(160, 300)]; EXPECT_WK_STREQ("", [webView stringByEvaluatingJavaScript:@"target.textContent"]); EXPECT_WK_STREQ("", [webView stringByEvaluatingJavaScript:@"output.textContent"]); EXPECT_TRUE([webView stringByEvaluatingJavaScript:@"observedMousedown"].boolValue); } struct DragStartData { BOOL containsFile { NO }; NSString *text { nil }; NSString *url { nil }; NSString *html { nil }; RetainPtr> customData; }; static DragStartData runDragStartDataTestCase(DragAndDropSimulator *simulator, NSString *elementID) { auto webView = [simulator webView]; NSString *tagName = [webView stringByEvaluatingJavaScript:[NSString stringWithFormat:@"document.getElementById('%@').tagName", elementID]]; if (![tagName isEqualToString:@"IMG"] && ![tagName isEqualToString:@"A"]) [webView selectElementWithID:elementID]; [simulator dragFromElementWithID:elementID to:CGPointMake(400, 400)]; RetainPtr> allData = adoptNS([[webView objectByEvaluatingJavaScript:@"allData"] mutableCopy]); DragStartData result; result.text = [allData objectForKey:@"text/plain"]; result.url = [allData objectForKey:@"text/uri-list"]; result.html = [allData objectForKey:@"text/html"]; result.containsFile = !![allData objectForKey:@"Files"]; [allData removeObjectForKey:@"text/plain"]; [allData removeObjectForKey:@"text/uri-list"]; [allData removeObjectForKey:@"text/html"]; [allData removeObjectForKey:@"Files"]; if ([allData count]) result.customData = WTFMove(allData); return result; } TEST(DragAndDropTests, DataTransferTypesOnDragStartForTextSelection) { auto simulator = adoptNS([[DragAndDropSimulator alloc] initWithWebViewFrame:CGRectMake(0, 0, 500, 500)]); [[simulator webView] synchronouslyLoadTestPageNamed:@"dragstart-data"]; auto result = runDragStartDataTestCase(simulator.get(), @"regular"); EXPECT_WK_STREQ("Regular text", result.text); EXPECT_NULL(result.url); EXPECT_TRUE([result.html containsString:@"Regular text"]); EXPECT_NULL(result.customData); EXPECT_FALSE(result.containsFile); result = runDragStartDataTestCase(simulator.get(), @"regular-url"); EXPECT_WK_STREQ("Regular text + URL data", result.text); EXPECT_WK_STREQ("https://www.google.com", result.url); EXPECT_TRUE([result.html containsString:@"Regular text + URL data"]); EXPECT_NULL(result.customData); EXPECT_FALSE(result.containsFile); result = runDragStartDataTestCase(simulator.get(), @"regular-custom"); EXPECT_WK_STREQ("Regular text + custom data", result.text); EXPECT_NULL(result.url); EXPECT_TRUE([result.html containsString:@"Regular text + custom data"]); EXPECT_WK_STREQ("Hello world", result.customData.get()[@"text/foo"]); EXPECT_FALSE(result.containsFile); result = runDragStartDataTestCase(simulator.get(), @"url"); EXPECT_NULL(result.text); EXPECT_WK_STREQ("https://www.google.com", result.url); EXPECT_NULL([result.html containsString:@"Regular text + custom data"]); EXPECT_NULL(result.customData); EXPECT_FALSE(result.containsFile); result = runDragStartDataTestCase(simulator.get(), @"custom"); EXPECT_NULL(result.text); EXPECT_NULL(result.url); EXPECT_NULL(result.html); EXPECT_WK_STREQ("Hello world", result.customData.get()[@"text/foo"]); EXPECT_FALSE(result.containsFile); } TEST(DragAndDropTests, DataTransferTypesOnDragStartForImage) { auto simulator = adoptNS([[DragAndDropSimulator alloc] initWithWebViewFrame:CGRectMake(0, 0, 500, 500)]); [[simulator webView] synchronouslyLoadTestPageNamed:@"dragstart-data"]; auto result = runDragStartDataTestCase(simulator.get(), @"image"); EXPECT_NULL(result.text); // The URL should be nil here because the image source is a file URL, so we shoud avoid exposing it to the page. EXPECT_NULL(result.url); #if PLATFORM(MAC) EXPECT_TRUE([result.html containsString:@" draggedText; #if USE(APPKIT) draggedText = [[simulator draggingInfo].draggingPasteboard stringForType:(__bridge NSString *)kUTTypeUTF8PlainText]; #else bool doneLoadingString = false; [[simulator sourceItemProviders].firstObject loadObjectOfClass:NSString.class completionHandler:[&](NSString *string, NSError *error) { draggedText = adoptNS([string copy]); doneLoadingString = true; }]; TestWebKitAPI::Util::run(&doneLoadingString); #endif EXPECT_WK_STREQ(draggedText.get(), "foobar"); } #endif // ENABLE(IMAGE_ANALYSIS) #endif // ENABLE(DRAG_SUPPORT) && !PLATFORM(MACCATALYST)