Merge pull request #11 from shivam198smart/master

Added Screenshot Functionality

Author: shankybnl

src/main/java/UITestFramework/CreateSession.java

@@ -20,9 +20,12 @@
 import org.apache.commons.exec.ExecuteException;
 import org.apache.log4j.PropertyConfigurator;
 import org.openqa.selenium.WebDriver;
+import org.openqa.selenium.WebDriverException;
 import org.openqa.selenium.remote.DesiredCapabilities;
 import org.testng.annotations.AfterMethod;
+import org.testng.annotations.AfterSuite;
 import org.testng.annotations.BeforeMethod;
+import org.testng.annotations.BeforeSuite;
 import org.testng.annotations.Parameters;
 
 
@@ -43,8 +46,14 @@ public class CreateSession  {
 	protected File file = new File("");
 	Properties configProp = new Properties();
 	String OS;
-
-
+	/** 
+	 * Returns the instance of the webdriver. 
+	 * @return webdriver instance 
+	 */
+	public WebDriver getDriver() {
+        return driver;
+}
+	
 
 	/** 
 	 * this method starts Appium server. Calls startAppiumServer method to start the session depending upon your OS.
@@ -143,7 +152,7 @@ public synchronized void androidDriver(String buildPath, Method methodName) thro
 		capabilities.setCapability("app", app.getAbsolutePath());
 		capabilities.setCapability(MobileCapabilityType.NO_RESET, false);
 		capabilities.setCapability("automationName", "UiAutomator2");
-		driver = new AndroidDriver( new URL("http://localhost:4723/wd/hub"), capabilities);
+		driver = new AndroidDriver( new URL("http://127.0.0.1:4723/wd/hub"), capabilities);
 
 	}
 

src/main/java/UITestFramework/retryLogic/TestListener.java

@@ -1,14 +1,25 @@
 package UITestFramework.retryLogic;
 
 
+import java.io.File;
+import java.io.IOException;
+import java.text.SimpleDateFormat;
+import java.util.Date;
 import java.util.Set;
+
+import org.apache.commons.io.FileUtils;
+import org.openqa.selenium.OutputType;
+import org.openqa.selenium.TakesScreenshot;
+import org.openqa.selenium.WebDriver;
 import org.testng.ITestContext;
 import org.testng.ITestListener;
 import org.testng.ITestNGMethod;
 import org.testng.ITestResult;
+import UITestFramework.CreateSession;
 
-public class TestListener implements ITestListener {
-	@Override
+public class TestListener extends CreateSession implements ITestListener {
+	
+	
 	public void onFinish(ITestContext context) {
 		Set<ITestResult> failedTests = context.getFailedTests().getAllResults();
 		for (ITestResult temp : failedTests) {
@@ -23,20 +34,37 @@ public void onFinish(ITestContext context) {
 		}
 	}
 
-	public void onTestStart(ITestResult result) { 
-
-	}
+	public void onTestStart(ITestResult result) {   }
 
 	public void onTestSuccess(ITestResult result) {   }
-
-	public void onTestFailure(ITestResult result) {   }
+   
+	@Override
+	public void onTestFailure(ITestResult result) { 
+		Object TestListener = result.getInstance();
+		WebDriver webDriver = ((CreateSession) TestListener).getDriver();
+		if (webDriver != null)
+		{
+			File scr = ((TakesScreenshot)webDriver).getScreenshotAs(OutputType.FILE);		
+			String filename =  new SimpleDateFormat("yyyyMMddhhmmss'.jpg'").format(new Date());
+    			File dest = new File("./Screenshots/" + filename); //Directory where Screenshot get saved.
+   			try 
+			{
+				FileUtils.copyFile(scr, dest);
+			} 
+			catch (IOException e) {
+				// TODO Auto-generated catch block
+				e.printStackTrace();
+			}	
+		}
+	   }
 
 	public void onTestSkipped(ITestResult result) {   }
 
 	public void onTestFailedButWithinSuccessPercentage(ITestResult result) {   }
 
-	public void onStart(ITestContext context) { 
-	}
-}  
+	public void onStart(ITestContext context) { }
+
+}
+
 
 

testng.xml

@@ -1,10 +1,12 @@
 <!DOCTYPE suite SYSTEM "http://testng.org/testng-1.0.dtd" >
 <suite name="mobile-automation-framework">
 	<parameter name="os" value="${os}" />
-
+	<listeners>
+	<listener class-name="UITestFramework.retryLogic.TestListener"/> 
+	</listeners>
 	<test name="automation-tests">
 		<packages>
-			<package name="tests.testngTests" />
+			<package name="tests.testngTests" />			
 		</packages>
 	</test>
-</suite>
+</suite>