Tools/TestWebKitAPI/Tests/WebKitCocoa/WKWebViewGetContents.mm - external/github.com/WebKit/webkit - Git at Google

 /*
  * Copyright (C) 2017 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 "PlatformUtilities.h"
 #import "Test.h"
 #import "TestNavigationDelegate.h"
 #import "TestWKWebView.h"
 #import <WebKit/WebKit.h>
 #import <wtf/cocoa/TypeCastsCocoa.h>

 TEST(WKWebView, GetContentsShouldReturnString)
 {
     RetainPtr<TestWKWebView> webView = adoptNS([[TestWKWebView alloc] initWithFrame:NSMakeRect(0, 0, 800, 600)]);

     [webView synchronouslyLoadTestPageNamed:@"simple"];

     __block bool finished = false;

     [webView _getContentsAsStringWithCompletionHandler:^(NSString *string, NSError *error) {
         EXPECT_NULL(error);
         EXPECT_WK_STREQ(@"Simple HTML file.", string);
         finished = true;
     }];

     TestWebKitAPI::Util::run(&finished);
 }

 TEST(WKWebView, GetContentsShouldFailWhenClosingPage)
 {
     auto webView = adoptNS([[TestWKWebView alloc] initWithFrame:NSMakeRect(0, 0, 800, 600)]);

     [webView synchronouslyLoadTestPageNamed:@"simple"];

     __block bool finished = false;

     [webView _getContentsAsStringWithCompletionHandlerKeepIPCConnectionAliveForTesting:^(NSString *string, NSError *error) {
         finished = true;
     }];

     [webView _close];

     TestWebKitAPI::Util::run(&finished);
 }

 TEST(WKWebView, GetContentsOfAllFramesShouldReturnString)
 {
     RetainPtr<TestWKWebView> webView = adoptNS([[TestWKWebView alloc] initWithFrame:NSMakeRect(0, 0, 800, 600)]);

     [webView synchronouslyLoadHTMLString:@"<body>beep<iframe srcdoc=\"meep\">herp</iframe><iframe srcdoc=\"moop\">derp</iframe></body>"];

     __block bool finished = false;

     [webView _getContentsOfAllFramesAsStringWithCompletionHandler:^(NSString *string) {
         EXPECT_WK_STREQ(@"beep\n\nmeep\n\nmoop", string);
         finished = true;
     }];

     TestWebKitAPI::Util::run(&finished);
 }

 TEST(WKWebView, GetContentsShouldReturnAttributedString)
 {
     RetainPtr<TestWKWebView> webView = adoptNS([[TestWKWebView alloc] initWithFrame:NSMakeRect(0, 0, 800, 600)]);

     [webView synchronouslyLoadHTMLString:@"<body bgcolor='red'>Hello <b>World!</b>"];

     __block bool finished = false;

     [webView _getContentsAsAttributedStringWithCompletionHandler:^(NSAttributedString *attributedString, NSDictionary<NSAttributedStringDocumentAttributeKey, id> *documentAttributes, NSError *error) {
         EXPECT_NOT_NULL(attributedString);
         EXPECT_NOT_NULL(documentAttributes);
         EXPECT_NULL(error);

         __block size_t i = 0;
         [attributedString enumerateAttributesInRange:NSMakeRange(0, attributedString.length) options:0 usingBlock:^(NSDictionary *attributes, NSRange attributeRange, BOOL *stop) {
             auto* substring = [attributedString attributedSubstringFromRange:attributeRange];

             if (!i) {
                 EXPECT_WK_STREQ(@"Hello ", substring.string);
 #if USE(APPKIT)
                 EXPECT_WK_STREQ(@"Times-Roman", dynamic_objc_cast<NSFont>(attributes[NSFontAttributeName]).fontName);
 #else
                 EXPECT_WK_STREQ(@"TimesNewRomanPSMT", dynamic_objc_cast<UIFont>(attributes[NSFontAttributeName]).fontName);
 #endif
             } else if (i == 1) {
                 EXPECT_WK_STREQ(@"World!", substring.string);
 #if USE(APPKIT)
                 EXPECT_WK_STREQ(@"Times-Bold", dynamic_objc_cast<NSFont>(attributes[NSFontAttributeName]).fontName);
 #else
                 EXPECT_WK_STREQ(@"TimesNewRomanPS-BoldMT", dynamic_objc_cast<UIFont>(attributes[NSFontAttributeName]).fontName);
 #endif
             } else
                 ASSERT_NOT_REACHED();

             ++i;
         }];

 #if USE(APPKIT)
         EXPECT_WK_STREQ(@"sRGB IEC61966-2.1 colorspace 1 0 0 1", dynamic_objc_cast<NSColor>(documentAttributes[NSBackgroundColorDocumentAttribute]).description);
 #else
         EXPECT_WK_STREQ(@"kCGColorSpaceModelRGB 1 0 0 1 ", dynamic_objc_cast<UIColor>(documentAttributes[NSBackgroundColorDocumentAttribute]).description);
 #endif

         finished = true;
     }];

     TestWebKitAPI::Util::run(&finished);
 }

 TEST(WKWebView, GetContentsWithOpticallySizedFontShouldReturnAttributedString)
 {
     RetainPtr<TestWKWebView> webView = adoptNS([[TestWKWebView alloc] initWithFrame:NSMakeRect(0, 0, 800, 600)]);

     [webView synchronouslyLoadHTMLString:@"<body style='font-family: system-ui; font-weight: 100; font-size: 16px; text-rendering: optimizeLegibility'>Hello</body>"];

     __block bool finished = false;

     [webView _getContentsAsAttributedStringWithCompletionHandler:^(NSAttributedString *attributedString, NSDictionary<NSAttributedStringDocumentAttributeKey, id> *documentAttributes, NSError *error) {
         EXPECT_NOT_NULL(attributedString);
         EXPECT_NOT_NULL(documentAttributes);
         EXPECT_NULL(error);

         __block size_t i = 0;
         [attributedString enumerateAttributesInRange:NSMakeRange(0, attributedString.length) options:0 usingBlock:^(NSDictionary *attributes, NSRange attributeRange, BOOL* stop) {
             auto *substring = [attributedString attributedSubstringFromRange:attributeRange];

             if (!i) {
                 EXPECT_WK_STREQ(@"Hello", substring.string);

 #if USE(APPKIT)
                 EXPECT_EQ([dynamic_objc_cast<NSFont>(attributes[NSFontAttributeName]) pointSize], 16);
 #else
                 EXPECT_EQ([dynamic_objc_cast<UIFont>(attributes[NSFontAttributeName]) pointSize], 16);
 #endif
             } else
                 ASSERT_NOT_REACHED();

             ++i;
         }];

         EXPECT_EQ(i, 1UL);

         finished = true;
     }];

     TestWebKitAPI::Util::run(&finished);
 }

 TEST(WKWebView, AttributedStringAccessibilityLabel)
 {
     auto webView = adoptNS([TestWKWebView new]);

     NSString *imagePath = [[NSBundle mainBundle] pathForResource:@"icon" ofType:@"png" inDirectory:@"TestWebKitAPI.resources"];
     [webView synchronouslyLoadHTMLString:[NSString stringWithFormat:@"<html><body><b>Hello</b> <img src='file://%@' width='100' height='100' alt='alt text'> <img src='file://%@' width='100' height='100' alt='aria label text'></body></html>", imagePath, imagePath]];

     __block bool finished = false;

     [webView _getContentsAsAttributedStringWithCompletionHandler:^(NSAttributedString *attributedString, NSDictionary<NSAttributedStringDocumentAttributeKey, id> *documentAttributes, NSError *error) {
         EXPECT_NOT_NULL(attributedString);
         EXPECT_NOT_NULL(documentAttributes);
         EXPECT_NULL(error);

         __block bool foundImage1 { NO };
         __block bool foundImage2 { NO };
         [attributedString enumerateAttribute:NSAttachmentAttributeName inRange:NSMakeRange(0, attributedString.length) options:0 usingBlock:^(id value, NSRange attributeRange, BOOL* stop) {
             if ([value isKindOfClass:NSTextAttachment.class]) {
                 if ([[value accessibilityLabel] isEqualToString:@"alt text"])
                     foundImage1 = YES;
                 if ([[value accessibilityLabel] isEqualToString:@"aria label text"])
                     foundImage2 = YES;
             }
         }];

         EXPECT_TRUE(foundImage1);
         EXPECT_TRUE(foundImage2);

         finished = true;
     }];

     TestWebKitAPI::Util::run(&finished);
 }

 TEST(WKWebView, AttributedStringAttributeTypes)
 {
     NSString *html = @"<html>"
     "<head>"
     "    <meta name='CreationTime' content='2023-12-01T12:23:34Z'/>"
     "    <meta name='Keywords' content='a b c'/>"
     "</head>"
     "<body>"
     "    <p style='text-shadow: 0 1px black'>text shadow paragraph</p>"
     "    <p style='display: inline; unicode-bidi: bidi-override'>bidi paragraph</p>"
     "</body>"
     "</html>";

     auto webView = adoptNS([TestWKWebView new]);
     [webView synchronouslyLoadHTMLString:html];

     __block bool finished { false };

     [webView _getContentsAsAttributedStringWithCompletionHandler:^(NSAttributedString *attributedString, NSDictionary<NSAttributedStringDocumentAttributeKey, id> *documentAttributes, NSError *error) {
         bool foundNSDate = [documentAttributes[@"NSCreationTimeDocumentAttribute"] isKindOfClass:NSDate.class];
         EXPECT_TRUE(foundNSDate);
         bool foundNSArrayOfNSStrings = [documentAttributes[@"NSKeywordsDocumentAttribute"] isEqualToArray:@[@"a", @"b", @"c"]];
         EXPECT_TRUE(foundNSArrayOfNSStrings);

         __block bool foundNSShadow { false };
         __block bool foundNSParagraphStyle { false };
         __block bool foundNSArrayOfNSNumbers { false };
         [attributedString enumerateAttributesInRange:NSMakeRange(0, [attributedString length]) options:0 usingBlock: ^(NSDictionary<NSAttributedStringKey, id> *attributes, NSRange range, BOOL *) {
             if ([attributes[@"NSShadow"] isKindOfClass:NSShadow.class])
                 foundNSShadow = true;
             if ([attributes[@"NSParagraphStyle"] isKindOfClass:NSParagraphStyle.class])
                 foundNSParagraphStyle = true;
             if ([attributes[@"NSWritingDirection"] isKindOfClass:NSArray.class])
                 foundNSArrayOfNSNumbers = [attributes[@"NSWritingDirection"][0] doubleValue] == 2;
         }];
         EXPECT_TRUE(foundNSShadow);
         EXPECT_TRUE(foundNSParagraphStyle);
         EXPECT_TRUE(foundNSArrayOfNSNumbers);
         finished = true;
     }];

     TestWebKitAPI::Util::run(&finished);
 }
	/*
	* Copyright (C) 2017 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 "PlatformUtilities.h"
	#import "Test.h"
	#import "TestNavigationDelegate.h"
	#import "TestWKWebView.h"
	#import <WebKit/WebKit.h>
	#import <wtf/cocoa/TypeCastsCocoa.h>

	TEST(WKWebView, GetContentsShouldReturnString)
	{
	RetainPtr<TestWKWebView> webView = adoptNS([[TestWKWebView alloc] initWithFrame:NSMakeRect(0, 0, 800, 600)]);

	[webView synchronouslyLoadTestPageNamed:@"simple"];

	__block bool finished = false;

	[webView _getContentsAsStringWithCompletionHandler:^(NSString string, NSError error) {
	EXPECT_NULL(error);
	EXPECT_WK_STREQ(@"Simple HTML file.", string);
	finished = true;
	}];

	TestWebKitAPI::Util::run(&finished);
	}

	TEST(WKWebView, GetContentsShouldFailWhenClosingPage)
	{
	auto webView = adoptNS([[TestWKWebView alloc] initWithFrame:NSMakeRect(0, 0, 800, 600)]);

	[webView synchronouslyLoadTestPageNamed:@"simple"];

	__block bool finished = false;

	[webView _getContentsAsStringWithCompletionHandlerKeepIPCConnectionAliveForTesting:^(NSString string, NSError error) {
	finished = true;
	}];

	[webView _close];

	TestWebKitAPI::Util::run(&finished);
	}

	TEST(WKWebView, GetContentsOfAllFramesShouldReturnString)
	{
	RetainPtr<TestWKWebView> webView = adoptNS([[TestWKWebView alloc] initWithFrame:NSMakeRect(0, 0, 800, 600)]);

	[webView synchronouslyLoadHTMLString:@"<body>beep<iframe srcdoc=\"meep\">herp</iframe><iframe srcdoc=\"moop\">derp</iframe></body>"];

	__block bool finished = false;

	[webView _getContentsOfAllFramesAsStringWithCompletionHandler:^(NSString *string) {
	EXPECT_WK_STREQ(@"beep\n\nmeep\n\nmoop", string);
	finished = true;
	}];

	TestWebKitAPI::Util::run(&finished);
	}

	TEST(WKWebView, GetContentsShouldReturnAttributedString)
	{
	RetainPtr<TestWKWebView> webView = adoptNS([[TestWKWebView alloc] initWithFrame:NSMakeRect(0, 0, 800, 600)]);

	[webView synchronouslyLoadHTMLString:@"<body bgcolor='red'>Hello <b>World!</b>"];

	__block bool finished = false;

	[webView _getContentsAsAttributedStringWithCompletionHandler:^(NSAttributedString attributedString, NSDictionary<NSAttributedStringDocumentAttributeKey, id> documentAttributes, NSError *error) {
	EXPECT_NOT_NULL(attributedString);
	EXPECT_NOT_NULL(documentAttributes);
	EXPECT_NULL(error);

	__block size_t i = 0;
	[attributedString enumerateAttributesInRange:NSMakeRange(0, attributedString.length) options:0 usingBlock:^(NSDictionary attributes, NSRange attributeRange, BOOL stop) {
	auto* substring = [attributedString attributedSubstringFromRange:attributeRange];

	if (!i) {
	EXPECT_WK_STREQ(@"Hello ", substring.string);
	#if USE(APPKIT)
	EXPECT_WK_STREQ(@"Times-Roman", dynamic_objc_cast<NSFont>(attributes[NSFontAttributeName]).fontName);
	#else
	EXPECT_WK_STREQ(@"TimesNewRomanPSMT", dynamic_objc_cast<UIFont>(attributes[NSFontAttributeName]).fontName);
	#endif
	} else if (i == 1) {
	EXPECT_WK_STREQ(@"World!", substring.string);
	#if USE(APPKIT)
	EXPECT_WK_STREQ(@"Times-Bold", dynamic_objc_cast<NSFont>(attributes[NSFontAttributeName]).fontName);
	#else
	EXPECT_WK_STREQ(@"TimesNewRomanPS-BoldMT", dynamic_objc_cast<UIFont>(attributes[NSFontAttributeName]).fontName);
	#endif
	} else
	ASSERT_NOT_REACHED();

	++i;
	}];

	#if USE(APPKIT)
	EXPECT_WK_STREQ(@"sRGB IEC61966-2.1 colorspace 1 0 0 1", dynamic_objc_cast<NSColor>(documentAttributes[NSBackgroundColorDocumentAttribute]).description);
	#else
	EXPECT_WK_STREQ(@"kCGColorSpaceModelRGB 1 0 0 1 ", dynamic_objc_cast<UIColor>(documentAttributes[NSBackgroundColorDocumentAttribute]).description);
	#endif

	finished = true;
	}];

	TestWebKitAPI::Util::run(&finished);
	}

	TEST(WKWebView, GetContentsWithOpticallySizedFontShouldReturnAttributedString)
	{
	RetainPtr<TestWKWebView> webView = adoptNS([[TestWKWebView alloc] initWithFrame:NSMakeRect(0, 0, 800, 600)]);

	[webView synchronouslyLoadHTMLString:@"<body style='font-family: system-ui; font-weight: 100; font-size: 16px; text-rendering: optimizeLegibility'>Hello</body>"];

	__block bool finished = false;

	[webView _getContentsAsAttributedStringWithCompletionHandler:^(NSAttributedString attributedString, NSDictionary<NSAttributedStringDocumentAttributeKey, id> documentAttributes, NSError *error) {
	EXPECT_NOT_NULL(attributedString);
	EXPECT_NOT_NULL(documentAttributes);
	EXPECT_NULL(error);

	__block size_t i = 0;
	[attributedString enumerateAttributesInRange:NSMakeRange(0, attributedString.length) options:0 usingBlock:^(NSDictionary attributes, NSRange attributeRange, BOOL stop) {
	auto *substring = [attributedString attributedSubstringFromRange:attributeRange];

	if (!i) {
	EXPECT_WK_STREQ(@"Hello", substring.string);

	#if USE(APPKIT)
	EXPECT_EQ([dynamic_objc_cast<NSFont>(attributes[NSFontAttributeName]) pointSize], 16);
	#else
	EXPECT_EQ([dynamic_objc_cast<UIFont>(attributes[NSFontAttributeName]) pointSize], 16);
	#endif
	} else
	ASSERT_NOT_REACHED();

	++i;
	}];

	EXPECT_EQ(i, 1UL);

	finished = true;
	}];

	TestWebKitAPI::Util::run(&finished);
	}

	TEST(WKWebView, AttributedStringAccessibilityLabel)
	{
	auto webView = adoptNS([TestWKWebView new]);

	NSString *imagePath = [[NSBundle mainBundle] pathForResource:@"icon" ofType:@"png" inDirectory:@"TestWebKitAPI.resources"];
	[webView synchronouslyLoadHTMLString:[NSString stringWithFormat:@"<html><body><b>Hello</b> <img src='file://%@' width='100' height='100' alt='alt text'> <img src='file://%@' width='100' height='100' alt='aria label text'></body></html>", imagePath, imagePath]];

	__block bool finished = false;

	[webView _getContentsAsAttributedStringWithCompletionHandler:^(NSAttributedString attributedString, NSDictionary<NSAttributedStringDocumentAttributeKey, id> documentAttributes, NSError *error) {
	EXPECT_NOT_NULL(attributedString);
	EXPECT_NOT_NULL(documentAttributes);
	EXPECT_NULL(error);

	__block bool foundImage1 { NO };
	__block bool foundImage2 { NO };
	[attributedString enumerateAttribute:NSAttachmentAttributeName inRange:NSMakeRange(0, attributedString.length) options:0 usingBlock:^(id value, NSRange attributeRange, BOOL* stop) {
	if ([value isKindOfClass:NSTextAttachment.class]) {
	if ([[value accessibilityLabel] isEqualToString:@"alt text"])
	foundImage1 = YES;
	if ([[value accessibilityLabel] isEqualToString:@"aria label text"])
	foundImage2 = YES;
	}
	}];

	EXPECT_TRUE(foundImage1);
	EXPECT_TRUE(foundImage2);

	finished = true;
	}];

	TestWebKitAPI::Util::run(&finished);
	}

	TEST(WKWebView, AttributedStringAttributeTypes)
	{
	NSString *html = @"<html>"
	"<head>"
	" <meta name='CreationTime' content='2023-12-01T12:23:34Z'/>"
	" <meta name='Keywords' content='a b c'/>"
	"</head>"
	"<body>"
	" <p style='text-shadow: 0 1px black'>text shadow paragraph</p>"
	" <p style='display: inline; unicode-bidi: bidi-override'>bidi paragraph</p>"
	"</body>"
	"</html>";

	auto webView = adoptNS([TestWKWebView new]);
	[webView synchronouslyLoadHTMLString:html];

	__block bool finished { false };

	[webView _getContentsAsAttributedStringWithCompletionHandler:^(NSAttributedString attributedString, NSDictionary<NSAttributedStringDocumentAttributeKey, id> documentAttributes, NSError *error) {
	bool foundNSDate = [documentAttributes[@"NSCreationTimeDocumentAttribute"] isKindOfClass:NSDate.class];
	EXPECT_TRUE(foundNSDate);
	bool foundNSArrayOfNSStrings = [documentAttributes[@"NSKeywordsDocumentAttribute"] isEqualToArray:@[@"a", @"b", @"c"]];
	EXPECT_TRUE(foundNSArrayOfNSStrings);

	__block bool foundNSShadow { false };
	__block bool foundNSParagraphStyle { false };
	__block bool foundNSArrayOfNSNumbers { false };
	[attributedString enumerateAttributesInRange:NSMakeRange(0, [attributedString length]) options:0 usingBlock: ^(NSDictionary<NSAttributedStringKey, id> attributes, NSRange range, BOOL ) {
	if ([attributes[@"NSShadow"] isKindOfClass:NSShadow.class])
	foundNSShadow = true;
	if ([attributes[@"NSParagraphStyle"] isKindOfClass:NSParagraphStyle.class])
	foundNSParagraphStyle = true;
	if ([attributes[@"NSWritingDirection"] isKindOfClass:NSArray.class])
	foundNSArrayOfNSNumbers = [attributes[@"NSWritingDirection"][0] doubleValue] == 2;
	}];
	EXPECT_TRUE(foundNSShadow);
	EXPECT_TRUE(foundNSParagraphStyle);
	EXPECT_TRUE(foundNSArrayOfNSNumbers);
	finished = true;
	}];

	TestWebKitAPI::Util::run(&finished);
	}