/**
* Copyright (c) 2009-2011 VMware, Inc. All Rights Reserved.
*
* Licensed 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 com.springsource.insight.plugin.ejb3;
import java.lang.annotation.Annotation;
import java.util.Date;
import javax.ejb.SessionSynchronization;
import javax.ejb.Stateful;
import javax.ejb.Stateless;
import org.junit.Test;
import com.springsource.insight.collection.test.OperationCollectionAspectTestSupport;
import com.springsource.insight.intercept.operation.Operation;
import com.springsource.insight.intercept.operation.OperationFields;
import com.springsource.insight.intercept.operation.OperationList;
import com.springsource.insight.util.StringFormatterUtils;
/**
*/
public class Ejb3LocalServiceOperationCollectionAspectTest extends OperationCollectionAspectTestSupport {
public Ejb3LocalServiceOperationCollectionAspectTest() {
super();
}
@Test
public void testEjb3StatefulLocalServiceAction() throws Exception {
runEjb3LocalServiceAction(new ExampleStatefulEjb3LocalServiceAction(), Stateful.class);
}
@Test
public void testEjb3StatelessLocalServiceAction() throws Exception {
runEjb3LocalServiceAction(new ExampleStatelessEjb3LocalServiceAction(), Stateless.class);
}
private <C extends ExampleEjb3LocalService & SessionSynchronization> void runEjb3LocalServiceAction(
final C testedService, final Class<? extends Annotation> annClass)
throws Exception {
final Class<?> testedServiceClass = testedService.getClass();
assertNotNull("Missing " + annClass.getSimpleName() + " annotation", testedServiceClass.getAnnotation(annClass));
testedService.afterBegin();
final Date[] invocationArgs = {new Date(System.currentTimeMillis())};
testedService.beforeCompletion();
testedService.invokeMe(invocationArgs[0]);
testedService.afterCompletion(true);
final Operation op = getLastEntered();
assertNotNull("No operation extracted", op);
assertEquals("Mismatched " + annClass.getSimpleName() + " EJB operation type(s)", Ejb3LocalServiceDefinitions.TYPE, op.getType());
// see JoinPointBreakDownSourceCodeLocationFinalizer#populateOperation
assertEquals("Mismatched " + annClass.getSimpleName() + " class name", testedServiceClass.getName(), op.get(OperationFields.CLASS_NAME, String.class));
assertEquals("Mismatched " + annClass.getSimpleName() + " invocation method", "invokeMe", op.get("methodName", String.class));
final OperationList argsList = op.get(OperationFields.ARGUMENTS, OperationList.class);
assertNotNull("Missing " + annClass.getSimpleName() + " invocation arguments", argsList);
assertEquals("Mismatched " + annClass.getSimpleName() + " num. of invocation arguments", invocationArgs.length, argsList.size());
for (int aIndex = 0; aIndex < invocationArgs.length; aIndex++) {
final Object expArg = invocationArgs[aIndex], actArg = argsList.get(aIndex);
final String expStr = StringFormatterUtils.formatObject(expArg),
actStr = StringFormatterUtils.formatObject(actArg);
assertEquals("Mismatched " + annClass.getSimpleName() + " invocation arguments #" + aIndex, expStr, actStr);
}
}
@Override
public Ejb3LocalServiceOperationCollectionAspect getAspect() {
return Ejb3LocalServiceOperationCollectionAspect.aspectOf();
}
}