pekko/akka-camel/src/test/java/akka/camel/ConsumerJavaTestBase.java

/**
 * Copyright (C) 2009-2012 Typesafe Inc. <http://www.typesafe.com>
 */

package akka.camel;

import static org.junit.Assert.assertEquals;

import java.util.concurrent.TimeUnit;

import org.junit.AfterClass;
import org.junit.Test;

import scala.concurrent.Await;
import scala.concurrent.ExecutionContext;
import scala.concurrent.util.Duration;
import scala.concurrent.util.FiniteDuration;
import akka.actor.ActorRef;
import akka.actor.ActorSystem;
import akka.actor.Props;
import akka.testkit.AkkaSpec;
import akka.testkit.JavaTestKit;


/**
 * @author Martin Krasser
 */
public class ConsumerJavaTestBase {

    static ActorSystem system = ActorSystem.create("test", AkkaSpec.testConf());

    @AfterClass
    public static void tearDownAfterClass() {
        system.shutdown();
    }

    @Test
    public void shouldHandleExceptionThrownByActorAndGenerateCustomResponse() throws Exception {
        new JavaTestKit(system) {{
            String result = new EventFilter<String>(Exception.class) {
                protected String run() {
                    FiniteDuration timeout = Duration.create(1, TimeUnit.SECONDS);
                    Camel camel = CamelExtension.get(system);
                    ExecutionContext executionContext = system.dispatcher();
                    try {
                        ActorRef ref = Await.result(
                                camel.activationFutureFor(system.actorOf(new Props(SampleErrorHandlingConsumer.class)), timeout, executionContext),
                                timeout);
                        return camel.template().requestBody("direct:error-handler-test-java", "hello", String.class);
                    }
                    catch (Exception e) {
                        return e.getMessage();
                    }
                }
            }.occurrences(1).exec();
            assertEquals("error: hello", result);
        }};
    }
}