Tools/TestWebKitAPI/Tests/ios/SelectionByWord.mm - WebKit - Git at Google

 /*
  * Copyright (C) 2019 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"

 #if PLATFORM(IOS_FAMILY)

 #import "PlatformUtilities.h"
 #import "TestInputDelegate.h"
 #import "TestWKWebView.h"
 #import "UIKitSPI.h"
 #import <wtf/BlockPtr.h>

 TEST(SelectionTests, ByWordAtEndOfDocument)
 {
     auto webView = adoptNS([[TestWKWebView alloc] init]);

     [webView synchronouslyLoadHTMLString:@"<body><p>Paragraph One</p><p style='-webkit-user-select: none'>Paragraph Two</p><p>Paragraph Three</p><p></p></body>"];

     [webView stringByEvaluatingJavaScript:@"getSelection().setPosition(document.body.lastChild, 0)"];

     EXPECT_TRUE([webView stringByEvaluatingJavaScript:@"getSelection().isCollapsed"].boolValue);

     // This method triggers the wordRangeFromPosition() code to be tested.
     [[webView textInputContentView] selectWordForReplacement];

     EXPECT_FALSE([webView stringByEvaluatingJavaScript:@"getSelection().isCollapsed"].boolValue);
     EXPECT_WK_STREQ([webView stringByEvaluatingJavaScript:@"getSelection().anchorNode.nodeValue"], "Paragraph Three");
     EXPECT_WK_STREQ([webView stringByEvaluatingJavaScript:@"getSelection().anchorOffset"], "10");
     EXPECT_WK_STREQ([webView stringByEvaluatingJavaScript:@"getSelection().focusNode.nodeValue"], "Paragraph Three");
     EXPECT_WK_STREQ([webView stringByEvaluatingJavaScript:@"getSelection().focusOffset"], "15");

     EXPECT_WK_STREQ([webView stringByEvaluatingJavaScript:@"getSelection().toString()"], "Three");
 }

 TEST(SelectionTests, SelectWordForReplacementWithDictationAlternative)
 {
     auto webView = adoptNS([[TestWKWebView alloc] initWithFrame:CGRectMake(0, 0, 320, 320)]);
     [webView synchronouslyLoadTestPageNamed:@"editable-responsive-body"];

     auto contentView = [webView textInputContentView];
     [contentView selectAll:nil];
     [contentView insertText:@"foo bar"];
     [webView waitForNextPresentationUpdate];

     auto alternatives = adoptNS([[NSTextAlternatives alloc] initWithPrimaryString:@"foo bar" alternativeStrings:@[ @"baz" ]]);
     [[webView textInputContentView] addTextAlternatives:alternatives.get()];
     [[webView textInputContentView] selectWordForReplacement];
     [webView waitForNextPresentationUpdate];

     EXPECT_WK_STREQ("foo bar", [webView selectedText]);
 }

 @interface SelectionChangeListener : NSObject <UITextInputDelegate>
 @property (nonatomic) dispatch_block_t selectionWillChangeHandler;
 @property (nonatomic) dispatch_block_t selectionDidChangeHandler;
 @end

 @implementation SelectionChangeListener {
     BlockPtr<void()> _selectionWillChangeHandler;
     BlockPtr<void()> _selectionDidChangeHandler;
 }

 - (void)setSelectionWillChangeHandler:(dispatch_block_t)handler
 {
     _selectionWillChangeHandler = makeBlockPtr(handler);
 }

 - (void)setSelectionDidChangeHandler:(dispatch_block_t)handler
 {
     _selectionDidChangeHandler = makeBlockPtr(handler);
 }

 - (dispatch_block_t)selectionWillChangeHandler
 {
     return _selectionWillChangeHandler.get();
 }

 - (dispatch_block_t)selectionDidChangeHandler
 {
     return _selectionDidChangeHandler.get();
 }

 - (void)selectionWillChange:(id <UITextInput>)textInput
 {
     if (_selectionWillChangeHandler)
         _selectionWillChangeHandler();
 }

 - (void)selectionDidChange:(id <UITextInput>)textInput
 {
     if (_selectionDidChangeHandler)
         _selectionDidChangeHandler();
 }

 - (void)textWillChange:(id <UITextInput>)textInput
 {
 }

 - (void)textDidChange:(id <UITextInput>)textInput
 {
 }

 @end

 TEST(SelectionTests, SelectedTextAfterSelectingWordForReplacement)
 {
     auto listener = adoptNS([[SelectionChangeListener alloc] init]);
     auto webView = adoptNS([[TestWKWebView alloc] init]);
     [webView synchronouslyLoadHTMLString:@"<body contenteditable><p>Hello</p></body><script>document.body.focus()</script>"];

     auto contentView = [webView textInputContentView];
     [contentView setInputDelegate:listener.get()];

     __block bool selectionWillChange = false;
     [listener setSelectionWillChangeHandler:^{
         selectionWillChange = true;
     }];

     __block bool selectionDidChange = false;
     [listener setSelectionDidChangeHandler:^{
         selectionDidChange = true;
     }];

     [contentView selectWordForReplacement];
     [webView waitForNextPresentationUpdate];

     TestWebKitAPI::Util::run(&selectionWillChange);
     TestWebKitAPI::Util::run(&selectionDidChange);
     EXPECT_WK_STREQ("Hello", [contentView textInRange:contentView.selectedTextRange]);
 }

 #endif // PLATFORM(IOS_FAMILY)
	/*
	* Copyright (C) 2019 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"

	#if PLATFORM(IOS_FAMILY)

	#import "PlatformUtilities.h"
	#import "TestInputDelegate.h"
	#import "TestWKWebView.h"
	#import "UIKitSPI.h"
	#import <wtf/BlockPtr.h>

	TEST(SelectionTests, ByWordAtEndOfDocument)
	{
	auto webView = adoptNS([[TestWKWebView alloc] init]);

	[webView synchronouslyLoadHTMLString:@"<body><p>Paragraph One</p><p style='-webkit-user-select: none'>Paragraph Two</p><p>Paragraph Three</p><p></p></body>"];

	[webView stringByEvaluatingJavaScript:@"getSelection().setPosition(document.body.lastChild, 0)"];

	EXPECT_TRUE([webView stringByEvaluatingJavaScript:@"getSelection().isCollapsed"].boolValue);

	// This method triggers the wordRangeFromPosition() code to be tested.
	[[webView textInputContentView] selectWordForReplacement];

	EXPECT_FALSE([webView stringByEvaluatingJavaScript:@"getSelection().isCollapsed"].boolValue);
	EXPECT_WK_STREQ([webView stringByEvaluatingJavaScript:@"getSelection().anchorNode.nodeValue"], "Paragraph Three");
	EXPECT_WK_STREQ([webView stringByEvaluatingJavaScript:@"getSelection().anchorOffset"], "10");
	EXPECT_WK_STREQ([webView stringByEvaluatingJavaScript:@"getSelection().focusNode.nodeValue"], "Paragraph Three");
	EXPECT_WK_STREQ([webView stringByEvaluatingJavaScript:@"getSelection().focusOffset"], "15");

	EXPECT_WK_STREQ([webView stringByEvaluatingJavaScript:@"getSelection().toString()"], "Three");
	}

	TEST(SelectionTests, SelectWordForReplacementWithDictationAlternative)
	{
	auto webView = adoptNS([[TestWKWebView alloc] initWithFrame:CGRectMake(0, 0, 320, 320)]);
	[webView synchronouslyLoadTestPageNamed:@"editable-responsive-body"];

	auto contentView = [webView textInputContentView];
	[contentView selectAll:nil];
	[contentView insertText:@"foo bar"];
	[webView waitForNextPresentationUpdate];

	auto alternatives = adoptNS([[NSTextAlternatives alloc] initWithPrimaryString:@"foo bar" alternativeStrings:@[ @"baz" ]]);
	[[webView textInputContentView] addTextAlternatives:alternatives.get()];
	[[webView textInputContentView] selectWordForReplacement];
	[webView waitForNextPresentationUpdate];

	EXPECT_WK_STREQ("foo bar", [webView selectedText]);
	}

	@interface SelectionChangeListener : NSObject <UITextInputDelegate>
	@property (nonatomic) dispatch_block_t selectionWillChangeHandler;
	@property (nonatomic) dispatch_block_t selectionDidChangeHandler;
	@end

	@implementation SelectionChangeListener {
	BlockPtr<void()> _selectionWillChangeHandler;
	BlockPtr<void()> _selectionDidChangeHandler;
	}

	- (void)setSelectionWillChangeHandler:(dispatch_block_t)handler
	{
	_selectionWillChangeHandler = makeBlockPtr(handler);
	}

	- (void)setSelectionDidChangeHandler:(dispatch_block_t)handler
	{
	_selectionDidChangeHandler = makeBlockPtr(handler);
	}

	- (dispatch_block_t)selectionWillChangeHandler
	{
	return _selectionWillChangeHandler.get();
	}

	- (dispatch_block_t)selectionDidChangeHandler
	{
	return _selectionDidChangeHandler.get();
	}

	- (void)selectionWillChange:(id <UITextInput>)textInput
	{
	if (_selectionWillChangeHandler)
	_selectionWillChangeHandler();
	}

	- (void)selectionDidChange:(id <UITextInput>)textInput
	{
	if (_selectionDidChangeHandler)
	_selectionDidChangeHandler();
	}

	- (void)textWillChange:(id <UITextInput>)textInput
	{
	}

	- (void)textDidChange:(id <UITextInput>)textInput
	{
	}

	@end

	TEST(SelectionTests, SelectedTextAfterSelectingWordForReplacement)
	{
	auto listener = adoptNS([[SelectionChangeListener alloc] init]);
	auto webView = adoptNS([[TestWKWebView alloc] init]);
	[webView synchronouslyLoadHTMLString:@"<body contenteditable><p>Hello</p></body><script>document.body.focus()</script>"];

	auto contentView = [webView textInputContentView];
	[contentView setInputDelegate:listener.get()];

	__block bool selectionWillChange = false;
	[listener setSelectionWillChangeHandler:^{
	selectionWillChange = true;
	}];

	__block bool selectionDidChange = false;
	[listener setSelectionDidChangeHandler:^{
	selectionDidChange = true;
	}];

	[contentView selectWordForReplacement];
	[webView waitForNextPresentationUpdate];

	TestWebKitAPI::Util::run(&selectionWillChange);
	TestWebKitAPI::Util::run(&selectionDidChange);
	EXPECT_WK_STREQ("Hello", [contentView textInRange:contentView.selectedTextRange]);
	}

	#endif // PLATFORM(IOS_FAMILY)