test/jdk/javax/swing/plaf/aqua/CustomComboBoxFocusTest.java - platform/libcore - Git at Google

 /*
  * Copyright (c) 2015, 2017, 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.
  */

 /**
  * @test
  * @key headful
  * @bug     8073001 8081764
  * @summary Test verifies that combo box with custom editor renders
  *          focus ring around arrow button correctly.
  * @library /lib/testlibrary
  * @build jdk.testlibrary.OSInfo
  * @run     main CustomComboBoxFocusTest
  */

 import java.awt.AWTException;
 import java.awt.Component;
 import java.awt.GridLayout;
 import java.awt.Point;
 import java.awt.Rectangle;
 import java.awt.Robot;
 import java.awt.event.ActionListener;
 import java.awt.event.FocusEvent;
 import java.awt.event.FocusListener;
 import java.awt.image.BufferedImage;
 import java.io.File;
 import java.io.IOException;
 import java.lang.reflect.InvocationTargetException;
 import java.util.concurrent.CountDownLatch;
 import javax.imageio.ImageIO;
 import javax.swing.ComboBoxEditor;
 import javax.swing.JComboBox;
 import javax.swing.JFrame;
 import javax.swing.JLabel;
 import javax.swing.JPanel;
 import javax.swing.JTextField;
 import javax.swing.SwingUtilities;
 import jdk.testlibrary.OSInfo;

 public class CustomComboBoxFocusTest {

     private static CustomComboBoxFocusTest test = null;

     public static void main(String[] args) {
         if (OSInfo.getOSType() != OSInfo.OSType.MACOSX) {
             System.out.println("Only Mac platform test. Test is skipped for other OS.");
             return;
         }

         try {
             SwingUtilities.invokeAndWait(new Runnable() {
                 public void run() {
                     test = new CustomComboBoxFocusTest();
                 }
             });
         } catch (InterruptedException | InvocationTargetException e ) {
             throw new RuntimeException("Test failed.", e);
         }

         SwingUtilities.invokeLater(test.init);

         try {
             System.out.println("Wait for screenshots...");
             test.testDone.await();
         } catch (InterruptedException e) {
             throw new RuntimeException("Test failed.", e);
         }
         System.out.println("Compare screenshots...");
         if (!test.match()) {
             throw new RuntimeException("Test failed.");
         }
         System.out.println("Test passed.");
     }

     private final JComboBox<String> ref = new JComboBox<String>() {
         public String toString() {
             return "reference";
         }
     };

     private final JComboBox<String> custom = new JComboBox<String>() {
         public String toString() {
             return "custom";
         }
     };

     private final JFrame frame;

     private CountDownLatch testDone = new CountDownLatch(1);

     private Robot robot;

     public CustomComboBoxFocusTest() {
         frame = new JFrame(System.getProperty("java.version"));

         try {
             robot = new Robot(frame.getGraphicsConfiguration().getDevice());

         } catch (AWTException e) {
             throw new RuntimeException("Test failed.", e);
         }
     }

     private boolean match() {
         final BufferedImage a = captureRef.img;
         final BufferedImage b = captureCustom.img;

         final int w = Math.min(a.getWidth(), b.getWidth());
         final int h = Math.min(a.getHeight(), b.getHeight());

         for (int y = 0; y < h; y++) {
             for (int x = 0; x < w; x++) {
                 if (a.getRGB(x, y) != b.getRGB(x, y)) {
                     return false;
                 }
             }
         }
         return true;
     }

     private JComboBox<String> getReference() {
         return ref;
     }

     private JComboBox<String> getCustom() {
         return custom;
     }

     private JFrame getFrame() {
         return frame;
     }

     private static abstract class Step implements Runnable {
         final public void run() {
             doStep();

             final Step next = nextStep();

             if (next != null) {
                 SwingUtilities.invokeLater(next);
             }
         }

         public abstract void doStep();

         public abstract Step nextStep();
     }

     private final Step init = new Step() {
         public void doStep() {
             final JFrame f = getFrame();

             final JPanel p = new JPanel(new GridLayout(4, 1, 20, 20));

             JComboBox<String> r = getReference();
             r.setEditable(true);
             r.addItem("One");

             JComboBox<String> c = getCustom();
             c.setEditable(true);
             c.addItem("One");
             final ComboBoxEditor e = new ComboBoxEditor() {
                 private JTextField text = new JTextField();

                 @Override
                 public Component getEditorComponent() {
                     return this.text;
                 }

                 @Override
                 public void setItem(Object o) {
                     text.setText(o == null ? "" : o.toString());
                 }

                 @Override
                 public Object getItem() {
                     return text.getText();
                 }

                 @Override
                 public void selectAll() {
                     text.selectAll();
                 }

                 @Override
                 public void addActionListener(ActionListener actionListener) {
                     text.addActionListener(actionListener);
                 }

                 @Override
                 public void removeActionListener(ActionListener actionListener) {
                     text.removeActionListener(actionListener);
                 }
             };
             c.setEditor(e);

             p.add(new JLabel("Reference"));
             p.add(r);
             p.add(c);
             p.add(new JLabel("Custom"));

             f.add(p);

             f.pack();
             f.setVisible(true);
         }

         public Step nextStep() {
             return focusRef;
         }
     };

     private class FocusStep extends Step {
         private final JComboBox<String> target;
         private final Step focusHandler;
         private final Step next;

         public FocusStep(JComboBox<String> t, Step h, Step n) {
             target = t;
             focusHandler = h;
             next = n;
         }

         public void doStep() {
             System.out.println("Request focus on " + target);
             final Component c = target.getEditor().getEditorComponent();

             c.addFocusListener(new FocusListener() {
                 @Override
                 public void focusGained(FocusEvent e) {
                     SwingUtilities.invokeLater(focusHandler);
                 }

                 @Override
                 public void focusLost(FocusEvent e) {

                 }
             });

             c.requestFocus();

         }

         public Step nextStep() {
             return next;
         }
     }


     private class CaptureStep extends Step {
         private final JComboBox<String> target;
         private BufferedImage img;
         private String fname;
         private final Step next;

         private final int timeout = 2000;

         public CaptureStep(JComboBox<String> t, String name, Step n) {
             target = t;
             next = n;
             fname = name;
         }

         public void doStep() {
             try {
                 Thread.sleep(timeout);
             } catch (InterruptedException e) {
             }
             System.out.println("Capture sceeenshot of " + target);

             Rectangle bounds = target.getBounds();
             Point p = target.getLocationOnScreen();
             System.out.println("Target bounds: " + bounds);
             System.out.println("Target location: " + p);

             bounds.x = p.x;
             bounds.y = p.y;

             img = robot.createScreenCapture(bounds);

             try {
                 ImageIO.write(img, "PNG", new File(fname + ".png"));
             } catch (IOException ioe) {
                 ioe.printStackTrace();
             }

         }

         public Step nextStep() {
             return next;
         }
     }

     private final Step done = new Step() {
         public void doStep() {
             JFrame f = getFrame();
             if (f != null) {
                 f.dispose();
             }
             System.out.println("Done");

             testDone.countDown();
         }

         public Step nextStep() {
             return null;
         }
     };

     private final CaptureStep captureCustom = new CaptureStep(getCustom(), "cb_custom", done);

     private final FocusStep focusCustom = new FocusStep(getCustom(), captureCustom, null);

     private final CaptureStep captureRef = new CaptureStep(getReference(), "cb_ref", focusCustom);

     private final FocusStep focusRef = new FocusStep(getReference(), captureRef, null);
 }
	/*
	* Copyright (c) 2015, 2017, 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.
	*/

	/**
	* @test
	* @key headful
	* @bug 8073001 8081764
	* @summary Test verifies that combo box with custom editor renders
	* focus ring around arrow button correctly.
	* @library /lib/testlibrary
	* @build jdk.testlibrary.OSInfo
	* @run main CustomComboBoxFocusTest
	*/

	import java.awt.AWTException;
	import java.awt.Component;
	import java.awt.GridLayout;
	import java.awt.Point;
	import java.awt.Rectangle;
	import java.awt.Robot;
	import java.awt.event.ActionListener;
	import java.awt.event.FocusEvent;
	import java.awt.event.FocusListener;
	import java.awt.image.BufferedImage;
	import java.io.File;
	import java.io.IOException;
	import java.lang.reflect.InvocationTargetException;
	import java.util.concurrent.CountDownLatch;
	import javax.imageio.ImageIO;
	import javax.swing.ComboBoxEditor;
	import javax.swing.JComboBox;
	import javax.swing.JFrame;
	import javax.swing.JLabel;
	import javax.swing.JPanel;
	import javax.swing.JTextField;
	import javax.swing.SwingUtilities;
	import jdk.testlibrary.OSInfo;

	public class CustomComboBoxFocusTest {

	private static CustomComboBoxFocusTest test = null;

	public static void main(String[] args) {
	if (OSInfo.getOSType() != OSInfo.OSType.MACOSX) {
	System.out.println("Only Mac platform test. Test is skipped for other OS.");
	return;
	}

	try {
	SwingUtilities.invokeAndWait(new Runnable() {
	public void run() {
	test = new CustomComboBoxFocusTest();
	}
	});
	} catch (InterruptedException \| InvocationTargetException e ) {
	throw new RuntimeException("Test failed.", e);
	}

	SwingUtilities.invokeLater(test.init);

	try {
	System.out.println("Wait for screenshots...");
	test.testDone.await();
	} catch (InterruptedException e) {
	throw new RuntimeException("Test failed.", e);
	}
	System.out.println("Compare screenshots...");
	if (!test.match()) {
	throw new RuntimeException("Test failed.");
	}
	System.out.println("Test passed.");
	}

	private final JComboBox<String> ref = new JComboBox<String>() {
	public String toString() {
	return "reference";
	}
	};

	private final JComboBox<String> custom = new JComboBox<String>() {
	public String toString() {
	return "custom";
	}
	};

	private final JFrame frame;

	private CountDownLatch testDone = new CountDownLatch(1);

	private Robot robot;

	public CustomComboBoxFocusTest() {
	frame = new JFrame(System.getProperty("java.version"));

	try {
	robot = new Robot(frame.getGraphicsConfiguration().getDevice());

	} catch (AWTException e) {
	throw new RuntimeException("Test failed.", e);
	}
	}

	private boolean match() {
	final BufferedImage a = captureRef.img;
	final BufferedImage b = captureCustom.img;

	final int w = Math.min(a.getWidth(), b.getWidth());
	final int h = Math.min(a.getHeight(), b.getHeight());

	for (int y = 0; y < h; y++) {
	for (int x = 0; x < w; x++) {
	if (a.getRGB(x, y) != b.getRGB(x, y)) {
	return false;
	}
	}
	}
	return true;
	}

	private JComboBox<String> getReference() {
	return ref;
	}

	private JComboBox<String> getCustom() {
	return custom;
	}

	private JFrame getFrame() {
	return frame;
	}

	private static abstract class Step implements Runnable {
	final public void run() {
	doStep();

	final Step next = nextStep();

	if (next != null) {
	SwingUtilities.invokeLater(next);
	}
	}

	public abstract void doStep();

	public abstract Step nextStep();
	}

	private final Step init = new Step() {
	public void doStep() {
	final JFrame f = getFrame();

	final JPanel p = new JPanel(new GridLayout(4, 1, 20, 20));

	JComboBox<String> r = getReference();
	r.setEditable(true);
	r.addItem("One");

	JComboBox<String> c = getCustom();
	c.setEditable(true);
	c.addItem("One");
	final ComboBoxEditor e = new ComboBoxEditor() {
	private JTextField text = new JTextField();

	@Override
	public Component getEditorComponent() {
	return this.text;
	}

	@Override
	public void setItem(Object o) {
	text.setText(o == null ? "" : o.toString());
	}

	@Override
	public Object getItem() {
	return text.getText();
	}

	@Override
	public void selectAll() {
	text.selectAll();
	}

	@Override
	public void addActionListener(ActionListener actionListener) {
	text.addActionListener(actionListener);
	}

	@Override
	public void removeActionListener(ActionListener actionListener) {
	text.removeActionListener(actionListener);
	}
	};
	c.setEditor(e);

	p.add(new JLabel("Reference"));
	p.add(r);
	p.add(c);
	p.add(new JLabel("Custom"));

	f.add(p);

	f.pack();
	f.setVisible(true);
	}

	public Step nextStep() {
	return focusRef;
	}
	};

	private class FocusStep extends Step {
	private final JComboBox<String> target;
	private final Step focusHandler;
	private final Step next;

	public FocusStep(JComboBox<String> t, Step h, Step n) {
	target = t;
	focusHandler = h;
	next = n;
	}

	public void doStep() {
	System.out.println("Request focus on " + target);
	final Component c = target.getEditor().getEditorComponent();

	c.addFocusListener(new FocusListener() {
	@Override
	public void focusGained(FocusEvent e) {
	SwingUtilities.invokeLater(focusHandler);
	}

	@Override
	public void focusLost(FocusEvent e) {

	}
	});

	c.requestFocus();

	}

	public Step nextStep() {
	return next;
	}
	}


	private class CaptureStep extends Step {
	private final JComboBox<String> target;
	private BufferedImage img;
	private String fname;
	private final Step next;

	private final int timeout = 2000;

	public CaptureStep(JComboBox<String> t, String name, Step n) {
	target = t;
	next = n;
	fname = name;
	}

	public void doStep() {
	try {
	Thread.sleep(timeout);
	} catch (InterruptedException e) {
	}
	System.out.println("Capture sceeenshot of " + target);

	Rectangle bounds = target.getBounds();
	Point p = target.getLocationOnScreen();
	System.out.println("Target bounds: " + bounds);
	System.out.println("Target location: " + p);

	bounds.x = p.x;
	bounds.y = p.y;

	img = robot.createScreenCapture(bounds);

	try {
	ImageIO.write(img, "PNG", new File(fname + ".png"));
	} catch (IOException ioe) {
	ioe.printStackTrace();
	}

	}

	public Step nextStep() {
	return next;
	}
	}

	private final Step done = new Step() {
	public void doStep() {
	JFrame f = getFrame();
	if (f != null) {
	f.dispose();
	}
	System.out.println("Done");

	testDone.countDown();
	}

	public Step nextStep() {
	return null;
	}
	};

	private final CaptureStep captureCustom = new CaptureStep(getCustom(), "cb_custom", done);

	private final FocusStep focusCustom = new FocusStep(getCustom(), captureCustom, null);

	private final CaptureStep captureRef = new CaptureStep(getReference(), "cb_ref", focusCustom);

	private final FocusStep focusRef = new FocusStep(getReference(), captureRef, null);
	}