Go 如何创建两个连接的io.ReadWriteClosers以用于测试目的

发布于04月22日

我正在try 对我的网络应用程序运行一些测试.我想避免 for each 测试创建一个插槽，所以我想使用标准库中的io.Pipe个.但它只提供io.PipeReader、io.PipeWriter对.我需要两个相连的io.ReadWriteClosers

func TestEndpoints() {
    // Create pipeEndpoint1 and pipeEndpoint2. Both should implement  
    // io.ReadWriteCloser

    e1 := makeEndpoint(pipeEndpoint1)
    e2 := makeEndpoint(pipeEndpoint2)

    // Run some test
}
 
func makeEndpoint(rwc io.ReadWriteCloser) *Endpoint {
    // 
}

有没有方法创建两个相连的io.ReadWriteClosers？

推荐答案

在标准库中似乎没有方法可以做到这一点，所以这就是我所做的:

type PipeBidirectional struct {
    r *io.PipeReader
    w *io.PipeWriter
}

func NewPipes() (*PipeBidirectional, *PipeBidirectional) {
    r1, w1 := io.Pipe()
    r2, w2 := io.Pipe()

    return &PipeBidirectional{r1, w2}, &PipeBidirectional{r2, w1}
}

func (p *PipeBidirectional) Read(b []byte) (int, error) {
    return p.r.Read(b)
}

func (p *PipeBidirectional) Write(b []byte) (int, error) {
    return p.w.Write(b)
}

func (p *PipeBidirectional) Close() error {
    e1 := p.r.Close()
    e2 := p.w.Close()
    if e1 != nil || e2 != nil {
        return fmt.Errorf("e1: %w e2: %w", e1, e2)
    }
    return nil
}