// Copyright (c) 2008-2009 Nokia Corporation and/or its subsidiary(-ies).

// All rights reserved.

// This component and the accompanying materials are made available

// under the terms of "Eclipse Public License v1.0"

// which accompanies this distribution, and is available

// at the URL "http://www.eclipse.org/legal/epl-v10.html".

//

// Initial Contributors:

// Nokia Corporation - initial contribution.

//

// Contributors:

//

// Description:

//

package com.symbian.pde.test.utils;

import java.io.File;

import java.io.FileNotFoundException;

import java.io.FileOutputStream;

import junit.framework.AssertionFailedError;

import junit.framework.TestCase;

import junit.framework.TestResult;

import org.apache.tools.ant.taskdefs.optional.junit.JUnitTest;

import org.apache.tools.ant.taskdefs.optional.junit.XMLJUnitResultFormatter;

import org.eclipse.jdt.internal.junit.model.ITestRunListener2;

public class PDETestListener implements ITestRunListener2 {

    private Object resultsCollector;

    private int totalNumberOfTests;

    private int testsRunCount;

    private int numberOfTestsPassed;

    private int numberOfTestsFailed;

    private int numberOfTestsWithError;

    private boolean testRunEnded = false;

    private XMLJUnitResultFormatter xmlResultsFormatter;

    private File outputFile;

    private String suiteName;

    private JUnitTest junitTestSuite;

    private TestCase currentTest;

    public PDETestListener(Object collector, String suite) {

        resultsCollector = collector;

        suiteName = suite;

        junitTestSuite = new JUnitTest(suiteName);

        junitTestSuite.setProperties(System.getProperties());

    }

    public void setOutputFile(String filename) {

        outputFile = new File(filename);

    }

    public File getOutputFile() {

        if (outputFile == null) {

            setOutputFile("TEST-" + suiteName + ".xml");

        }

        return outputFile;

    }

    public boolean failed() {

        return ((numberOfTestsFailed + numberOfTestsWithError) > 0) || (testRunEnded && (testsRunCount == 0));

    }

    public int count() {

        return testsRunCount;

    }

    private XMLJUnitResultFormatter getXMLJUnitResultFormatter() {

        if (xmlResultsFormatter == null) {

            xmlResultsFormatter = new XMLJUnitResultFormatter();

            try {

                xmlResultsFormatter.setOutput(new FileOutputStream(getOutputFile()));

            } catch (FileNotFoundException e) {

                e.printStackTrace();

            }

        }

        return xmlResultsFormatter;

    }

    public synchronized void testRunStarted(int testCount) {

        totalNumberOfTests = testCount;

        testsRunCount = 0;

        numberOfTestsPassed = 0;

        numberOfTestsFailed = 0;

        numberOfTestsWithError = 0;

        testRunEnded = false;

        getXMLJUnitResultFormatter().startTestSuite(junitTestSuite);

        System.out.println("PDE Test Run Started - running " + totalNumberOfTests + " tests ...");

    }

    public synchronized void testRunEnded(long elapsedTime) {

        testRunEnded = true;

        junitTestSuite.setCounts(testsRunCount, numberOfTestsFailed, numberOfTestsWithError);

        junitTestSuite.setRunTime(elapsedTime);

        getXMLJUnitResultFormatter().endTestSuite(junitTestSuite);

        System.out.println("Test Run Ended   - " + (failed() ? "FAILED" : "PASSED") + " - Total: " + totalNumberOfTests

                + " (Errors: " + numberOfTestsWithError

                + ", Failed: " + numberOfTestsFailed

                + ", Passed: " + numberOfTestsPassed + "), duration " + elapsedTime + "ms.");

        synchronized (resultsCollector) {

            resultsCollector.notifyAll();

        }

    }

    public synchronized void testRunStopped(long elapsedTime) {

        System.out.println("Test Run Stopped");

        testRunEnded(elapsedTime);

    }

    public synchronized void testRunTerminated() {

        System.out.println("Test Run Terminated");

        testRunEnded(0);

    }

    public synchronized void testStarted(String testId, String testName) {

        testsRunCount++;

        currentTest = new WrapperTestCase(testName);

        getXMLJUnitResultFormatter().startTest(currentTest);

        System.out.println("  Test Started - " + count() + " - " + testName);

    }

    public synchronized void testEnded(String testId, String testName) {

        numberOfTestsPassed = count() - (numberOfTestsFailed + numberOfTestsWithError);

        getXMLJUnitResultFormatter().endTest(currentTest);

        System.out.println("  Test Ended   - " + count() + " - " + testName);

    }

    public synchronized void testFailed(int status, String testId, String testName, String trace, String expected, String actual) {

        String statusMessage = String.valueOf(status);

        if (status == ITestRunListener2.STATUS_OK) {

            numberOfTestsPassed++;

            statusMessage = "OK";

        } else if (status == ITestRunListener2.STATUS_FAILURE) {

            numberOfTestsFailed++;

            statusMessage = "FAILED";

            getXMLJUnitResultFormatter().addFailure(currentTest, new AssertionFailedError(trace));

        } else if (status == ITestRunListener2.STATUS_ERROR) {

            numberOfTestsWithError++;

            statusMessage = "ERROR";

            getXMLJUnitResultFormatter().addError(currentTest, new Exception(trace));

        }

        System.out.println("  Test Failed  - " + count() + " - " + testName + " - status: " + statusMessage

                + ", trace: " + trace + ", expected: " + expected + ", actual: " + actual);

	}

	public synchronized void testReran(String testId, String testClass, String testName, int status, String trace, String expected, String actual) {

        String statusMessage = String.valueOf(status);

        if (status == ITestRunListener2.STATUS_OK) {

            statusMessage = "OK";

        } else if (status == ITestRunListener2.STATUS_FAILURE) {

            statusMessage = "FAILED";

        } else if (status == ITestRunListener2.STATUS_ERROR) {

            statusMessage = "ERROR";

        }

        System.out.println("  Test ReRan   - " + testName + " - test class: " + testClass + ", status: " + statusMessage

                + ", trace: " + trace + ", expected: " + expected + ", actual: " + actual);

	}

	public synchronized void testTreeEntry(String description) {

        System.out.println("Test Tree Entry - Description: " + description);

    }

	class WrapperTestCase extends TestCase {

        public WrapperTestCase(String name) {

            super(name);

        }

        public int countTestCases() {

            return 1;

        }

        public void run(TestResult result) {

        }

    }

}