common.MyListener.java Source code

Introduction

Here is the source code for common.MyListener.java

Source

/*******************************************************************************
 * Licensed to the Apache Software Foundation (ASF) under one
 * or more contributor license agreements.  See the NOTICE file
 * distributed with this work for additional information
 * regarding copyright ownership.  The ASF licenses this file
 * to you under the Apache License, Version 2.0 (the
 * "License"); you may not use this file except in compliance
 * with the License.  You may obtain a copy of the License at
 *
 *   http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing,
 * software distributed under the License is distributed on an
 * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
 * KIND, either express or implied.  See the License for the
 * specific language governing permissions and limitations
 * under the License.
 *******************************************************************************/
package common;

import java.io.File;
import java.io.IOException;
import java.io.PrintWriter;
import java.io.StringWriter;
import java.text.DateFormat;
import java.text.SimpleDateFormat;
import java.util.Date;
import java.util.LinkedHashMap;
import java.util.Map;

import org.apache.commons.io.FileUtils;
import org.apache.log4j.Logger;
import org.openqa.selenium.OutputType;
import org.openqa.selenium.TakesScreenshot;
import org.openqa.selenium.WebDriver;
import org.openqa.selenium.remote.Augmenter;
import org.testng.Assert;
import org.testng.ITestContext;
import org.testng.ITestResult;
import org.testng.TestListenerAdapter;

/**
 * 
 * @author Vishshady
 *
 */
public class MyListener extends TestListenerAdapter implements IDriver {
    private static Logger log = Logger.getLogger(MyListener.class);
    Parameters p = new Parameters();
    LinkedHashMap<String, Object[]> testPassed;
    LinkedHashMap<String, Object[]> testFailed;
    LinkedHashMap<String, Object[]> testSkipped;
    LinkedHashMap<String, StringBuilder> testLogs;
    private ITestContext testContext = null;
    String t = null;
    long time = 0;
    int totalTime;

    @Override
    public void onStart(ITestContext context) {
        totalTime = (int) System.currentTimeMillis();
        testContext = context;
        testPassed = new LinkedHashMap<String, Object[]>();
        testFailed = new LinkedHashMap<String, Object[]>();
        testSkipped = new LinkedHashMap<String, Object[]>();
        testLogs = new LinkedHashMap<String, StringBuilder>();
    }

    @Override
    public void onFinish(ITestContext context) {
        totalTime = (int) (System.currentTimeMillis() - totalTime);
        totalTime = totalTime / 1000;
        CustomReport reporter = new CustomReport(p);
        try {
            reporter.startReport();
            reporter.writeExecutionSummary(testContext, timeConvertor(totalTime));
            reporter.writeExecutionDetail(testPassed, testFailed, testSkipped, testLogs);
            reporter.finishReport();
        } catch (IOException e) {
            e.printStackTrace();
        }
    }

    @Override
    public synchronized void onTestFailure(ITestResult tr) {
        try {
            time = tr.getEndMillis() - tr.getStartMillis();
            time = time / 1000;
            String testName = null;
            DateFormat dateFormat = new SimpleDateFormat("dd_MMM_yyyy__hh_mm_ssaa");
            String destFile = null;
            for (Map.Entry<String, String> entry : MyTestContext.getTestStats().entrySet()) {
                testName = entry.getKey();
                destFile = testName + "_" + dateFormat.format(new Date()) + ".png";
                String[] err = null;
                err = tr.getThrowable().getMessage().split("\\n");
                err[0] = err[0] + "\n. Screen shot saved at /screenshots/" + destFile;
                testFailed.put(testName, new Object[] { entry.getValue(), err[0], MyTestContext.getStep(), time });
                t = entry.getKey();
            }
            addTestLogs(t);
            MyTestContext.testStats.remove();
            WebDriver driver = new Augmenter().augment(driver());
            File scrFile = ((TakesScreenshot) driver).getScreenshotAs(OutputType.FILE);
            String destDir = "screenshots";
            new File(destDir).mkdirs();
            FileUtils.copyFile(scrFile, new File(destDir + "/" + destFile));
        } catch (Exception e) {
            StringWriter s = new StringWriter();
            PrintWriter p = new PrintWriter(s, true);
            e.printStackTrace(p);
            log.error(s.getBuffer().toString());
            Assert.fail("Check log file for error details");
        }
    }

    @Override
    public void onTestSkipped(ITestResult tr) {
        time = tr.getEndMillis() - tr.getStartMillis();
        time = time / 1000;
        for (Map.Entry<String, String> entry : MyTestContext.getTestStats().entrySet()) {
            testSkipped.put(entry.getKey(), new Object[] { entry.getValue(), tr.getThrowable(), time });
            t = entry.getKey();
        }
        addTestLogs(t);
        MyTestContext.testStats.remove();
    }

    @Override
    public void onTestSuccess(ITestResult tr) {
        time = tr.getEndMillis() - tr.getStartMillis();
        time = time / 1000;
        for (Map.Entry<String, String> entry : MyTestContext.getTestStats().entrySet()) {
            testPassed.put(entry.getKey(), new Object[] { entry.getValue(), time });
            t = entry.getKey();
        }
        addTestLogs(t);
        MyTestContext.testStats.remove();
    }

    public void addTestLogs(String s) {
        testLogs.put(s, MyTestContext.getMessages());
        MyTestContext.messages.remove();
    }

    public String timeConvertor(int time) {
        int secs = time % 60;
        int totalMins = time / 60;
        int mins = totalMins % 60;
        int hrs = totalMins / 60;
        return (hrs + "H : " + mins + "M : " + secs + "S");
    }

}