如何在Citrus中测试套接字服务器?

时间:2015-12-30 18:34:33

标签: java sockets junit netty citrus-framework

我想对使用Netty构建的套接字服务器进行一些单元测试。

套接字服务器具有以下简单代码:

import io.netty.bootstrap.ServerBootstrap;
import io.netty.channel.ChannelFuture;
import io.netty.channel.ChannelInitializer;
import io.netty.channel.ChannelOption;
import io.netty.channel.EventLoopGroup;
import io.netty.channel.nio.NioEventLoopGroup;
import io.netty.channel.socket.SocketChannel;
import io.netty.channel.socket.nio.NioServerSocketChannel;

public class SocketServer implements Runnable {

    private int port;

    private EventLoopGroup bossGroup;
    private EventLoopGroup workerGroup;

    private ChannelFuture channelFuture;
    private ServerBootstrap bootstrap;

    public SocketServer(int port) {
        this.port = port;
        this.bossGroup = new NioEventLoopGroup();
        this.workerGroup = new NioEventLoopGroup();
    }

    public int getPort() {
        return port;
    }

    @Override
    public void run() {
        try {
            bootstrap = new ServerBootstrap();
            bootstrap
                    .group(bossGroup, workerGroup)
                    .channel(NioServerSocketChannel.class)
                    .childHandler(new ChannelInitializer<SocketChannel>() {
                        @Override
                        public void initChannel(SocketChannel ch)
                                throws Exception {
                            ch.pipeline()
                                    .addLast(new ReceiveMessageServerHandler())
                                    .addLast(new ParseMessageServerHandler());
                        }
                    }).option(ChannelOption.SO_BACKLOG, 128).childOption(ChannelOption.SO_KEEPALIVE, true);

            // Bind and start to accept incoming connections.
            channelFuture = bootstrap.bind(port).sync();

            // Wait until the server socket is closed
            channelFuture.channel().closeFuture().sync();

        } catch (InterruptedException e) {
            e.printStackTrace();

        } finally {
            workerGroup.shutdownGracefully();
            bossGroup.shutdownGracefully();
        }
    }

    public void shutdown() throws InterruptedException {
        channelFuture.channel().close();
    }

}

在MessageHandlers上,我首先收到由&#39; \ n&#39;分隔的短信。我非常需要一个telnet客户端。

我想测试一下,我可以向服务器发送不同的消息,并且我会在一段时间内收到某些预期的响应。

我尝试使用Citrus Framework,但我无法获得任何结果,因为它没有提供正确的纯文本协议(我尝试过Rest,Soap等,但它们对我没有好处)。我在Citrus Reference 2.4中找不到答案。

Citrus 2.4 Reference - HTML Version

2 个答案:

答案 0 :(得分:2)

您也可以使用Citrus Apache Camel集成。你需要柑橘 - 骆驼模块和camel-netty(http://camel.apache.org/netty.html)或camel-netty4(http://camel.apache.org/netty4.html)依赖。然后,您可以直接使用Camel Netty组件发送消息:

@Test
@CitrusTest
public void sendNettyMessageTest() {
    status(Status.DRAFT);
    //sends data to server
    send("camel:netty4:tcp://localhost:9123").payload("Message 1");
    send("camel:netty4:tcp://localhost:9123").payload("Message 2");
}

这是在Citrus中使用动态端点,其中端点配置是在测试运行时创建的。所以几乎不需要额外的配置!

答案 1 :(得分:1)

我能够在发布问题之前解决问题(我昨天写了这个问题,但直到今天我才发布它......我找到了答案)。

所以我最终可以用Citrus Framework和Spring Integration来解决这个问题。

在与同事一起阅读之后,我能够使用Spring Integration TCP适配器作为Citrus端点的通道(来自java.nio的SocketChannel)。

Spring Integration Reference - IP

在这里您可以看到我使用的citrus-context.xml配置:

<?xml version="1.0" encoding="UTF-8"?>
<beans xmlns="http://www.springframework.org/schema/beans"
    xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance" 
    xmlns:citrus="http://www.citrusframework.org/schema/config"
    xmlns:int="http://www.springframework.org/schema/integration"
    xmlns:int-ip="http://www.springframework.org/schema/integration/ip"
    xmlns:context="http://www.springframework.org/schema/context"
    xsi:schemaLocation="http://www.springframework.org/schema/beans http://www.springframework.org/schema/beans/spring-beans.xsd
        http://www.springframework.org/schema/context http://www.springframework.org/schema/context/spring-context.xsd 
        http://www.citrusframework.org/schema/config http://www.citrusframework.org/schema/config/citrus-config.xsd
        http://www.springframework.org/schema/integration  http://www.springframework.org/schema/integration/spring-integration.xsd
        http://www.springframework.org/schema/integration/ip  http://www.springframework.org/schema/integration/ip/spring-integration-ip.xsd">

    <citrus:channel-endpoint id="citrusServiceEndpoint"
        channel-name="input" />

    <int-ip:tcp-connection-factory id="client"
        type="client" host="localhost" port="9123" single-use="true"
        so-timeout="10000" using-nio="true" />

    <int:channel id="input" />

    <int-ip:tcp-outbound-channel-adapter
        id="outboundClient" channel="input" connection-factory="client" />

</beans>

在我的柑橘测试中,我能够根据需要发送消息:

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

import com.consol.citrus.TestCaseMetaInfo.Status;
import com.consol.citrus.annotations.CitrusTest;
import com.consol.citrus.dsl.junit.JUnit4CitrusTestDesigner;

public class MyFirstTest extends JUnit4CitrusTestDesigner {

    private static final int PORT = 9123;
    private static SocketServer socketServer;
    private static Thread socketThread;

    @BeforeClass
    public static void setUp() throws Exception {
        socketServer = new SocketServer(PORT);
        socketThread = new Thread(socketServer);
        socketThread.start();
    }

    @AfterClass
    public static void tearDown() throws Exception {
        socketServer.shutdown();
        socketThread.join();
    }

    @Test
    @CitrusTest(name = "sendSpringIntegrationMessageTest")
    public void sendSpringIntegrationMessageTest() {
        status(Status.DRAFT);
        send("citrusServiceEndpoint").payload("Message 1");
        send("citrusServiceEndpoint").payload("Message 2");
    }

}

我希望这可以帮助任何可能遇到与我相同问题的人。