src/main/org/apache/ant/antunit/junit3/JUnitNotificationAdapter.java

/*
 * 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 org.apache.ant.antunit.junit3;

import java.util.Enumeration;
import java.util.HashMap;
import java.util.Map;

import junit.framework.Test;
import junit.framework.TestResult;

import org.apache.ant.antunit.AntUnitExecutionNotifier;
import org.apache.ant.antunit.AssertionFailedException;

/**
 * Adapt AntUnitExecutionNotifier events into JUnit3 TestResult events
 */
class JUnitNotificationAdapter implements AntUnitExecutionNotifier {

    private final TestResult junitTestResult;
    private Map testByTarget = new HashMap(); 

    public JUnitNotificationAdapter(TestResult testResult, Enumeration tests) {
        this.junitTestResult = testResult;
        while(tests.hasMoreElements()) {
            AntUnitTestCase test = (AntUnitTestCase) tests.nextElement();
            testByTarget.put(test.getTarget(), test);
        }
    }

    public void fireStartTest(String targetName) {
        //TODO : if it is null, eclipse stop the unit test (add a unit test)
        junitTestResult.startTest((Test) testByTarget.get(targetName));
    }
    
    public void fireEndTest(String targetName) {
        junitTestResult.endTest((Test) testByTarget.get(targetName));
    }

    public void fireError(String targetName, Throwable t) {
        junitTestResult.addError((Test) testByTarget.get(targetName), t);
    }

    public void fireFail(String targetName, AssertionFailedException ae) {
        //I don't see how to transform the AntUnit assertion exception into 
        //junit assertion exception (we would loose the stack trace).
        //So failures will be reported as errors
        junitTestResult.addError((Test) testByTarget.get(targetName), ae);
    }

}