PrintScreen code samples moved to core-java module

core-java/src/main/java/com/baeldung/printscreen/Screenshot.java

@@ -0,0 +1,38 @@
+package com.baeldung.printscreen;
+
+import javax.imageio.ImageIO;
+import java.awt.*;
+import java.awt.image.BufferedImage;
+import java.io.File;
+
+public class Screenshot {
+
+    private String filePath;
+    private String filenamePrefix;
+    private String fileType;
+    private int timeToWait;
+
+    public Screenshot(String filePath, String filenamePrefix,
+                      String fileType, int timeToWait) {
+        this.filePath = filePath;
+        this.filenamePrefix = filenamePrefix;
+        this.fileType = fileType;
+        this.timeToWait = timeToWait;
+    }
+
+    public void getScreenshot() throws Exception {
+        Thread.sleep(timeToWait);
+        Rectangle rectangle = new Rectangle(Toolkit.getDefaultToolkit().getScreenSize());
+        Robot robot = new Robot();
+        BufferedImage img = robot.createScreenCapture(rectangle);
+        ImageIO.write(img, fileType, setupFileNamePath());
+    }
+
+    private File setupFileNamePath() {
+        return new File(filePath + filenamePrefix + "." + fileType);
+    }
+
+    private Rectangle getScreenSizedRectangle(final Dimension d) {
+        return new Rectangle(0, 0, d.width, d.height);
+    }
+}

core-java/src/test/java/com/baeldung/printscreen/ScreenshotTest.java

@@ -0,0 +1,39 @@
+package com.baeldung.printscreen;
+
+import org.junit.After;
+import org.junit.Before;
+import org.junit.Test;
+
+import java.io.File;
+
+import static org.junit.Assert.*;
+
+
+public class ScreenshotTest {
+
+    private Screenshot screenshot;
+    private String filePath;
+    private String fileName;
+    private String fileType;
+    private File file;
+
+    @Before
+    public void setUp() throws Exception {
+        filePath = "";
+        fileName = "Screenshot";
+        fileType = "jpg";
+        file = new File(filePath + fileName + "." + fileType);
+        screenshot = new Screenshot(filePath, fileName, fileType, 2000);
+    }
+
+    @Test
+    public void testGetScreenshot() throws Exception {
+        screenshot.getScreenshot();
+        assertTrue(file.exists());
+    }
+
+    @After
+    public void tearDown() throws Exception {
+        file.delete();
+    }
+}