在Go中模拟tcp连接

Question

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

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

是否有为此的数据结构，或者一种简单的制作方法？

Answer 1

不知道在提出问题时是否存在这种情况，但您可能需要 net.Pipe() 它为您提供两个全双工 net.Conn 实例，它们链接到每个实例其他

Answer 2

编辑：我已将这个答案打包到一个包中，这使事情变得更简单 - 请参阅此处：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,
    }
}

当模拟“服务器”连接时，只需传递 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 3

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

type CloseableBuffer struct {
    bytes.Buffer
}

并定义一个 Close 方法：

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

Answer 4

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

您只需要模拟那些会增加测试时间、阻止测试并行运行（使用硬编码文件名等共享资源）或可能导致中断（您可能会耗尽连接限制或端口，但在大多数情况下）的内容。当您单独运行测试时，这不是问题）。

不模拟的优点是可以更精确地测试您想要用真实事物测试的内容。

https://www.accenture.com/us-en/blogs/software-engineering-blog/to-mock-or-not-to-mock-is-that-even-a-question< /a>

您可以编写一个模拟服务器，在测试中的 goroutine 中运行它，然后使用真实的 net.Conn 连接到它，而不是模拟 net.Conn

一个快速而肮脏的示例：

port := someRandomPort()
srv := &http.Server{Addr: port}
go func(msg string) {
    http.HandleFunc("/hello", myHandleFUnc)
    srv.ListenAndServe()
}
myTestCodeUsingConn(port)
srv.Shutdown(context.TODO())