InterceptorExecutorTest.java

package netflix.karyon.server.interceptor;

import io.netty.buffer.ByteBuf;
import io.netty.buffer.Unpooled;
import netflix.karyon.server.MockChannelHandlerContext;
import netflix.karyon.transport.interceptor.InterceptorExecutor;
import netflix.karyon.transport.interceptor.InterceptorKey;
import netflix.karyon.transport.interceptor.InterceptorSupport;
import netflix.karyon.transport.interceptor.KeyEvaluationContext;
import org.junit.Assert;
import org.junit.Test;
import rx.functions.Action0;

import java.util.concurrent.CountDownLatch;
import java.util.concurrent.TimeUnit;

/**
 * @author Nitesh Kant
 */
public class InterceptorExecutorTest {

    @Test
    public void testInbound() throws Exception {

        InterceptorKey<ByteBuf, KeyEvaluationContext> key = new MockKey(true);
        TestableInboundInterceptor interceptor1 = new TestableInboundInterceptor(key);
        TestableInboundInterceptor interceptor2 = new TestableInboundInterceptor(key);
        TestableInboundInterceptor interceptor3 = new TestableInboundInterceptor(key);

        InterceptorSupport<ByteBuf, ByteBuf, KeyEvaluationContext> support =
                new InterceptorSupport<ByteBuf, ByteBuf, KeyEvaluationContext>();
        support.forKey(key).intercept(interceptor1);
        support.forKey(key).intercept(interceptor2);
        support.forKey(key).intercept(interceptor3);

        TestableRequestRouter<ByteBuf, ByteBuf> router = new TestableRequestRouter<ByteBuf, ByteBuf>();

        InterceptorExecutor<ByteBuf, ByteBuf, KeyEvaluationContext> executor =
                new InterceptorExecutor<ByteBuf, ByteBuf, KeyEvaluationContext>(support, router);

        executeAndAwait(executor);

        Assert.assertTrue("1st interceptor did not get invoked.", interceptor1.isReceivedACall());
        Assert.assertTrue("2rd interceptor did not get invoked.", interceptor2.isReceivedACall());
        Assert.assertTrue("3rd interceptor did not get invoked.", interceptor3.isReceivedACall());
        Assert.assertTrue("Tail interceptor did not get invoked.", router.isReceivedACall());
    }

    @Test
    public void testOutbound() throws Exception {


        InterceptorKey<ByteBuf, KeyEvaluationContext> key = new MockKey(true);
        TestableOutboundInterceptor interceptor1 = new TestableOutboundInterceptor(key);
        TestableOutboundInterceptor interceptor2 = new TestableOutboundInterceptor(key);
        TestableOutboundInterceptor interceptor3 = new TestableOutboundInterceptor(key);

        InterceptorSupport<ByteBuf, ByteBuf, KeyEvaluationContext> support =
                new InterceptorSupport<ByteBuf, ByteBuf, KeyEvaluationContext>();
        support.forKey(key).intercept(interceptor1);
        support.forKey(key).intercept(interceptor2);
        support.forKey(key).intercept(interceptor3);

        TestableRequestRouter<ByteBuf, ByteBuf> router = new TestableRequestRouter<ByteBuf, ByteBuf>();

        InterceptorExecutor<ByteBuf, ByteBuf, KeyEvaluationContext> executor =
                new InterceptorExecutor<ByteBuf, ByteBuf, KeyEvaluationContext>(support, router);

        executeAndAwait(executor);

        Assert.assertTrue("1st interceptor did not get invoked.", interceptor1.isReceivedACall());
        Assert.assertTrue("2rd interceptor did not get invoked.", interceptor2.isReceivedACall());
        Assert.assertTrue("3rd interceptor did not get invoked.", interceptor3.isReceivedACall());
        Assert.assertTrue("Tail interceptor did not get invoked.", router.isReceivedACall());
    }

    @Test
    public void testUnsubscribe() throws Exception {

        TestableRequestRouter<ByteBuf, ByteBuf> router = new TestableRequestRouter<ByteBuf, ByteBuf>();

        InterceptorSupport<ByteBuf, ByteBuf, KeyEvaluationContext> support = new InterceptorSupport<ByteBuf, ByteBuf, KeyEvaluationContext>();

        InterceptorExecutor<ByteBuf, ByteBuf, KeyEvaluationContext> executor =
                new InterceptorExecutor<ByteBuf, ByteBuf, KeyEvaluationContext>(support, router);

        executeAndAwait(executor);

        Assert.assertTrue("Router did not get invoked.", router.isReceivedACall());
        Assert.assertTrue("Router did not get unsubscribed.", router.isUnsubscribed());
    }

    protected void executeAndAwait(InterceptorExecutor<ByteBuf, ByteBuf, KeyEvaluationContext> executor)
            throws InterruptedException {
        final CountDownLatch completionLatch = new CountDownLatch(1);

        executor.execute(Unpooled.buffer(), Unpooled.buffer(),
                         new KeyEvaluationContext(new MockChannelHandlerContext("mock").channel()))
                .doOnCompleted(new Action0() {
                    @Override
                    public void call() {
                        completionLatch.countDown();
                    }
                })
                .subscribe();

        completionLatch.await(1, TimeUnit.MINUTES);
    }

}