Add bufcon.DialContext(context.Context,string)

for use with grpc.WithContextDial DialOption

Signed-off-by: Ed Warnicke <hagbard@gmail.com>

test/bufconn/bufconn.go

@@ -21,6 +21,7 @@
 package bufconn
 
 import (
+	"context"
 	"fmt"
 	"io"
 	"net"
@@ -95,6 +96,16 @@ func (l *Listener) Dial() (net.Conn, error) {
 	}
 }
 
+// DialContext provides a function usable with the grpc.WithContextDial DialOption
+func (l *Listener) DialContext(ctx context.Context, _ string) (net.Conn, error) {
+	select {
+	case <-ctx.Done():
+		return nil, ctx.Err()
+	default:
+	}
+	return l.Dial()
+}
+
 type pipe struct {
 	mu sync.Mutex