Netty 上的网络代理的单元测试

Question

我试图从netty的标准示例中为HexDumpProxyFrontendHandler创建一个单元测试。但是使用嵌入式通道时遇到麻烦。这是我的代码：

public class HexDumpProxyFrontendHandlerTest {
    @Test
    public void test1() throws Exception {
        HttpRequest request = new DefaultFullHttpRequest(HttpVersion.HTTP_1_1, HttpMethod.GET, "/");
        HttpUtil.set100ContinueExpected(request, true);
        EmbeddedChannel channel = new EmbeddedChannel(new HexDumpProxyFrontendHandler("127.0.0.1", 8080));
        channel.writeInbound(request);
    }
}

我得到了java.nio.channels.ClosedChannelException，因为我理解代码试图连接到另一台服务器 - 127.0.0.1:8080.

我想知道有没有办法用嵌入式通道测试这段代码？或者我应该创建某种模拟服务器

Answer 1

我开始使用Moco来创建目标http服务器。它是用于创建存根服务器的特殊框架。我在另一个线程中运行代理并通过网络流读取数据。现在测试看起来像：

private static final URI TARGET_LINK = URI.create("http://127.0.0.1:12306/target");
private Runner runner;
@Before
public void setUp() throws Exception {
    HttpServer server = httpServer(TARGET_LINK.getPort());
    server.request(and(by(uri("/target")), by(version(VERSION_1_1))))
            .response(with(text(TARGET_RESPONSE_BASE64)), header("Content-Type", "base64"));
    server.response(DEFAULT_RESPONSE);
    server.request(and(by(uri("/binary")), by(version(VERSION_1_1))))
            .response(with(text(BINARY_RESPONSE_BASE64)), header("Content-Type", "base64"));
    runner = runner(server);
    runner.start();
}

@After
public void tearDown() {
    runner.stop();
}
@Test
public void testTarget() throws Exception {
    BinaryProxy proxy = startProxy(TARGET_LINK.toASCIIString());
    byte[] response = getServerResponse();
    proxy.stop();
    assertArrayEquals(wrapLine(TARGET_REPLY), response);
}
private BinaryProxy startProxy(String link) throws InterruptedException {
    final BinaryProxy binaryProxy = new BinaryProxy(LISTEN_PORT, link);
    Thread thread = new Thread(new Runnable() {
        @Override
        public void run() {
            try {
                binaryProxy.start();
            } catch (InterruptedException e) {
                e.printStackTrace();
            }
        }
    });
    thread.start();
    TimeUnit.MILLISECONDS.sleep(400);
    return binaryProxy;
}
private byte[] getServerResponse() throws IOException {
    Socket server = new Socket(LISTEN_ADDRESS, LISTEN_PORT);
    OutputStream out = server.getOutputStream();
    out.write(TEST_MESSAGE);
    out.flush();
    InputStream is = server.getInputStream();
    ByteArrayOutputStream buffer = new ByteArrayOutputStream();
    int nRead;
    byte[] data = new byte[512];
    while ((nRead = is.read(data, 0, data.length)) != -1) {
        buffer.write(data, 0, nRead);
    }
    buffer.flush();
    return buffer.toByteArray();
}

我知道这不是单元测试，但我没有找到比这更好的方法。

我对EmbeddedChannel类的所有实验都失败了。我只能将这个类与后端处理程序一起使用。