24 lines
562 B
Go
24 lines
562 B
Go
package mockablex
|
|
|
|
import (
|
|
"context"
|
|
"net"
|
|
)
|
|
|
|
// dialer is the interface we expect from a dialer
|
|
type dialer interface {
|
|
DialContext(ctx context.Context, network, address string) (net.Conn, error)
|
|
}
|
|
|
|
// Dialer is a mockable Dialer.
|
|
type Dialer struct {
|
|
MockDialContext func(ctx context.Context, network, address string) (net.Conn, error)
|
|
}
|
|
|
|
// DialContext implements Dialer.DialContext.
|
|
func (d Dialer) DialContext(ctx context.Context, network, address string) (net.Conn, error) {
|
|
return d.MockDialContext(ctx, network, address)
|
|
}
|
|
|
|
var _ dialer = Dialer{}
|