72 lines
1.3 KiB
Go
72 lines
1.3 KiB
Go
|
package dialer
|
||
|
|
|
||
|
|
import (
|
||
|
|
"context"
|
||
|
|
"io"
|
||
|
|
"net"
|
||
|
|
"time"
|
||
|
|
)
|
||
|
|
|
||
|
|
type FakeDialer struct {
|
||
|
|
Conn net.Conn
|
||
|
|
Err error
|
||
|
|
}
|
||
|
|
|
||
|
|
func (d FakeDialer) DialContext(ctx context.Context, network, address string) (net.Conn, error) {
|
||
|
|
time.Sleep(10 * time.Microsecond)
|
||
|
|
return d.Conn, d.Err
|
||
|
|
}
|
||
|
|
|
||
|
|
type FakeConn struct {
|
||
|
|
ReadError error
|
||
|
|
ReadData []byte
|
||
|
|
SetDeadlineError error
|
||
|
|
SetReadDeadlineError error
|
||
|
|
SetWriteDeadlineError error
|
||
|
|
WriteError error
|
||
|
|
}
|
||
|
|
|
||
|
|
func (c *FakeConn) Read(b []byte) (int, error) {
|
||
|
|
if len(c.ReadData) > 0 {
|
||
|
|
n := copy(b, c.ReadData)
|
||
|
|
c.ReadData = c.ReadData[n:]
|
||
|
|
return n, nil
|
||
|
|
}
|
||
|
|
if c.ReadError != nil {
|
||
|
|
return 0, c.ReadError
|
||
|
|
}
|
||
|
|
return 0, io.EOF
|
||
|
|
}
|
||
|
|
|
||
|
|
func (c *FakeConn) Write(b []byte) (n int, err error) {
|
||
|
|
if c.WriteError != nil {
|
||
|
|
return 0, c.WriteError
|
||
|
|
}
|
||
|
|
n = len(b)
|
||
|
|
return
|
||
|
|
}
|
||
|
|
|
||
|
|
func (*FakeConn) Close() (err error) {
|
||
|
|
return
|
||
|
|
}
|
||
|
|
|
||
|
|
func (*FakeConn) LocalAddr() net.Addr {
|
||
|
|
return &net.TCPAddr{}
|
||
|
|
}
|
||
|
|
|
||
|
|
func (*FakeConn) RemoteAddr() net.Addr {
|
||
|
|
return &net.TCPAddr{}
|
||
|
|
}
|
||
|
|
|
||
|
|
func (c *FakeConn) SetDeadline(t time.Time) (err error) {
|
||
|
|
return c.SetDeadlineError
|
||
|
|
}
|
||
|
|
|
||
|
|
func (c *FakeConn) SetReadDeadline(t time.Time) (err error) {
|
||
|
|
return c.SetReadDeadlineError
|
||
|
|
}
|
||
|
|
|
||
|
|
func (c *FakeConn) SetWriteDeadline(t time.Time) (err error) {
|
||
|
|
return c.SetWriteDeadlineError
|
||
|
|
}
|