34 lines
979 B
Go
34 lines
979 B
Go
package testsupport
|
|
|
|
import "context"
|
|
|
|
type InmemoryConnection struct {
|
|
ctx context.Context
|
|
frontToBack chan ([]byte)
|
|
backToFront chan ([]byte)
|
|
addr string
|
|
}
|
|
|
|
func NewInmemoryConnection(ctx context.Context, addr string) *InmemoryConnection {
|
|
pipe := InmemoryConnection{
|
|
ctx: ctx,
|
|
// arbitrary unbuffered channel, unbuffered is more similar to TCP connections.
|
|
frontToBack: make(chan []byte),
|
|
backToFront: make(chan []byte),
|
|
addr: addr,
|
|
}
|
|
return &pipe
|
|
}
|
|
|
|
func (bitpipe *InmemoryConnection) Front() *ChannelReadWriter {
|
|
return pipe(bitpipe.ctx, bitpipe.backToFront, bitpipe.frontToBack, bitpipe.addr)
|
|
}
|
|
|
|
func (bitpipe *InmemoryConnection) Back() *ChannelReadWriter {
|
|
return pipe(bitpipe.ctx, bitpipe.frontToBack, bitpipe.backToFront, bitpipe.addr)
|
|
}
|
|
|
|
func pipe(ctx context.Context, receiveBuffer <-chan []byte, sendBuffer chan<- []byte, remoteAddr string) *ChannelReadWriter {
|
|
return NewChannelReadWriter(ctx, receiveBuffer, sendBuffer)
|
|
}
|