我正在开发网络应用程序,我想正确进行单元测试。这次我们会做的,你知道吗? :)

我在测试网络连接时遇到了麻烦。

在我的应用程序中,我使用普通的java.net.Socket

例如:

import java.io.IOException;
import java.io.OutputStream;
import java.net.Socket;
import java.net.UnknownHostException;

public class Message {
    byte[] payload;

    public Message(byte[] payload) {
        this.payload = payload;
    }

    public boolean sendTo(String hostname, int port) {
        boolean sent = false;

        try {
            Socket socket = new Socket(hostname, port);

            OutputStream out = socket.getOutputStream();

            out.write(payload);

            socket.close();

            sent = true;
        } catch (UnknownHostException e) {
        } catch (IOException e) {
        }

        return sent;
    }
}

我读过有关 mock 的文章,但不确定如何应用。

最佳答案

如果要测试代码,请执行以下操作。

首先,重构代码,以便在要测试的方法中不直接实例化Socket。下面的示例显示了我可以想到的最小更改。将来的更改可能会将Socket的创建排除在一个完全独立的类之外,但是我喜欢一些小步骤,而且我不喜欢在未经测试的代码上进行大的更改。

public boolean sendTo(String hostname, int port) {
    boolean sent = false;

    try {
        Socket socket = createSocket();
        OutputStream out = socket.getOutputStream();
        out.write(payload);
        socket.close();
        sent = true;
    } catch (UnknownHostException e) {
        // TODO
    } catch (IOException e) {
        // TODO
    }

    return sent;
}

protected Socket createSocket() {
    return new Socket();
}

既然套接字创建逻辑不在您要测试的方法之外,您就可以开始进行模拟,并加入套接字的创建中。
public class MessageTest {
    @Test
    public void testSimplePayload() () {
        byte[] emptyPayload = new byte[1001];

        // Using Mockito
        final Socket socket = mock(Socket.class);
        final ByteArrayOutputStream byteArrayOutputStream = new ByteArrayOutputStream();
        when(socket.getOutputStream()).thenReturn(byteArrayOutputStream);

        Message text = new Message(emptyPayload) {
            @Override
            protected Socket createSocket() {
                return socket;
            }
        };

        Assert.assertTrue("Message sent successfully", text.sendTo("localhost", "1234"));
        Assert.assertEquals("whatever you wanted to send".getBytes(), byteArrayOutputStream.toByteArray());
    }
}

在要测试的单元上覆盖单个方法对于测试确实很有用,尤其是在具有可怕依赖关系的丑陋代码中。显然,最好的解决方案是解决依赖关系(在这种情况下,我认为Message并不依赖Socket,或者可能会像glowcoder所建议的那样存在Messager接口(interface)),但是最好以尽可能小的步骤迈向解决方案。

09-30 14:03
查看更多