/*

 * Copyright (c) 2007, 2014, Oracle and/or its affiliates. All rights reserved.

 * 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.

 */

import java.awt.*;

import java.awt.event.*;

import static jdk.testlibrary.Asserts.assertTrue;

/*

 * @test

 * @summary Make sure that modifier key mask is set when robot press

 * some key with one or more modifiers.

 *

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

 * @build ExtendedRobot

 * @run main ModifierRobotKeyTest

 */

public class ModifierRobotKeyTest extends KeyAdapter {

    private boolean focusGained = false;

    private boolean startTest = false;

    private ExtendedRobot robot;

    private Frame frame;

    private Canvas canvas;

    private volatile boolean tempPress = false;

    private int[] textKeys, modifierKeys, inputMasks;

    private boolean[] modifierStatus, textStatus;

    private final static int waitDelay = 5000;

    private Object tempLock = new Object();

    private Object keyLock = new Object();

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

        ModifierRobotKeyTest test = new ModifierRobotKeyTest();

        test.doTest();

    }

    public ModifierRobotKeyTest() throws Exception {

        modifierKeys =  new int[3];

        modifierKeys[0] = KeyEvent.VK_SHIFT;

        modifierKeys[1] = KeyEvent.VK_CONTROL;

        modifierKeys[2] = KeyEvent.VK_ALT;

        inputMasks = new int[3];

        inputMasks[0] =  InputEvent.SHIFT_MASK;

        inputMasks[1] =  InputEvent.CTRL_MASK;

        inputMasks[2] =  InputEvent.ALT_MASK;

        modifierStatus = new boolean[modifierKeys.length];

        textKeys = new int[2];

        textKeys[0] = KeyEvent.VK_A;

        String os = System.getProperty("os.name").toLowerCase();

        if (os.contains("solaris") || os.contains("sunos"))

            textKeys[1] = KeyEvent.VK_S;

        else if (os.contains("os x"))

            textKeys[1] = KeyEvent.VK_K;

        else

            textKeys[1] = KeyEvent.VK_I;

        textStatus = new boolean[textKeys.length];

        EventQueue.invokeAndWait( () -> { initializeGUI(); });

    }

    public void keyPressed(KeyEvent event) {

        tempPress = true;

        synchronized (tempLock) { tempLock.notifyAll(); }

        if (! startTest) {

            return;

        }

        for (int x = 0; x < inputMasks.length; x++) {

            if ((event.getModifiers() & inputMasks[x]) != 0) {

                System.out.println("Modifier set: " + event.getKeyModifiersText(inputMasks[x]));

                modifierStatus[x] = true;

            }

        }

        for (int x = 0; x < textKeys.length; x++) {

            if (event.getKeyCode() == textKeys[x]) {

                System.out.println("Text set: " + event.getKeyText(textKeys[x]));

                textStatus[x] = true;

            }

        }

        synchronized (keyLock) { keyLock.notifyAll(); }

    }

    private void initializeGUI() {

        frame = new Frame("Test frame");

        canvas = new Canvas();

        canvas.addFocusListener(new FocusAdapter() {

            public void focusGained(FocusEvent event) { focusGained = true; }

        });

        canvas.addKeyListener(this);

        frame.setLayout(new BorderLayout());

        frame.add(canvas);

        frame.setSize(200, 200);

        frame.setVisible(true);

    }

    public void doTest() throws Exception {

        robot = new ExtendedRobot();

        robot.mouseMove((int) frame.getLocationOnScreen().getX() + frame.getSize().width / 2,

                        (int) frame.getLocationOnScreen().getY() + frame.getSize().height / 2);

        robot.click(MouseEvent.BUTTON1_MASK);

        robot.waitForIdle();

        assertTrue(focusGained, "FAIL: Canvas gained focus!");

        for (int i = 0; i < modifierKeys.length; i++) {

            for (int j = 0; j < textKeys.length; j++) {

                tempPress = false;

                robot.keyPress(modifierKeys[i]);

                robot.waitForIdle();

                if (! tempPress) {

                    synchronized (tempLock) { tempLock.wait(waitDelay); }

                }

                assertTrue(tempPress, "FAIL: keyPressed triggered for i=" + i);

                resetStatus();

                startTest = true;

                robot.keyPress(textKeys[j]);

                robot.waitForIdle();

                if (! modifierStatus[i] || ! textStatus[j]) {

                    synchronized (keyLock) { keyLock.wait(waitDelay); }

                }

                assertTrue(modifierStatus[i] && textStatus[j],

                        "FAIL: KeyEvent not proper!"+

                        "Key checked: i=" + i + "; j=" + j+

                        "ModifierStatus = " + modifierStatus[i]+

                        "TextStatus = " + textStatus[j]);

                startTest = false;

                robot.keyRelease(textKeys[j]);

                robot.waitForIdle();

                robot.keyRelease(modifierKeys[i]);

                robot.waitForIdle();

            }

        }

        for (int i = 0; i < modifierKeys.length; i++) {

            for (int j = i + 1; j < modifierKeys.length; j++) {

                for (int k = 0; k < textKeys.length; k++) {

                    tempPress = false;

                    robot.keyPress(modifierKeys[i]);

                    robot.waitForIdle();

                    if (! tempPress) {

                        synchronized (tempLock) { tempLock.wait(waitDelay); }

                    }

                    assertTrue(tempPress, "FAIL: MultiKeyTest: keyPressed triggered for i=" + i);

                    tempPress = false;

                    robot.keyPress(modifierKeys[j]);

                    robot.waitForIdle();

                    if (! tempPress) {

                        synchronized (tempLock) { tempLock.wait(waitDelay); }

                    }

                    assertTrue(tempPress, "FAIL: MultiKeyTest keyPressed triggered for j=" + j);

                    resetStatus();

                    startTest = true;

                    robot.keyPress(textKeys[k]);

                    robot.waitForIdle();

                    if (! modifierStatus[i] || ! modifierStatus[j] || ! textStatus[k]) {

                        synchronized (keyLock) {

                            keyLock.wait(waitDelay);

                        }

                    }

                    assertTrue(modifierStatus[i] && modifierStatus[j] && textStatus[k],

                            "FAIL: KeyEvent not proper!"+

                            "Key checked: i=" + i + "; j=" + j + "; k=" + k+

                            "Modifier1Status = " + modifierStatus[i]+

                            "Modifier2Status = " + modifierStatus[j]+

                            "TextStatus = " + textStatus[k]);

                    startTest = false;

                    robot.keyRelease(textKeys[k]);

                    robot.waitForIdle();

                    robot.keyRelease(modifierKeys[j]);

                    robot.waitForIdle();

                    robot.keyRelease(modifierKeys[i]);

                    robot.waitForIdle();

                }

            }

        }

        frame.dispose();

    }

    private void resetStatus() {

        for (int i = 0; i < modifierStatus.length; i++) {

            modifierStatus[i] = false;

        }

        for (int i = 0; i < textStatus.length; i++) {

            textStatus[i] = false;

        }

    }

}