1 /*
2 * Copyright (c) 2001, 2015, Oracle and/or its affiliates. All rights reserved.
3 * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
4 *
5 * This code is free software; you can redistribute it and/or modify it
6 * under the terms of the GNU General Public License version 2 only, as
7 * published by the Free Software Foundation. Oracle designates this
8 * particular file as subject to the "Classpath" exception as provided
9 * by Oracle in the LICENSE file that accompanied this code.
10 *
11 * This code is distributed in the hope that it will be useful, but WITHOUT
12 * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
13 * FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License
14 * version 2 for more details (a copy is included in the LICENSE file that
15 * accompanied this code).
16 *
17 * You should have received a copy of the GNU General Public License version
18 * 2 along with this work; if not, write to the Free Software Foundation,
19 * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA.
20 *
21 * Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA
22 * or visit www.oracle.com if you need additional information or have any
23 * questions.
24 */
25
26 import java.awt.datatransfer.DataFlavor;
27 import java.awt.datatransfer.SystemFlavorMap;
28 import java.util.*;
29 import java.nio.charset.Charset;
30
31 /*
32 * @test
33 * @summary To test SystemFlavorMap method:
34 * addFlavorForUnencodedNative(String nat, DataFlavor flav)
35 * with valid natives and DataFlavors. This stress test will
36 * define numerous mappings of valid String natives and
37 * DataFlavors. The mappings will be verified by examining
38 * that all entries are present, and order is maintained.
39 * @author Rick Reynaga (rick.reynaga@eng.sun.com) area=Clipboard
40 * @author dmitriy.ermashov@oracle.com
41 * @modules java.datatransfer
42 * @run main AddFlavorTest
43 */
44
45 public class AddFlavorTest {
46
47 SystemFlavorMap flavorMap;
48 Hashtable<String, List<DataFlavor>> hashVerify;
49
50 public static void main (String[] args) throws Exception {
51 new AddFlavorTest().doTest();
52 }
53
54 void doTest() throws Exception {
55 flavorMap = (SystemFlavorMap)SystemFlavorMap.getDefaultFlavorMap();
56
57 // Test addFlavorForUnencodedNative(String nat, DataFlavor flav);
58 //
59 // Enumerate through all the system defined String natives,
60 // and for each String native, define it again to the SystemFlavorMap
61 // with a slightly modified String native (name).
62 //
63 // As a list of DataFlavors will be returned for each String native,
64 // the method addFlavorForUnencodedNative will be called for each
65 // DataFlavor in the list.
66 hashVerify = new Hashtable();
67
68 for (String key : flavorMap.getFlavorsForNatives(null).keySet()) {
69 Set<DataFlavor> flavorsSet = new HashSet<>(flavorMap.getFlavorsForNative(key));
70 // construct a unique String native
71 key = key.concat("TEST");
72
73 for (DataFlavor element : flavorsSet) {
74 flavorMap.addFlavorForUnencodedNative(key, element);
75 }
76 hashVerify.put(key, new Vector(flavorsSet));
77 }
78
79 // Assertions: After establishing "new" mappings, verify that the defined
80 // DataFlavors can be retrieved and that the List is preserved.
81 verifyNewMappings();
82 }
83
84 // Verify getFlavorsForNative(String nat) is returning the correct list
85 // of DataFlavors (for the new mappings).
86 void verifyNewMappings() {
87 // Enumerate through all natives
88 System.out.println("*** native size = " + hashVerify.size());
89 for (Enumeration e = hashVerify.keys() ; e.hasMoreElements() ;) {
90 String key = (String)e.nextElement();
91 compareFlavors(hashVerify.get(key), flavorMap.getFlavorsForNative(key), key);
92 compareFlavors(flavorMap.getFlavorsForNative(key), hashVerify.get(key), key);
93 }
94 }
95
96 void compareFlavors(List<DataFlavor> flavors1, List<DataFlavor> flavors2, String key){
97 for (DataFlavor flavor1 : flavors1) {
98 boolean result = false;
99 for (DataFlavor flavor2 : flavors2) {
100 if (flavor1.equals(flavor2)) result = true;
101 }
102 if (!result)
103 throw new RuntimeException("\n*** Error in verifyNewMappings()" +
104 "\nmethod1: addFlavorForUnencodedNative(String nat, DataFlavor flav)" +
105 "\nmethod2: List getFlavorsForNative(String nat)" +
106 "\nString native: " + key +
107 "\nAfter adding several mappings with addFlavorForUnencodedNative," +
108 "\nthe returned list did not match the mappings that were added." +
109 "\nEither the mapping was not included in the list, or the order was incorect.");
110 }
111
112 }
113
114 Set<DataFlavor> convertMimeTypeToDataFlavors(String baseType) throws Exception {
115 Set<DataFlavor> result = new LinkedHashSet<>();
116
117 for (String charset : getStandardEncodings()) {
118 for (String txtClass : new String[]{"java.io.InputStream", "java.nio.ByteBuffer", "\"[B\""}) {
119 String mimeType = baseType + ";charset=" + charset + ";class=" + txtClass;
120
121 if ("text/html".equals(baseType)) {
122 for (String documentType : new String[]{"all", "selection", "fragment"})
123 result.add(new DataFlavor(mimeType + ";document=" + documentType));
124 } else {
125 DataFlavor df = new DataFlavor(mimeType);
126 if (df.equals(DataFlavor.plainTextFlavor))
127 df = DataFlavor.plainTextFlavor;
128 result.add(df);
129 }
130 }
131 }
132 return result;
133 }
134
135 Set<String> getStandardEncodings() {
136 Set<String> tempSet = new HashSet<>();
137 tempSet.add("US-ASCII");
138 tempSet.add("ISO-8859-1");
139 tempSet.add("UTF-8");
140 tempSet.add("UTF-16BE");
141 tempSet.add("UTF-16LE");
142 tempSet.add("UTF-16");
143 tempSet.add(Charset.defaultCharset().name());
144 return tempSet;
145 }
146 }
147