diff options
Diffstat (limited to 'internal/middleware/limithandler/limithandler_test.go')
-rw-r--r-- | internal/middleware/limithandler/limithandler_test.go | 5 |
1 files changed, 4 insertions, 1 deletions
diff --git a/internal/middleware/limithandler/limithandler_test.go b/internal/middleware/limithandler/limithandler_test.go index 4a6e6ffc8..ee5e24b10 100644 --- a/internal/middleware/limithandler/limithandler_test.go +++ b/internal/middleware/limithandler/limithandler_test.go @@ -207,12 +207,15 @@ func runServer(t *testing.T, s *server, opt ...grpc.ServerOption) (*grpc.Server, go grpcServer.Serve(lis) - return grpcServer, "unix://" + serverSocketPath + return grpcServer, serverSocketPath } func newClient(t *testing.T, serverSocketPath string) (pb.TestClient, *grpc.ClientConn) { connOpts := []grpc.DialOption{ grpc.WithInsecure(), + grpc.WithDialer(func(addr string, timeout time.Duration) (net.Conn, error) { + return net.DialTimeout("unix", addr, timeout) + }), } conn, err := grpc.Dial(serverSocketPath, connOpts...) if err != nil { |