From 6a93395c8a9bf3210aae1fe73f1ab7958ec71c53 Mon Sep 17 00:00:00 2001
From: YardenNaveh <yardennaveh@gmail.com>
Date: Tue, 10 Oct 2017 17:35:26 -0700
Subject: [PATCH] Create CompareScreens.java

Code Example for use:

WebDriver driver = new ChromeDriver();

CompareScreens comparingScreens = new CompareScreens("API_KEY", "appName", "testName", new RectangleSize(width, height));

driver.get("https://applitools.com/helloworld");
comparingScreens.setScreenshotAsBaseline(driver);
driver.get("https://applitools.com/helloworld?diff2");
comparingScreens.compareScreenshotAgainstBaseline(driver);

driver.close();
---
 CompareScreens.java | 114 ++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 114 insertions(+)
 create mode 100644 CompareScreens.java

diff --git a/CompareScreens.java b/CompareScreens.java
new file mode 100644
index 0000000..3fa56be
--- /dev/null
+++ b/CompareScreens.java
@@ -0,0 +1,114 @@
+package test;
+
+import java.rmi.UnexpectedException;
+
+import org.openqa.selenium.WebDriver;
+
+import com.applitools.eyes.BatchInfo;
+import com.applitools.eyes.selenium.Eyes;
+import com.applitools.eyes.RectangleSize;
+import com.applitools.eyes.selenium.StitchMode;
+import com.applitools.eyes.TestResults;
+
+public class CompareScreens {
+	
+	// Private Fields
+	private Eyes eyes;
+	
+	private BatchInfo baselinesBatch;
+	private BatchInfo comparisonsBatch;
+	
+	private String appName;
+	private String testName;
+	private RectangleSize viewportSize;
+	
+	private boolean isBaselineSet = false;
+	
+	private int numberOfMisMatches = 0;
+	private int numberOfMatches = 0;
+
+	// Getters
+	public int getNumberOfMisMatches() {
+		return numberOfMisMatches;
+	}
+
+	public int getNumberOfMatches() {
+		return numberOfMatches;
+	}
+	
+	public Eyes getEyes()
+	{
+		return this.eyes;
+	}
+
+	// Constructors
+	public CompareScreens(String apiKey, String baselinesBatchName, String comparisonsBatchName, String appName, String testName,
+			RectangleSize viewportSize) {
+
+		eyes = new Eyes();
+
+		eyes.setForceFullPageScreenshot(true);
+		eyes.setStitchMode(StitchMode.CSS);
+		eyes.setApiKey(apiKey);
+
+		this.baselinesBatch = new BatchInfo(baselinesBatchName);
+		this.comparisonsBatch = new BatchInfo(comparisonsBatchName);
+
+		int baselinesBatchID = (int) (Integer.MAX_VALUE * Math.random());
+		int comparisonsBatchID = (int) (Integer.MAX_VALUE * Math.random());
+
+		this.baselinesBatch.setId(Integer.toString(baselinesBatchID));
+		this.comparisonsBatch.setId(Integer.toString(comparisonsBatchID));
+
+		this.appName = appName;
+		this.testName = testName;
+		this.viewportSize = viewportSize;
+
+	}
+
+	public CompareScreens(String apiKey, String appName, String testName, RectangleSize viewportSize) {
+		this(apiKey, "Compare Screens - Baselines", "Compare Screens - comparisons", appName, testName, viewportSize);
+	}
+
+	// Public Methods
+	public void setScreenshotAsBaseline(WebDriver driver) {
+
+		eyes.setSaveFailedTests(true);
+		eyes.setBatch(this.baselinesBatch);
+
+		if (validatePage(driver) != null)
+			this.isBaselineSet = true;
+	}
+
+	public void compareScreenshotAgainstBaseline(WebDriver driver) throws UnexpectedException {
+
+		if (!this.isBaselineSet)
+			throw new UnexpectedException(
+					"A baseline was not set for comparison. Please call the setBaseline method first.");
+		
+		eyes.setSaveFailedTests(false);
+		eyes.setBatch(this.comparisonsBatch);
+
+		TestResults testResults = validatePage(driver);
+
+		this.numberOfMisMatches += testResults.getMismatches();
+		this.numberOfMatches += testResults.getMatches();
+	}
+
+	// Private Methods
+	private TestResults validatePage(WebDriver driver) {
+
+		TestResults testResults = null;
+
+		try {
+			eyes.open(driver, this.appName, this.testName, this.viewportSize);
+			eyes.checkWindow();
+			testResults = eyes.close(false);
+
+		} finally {
+			eyes.abortIfNotClosed();
+		}
+
+		return testResults;
+	}
+}