1 /*
   2  * Copyright (c) 2018, Oracle and/or its affiliates. All rights reserved.
   3  * 
   4  * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
   5  *
   6  * The contents of this file are subject to the terms of either the Universal Permissive License
   7  * v 1.0 as shown at http://oss.oracle.com/licenses/upl
   8  *
   9  * or the following license:
  10  *
  11  * Redistribution and use in source and binary forms, with or without modification, are permitted
  12  * provided that the following conditions are met:
  13  * 
  14  * 1. Redistributions of source code must retain the above copyright notice, this list of conditions
  15  * and the following disclaimer.
  16  * 
  17  * 2. Redistributions in binary form must reproduce the above copyright notice, this list of
  18  * conditions and the following disclaimer in the documentation and/or other materials provided with
  19  * the distribution.
  20  * 
  21  * 3. Neither the name of the copyright holder nor the names of its contributors may be used to
  22  * endorse or promote products derived from this software without specific prior written permission.
  23  * 
  24  * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" AND ANY EXPRESS OR
  25  * IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND
  26  * FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR
  27  * CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
  28  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
  29  * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY,
  30  * WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY
  31  * WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
  32  */
  33 package org.openjdk.jmc.test.jemmy;
  34 
  35 import java.awt.AWTException;
  36 import java.awt.Rectangle;
  37 import java.awt.Robot;
  38 import java.awt.Toolkit;
  39 import java.awt.image.BufferedImage;
  40 import java.io.File;
  41 import java.io.IOException;
  42 
  43 import javax.imageio.ImageIO;
  44 
  45 import org.eclipse.swt.widgets.Display;
  46 
  47 import org.openjdk.jmc.ui.MCPathEditorInput;
  48 import org.openjdk.jmc.ui.WorkbenchToolkit;
  49 
  50 public class TestHelper {
  51 
  52         public static File createResultFile(String prefix, String suffix, boolean deleteTempOnExit) throws IOException {
  53                 String resultDir = System.getProperty("results.dir"); //$NON-NLS-1$
  54                 File resultFile;
  55                 if (resultDir != null) {
  56                         resultFile = new File(resultDir, prefix + '.' + System.currentTimeMillis() + '.' + suffix);
  57                 } else {
  58                         resultFile = File.createTempFile(prefix, suffix);
  59                         if (deleteTempOnExit) {
  60                                 resultFile.deleteOnExit();
  61                         }
  62                 }
  63                 return resultFile;
  64         }
  65 
  66         public static void dumpScreen(String name) {
  67 
  68                 Robot robot;
  69                 try {
  70                         robot = new Robot();
  71                         BufferedImage screenShot = robot
  72                                         .createScreenCapture(new Rectangle(Toolkit.getDefaultToolkit().getScreenSize()));
  73                         ImageIO.write(screenShot, "PNG", TestHelper.createResultFile(name, "dump.png", false)); //$NON-NLS-1$ //$NON-NLS-2$
  74                 } catch (IOException e) {
  75                         e.printStackTrace();
  76                 } catch (AWTException e) {
  77                         e.printStackTrace();
  78                 }
  79         }
  80 
  81         public static void openJfr(final String fileName) {
  82                 Display.getDefault().syncExec(new Runnable() {
  83                         @Override
  84                         public void run() {
  85                                 WorkbenchToolkit.openEditor(new MCPathEditorInput(new File(fileName)));
  86                         }
  87                 });
  88         }
  89 
  90 }