SwingSet/src/ButtonDemoTest.java

/*
 * Copyright (c) 2011, 2016, 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 com.sun.swingset3.demos.JHyperlink;
import com.sun.swingset3.demos.button.ButtonDemo;
import java.util.concurrent.ArrayBlockingQueue;
import java.util.concurrent.BlockingQueue;
import javax.swing.ButtonModel;
import javax.swing.JButton;
import javax.swing.event.ChangeEvent;
import static org.testng.AssertJUnit.*;
import org.testng.annotations.Test;
import org.jemmy2ext.JemmyExt.ByToolTipChooser;
import static org.jemmy2ext.JemmyExt.EXACT_STRING_COMPARATOR;
import org.netbeans.jemmy.ClassReference;
import org.netbeans.jemmy.operators.JButtonOperator;
import org.netbeans.jemmy.operators.JFrameOperator;
import static com.sun.swingset3.demos.button.ButtonDemo.*;
import org.jemmy2ext.JemmyExt;
import org.jemmy2ext.JemmyExt.MultiThreadedTryCatch;
import static org.jemmy2ext.JemmyExt.captureDebugInfoOnFail;

/*
 * @test
 * @key headful
 * @summary Verifies buttons on SwingSet3 ButtonDemo page by clicking each button
 *          and checking model change events. It also verifies tooltips and text
 *          on buttons before and after click.
 *
 * @library /sanity/client/lib/jemmy/src
 * @library /sanity/client/lib/Jemmy2Ext/src
 * @library /sanity/client/lib/SwingSet3/src
 * @build org.jemmy2ext.JemmyExt
 * @build com.sun.swingset3.demos.button.ButtonDemo
 * @run testng ButtonDemoTest
 */
public class ButtonDemoTest {

    private static final String[] BUTTON_TEXT_AFTER = {
        DO_IT_AGAIN,};

    private static final String[] BUTTON_TEXT_BEFORE = {
        DO_IT,
        "",
        FIND,
        GO,
        CONNECT,
        "",
        GET_MORE_INFO,
        null
    };

    private static final String[] BUTTON_TOOLTIP = {
        SIMPLE_BUTTON,
        IMAGE_BUTTON,
        BUTTON_WITH_TEXT_AND_IMAGE,
        BUTTON_WITH_BACKGROUND_COLOR,
        BUTTON_WITH_NO_BORDER,
        BUTTON_WITH_ROLLOVER_IMAGE,
        JAVA_SE_URL,
        JAVA_BLOGS_URL
    };

    private static final String[] GOLDEN = {
        "isArmed = false, isEnabled = true, isPressed = false, isSelected = false",
        "isArmed = true, isEnabled = true, isPressed = false, isSelected = false",
        "isArmed = true, isEnabled = true, isPressed = true, isSelected = false",
        "isArmed = true, isEnabled = true, isPressed = false, isSelected = false",
        "isArmed = false, isEnabled = true, isPressed = false, isSelected = false"
    };

    @Test
    public void test() throws Exception {

        captureDebugInfoOnFail(() -> {

            new ClassReference(ButtonDemo.class.getCanonicalName()).startApplication();

            JFrameOperator mainFrame = new JFrameOperator(DEMO_TITLE);
            mainFrame.setComparator(EXACT_STRING_COMPARATOR);

            // Check all the buttons
            for (int i = 0; i < BUTTON_TOOLTIP.length; i++) {
                String tooltip = BUTTON_TOOLTIP[i];

                JButtonOperator button = new JButtonOperator(mainFrame, new ByToolTipChooser(tooltip));

                assertEquals(BUTTON_TEXT_BEFORE[i], button.getText());

                // Two buttons are hyperlinks, we don't want to click them
                if (!button.getSource().getClass().equals(JHyperlink.class)) {
                    checkButton(button);
                }

                if (BUTTON_TEXT_AFTER.length > i) {
                    assertEquals(BUTTON_TEXT_AFTER[i], button.getText());
                } else {
                    assertEquals(BUTTON_TEXT_BEFORE[i], button.getText());
                }
            }

        });
    }

    private void checkButton(JButtonOperator button) throws Exception {
        MultiThreadedTryCatch tryCatch = new JemmyExt.MultiThreadedTryCatch();
        try {
            BlockingQueue<String> modelStateChanges = new ArrayBlockingQueue<>(GOLDEN.length);
            button.getQueueTool().invokeAndWait(() -> {
                try {
                    JButton jButton = (JButton) button.getSource();
                    ButtonModel model = jButton.getModel();
                    String line = toString(model);
                    System.out.println("Inital: " + line);
                    modelStateChanges.add(line);
                    model.addChangeListener((ChangeEvent e) -> {
                        try {
                            String line2 = toString(model);
                            System.out.println("ChangeEvent: " + line2);

                            // We are only interested in the first GOLDEN.length events
                            if (modelStateChanges.remainingCapacity() > 0) {
                                modelStateChanges.add(line2);
                            }
                        } catch (RuntimeException | Error t) {
                            tryCatch.register(t);
                        }
                    });
                } catch (Error error) {
                    // All exceptions are already handled by Jemmy but Errors are not
                    tryCatch.register(error);
                    throw error;
                }
            });

            assertEquals("Initial state check", GOLDEN[0], modelStateChanges.take());

            button.clickMouse();

            for (int state = 1; state < GOLDEN.length; state++) {
                assertEquals(GOLDEN[state], modelStateChanges.take());
            }
        } catch (RuntimeException | Error | InterruptedException t) {
            tryCatch.registerRoot(t);
        } finally {
            tryCatch.throwRegistered();
        }
    }

    private static String toString(ButtonModel model) {
        return "isArmed = " + model.isArmed()
                + ", isEnabled = " + model.isEnabled()
                + ", isPressed = " + model.isPressed()
                + ", isSelected = " + model.isSelected();
    }

}