c3adebb3bfc39b529080b41e54e8f5772df504e1
[WebKit-https.git] / Tools / TestWebKitAPI / Tests / ios / UIPasteboardTests.mm
1 /*
2  * Copyright (C) 2017 Apple Inc. All rights reserved.
3  *
4  * Redistribution and use in source and binary forms, with or without
5  * modification, are permitted provided that the following conditions
6  * are met:
7  * 1. Redistributions of source code must retain the above copyright
8  *    notice, this list of conditions and the following disclaimer.
9  * 2. Redistributions in binary form must reproduce the above copyright
10  *    notice, this list of conditions and the following disclaimer in the
11  *    documentation and/or other materials provided with the distribution.
12  *
13  * THIS SOFTWARE IS PROVIDED BY APPLE INC. AND ITS CONTRIBUTORS ``AS IS''
14  * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO,
15  * THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
16  * PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL APPLE INC. OR ITS CONTRIBUTORS
17  * BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
18  * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
19  * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
20  * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
21  * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
22  * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF
23  * THE POSSIBILITY OF SUCH DAMAGE.
24  */
25
26 #include "config.h"
27
28 #if PLATFORM(IOS) && WK_API_ENABLED
29
30 #import "PlatformUtilities.h"
31 #import "TestWKWebView.h"
32 #import <MobileCoreServices/MobileCoreServices.h>
33 #import <UIKit/UIPasteboard.h>
34 #import <WebKit/WKPreferencesPrivate.h>
35 #import <WebKit/WKWebViewPrivate.h>
36
37 namespace TestWebKitAPI {
38
39 NSData *dataForPasteboardType(CFStringRef type)
40 {
41     return [[UIPasteboard generalPasteboard] dataForPasteboardType:(NSString *)type inItemSet:[NSIndexSet indexSetWithIndex:0]].firstObject;
42 }
43
44 RetainPtr<TestWKWebView> setUpWebViewForPasteboardTests()
45 {
46     [UIPasteboard generalPasteboard].items = @[];
47     EXPECT_TRUE(!dataForPasteboardType(kUTTypeUTF8PlainText).length);
48     EXPECT_TRUE(!dataForPasteboardType(kUTTypeUTF16PlainText).length);
49
50     auto webView = adoptNS([[TestWKWebView alloc] initWithFrame:CGRectMake(0, 0, 320, 500)]);
51     [webView configuration].preferences._javaScriptCanAccessClipboard = YES;
52     [webView synchronouslyLoadTestPageNamed:@"rich-and-plain-text"];
53     return webView;
54 }
55
56 TEST(UIPasteboardTests, CopyPlainTextWritesConcreteTypes)
57 {
58     auto webView = setUpWebViewForPasteboardTests();
59     [webView stringByEvaluatingJavaScript:@"selectPlainText()"];
60     [webView stringByEvaluatingJavaScript:@"document.execCommand('copy')"];
61
62     auto utf8Result = adoptNS([[NSString alloc] initWithData:dataForPasteboardType(kUTTypeUTF8PlainText) encoding:NSUTF8StringEncoding]);
63     auto utf16Result = adoptNS([[NSString alloc] initWithData:dataForPasteboardType(kUTTypeUTF16PlainText) encoding:NSUTF16StringEncoding]);
64     EXPECT_WK_STREQ("Hello world", [utf8Result UTF8String]);
65     EXPECT_WK_STREQ("Hello world", [utf16Result UTF8String]);
66 }
67
68 TEST(UIPasteboardTests, CopyRichTextWritesConcreteTypes)
69 {
70     auto webView = setUpWebViewForPasteboardTests();
71     [webView stringByEvaluatingJavaScript:@"selectRichText()"];
72     [webView stringByEvaluatingJavaScript:@"document.execCommand('copy')"];
73
74     auto utf8Result = adoptNS([[NSString alloc] initWithData:dataForPasteboardType(kUTTypeUTF8PlainText) encoding:NSUTF8StringEncoding]);
75     auto utf16Result = adoptNS([[NSString alloc] initWithData:dataForPasteboardType(kUTTypeUTF16PlainText) encoding:NSUTF16StringEncoding]);
76     EXPECT_WK_STREQ("Hello world", [utf8Result UTF8String]);
77     EXPECT_WK_STREQ("Hello world", [utf16Result UTF8String]);
78 }
79
80 } // namespace TestWebKitAPI
81
82 #endif