1 /*
   2  * Copyright (c) 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.
   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   @bug 4422345 8039083
  27   @summary tests serialization of DragSourceListeners
  28   @author das@sparc.spb.su area=dnd
  29   @library ../../../../lib/testlibrary
  30   @build jdk.testlibrary.Asserts
  31   @run main/othervm DragSourceListenerSerializationTest
  32 */
  33 
  34 import java.awt.Button;
  35 import java.awt.Component;
  36 import java.awt.Cursor;
  37 import java.awt.Point;
  38 import java.awt.Toolkit;
  39 import java.awt.datatransfer.StringSelection;
  40 import java.awt.dnd.DnDConstants;
  41 import java.awt.dnd.DragGestureEvent;
  42 import java.awt.dnd.DragGestureRecognizer;
  43 import java.awt.dnd.DragSource;
  44 import java.awt.dnd.DragSourceAdapter;
  45 import java.awt.dnd.DragSourceContext;
  46 import java.awt.dnd.DragSourceListener;
  47 import java.awt.dnd.DragSourceMotionListener;
  48 import java.awt.event.InputEvent;
  49 import java.awt.event.MouseEvent;
  50 import java.io.ByteArrayInputStream;
  51 import java.io.ByteArrayOutputStream;
  52 import java.io.ObjectInputStream;
  53 import java.io.ObjectOutputStream;
  54 import java.io.Serializable;
  55 import java.util.Arrays;
  56 import java.util.TooManyListenersException;
  57 import java.util.stream.Collectors;
  58 import java.util.stream.Stream;
  59 
  60 import static jdk.testlibrary.Asserts.assertEquals;
  61 
  62 public class DragSourceListenerSerializationTest {
  63     public static void main(String[] args) throws Exception {
  64         DragSource ds = new DragSource();
  65         TestDragSourceAdapter dsa1 = new TestDragSourceAdapter(1);
  66         TestDragSourceAdapter dsa2 = new TestDragSourceAdapter(2);
  67         Component c = new Button();
  68         DragGestureRecognizer dgr = ds.createDefaultDragGestureRecognizer(c,
  69                 DnDConstants.ACTION_COPY,
  70                 e -> e.startDrag(null, null));
  71         MouseEvent me = new MouseEvent(c, MouseEvent.MOUSE_PRESSED, 0,
  72                 InputEvent.CTRL_MASK, 100, 100, 0, false);
  73         DragGestureEvent dge = new DragGestureEvent(dgr, DnDConstants.ACTION_COPY,
  74                 new Point(100, 100),
  75                 Arrays.asList(me));
  76         DragSourceContext dsc = new DragSourceContext(
  77                 dge,
  78                 new Cursor(Cursor.HAND_CURSOR),
  79                 null, null, new StringSelection("TEXT"), null);
  80 
  81         ds.addDragSourceListener(dsa1);
  82         ds.addDragSourceListener(dsa2);
  83         ds.addDragSourceListener(dsa2);
  84         ds.addDragSourceMotionListener(dsa1);
  85         ds.addDragSourceMotionListener(dsa1);
  86         ds.addDragSourceMotionListener(dsa2);
  87         dsc.addDragSourceListener(dsa2);
  88 
  89         byte[] serialized;
  90         try (ByteArrayOutputStream bos = new ByteArrayOutputStream();
  91              ObjectOutputStream oos = new ObjectOutputStream(bos)) {
  92             oos.writeObject(dsc);
  93             serialized = bos.toByteArray();
  94         }
  95 
  96         DragSourceContext dsc_copy;
  97         try (ByteArrayInputStream bis = new ByteArrayInputStream(serialized);
  98              ObjectInputStream ois = new ObjectInputStream(bis)) {
  99             dsc_copy = (DragSourceContext) ois.readObject();
 100         }
 101 
 102         try {
 103             dsc_copy.addDragSourceListener(dsa1);
 104             throw new RuntimeException("Test failed. Listener addition succeeded");
 105         } catch (TooManyListenersException ignored) {
 106         }
 107 
 108         try {
 109             dsc_copy.addDragSourceListener(dsa2);
 110             throw new RuntimeException("Test failed. Listener addition succeeded");
 111         } catch (TooManyListenersException ignored) {
 112         }
 113 
 114         try (ByteArrayOutputStream bos = new ByteArrayOutputStream();
 115              ObjectOutputStream oos = new ObjectOutputStream(bos)) {
 116             oos.writeObject(ds);
 117             serialized = bos.toByteArray();
 118         }
 119 
 120         DragSource ds_copy;
 121         try (ByteArrayInputStream bis = new ByteArrayInputStream(serialized);
 122              ObjectInputStream ois = new ObjectInputStream(bis)) {
 123              ds_copy = (DragSource) ois.readObject();
 124         }
 125 
 126         DragSourceListener[] dsls = ds_copy.getDragSourceListeners();
 127         assertEquals(3, dsls.length, "DragSourceListeners number");
 128         assertEquals(1, Stream.of(dsls).filter(dsa1::equals).collect(Collectors.counting()).intValue());
 129         assertEquals(2, Stream.of(dsls).filter(dsa2::equals).collect(Collectors.counting()).intValue());
 130 
 131         DragSourceMotionListener[] dsmls = ds_copy.getDragSourceMotionListeners();
 132         assertEquals(3, dsmls.length, "DragSourceMotionListeners number");
 133         assertEquals(2, Stream.of(dsmls).filter(dsa1::equals).collect(Collectors.counting()).intValue());
 134         assertEquals(1, Stream.of(dsmls).filter(dsa2::equals).collect(Collectors.counting()).intValue());
 135     }
 136 }
 137 
 138 class TestDragSourceAdapter extends DragSourceAdapter implements Serializable {
 139     final int id;
 140 
 141     TestDragSourceAdapter(int id) {
 142         this.id = id;
 143     }
 144 
 145     public int getId() {
 146         return id;
 147     }
 148 
 149     public boolean equals(Object obj) {
 150         if (obj instanceof TestDragSourceAdapter) {
 151             TestDragSourceAdapter tdsa = (TestDragSourceAdapter) obj;
 152             return tdsa.getId() == getId();
 153         }
 154         return false;
 155     }
 156 }