/*

 * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.

 *

 * This code is free software; you can redistribute it and/or modify it

 * under the terms of the GNU General Public License version 2 only, as

 * published by the Free Software Foundation.

 *

 * This code is distributed in the hope that it will be useful, but WITHOUT

 * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or

 * FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License

 * version 2 for more details (a copy is included in the LICENSE file that

 * accompanied this code).

 *

 * You should have received a copy of the GNU General Public License version

 * 2 along with this work; if not, write to the Free Software Foundation,

 * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA.

 *

 * Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA

 * or visit www.oracle.com if you need additional information or have any

 * questions.

 */

/*

  @test

  @bug 4422345 8039083

  @summary tests serialization of DragSourceListeners

  @author das@sparc.spb.su area=dnd

  @library ../../../../lib/testlibrary

  @build jdk.testlibrary.Asserts

  @run main/othervm DragSourceListenerSerializationTest

*/

import java.awt.Button;

import java.awt.Component;

import java.awt.Cursor;

import java.awt.Point;

import java.awt.Toolkit;

import java.awt.datatransfer.StringSelection;

import java.awt.dnd.DnDConstants;

import java.awt.dnd.DragGestureEvent;

import java.awt.dnd.DragGestureRecognizer;

import java.awt.dnd.DragSource;

import java.awt.dnd.DragSourceAdapter;

import java.awt.dnd.DragSourceContext;

import java.awt.dnd.DragSourceListener;

import java.awt.dnd.DragSourceMotionListener;

import java.awt.event.InputEvent;

import java.awt.event.MouseEvent;

import java.io.ByteArrayInputStream;

import java.io.ByteArrayOutputStream;

import java.io.ObjectInputStream;

import java.io.ObjectOutputStream;

import java.io.Serializable;

import java.util.Arrays;

import java.util.TooManyListenersException;

import java.util.stream.Collectors;

import java.util.stream.Stream;

import static jdk.testlibrary.Asserts.assertEquals;

public class DragSourceListenerSerializationTest {

    public static void main(String[] args) throws Exception {

        DragSource ds = new DragSource();

        TestDragSourceAdapter dsa1 = new TestDragSourceAdapter(1);

        TestDragSourceAdapter dsa2 = new TestDragSourceAdapter(2);

        Component c = new Button();

        DragGestureRecognizer dgr = ds.createDefaultDragGestureRecognizer(c,

                DnDConstants.ACTION_COPY,

                e -> e.startDrag(null, null));

        MouseEvent me = new MouseEvent(c, MouseEvent.MOUSE_PRESSED, 0,

                InputEvent.CTRL_MASK, 100, 100, 0, false);

        DragGestureEvent dge = new DragGestureEvent(dgr, DnDConstants.ACTION_COPY,

                new Point(100, 100),

                Arrays.asList(me));

        DragSourceContext dsc = new DragSourceContext(

                dge,

                new Cursor(Cursor.HAND_CURSOR),

                null, null, new StringSelection("TEXT"), null);

        ds.addDragSourceListener(dsa1);

        ds.addDragSourceListener(dsa2);

        ds.addDragSourceListener(dsa2);

        ds.addDragSourceMotionListener(dsa1);

        ds.addDragSourceMotionListener(dsa1);

        ds.addDragSourceMotionListener(dsa2);

        dsc.addDragSourceListener(dsa2);

        byte[] serialized;

        try (ByteArrayOutputStream bos = new ByteArrayOutputStream();

             ObjectOutputStream oos = new ObjectOutputStream(bos)) {

            oos.writeObject(dsc);

            serialized = bos.toByteArray();

        }

        DragSourceContext dsc_copy;

        try (ByteArrayInputStream bis = new ByteArrayInputStream(serialized);

             ObjectInputStream ois = new ObjectInputStream(bis)) {

            dsc_copy = (DragSourceContext) ois.readObject();

        }

        try {

            dsc_copy.addDragSourceListener(dsa1);

            throw new RuntimeException("Test failed. Listener addition succeeded");

        } catch (TooManyListenersException ignored) {

        }

        try {

            dsc_copy.addDragSourceListener(dsa2);

            throw new RuntimeException("Test failed. Listener addition succeeded");

        } catch (TooManyListenersException ignored) {

        }

        try (ByteArrayOutputStream bos = new ByteArrayOutputStream();

             ObjectOutputStream oos = new ObjectOutputStream(bos)) {

            oos.writeObject(ds);

            serialized = bos.toByteArray();

        }

        DragSource ds_copy;

        try (ByteArrayInputStream bis = new ByteArrayInputStream(serialized);

             ObjectInputStream ois = new ObjectInputStream(bis)) {

             ds_copy = (DragSource) ois.readObject();

        }

        DragSourceListener[] dsls = ds_copy.getDragSourceListeners();

        assertEquals(3, dsls.length, "DragSourceListeners number");

        assertEquals(1, Stream.of(dsls).filter(dsa1::equals).collect(Collectors.counting()).intValue());

        assertEquals(2, Stream.of(dsls).filter(dsa2::equals).collect(Collectors.counting()).intValue());

        DragSourceMotionListener[] dsmls = ds_copy.getDragSourceMotionListeners();

        assertEquals(3, dsmls.length, "DragSourceMotionListeners number");

        assertEquals(2, Stream.of(dsmls).filter(dsa1::equals).collect(Collectors.counting()).intValue());

        assertEquals(1, Stream.of(dsmls).filter(dsa2::equals).collect(Collectors.counting()).intValue());

    }

}

class TestDragSourceAdapter extends DragSourceAdapter implements Serializable {

    final int id;

    TestDragSourceAdapter(int id) {

        this.id = id;

    }

    public int getId() {

        return id;

    }

    public boolean equals(Object obj) {

        if (obj instanceof TestDragSourceAdapter) {

            TestDragSourceAdapter tdsa = (TestDragSourceAdapter) obj;

            return tdsa.getId() == getId();

        }

        return false;

    }

}