JetBrainsRuntime/test/jdk/jb/sun/awt/macos/InputMethodTest/InputMethodTest.java

/*
 * Copyright 2000-2023 JetBrains s.r.o.
 * 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 sun.lwawt.macosx.LWCToolkit;

import javax.swing.*;
import java.awt.*;
import java.awt.event.InputEvent;
import java.awt.event.KeyEvent;
import java.awt.event.KeyListener;
import java.util.ArrayList;
import java.util.List;

public class InputMethodTest {
    private static JFrame frame;
    private static JTextArea textArea;
    private static Robot robot;
    private static String currentTest = "";
    private static String currentSection = "";
    private static String initialLayout;
    private static boolean success = true;
    private static int lastKeyCode = -1;

    private enum TestCases {
        DeadKeysTest (new DeadKeysTest()),
        KeyCodesTest (new KeyCodesTest()),
        PinyinQuotesTest (new PinyinQuotesTest()),
        PinyinFullWidthPunctuationTest (new PinyinFullWidthPunctuationTest()),
        PinyinHalfWidthPunctuationTest (new PinyinHalfWidthPunctuationTest())
        ;

        private Runnable test;

        TestCases(Runnable runnable) {
            test = runnable;
        }

        public void run() {
            test.run();
        }
    }

    public static void main(String[] args) {
        init();
        for (String arg : args) {
            runTest(arg);
        }
        LWCToolkit.switchKeyboardLayout(initialLayout);
        System.exit(success ? 0 : 1);
    }

    private static void init() {
        try {
            robot = new Robot();
            robot.setAutoDelay(100);
        } catch (AWTException e) {
            e.printStackTrace();
            System.exit(1);
        }

        initialLayout = LWCToolkit.getKeyboardLayoutId();

        frame = new JFrame("InputMethodTest");
        frame.setVisible(true);
        frame.setSize(300, 300);
        frame.setLocationRelativeTo(null);

        textArea = new JTextArea();
        textArea.addKeyListener(new KeyListener() {
            @Override
            public void keyTyped(KeyEvent keyEvent) {}

            @Override
            public void keyPressed(KeyEvent keyEvent) {
                lastKeyCode = keyEvent.getKeyCode();
            }

            @Override
            public void keyReleased(KeyEvent keyEvent) {}
        });

        frame.setLayout(new BorderLayout());
        frame.getContentPane().add(textArea, BorderLayout.CENTER);

        textArea.grabFocus();
        try {
            Thread.sleep(500);
        } catch (InterruptedException ignored) {}
    }

    private static void runTest(String name) {
        currentTest = name;
        TestCases.valueOf(name).run();
    }

    public static void section(String description) {
        currentSection = description;
        textArea.setText("");
        frame.setTitle(currentTest + ": " + description);
    }

    public static void layout(String name) {
        LWCToolkit.switchKeyboardLayout(name);
    }

    public static void type(int key, int modifiers) {
        lastKeyCode = -1;
        List<Integer> modKeys = new ArrayList<>();

        if ((modifiers & InputEvent.ALT_DOWN_MASK) != 0) {
            modKeys.add(KeyEvent.VK_ALT);
        }

        if ((modifiers & InputEvent.CTRL_DOWN_MASK) != 0) {
            modKeys.add(KeyEvent.VK_CONTROL);
        }

        if ((modifiers & InputEvent.SHIFT_DOWN_MASK) != 0) {
            modKeys.add(KeyEvent.VK_SHIFT);
        }

        if ((modifiers & InputEvent.META_DOWN_MASK) != 0) {
            modKeys.add(KeyEvent.VK_META);
        }

        for (var modKey : modKeys) {
            robot.keyPress(modKey);
        }

        robot.keyPress(key);
        robot.keyRelease(key);

        for (var modKey : modKeys) {
            robot.keyRelease(modKey);
        }
    }

    public static void expect(String expectedValue) {
        var actualValue = textArea.getText();
        if (actualValue.equals(expectedValue)) {
            System.out.printf("Test %s (%s) passed, got '%s'\n", currentTest, currentSection, actualValue);
        } else {
            success = false;
            System.out.printf("Test %s (%s) failed, expected '%s', got '%s'\n", currentTest, currentSection, expectedValue, actualValue);
        }
    }

    public static void expectKeyCode(int keyCode) {
        if (lastKeyCode == keyCode) {
            System.out.printf("Test %s (%s) passed, got key code %d\n", currentTest, currentSection, keyCode);
        } else {
            success = false;
            System.out.printf("Test %s (%s) failed, expected key code %d, got %d\n", currentTest, currentSection, keyCode, lastKeyCode);
        }
    }

    public static void expectTrue(boolean value, String comment) {
        if (value) {
            System.out.printf("Test %s (%s) passed: %s\n", currentTest, currentSection, comment);
        } else {
            success = false;
            System.out.printf("Test %s (%s) failed: %s\n", currentTest, currentSection, comment);
        }
    }
}