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
26import java.awt.datatransfer.DataFlavor;
27import java.awt.datatransfer.SystemFlavorMap;
28import java.util.*;
29import 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
45public 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
148