import io.netty.bootstrap.ServerBootstrap;
import io.netty.channel.ChannelInboundHandlerAdapter;
import io.netty.channel.ChannelInitializer;
import io.netty.channel.EventLoopGroup;
import io.netty.channel.nio.NioEventLoopGroup;
import io.netty.channel.socket.SocketChannel;
import io.netty.channel.socket.nio.NioServerSocketChannel;
import org.junit.Test;

import static java.util.concurrent.TimeUnit.MILLISECONDS;
import static org.junit.Assert.assertTrue;

public class NettyNioTest {
  @Test
  public void testShutdown() throws Exception {
    EventLoopGroup serverEventLoopGroup = new NioEventLoopGroup();
    ServerBootstrap serverBootstrap = new ServerBootstrap();
    serverBootstrap.group(serverEventLoopGroup)
            .channel(NioServerSocketChannel.class)
            .childHandler(new ChannelInitializer<SocketChannel>() {
              @Override
              public void initChannel(SocketChannel ch) throws Exception {
                ch.pipeline().addLast(new ChannelInboundHandlerAdapter());
              }
            });

    serverBootstrap.bind(0).sync();

    final long t0 = System.currentTimeMillis();
    serverEventLoopGroup.shutdownGracefully(50, 5000, MILLISECONDS).sync();
    final long t = System.currentTimeMillis() - t0;
    assertTrue("Shutdown took " + t + " ms", t < 200);
  }
}