566c6b246a
This diff addresses another point of https://github.com/ooni/probe/issues/1956: > - [ ] observe that we're still using a bunch of private interfaces for common interfaces such as the `Dialer`, so we can get rid of these private interfaces and always use the ones in `model`, which allows us to remove a bunch of legacy wrappers Additional cleanups may still be possible. The more I cleanup, the more I see there's extra legacy code we can dispose of (which seems good?).
74 lines
1.3 KiB
Go
74 lines
1.3 KiB
Go
package hirl_test
|
|
|
|
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
|
|
}
|
|
|
|
func (d FakeDialer) CloseIdleConnections() {}
|
|
|
|
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
|
|
}
|