/**
* 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.openejb.client;
/**
* @version $Revision$ $Date$
*/
import junit.framework.TestCase;
import java.io.ByteArrayInputStream;
import java.io.ByteArrayOutputStream;
import java.io.IOException;
import java.io.ObjectInputStream;
import java.io.ObjectOutputStream;
import java.io.PrintStream;
public class ThrowableArtifactTest extends TestCase {
public void testThrowableArtifact() throws Throwable {
Throwable exception = new NullPointerException("ONE");
exception = throwCatchReturn(exception);
exception = new IllegalArgumentException("TWO", exception);
exception = throwCatchReturn(exception);
exception = new UnsupportedOperationException("THREE", exception);
exception = throwCatchReturn(exception);
exception = new IllegalStateException("FOUR", exception);
exception = throwCatchReturn(exception);
exception = new BadException("FIVE", exception);
exception = throwCatchReturn(exception);
String expectedStackTrace = getPrintedStackTrace(exception);
ThrowableArtifact artifact = marshal(new ThrowableArtifact(exception));
exception = throwCatchReturn(artifact.getThrowable().getCause());
String actualStackTrace = getPrintedStackTrace(exception);
assertEquals("stackTrace", expectedStackTrace, actualStackTrace);
}
private String getPrintedStackTrace(Throwable exception) {
ByteArrayOutputStream baos = new ByteArrayOutputStream();
PrintStream printStream = new PrintStream(baos);
exception.printStackTrace(printStream);
printStream.flush();
String stackTrace = new String(baos.toByteArray());
return stackTrace;
}
private Throwable throwCatchReturn(Throwable exception) {
try {
throw exception;
} catch (Throwable e) {
return exception;
}
}
private ThrowableArtifact marshal(ThrowableArtifact artifact) throws IOException, ClassNotFoundException {
ByteArrayOutputStream baos = new ByteArrayOutputStream();
ObjectOutputStream out = new ObjectOutputStream(baos);
out.writeObject(artifact);
out.close();
ByteArrayInputStream bais = new ByteArrayInputStream(baos.toByteArray());
ObjectInputStream in = new ObjectInputStream(bais);
return (ThrowableArtifact) in.readObject();
}
public static class BadException extends Exception {
private final Object data = new NotSerializableObject();
public BadException(String message, Throwable throwable) {
super(message, throwable);
}
}
public static class NotSerializableObject {
}
}