Question

在Go中，TCP连接（net.Conn）是io.ReadWriteCloser。我想通过模拟TCP连接来测试我的网络代码。我有两个要求：

要读取的数据存储在字符串
每当写入数据时，我希望它存储在某种缓冲区中，以后我可以访问

是否有这样的数据结构，或者制作一个简单的方法？

Answer 1

编辑：我已将此答案推广到一个让事情变得更简单的方案中 - 请参阅此处：https://github.com/jordwest/mock-conn

虽然Ivan的解决方案适用于简单的情况，但请记住，真正的TCP连接实际上是两个缓冲区，或者说是管道。例如：

 Server   |   Client
 ---------+---------
  reads <===  writes
 writes ===>  reads

如果您使用服务器读取和写入的单个缓冲区，您最终可能会与服务器通信。

这是一个允许您将MockConn类型作为ReadWriteCloser传递到服务器的解决方案。 Read，Write和Close函数只是代理服务器端管道上的函数。

type MockConn struct {
    ServerReader *io.PipeReader
    ServerWriter *io.PipeWriter

    ClientReader *io.PipeReader
    ClientWriter *io.PipeWriter
}

func (c MockConn) Close() error {
    if err := c.ServerWriter.Close(); err != nil {
        return err
    }
    if err := c.ServerReader.Close(); err != nil {
        return err
    }
    return nil
}

func (c MockConn) Read(data []byte) (n int, err error)  { return c.ServerReader.Read(data) }
func (c MockConn) Write(data []byte) (n int, err error) { return c.ServerWriter.Write(data) }

func NewMockConn() MockConn {
    serverRead, clientWrite := io.Pipe()
    clientRead, serverWrite := io.Pipe()

    return MockConn{
        ServerReader: serverRead,
        ServerWriter: serverWrite,
        ClientReader: clientRead,
        ClientWriter: clientWrite,
    }
}

嘲笑服务器＆＃39;连接，只需传递MockConn代替您使用net.Conn的位置（这显然只实现ReadWriteCloser接口，如果需要支持，可以轻松地为LocalAddr()等添加虚拟方法完整的net.Conn界面）

在您的测试中，您可以根据需要通过读取ClientReader和ClientWriter字段来充当客户端：

func TestTalkToServer(t *testing.T) {
    /*
     * Assumes that NewMockConn has already been called and
     * the server is waiting for incoming data
     */

    // Send a message to the server
    fmt.Fprintf(mockConn.ClientWriter, "Hello from client!\n")

    // Wait for the response from the server
    rd := bufio.NewReader(mockConn.ClientReader)
    line, err := rd.ReadString('\n')

    if line != "Hello from server!" {
        t.Errorf("Server response not as expected: %s\n", line)
    }
}

Answer 2

为什么不使用bytes.Buffer？它是io.ReadWriter并且有String方法来获取存储的数据。如果您需要将其设为io.ReadWriteCloser，则可以定义自己的类型：

type CloseableBuffer struct {
    bytes.Buffer
}

并定义Close方法：

func (b *CloseableBuffer) Close() error {
    return nil
}

Answer 3

当问到这个问题时，不知道是否存在此问题，但您可能希望noexcept为您提供两个相互关联的全双工net.Pipe()实例

Answer 4

大多数情况下，您不需要模拟net.Conn。

您总是可以在测试代码中运行一个模仿另一面的goroutine。

在Go中模拟tcp连接

4 个答案: