/**
* 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.core.mdb;
import org.apache.openejb.junit.ApplicationComposer;
import org.apache.openejb.testing.Classes;
import org.apache.openejb.testing.Configuration;
import org.apache.openejb.testing.SimpleLog;
import org.apache.openejb.testng.PropertiesBuilder;
import org.apache.openejb.util.NetworkUtil;
import org.junit.Test;
import org.junit.runner.RunWith;
import javax.annotation.Resource;
import javax.ejb.MessageDriven;
import javax.interceptor.AroundInvoke;
import javax.interceptor.Interceptor;
import javax.interceptor.InterceptorBinding;
import javax.interceptor.InvocationContext;
import javax.jms.Connection;
import javax.jms.ConnectionFactory;
import javax.jms.Message;
import javax.jms.MessageListener;
import javax.jms.MessageProducer;
import javax.jms.Queue;
import javax.jms.Session;
import javax.jms.TextMessage;
import java.io.Serializable;
import java.lang.annotation.ElementType;
import java.lang.annotation.Retention;
import java.lang.annotation.RetentionPolicy;
import java.lang.annotation.Target;
import java.util.Properties;
import java.util.concurrent.CountDownLatch;
import java.util.concurrent.TimeUnit;
import static org.junit.Assert.assertEquals;
import static org.junit.Assert.assertNotNull;
import static org.junit.Assert.assertTrue;
@SimpleLog
@Classes(cdi = true, innerClassesAsBean = true, cdiInterceptors = MDBCdiInterceptorTest.InInterceptor.class)
@RunWith(ApplicationComposer.class)
public class MDBCdiInterceptorTest {
@Resource(name = "target")
private Queue queue;
@Resource
private ConnectionFactory connectionFactory;
@Test
public void sendMessage() throws Exception {
Connection connection = null;
Session session = null;
MessageProducer producer = null;
try {
connection = connectionFactory.createConnection();
session = connection.createSession(false, Session.AUTO_ACKNOWLEDGE);
final TextMessage requestMessage = session.createTextMessage("sent");
producer = session.createProducer(queue);
producer.send(requestMessage);
// wait
MyInterceptedMdb.latch.await(1, TimeUnit.MINUTES);
assertNotNull(InInterceptor.called);
assertEquals("sent", TextMessage.class.cast(InInterceptor.called).getText());
assertTrue(MyInterceptedMdb.called);
} finally {
MdbUtil.close(producer);
MdbUtil.close(session);
MdbUtil.close(connection);
}
}
@Configuration
public Properties p() {
return new PropertiesBuilder()
.p("Default JMS Resource Adapter.BrokerXmlConfig", "broker:(tcp://localhost:" + NetworkUtil.getNextAvailablePort() + ")?useJmx=false")
.p("target", "new://Resource?type=Queue")
.build();
}
@In
@MessageDriven(activationConfig = {
@javax.ejb.ActivationConfigProperty(propertyName = "maxSessions", propertyValue = "1"),
@javax.ejb.ActivationConfigProperty(propertyName = "maxMessagesPerSessions", propertyValue = "1"),
@javax.ejb.ActivationConfigProperty(propertyName = "destination", propertyValue = "target")
})
public static class MyInterceptedMdb implements MessageListener {
private static volatile boolean called = false;
private static final CountDownLatch latch = new CountDownLatch(1);
@Override
public void onMessage(final Message message) {
called = true;
latch.countDown();
}
}
@InterceptorBinding
@Target(ElementType.TYPE)
@Retention(RetentionPolicy.RUNTIME)
public static @interface In {
}
@Interceptor
@In
public static class InInterceptor implements Serializable {
private static volatile Message called;
@AroundInvoke
public Object aroundInvoke(final InvocationContext context) throws Exception {
called = Message.class.cast(context.getParameters()[0]);
return context.proceed();
}
}
}