AddFlavorTest.java revision 10341:c6a4534a458d 
1/*
2 * Copyright (c) 2001, 2014, 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 sun.awt.datatransfer.DataTransferer;
27
28import java.awt.datatransfer.DataFlavor;
29import java.awt.datatransfer.SystemFlavorMap;
30import java.util.*;
31
32/*
33 * @test
34 * @summary To test SystemFlavorMap method:
35 *          addFlavorForUnencodedNative(String nat, DataFlavor flav)
36 *          with valid natives and DataFlavors. This stress test will
37 *          define numerous mappings of valid String natives and
38 *          DataFlavors.  The mappings will be verified by examining
39 *          that all entries are present, and order is maintained.
40 * @author Rick Reynaga (rick.reynaga@eng.sun.com) area=Clipboard
41 * @author dmitriy.ermashov@oracle.com
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            // This part is valid only for X-based graphical systems
77            if (!System.getProperty("os.name").startsWith("Win") && !System.getProperty("os.name").startsWith("Mac") ) {
78                if (key.contains("/")) {
79                    Set<DataFlavor> fls = DataTransferer.getInstance().getPlatformMappingsForNative(key);
80                    flavorsSet.addAll(fls);
81                    if (!fls.isEmpty() && key.startsWith("text/"))
82                        flavorsSet.addAll(convertMimeTypeToDataFlavors(key));
83                }
84            }
85            hashVerify.put(key, new Vector(flavorsSet));
86        }
87
88        // Assertions: After establishing "new" mappings, verify that the defined
89        //             DataFlavors can be retrieved and that the List is preserved.
90        verifyNewMappings();
91    }
92
93    // Verify getFlavorsForNative(String nat) is returning the correct list
94    // of DataFlavors (for the new mappings).
95    void verifyNewMappings() {
96        // Enumerate through all natives
97        System.out.println("*** native size = " + hashVerify.size());
98        for (Enumeration e = hashVerify.keys() ; e.hasMoreElements() ;) {
99            String key = (String)e.nextElement();
100            compareFlavors(hashVerify.get(key), flavorMap.getFlavorsForNative(key), key);
101            compareFlavors(flavorMap.getFlavorsForNative(key), hashVerify.get(key), key);
102        }
103    }
104
105    void compareFlavors(List<DataFlavor> flavors1, List<DataFlavor> flavors2, String key){
106        DataTransferer.DataFlavorComparator comparator = new DataTransferer.DataFlavorComparator();
107        for (DataFlavor flavor1 : flavors1) {
108            boolean result = false;
109            for (DataFlavor flavor2 : flavors2) {
110                if (comparator.compare(flavor1, flavor2) == 0)
111                    result = true;
112            }
113            if (!result)
114                throw new RuntimeException("\n*** Error in verifyNewMappings()" +
115                        "\nmethod1: addFlavorForUnencodedNative(String nat, DataFlavor flav)"  +
116                        "\nmethod2: List getFlavorsForNative(String nat)" +
117                        "\nString native: " + key +
118                        "\nAfter adding several mappings with addFlavorForUnencodedNative," +
119                        "\nthe returned list did not match the mappings that were added." +
120                        "\nEither the mapping was not included in the list, or the order was incorect.");
121        }
122
123    }
124
125    Set<DataFlavor> convertMimeTypeToDataFlavors(String baseType) throws Exception {
126        Set<DataFlavor> result = new LinkedHashSet<>();
127
128        for (String charset : DataTransferer.standardEncodings()) {
129            for (String txtClass : new String[]{"java.io.InputStream", "java.nio.ByteBuffer", "\"[B\""}) {
130                String mimeType = baseType + ";charset=" + charset + ";class=" + txtClass;
131
132                if ("text/html".equals(baseType)) {
133                    for (String documentType : new String[]{"all", "selection", "fragment"})
134                        result.add(new DataFlavor(mimeType + ";document=" + documentType));
135                } else {
136                    DataFlavor df = new DataFlavor(mimeType);
137                    if (df.equals(DataFlavor.plainTextFlavor))
138                        df = DataFlavor.plainTextFlavor;
139                    result.add(df);
140                }
141            }
142        }
143        return result;
144    }
145}
146
147