1 /*
   2  * Copyright (c) 2014, 2016, 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.
   8  *
   9  * This code is distributed in the hope that it will be useful, but WITHOUT
  10  * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
  11  * FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License
  12  * version 2 for more details (a copy is included in the LICENSE file that
  13  * accompanied this code).
  14  *
  15  * You should have received a copy of the GNU General Public License version
  16  * 2 along with this work; if not, write to the Free Software Foundation,
  17  * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA.
  18  *
  19  * Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA
  20  * or visit www.oracle.com if you need additional information or have any
  21  * questions.
  22  */
  23 
  24 /*
  25   @test
  26   @key headful
  27   @bug 4422345 8039083
  28   @summary tests serialization of DragSourceListeners
  29   @author das@sparc.spb.su area=dnd
  30   @library ../../../../lib/testlibrary
  31   @build jdk.testlibrary.Asserts
  32   @run main/othervm DragSourceListenerSerializationTest
  33 */
  34 
  35 import java.awt.Button;
  36 import java.awt.Component;
  37 import java.awt.Cursor;
  38 import java.awt.Point;
  39 import java.awt.Toolkit;
  40 import java.awt.datatransfer.StringSelection;
  41 import java.awt.dnd.DnDConstants;
  42 import java.awt.dnd.DragGestureEvent;
  43 import java.awt.dnd.DragGestureRecognizer;
  44 import java.awt.dnd.DragSource;
  45 import java.awt.dnd.DragSourceAdapter;
  46 import java.awt.dnd.DragSourceContext;
  47 import java.awt.dnd.DragSourceListener;
  48 import java.awt.dnd.DragSourceMotionListener;
  49 import java.awt.event.InputEvent;
  50 import java.awt.event.MouseEvent;
  51 import java.io.ByteArrayInputStream;
  52 import java.io.ByteArrayOutputStream;
  53 import java.io.ObjectInputStream;
  54 import java.io.ObjectOutputStream;
  55 import java.io.Serializable;
  56 import java.util.Arrays;
  57 import java.util.TooManyListenersException;
  58 import java.util.stream.Collectors;
  59 import java.util.stream.Stream;
  60 
  61 import static jdk.testlibrary.Asserts.assertEquals;
  62 
  63 public class DragSourceListenerSerializationTest {
  64     public static void main(String[] args) throws Exception {
  65         DragSource ds = new DragSource();
  66         TestDragSourceAdapter dsa1 = new TestDragSourceAdapter(1);
  67         TestDragSourceAdapter dsa2 = new TestDragSourceAdapter(2);
  68         Component c = new Button();
  69         DragGestureRecognizer dgr = ds.createDefaultDragGestureRecognizer(c,
  70                 DnDConstants.ACTION_COPY,
  71                 e -> e.startDrag(null, null));
  72         MouseEvent me = new MouseEvent(c, MouseEvent.MOUSE_PRESSED, 0,
  73                 InputEvent.CTRL_MASK, 100, 100, 0, false);
  74         DragGestureEvent dge = new DragGestureEvent(dgr, DnDConstants.ACTION_COPY,
  75                 new Point(100, 100),
  76                 Arrays.asList(me));
  77         DragSourceContext dsc = new DragSourceContext(
  78                 Toolkit.getDefaultToolkit().createDragSourceContextPeer(dge),
  79                 dge,
  80                 new Cursor(Cursor.HAND_CURSOR),
  81                 null, null, new StringSelection("TEXT"), null);
  82 
  83         ds.addDragSourceListener(dsa1);
  84         ds.addDragSourceListener(dsa2);
  85         ds.addDragSourceListener(dsa2);
  86         ds.addDragSourceMotionListener(dsa1);
  87         ds.addDragSourceMotionListener(dsa1);
  88         ds.addDragSourceMotionListener(dsa2);
  89         dsc.addDragSourceListener(dsa2);
  90 
  91         byte[] serialized;
  92         try (ByteArrayOutputStream bos = new ByteArrayOutputStream();
  93              ObjectOutputStream oos = new ObjectOutputStream(bos)) {
  94             oos.writeObject(dsc);
  95             serialized = bos.toByteArray();
  96         }
  97 
  98         DragSourceContext dsc_copy;
  99         try (ByteArrayInputStream bis = new ByteArrayInputStream(serialized);
 100              ObjectInputStream ois = new ObjectInputStream(bis)) {
 101             dsc_copy = (DragSourceContext) ois.readObject();
 102         }
 103 
 104         try {
 105             dsc_copy.addDragSourceListener(dsa1);
 106             throw new RuntimeException("Test failed. Listener addition succeeded");
 107         } catch (TooManyListenersException ignored) {
 108         }
 109 
 110         try {
 111             dsc_copy.addDragSourceListener(dsa2);
 112             throw new RuntimeException("Test failed. Listener addition succeeded");
 113         } catch (TooManyListenersException ignored) {
 114         }
 115 
 116         try (ByteArrayOutputStream bos = new ByteArrayOutputStream();
 117              ObjectOutputStream oos = new ObjectOutputStream(bos)) {
 118             oos.writeObject(ds);
 119             serialized = bos.toByteArray();
 120         }
 121 
 122         DragSource ds_copy;
 123         try (ByteArrayInputStream bis = new ByteArrayInputStream(serialized);
 124              ObjectInputStream ois = new ObjectInputStream(bis)) {
 125              ds_copy = (DragSource) ois.readObject();
 126         }
 127 
 128         DragSourceListener[] dsls = ds_copy.getDragSourceListeners();
 129         assertEquals(3, dsls.length, "DragSourceListeners number");
 130         assertEquals(1, Stream.of(dsls).filter(dsa1::equals).collect(Collectors.counting()).intValue());
 131         assertEquals(2, Stream.of(dsls).filter(dsa2::equals).collect(Collectors.counting()).intValue());
 132 
 133         DragSourceMotionListener[] dsmls = ds_copy.getDragSourceMotionListeners();
 134         assertEquals(3, dsmls.length, "DragSourceMotionListeners number");
 135         assertEquals(2, Stream.of(dsmls).filter(dsa1::equals).collect(Collectors.counting()).intValue());
 136         assertEquals(1, Stream.of(dsmls).filter(dsa2::equals).collect(Collectors.counting()).intValue());
 137     }
 138 }
 139 
 140 class TestDragSourceAdapter extends DragSourceAdapter implements Serializable {
 141     final int id;
 142 
 143     TestDragSourceAdapter(int id) {
 144         this.id = id;
 145     }
 146 
 147     public int getId() {
 148         return id;
 149     }
 150 
 151     public boolean equals(Object obj) {
 152         if (obj instanceof TestDragSourceAdapter) {
 153             TestDragSourceAdapter tdsa = (TestDragSourceAdapter) obj;
 154             return tdsa.getId() == getId();
 155         }
 156         return false;
 157     }
 158 }