Google Git
Sign in
webkit / WebKit / 15e4da4b5ce83fcfe6de6863df44c60571019d5f / . / Tools / TestWebKitAPI / Tests / WebKitCocoa / WKWebViewSnapshot.mm
blob: e95041b61e94a31af58df52b379c0014faf668a9 [file] [log] [blame]
/*
* Copyright (C) 2016 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 <WebKit/WKSnapshotConfiguration.h>
#import <wtf/BlockPtr.h>
#import <wtf/RetainPtr.h>
static bool isDone;
#if PLATFORM(MAC)
typedef NSImage *PlatformImage;
typedef NSWindow *PlatformWindow;
static RetainPtr<CGImageRef> convertToCGImage(NSImage *image)
{
return [image CGImageForProposedRect:nil context:nil hints:nil];
}
#else
typedef UIImage *PlatformImage;
typedef UIWindow *PlatformWindow;
static RetainPtr<CGImageRef> convertToCGImage(UIImage *image)
{
return image.CGImage;
}
#endif
static NSInteger getPixelIndex(NSInteger x, NSInteger y, NSInteger width)
{
return (y * width + x) * 4;
}
@interface TestSnapshotWrapper : NSObject
@property (nonatomic, readonly) RetainPtr<CGImageRef> image;
@property (nonatomic, readonly) RetainPtr<NSError> error;
@end
@implementation TestSnapshotWrapper
- (void)takeSnapshotWithWebView:(WKWebView *)webView configuration:(WKSnapshotConfiguration *)configuration completionHandler:(void (^)(void))completionHandler
{
auto completionBlock = makeBlockPtr(completionHandler);
[webView takeSnapshotWithConfiguration:configuration completionHandler:^(PlatformImage snapshotImage, NSError *error) {
_image = convertToCGImage(snapshotImage);
_error = error;
completionBlock();
}];
}
@end
TEST(WKWebView, SnapshotImageError)
{
CGFloat viewWidth = 800;
CGFloat viewHeight = 600;
RetainPtr<WKWebView> webView = adoptNS([[WKWebView alloc] initWithFrame:NSMakeRect(0, 0, viewWidth, viewHeight)]);
[webView loadHTMLString:@"<body style='background-color: red;'></body>" baseURL:nil];
[webView _test_waitForDidFinishNavigation];
[webView _killWebContentProcessAndResetState];
RetainPtr<WKSnapshotConfiguration> snapshotConfiguration = adoptNS([[WKSnapshotConfiguration alloc] init]);
[snapshotConfiguration setRect:NSMakeRect(0, 0, viewWidth, viewHeight)];
[snapshotConfiguration setSnapshotWidth:@(viewWidth)];
isDone = false;
[webView takeSnapshotWithConfiguration:snapshotConfiguration.get() completionHandler:^(PlatformImage snapshotImage, NSError *error) {
EXPECT_NULL(snapshotImage);
EXPECT_WK_STREQ(@"WKErrorDomain", error.domain);
isDone = true;
}];
TestWebKitAPI::Util::run(&isDone);
}
TEST(WKWebView, SnapshotImageEmptyRect)
{
CGFloat viewWidth = 800;
CGFloat viewHeight = 600;
RetainPtr<WKWebView> webView = adoptNS([[WKWebView alloc] initWithFrame:NSMakeRect(0, 0, viewWidth, viewHeight)]);
[webView loadHTMLString:@"<body style='background-color: red;'></body>" baseURL:nil];
[webView _test_waitForDidFinishNavigation];
auto pid = [webView _webProcessIdentifier];
RetainPtr<WKSnapshotConfiguration> snapshotConfiguration = adoptNS([[WKSnapshotConfiguration alloc] init]);
[snapshotConfiguration setRect:NSMakeRect(0, 0, 0, 0)];
[snapshotConfiguration setSnapshotWidth:@(viewWidth)];
isDone = false;
[webView takeSnapshotWithConfiguration:snapshotConfiguration.get() completionHandler:^(PlatformImage snapshotImage, NSError *error) {
EXPECT_NULL(error);
EXPECT_EQ(0, snapshotImage.size.width);
EXPECT_EQ(0, snapshotImage.size.height);
isDone = true;
}];
TestWebKitAPI::Util::run(&isDone);
EXPECT_EQ(pid, [webView _webProcessIdentifier]); // Make sure the WebProcess did not crash.
}
TEST(WKWebView, SnapshotImageZeroWidth)
{
CGFloat viewWidth = 800;
CGFloat viewHeight = 600;
RetainPtr<WKWebView> webView = adoptNS([[WKWebView alloc] initWithFrame:NSMakeRect(0, 0, viewWidth, viewHeight)]);
[webView loadHTMLString:@"<body style='background-color: red;'></body>" baseURL:nil];
[webView _test_waitForDidFinishNavigation];
auto pid = [webView _webProcessIdentifier];
RetainPtr<WKSnapshotConfiguration> snapshotConfiguration = adoptNS([[WKSnapshotConfiguration alloc] init]);
[snapshotConfiguration setRect:NSMakeRect(0, 0, viewWidth, viewHeight)];
[snapshotConfiguration setSnapshotWidth:@(0.0)]; // Will fall back to the view width.
isDone = false;
[webView takeSnapshotWithConfiguration:snapshotConfiguration.get() completionHandler:^(PlatformImage snapshotImage, NSError *error) {
EXPECT_NULL(error);
EXPECT_EQ(viewWidth, snapshotImage.size.width);
isDone = true;
}];
TestWebKitAPI::Util::run(&isDone);
EXPECT_EQ(pid, [webView _webProcessIdentifier]); // Make sure the WebProcess did not crash.
}
TEST(WKWebView, SnapshotImageZeroSizeView)
{
CGFloat viewWidth = 0;
CGFloat viewHeight = 0;
RetainPtr<WKWebView> webView = adoptNS([[WKWebView alloc] initWithFrame:NSMakeRect(0, 0, viewWidth, viewHeight)]);
[webView loadHTMLString:@"<body style='background-color: red;'></body>" baseURL:nil];
[webView _test_waitForDidFinishNavigation];
auto pid = [webView _webProcessIdentifier];
RetainPtr<WKSnapshotConfiguration> snapshotConfiguration = adoptNS([[WKSnapshotConfiguration alloc] init]);
[snapshotConfiguration setRect:NSMakeRect(0, 0, viewWidth, viewHeight)];
[snapshotConfiguration setSnapshotWidth:@(100)];
isDone = false;
[webView takeSnapshotWithConfiguration:snapshotConfiguration.get() completionHandler:^(PlatformImage snapshotImage, NSError *error) {
EXPECT_NULL(error);
EXPECT_EQ(0, snapshotImage.size.width);
EXPECT_EQ(0, snapshotImage.size.height);
isDone = true;
}];
TestWebKitAPI::Util::run(&isDone);
EXPECT_EQ(pid, [webView _webProcessIdentifier]); // Make sure the WebProcess did not crash.
}
TEST(WKWebView, SnapshotImageZeroSizeViewNoConfiguration)
{
CGFloat viewWidth = 0;
CGFloat viewHeight = 0;
RetainPtr<WKWebView> webView = adoptNS([[WKWebView alloc] initWithFrame:NSMakeRect(0, 0, viewWidth, viewHeight)]);
[webView loadHTMLString:@"<body style='background-color: red;'></body>" baseURL:nil];
[webView _test_waitForDidFinishNavigation];
auto pid = [webView _webProcessIdentifier];
isDone = false;
[webView takeSnapshotWithConfiguration:nil completionHandler:^(PlatformImage snapshotImage, NSError *error) {
EXPECT_NULL(error);
EXPECT_EQ(0, snapshotImage.size.width);
EXPECT_EQ(0, snapshotImage.size.height);
isDone = true;
}];
TestWebKitAPI::Util::run(&isDone);
EXPECT_EQ(pid, [webView _webProcessIdentifier]); // Make sure the WebProcess did not crash.
}
TEST(WKWebView, SnapshotImageEmptyWithOutOfScopeCompletionHandler)
{
CGFloat viewWidth = 0;
CGFloat viewHeight = 0;
RetainPtr<WKWebView> webView = adoptNS([[WKWebView alloc] initWithFrame:NSMakeRect(0, 0, viewWidth, viewHeight)]);
[webView loadHTMLString:@"<body style='background-color: red;'></body>" baseURL:nil];
[webView _test_waitForDidFinishNavigation];
auto pid = [webView _webProcessIdentifier];
RetainPtr<WKSnapshotConfiguration> snapshotConfiguration = adoptNS([[WKSnapshotConfiguration alloc] init]);
[snapshotConfiguration setRect:NSMakeRect(0, 0, 0, 0)];
[snapshotConfiguration setSnapshotWidth:@(viewWidth)];
isDone = false;
auto snapshotWrapper = adoptNS([[TestSnapshotWrapper alloc] init]);
[snapshotWrapper takeSnapshotWithWebView:webView.get() configuration:snapshotConfiguration.get() completionHandler:^{
isDone = true;
}];
TestWebKitAPI::Util::run(&isDone);
EXPECT_NULL([snapshotWrapper error]);
auto image = [snapshotWrapper image].get();
EXPECT_EQ(0UL, CGImageGetWidth(image));
EXPECT_EQ(0UL, CGImageGetHeight(image));
EXPECT_EQ(pid, [webView _webProcessIdentifier]); // Make sure the WebProcess did not crash.
}
TEST(WKWebView, SnapshotImageBaseCase)
{
NSInteger viewWidth = 800;
NSInteger viewHeight = 600;
RetainPtr<WKWebView> webView = adoptNS([[WKWebView alloc] initWithFrame:NSMakeRect(0, 0, viewWidth, viewHeight)]);
RetainPtr<PlatformWindow> window;
CGFloat backingScaleFactor;
#if PLATFORM(MAC)
window = adoptNS([[NSWindow alloc] initWithContentRect:[webView frame] styleMask:NSBorderlessWindowMask backing:NSBackingStoreBuffered defer:NO]);
[[window contentView] addSubview:webView.get()];
backingScaleFactor = [window backingScaleFactor];
#elif PLATFORM(IOS_FAMILY)
window = adoptNS([[UIWindow alloc] initWithFrame:[webView frame]]);
[window addSubview:webView.get()];
backingScaleFactor = [[window screen] scale];
#endif
[webView loadHTMLString:@"<body style='background-color:red;'><div style='background-color:blue; position:absolute; width:100px; height:100px; top:50px; left:50px'></div></body>" baseURL:nil];
[webView _test_waitForDidFinishNavigation];
RetainPtr<WKSnapshotConfiguration> snapshotConfiguration = adoptNS([[WKSnapshotConfiguration alloc] init]);
[snapshotConfiguration setRect:NSMakeRect(0, 0, viewWidth, viewHeight)];
[snapshotConfiguration setSnapshotWidth:@(viewWidth)];
isDone = false;
[webView takeSnapshotWithConfiguration:snapshotConfiguration.get() completionHandler:^(PlatformImage snapshotImage, NSError *error) {
EXPECT_NULL(error);
EXPECT_EQ(viewWidth, snapshotImage.size.width);
RetainPtr<CGImageRef> cgImage = convertToCGImage(snapshotImage);
RetainPtr<CGColorSpaceRef> colorSpace = adoptCF(CGColorSpaceCreateDeviceRGB());
NSInteger viewWidthInPixels = viewWidth * backingScaleFactor;
NSInteger viewHeightInPixels = viewHeight * backingScaleFactor;
uint8_t *rgba = (unsigned char *)calloc(viewWidthInPixels * viewHeightInPixels * 4, sizeof(unsigned char));
auto context = adoptCF(CGBitmapContextCreate(rgba, viewWidthInPixels, viewHeightInPixels, 8, 4 * viewWidthInPixels, colorSpace.get(), kCGImageAlphaPremultipliedLast | kCGBitmapByteOrder32Big));
CGContextDrawImage(context.get(), CGRectMake(0, 0, viewWidthInPixels, viewHeightInPixels), cgImage.get());
NSInteger pixelIndex = getPixelIndex(0, 0, viewWidthInPixels);
EXPECT_EQ(255, rgba[pixelIndex]);
EXPECT_EQ(0, rgba[pixelIndex + 1]);
EXPECT_EQ(0, rgba[pixelIndex + 2]);
// Inside the blue div (50, 50, 100, 100)
pixelIndex = getPixelIndex(55 * backingScaleFactor, 55 * backingScaleFactor, viewWidthInPixels);
EXPECT_EQ(0, rgba[pixelIndex]);
EXPECT_EQ(0, rgba[pixelIndex + 1]);
EXPECT_EQ(255, rgba[pixelIndex + 2]);
pixelIndex = getPixelIndex(155 * backingScaleFactor, 155 * backingScaleFactor, viewWidthInPixels);
EXPECT_EQ(255, rgba[pixelIndex]);
EXPECT_EQ(0, rgba[pixelIndex + 1]);
EXPECT_EQ(0, rgba[pixelIndex + 2]);
free(rgba);
isDone = true;
}];
TestWebKitAPI::Util::run(&isDone);
}
TEST(WKWebView, SnapshotImageScale)
{
CGFloat viewWidth = 800;
CGFloat viewHeight = 600;
CGFloat scaleFactor = 2;
RetainPtr<WKWebView> webView = adoptNS([[WKWebView alloc] initWithFrame:NSMakeRect(0, 0, viewWidth, viewHeight)]);
[webView loadHTMLString:@"<body style='background-color: red;'></body>" baseURL:nil];
[webView _test_waitForDidFinishNavigation];
RetainPtr<WKSnapshotConfiguration> snapshotConfiguration = adoptNS([[WKSnapshotConfiguration alloc] init]);
[snapshotConfiguration setRect:NSMakeRect(0, 0, viewWidth, viewHeight)];
[snapshotConfiguration setSnapshotWidth:@(viewWidth * scaleFactor)];
isDone = false;
[webView takeSnapshotWithConfiguration:snapshotConfiguration.get() completionHandler:^(PlatformImage snapshotImage, NSError *error) {
EXPECT_NULL(error);
EXPECT_NOT_NULL(snapshotImage);
EXPECT_EQ(viewWidth * scaleFactor, snapshotImage.size.width);
EXPECT_EQ(viewHeight * scaleFactor, snapshotImage.size.height);
isDone = true;
}];
TestWebKitAPI::Util::run(&isDone);
}
TEST(WKWebView, SnapshotImageNilConfiguration)
{
CGFloat viewWidth = 800;
CGFloat viewHeight = 600;
RetainPtr<WKWebView> webView = adoptNS([[WKWebView alloc] initWithFrame:NSMakeRect(0, 0, viewWidth, viewHeight)]);
[webView loadHTMLString:@"<body style='background-color: red;'></body>" baseURL:nil];
[webView _test_waitForDidFinishNavigation];
isDone = false;
[webView takeSnapshotWithConfiguration:nil completionHandler:^(PlatformImage snapshotImage, NSError *error) {
EXPECT_NULL(error);
EXPECT_NOT_NULL(snapshotImage);
EXPECT_EQ([webView bounds].size.width, snapshotImage.size.width);
isDone = true;
}];
TestWebKitAPI::Util::run(&isDone);
}
TEST(WKWebView, SnapshotImageUninitializedConfiguration)
{
CGFloat viewWidth = 800;
CGFloat viewHeight = 600;
RetainPtr<WKWebView> webView = adoptNS([[WKWebView alloc] initWithFrame:NSMakeRect(0, 0, viewWidth, viewHeight)]);
[webView loadHTMLString:@"<body style='background-color: red;'></body>" baseURL:nil];
[webView _test_waitForDidFinishNavigation];
RetainPtr<WKSnapshotConfiguration> snapshotConfiguration = adoptNS([[WKSnapshotConfiguration alloc] init]);
isDone = false;
[webView takeSnapshotWithConfiguration:snapshotConfiguration.get() completionHandler:^(PlatformImage snapshotImage, NSError *error) {
EXPECT_NULL(error);
EXPECT_NOT_NULL(snapshotImage);
EXPECT_EQ([webView bounds].size.width, snapshotImage.size.width);
isDone = true;
}];
TestWebKitAPI::Util::run(&isDone);
}
TEST(WKWebView, SnapshotImageUninitializedSnapshotWidth)
{
CGFloat viewWidth = 800;
CGFloat viewHeight = 600;
RetainPtr<WKWebView> webView = adoptNS([[WKWebView alloc] initWithFrame:NSMakeRect(0, 0, viewWidth, viewHeight)]);
[webView loadHTMLString:@"<body style='background-color: red;'></body>" baseURL:nil];
[webView _test_waitForDidFinishNavigation];
RetainPtr<WKSnapshotConfiguration> snapshotConfiguration = adoptNS([[WKSnapshotConfiguration alloc] init]);
[snapshotConfiguration setRect:NSMakeRect(0, 0, viewWidth, viewHeight)];
isDone = false;
[webView takeSnapshotWithConfiguration:snapshotConfiguration.get() completionHandler:^(PlatformImage snapshotImage, NSError *error) {
EXPECT_NULL(error);
EXPECT_NOT_NULL(snapshotImage);
EXPECT_EQ([snapshotConfiguration rect].size.width, snapshotImage.size.width);
isDone = true;
}];
TestWebKitAPI::Util::run(&isDone);
}
TEST(WKWebView, SnapshotImageLargeAsyncDecoding)
{
NSInteger viewWidth = 800;
NSInteger viewHeight = 600;
RetainPtr<WKWebView> webView = adoptNS([[WKWebView alloc] initWithFrame:NSMakeRect(0, 0, viewWidth, viewHeight)]);
NSURL *fileURL = [[NSBundle mainBundle] URLForResource:@"large-red-square-image" withExtension:@"html" subdirectory:@"TestWebKitAPI.resources"];
[webView loadFileURL:fileURL allowingReadAccessToURL:fileURL];
[webView _test_waitForDidFinishNavigation];
RetainPtr<WKSnapshotConfiguration> snapshotConfiguration = adoptNS([[WKSnapshotConfiguration alloc] init]);
[snapshotConfiguration setRect:NSMakeRect(0, 0, viewWidth, viewHeight)];
[snapshotConfiguration setSnapshotWidth:@(viewWidth)];
isDone = false;
[webView takeSnapshotWithConfiguration:snapshotConfiguration.get() completionHandler:^(PlatformImage snapshotImage, NSError *error) {
EXPECT_NULL(error);
EXPECT_EQ(viewWidth, snapshotImage.size.width);
RetainPtr<CGImageRef> cgImage = convertToCGImage(snapshotImage);
RetainPtr<CGColorSpaceRef> colorSpace = adoptCF(CGColorSpaceCreateDeviceRGB());
uint8_t *rgba = (unsigned char *)calloc(viewWidth * viewHeight * 4, sizeof(unsigned char));
auto context = adoptCF(CGBitmapContextCreate(rgba, viewWidth, viewHeight, 8, 4 * viewWidth, colorSpace.get(), kCGImageAlphaPremultipliedLast | kCGBitmapByteOrder32Big));
CGContextDrawImage(context.get(), CGRectMake(0, 0, viewWidth, viewHeight), cgImage.get());
// Top-left corner of the div (0, 0, 100, 100)
NSInteger pixelIndex = getPixelIndex(0, 0, viewWidth);
EXPECT_EQ(255, rgba[pixelIndex]);
EXPECT_EQ(0, rgba[pixelIndex + 1]);
EXPECT_EQ(0, rgba[pixelIndex + 2]);
// Right-bottom corner of the div (0, 0, 100, 100)
pixelIndex = getPixelIndex(99, 99, viewWidth);
EXPECT_EQ(255, rgba[pixelIndex]);
EXPECT_EQ(0, rgba[pixelIndex + 1]);
EXPECT_EQ(0, rgba[pixelIndex + 2]);
// Outside the div (0, 0, 100, 100)
pixelIndex = getPixelIndex(100, 100, viewWidth);
EXPECT_EQ(255, rgba[pixelIndex]);
EXPECT_EQ(255, rgba[pixelIndex + 1]);
EXPECT_EQ(255, rgba[pixelIndex + 2]);
free(rgba);
isDone = true;
}];
TestWebKitAPI::Util::run(&isDone);
}
TEST(WKWebView, SnapshotAfterScreenUpdates)
{
// The API tests currently cannot truly test SnapshotConfiguration.afterScreenUpdates since it is only needed
// on iOS devices, and we do not currently run API tests on iOS devices. So we expect this test to pass with
// afterScreenUpdates set to YES or NO on the configuration. On device, afterScreenUpdates must be YES in order
// pass this test.
NSInteger viewWidth = 800;
NSInteger viewHeight = 600;
RetainPtr<WKWebView> webView = adoptNS([[WKWebView alloc] initWithFrame:NSMakeRect(0, 0, viewWidth, viewHeight)]);
RetainPtr<PlatformWindow> window;
CGFloat backingScaleFactor;
#if PLATFORM(MAC)
window = adoptNS([[NSWindow alloc] initWithContentRect:[webView frame] styleMask:NSBorderlessWindowMask backing:NSBackingStoreBuffered defer:NO]);
[[window contentView] addSubview:webView.get()];
backingScaleFactor = [window backingScaleFactor];
#elif PLATFORM(IOS_FAMILY)
window = adoptNS([[UIWindow alloc] initWithFrame:[webView frame]]);
[window addSubview:webView.get()];
backingScaleFactor = [[window screen] scale];
#endif
[webView loadHTMLString:@"<body style='margin:0'><div id='change-me' style='background-color:red; position:fixed; width:100%; height:100%'></div></body>" baseURL:nil];
[webView _test_waitForDidFinishNavigation];
RetainPtr<WKSnapshotConfiguration> snapshotConfiguration = adoptNS([[WKSnapshotConfiguration alloc] init]);
[snapshotConfiguration setRect:NSMakeRect(0, 0, viewWidth, viewHeight)];
[snapshotConfiguration setSnapshotWidth:@(viewWidth)];
[snapshotConfiguration setAfterScreenUpdates:YES];
[webView evaluateJavaScript:@"var div = document.getElementById('change-me');div.style.backgroundColor = 'blue';" completionHandler:nil];
isDone = false;
[webView takeSnapshotWithConfiguration:snapshotConfiguration.get() completionHandler:^(PlatformImage snapshotImage, NSError *error) {
EXPECT_NULL(error);
EXPECT_EQ(viewWidth, snapshotImage.size.width);
RetainPtr<CGImageRef> cgImage = convertToCGImage(snapshotImage);
RetainPtr<CGColorSpaceRef> colorSpace = adoptCF(CGColorSpaceCreateDeviceRGB());
NSInteger viewWidthInPixels = viewWidth * backingScaleFactor;
NSInteger viewHeightInPixels = viewHeight * backingScaleFactor;
uint8_t *rgba = (unsigned char *)calloc(viewWidthInPixels * viewHeightInPixels * 4, sizeof(unsigned char));
auto context = adoptCF(CGBitmapContextCreate(rgba, viewWidthInPixels, viewHeightInPixels, 8, 4 * viewWidthInPixels, colorSpace.get(), kCGImageAlphaPremultipliedLast | kCGBitmapByteOrder32Big));
CGContextDrawImage(context.get(), CGRectMake(0, 0, viewWidthInPixels, viewHeightInPixels), cgImage.get());
NSInteger pixelIndex = getPixelIndex(0, 0, viewWidthInPixels);
EXPECT_EQ(0, rgba[pixelIndex]);
EXPECT_EQ(0, rgba[pixelIndex + 1]);
EXPECT_EQ(255, rgba[pixelIndex + 2]);
free(rgba);
isDone = true;
}];
TestWebKitAPI::Util::run(&isDone);
}
TEST(WKWebView, SnapshotWithoutAfterScreenUpdates)
{
// SnapshotConfiguration.afterScreenUpdates tests currently cannot truly test this API since it is only needed
// on iOS devices, and we do not currently run API tests on iOS devices. The expectations below are based on
// what we expect in the simulator and on macOS, which is that setting afterScreenUpdates to NO will still
// result in a snapshot that includes the recent screen updates. If we get these tests running on iOS device,
// then we would expect the pixels to be red instead of blue.
NSInteger viewWidth = 800;
NSInteger viewHeight = 600;
RetainPtr<WKWebView> webView = adoptNS([[WKWebView alloc] initWithFrame:NSMakeRect(0, 0, viewWidth, viewHeight)]);
RetainPtr<PlatformWindow> window;
CGFloat backingScaleFactor;
#if PLATFORM(MAC)
window = adoptNS([[NSWindow alloc] initWithContentRect:[webView frame] styleMask:NSBorderlessWindowMask backing:NSBackingStoreBuffered defer:NO]);
[[window contentView] addSubview:webView.get()];
backingScaleFactor = [window backingScaleFactor];
#elif PLATFORM(IOS_FAMILY)
window = adoptNS([[UIWindow alloc] initWithFrame:[webView frame]]);
[window addSubview:webView.get()];
backingScaleFactor = [[window screen] scale];
#endif
[webView loadHTMLString:@"<body style='margin:0'><div id='change-me' style='background-color:red; position:fixed; width:100%; height:100%'></div></body>" baseURL:nil];
[webView _test_waitForDidFinishNavigation];
RetainPtr<WKSnapshotConfiguration> snapshotConfiguration = adoptNS([[WKSnapshotConfiguration alloc] init]);
[snapshotConfiguration setRect:NSMakeRect(0, 0, viewWidth, viewHeight)];
[snapshotConfiguration setSnapshotWidth:@(viewWidth)];
[snapshotConfiguration setAfterScreenUpdates:NO];
[webView evaluateJavaScript:@"var div = document.getElementById('change-me');div.style.backgroundColor = 'blue';" completionHandler:nil];
isDone = false;
[webView takeSnapshotWithConfiguration:snapshotConfiguration.get() completionHandler:^(PlatformImage snapshotImage, NSError *error) {
EXPECT_NULL(error);
EXPECT_EQ(viewWidth, snapshotImage.size.width);
RetainPtr<CGImageRef> cgImage = convertToCGImage(snapshotImage);
RetainPtr<CGColorSpaceRef> colorSpace = adoptCF(CGColorSpaceCreateDeviceRGB());
NSInteger viewWidthInPixels = viewWidth * backingScaleFactor;
NSInteger viewHeightInPixels = viewHeight * backingScaleFactor;
uint8_t *rgba = (unsigned char *)calloc(viewWidthInPixels * viewHeightInPixels * 4, sizeof(unsigned char));
auto context = adoptCF(CGBitmapContextCreate(rgba, viewWidthInPixels, viewHeightInPixels, 8, 4 * viewWidthInPixels, colorSpace.get(), kCGImageAlphaPremultipliedLast | kCGBitmapByteOrder32Big));
CGContextDrawImage(context.get(), CGRectMake(0, 0, viewWidthInPixels, viewHeightInPixels), cgImage.get());
NSInteger pixelIndex = getPixelIndex(0, 0, viewWidthInPixels);
EXPECT_EQ(0, rgba[pixelIndex]);
EXPECT_EQ(0, rgba[pixelIndex + 1]);
EXPECT_EQ(255, rgba[pixelIndex + 2]);
free(rgba);
isDone = true;
}];
TestWebKitAPI::Util::run(&isDone);
}
// FIXME: Confirm that this works on iOS too.
#if PLATFORM(MAC)
TEST(WKWebView, SnapshotWebGL)
{
NSInteger viewWidth = 800;
NSInteger viewHeight = 600;
RetainPtr<WKWebView> webView = adoptNS([[WKWebView alloc] initWithFrame:NSMakeRect(0, 0, viewWidth, viewHeight)]);
RetainPtr<PlatformWindow> window;
CGFloat backingScaleFactor;
#if PLATFORM(MAC)
window = adoptNS([[NSWindow alloc] initWithContentRect:[webView frame] styleMask:NSBorderlessWindowMask backing:NSBackingStoreBuffered defer:NO]);
[[window contentView] addSubview:webView.get()];
backingScaleFactor = [window backingScaleFactor];
#elif PLATFORM(IOS_FAMILY)
window = adoptNS([[UIWindow alloc] initWithFrame:[webView frame]]);
[window addSubview:webView.get()];
backingScaleFactor = [[window screen] scale];
#endif
[webView loadHTMLString:@"<style>body{margin:0;padding:0}</style><canvas></canvas><script>window.addEventListener('load', () => { let ctx = document.querySelector('canvas').getContext('webgl'); ctx.clearColor(0, 1, 0, 1); ctx.clear(ctx.COLOR_BUFFER_BIT); }, false);</script>" baseURL:nil];
[webView _test_waitForDidFinishNavigation];
RetainPtr<WKSnapshotConfiguration> snapshotConfiguration = adoptNS([[WKSnapshotConfiguration alloc] init]);
[snapshotConfiguration setRect:NSMakeRect(0, 0, viewWidth, viewHeight)];
[snapshotConfiguration setSnapshotWidth:@(viewWidth)];
[snapshotConfiguration setAfterScreenUpdates:YES];
isDone = false;
[webView takeSnapshotWithConfiguration:snapshotConfiguration.get() completionHandler:^(PlatformImage snapshotImage, NSError *error) {
EXPECT_NULL(error);
EXPECT_EQ(viewWidth, snapshotImage.size.width);
RetainPtr<CGImageRef> cgImage = convertToCGImage(snapshotImage);
RetainPtr<CGColorSpaceRef> colorSpace = adoptCF(CGColorSpaceCreateDeviceRGB());
NSInteger viewWidthInPixels = viewWidth * backingScaleFactor;
NSInteger viewHeightInPixels = viewHeight * backingScaleFactor;
uint8_t *rgba = (unsigned char *)calloc(viewWidthInPixels * viewHeightInPixels * 4, sizeof(unsigned char));
auto context = adoptCF(CGBitmapContextCreate(rgba, viewWidthInPixels, viewHeightInPixels, 8, 4 * viewWidthInPixels, colorSpace.get(), kCGImageAlphaPremultipliedLast | kCGBitmapByteOrder32Big));
CGContextDrawImage(context.get(), CGRectMake(0, 0, viewWidthInPixels, viewHeightInPixels), cgImage.get());
NSInteger pixelIndex = getPixelIndex(0, 0, viewWidthInPixels);
EXPECT_EQ(0, rgba[pixelIndex]);
EXPECT_EQ(255, rgba[pixelIndex + 1]);
EXPECT_EQ(0, rgba[pixelIndex + 2]);
EXPECT_EQ(255, rgba[pixelIndex + 3]);
free(rgba);
isDone = true;
}];
TestWebKitAPI::Util::run(&isDone);
}
#endif