diff options
Diffstat (limited to 'internal/server/auth_test.go')
| -rw-r--r-- | internal/server/auth_test.go | 7 |
1 files changed, 6 insertions, 1 deletions
diff --git a/internal/server/auth_test.go b/internal/server/auth_test.go index be711ac3b..b50445a84 100644 --- a/internal/server/auth_test.go +++ b/internal/server/auth_test.go @@ -3,6 +3,7 @@ package server import ( "net" "testing" + "time" "github.com/stretchr/testify/assert" "github.com/stretchr/testify/require" @@ -145,6 +146,10 @@ func (brokenAuth) GetRequestMetadata(netctx.Context, ...string) (map[string]stri } func dial(serverSocketPath string, opts []grpc.DialOption) (*grpc.ClientConn, error) { + opts = append(opts, grpc.WithDialer(func(addr string, timeout time.Duration) (net.Conn, error) { + return net.DialTimeout("unix", addr, timeout) + })) + return grpc.Dial(serverSocketPath, opts...) } @@ -166,5 +171,5 @@ func runServer(t *testing.T) (*grpc.Server, string) { require.NoError(t, err) go srv.Serve(listener) - return srv, "unix://" + serverSocketPath + return srv, serverSocketPath } |