mirror of
https://github.com/fnproject/fn.git
synced 2022-10-28 21:29:17 +03:00
e637661ea2
* Adding a way to inject a request ID It is very useful to associate a request ID to each incoming request, this change allows to provide a function to do that via Server Option. The change comes with a default function which will generate a new request ID. The request ID is put in the request context along with a common logger which always logs the request-id We add gRPC interceptors to the server so it can get the request ID out of the gRPC metadata and put it in the common logger stored in the context so as all the log lines using the common logger from the context will have the request ID logged
112 lines
3.3 KiB
Go
112 lines
3.3 KiB
Go
// Copyright 2016 Michal Witkowski. All Rights Reserved.
|
|
// See LICENSE for licensing terms.
|
|
|
|
package grpc_testing
|
|
|
|
import (
|
|
"net"
|
|
"time"
|
|
|
|
"flag"
|
|
"path"
|
|
"runtime"
|
|
|
|
pb_testproto "github.com/grpc-ecosystem/go-grpc-middleware/testing/testproto"
|
|
"github.com/stretchr/testify/require"
|
|
"github.com/stretchr/testify/suite"
|
|
"golang.org/x/net/context"
|
|
"google.golang.org/grpc"
|
|
"google.golang.org/grpc/credentials"
|
|
)
|
|
|
|
var (
|
|
flagTls = flag.Bool("use_tls", true, "whether all gRPC middleware tests should use tls")
|
|
)
|
|
|
|
func getTestingCertsPath() string {
|
|
_, callerPath, _, _ := runtime.Caller(0)
|
|
return path.Join(path.Dir(callerPath), "certs")
|
|
}
|
|
|
|
// InterceptorTestSuite is a testify/Suite that starts a gRPC PingService server and a client.
|
|
type InterceptorTestSuite struct {
|
|
suite.Suite
|
|
|
|
TestService pb_testproto.TestServiceServer
|
|
ServerOpts []grpc.ServerOption
|
|
ClientOpts []grpc.DialOption
|
|
|
|
ServerListener net.Listener
|
|
Server *grpc.Server
|
|
clientConn *grpc.ClientConn
|
|
Client pb_testproto.TestServiceClient
|
|
}
|
|
|
|
func (s *InterceptorTestSuite) SetupSuite() {
|
|
var err error
|
|
s.ServerListener, err = net.Listen("tcp", "127.0.0.1:0")
|
|
require.NoError(s.T(), err, "must be able to allocate a port for serverListener")
|
|
if *flagTls {
|
|
creds, err := credentials.NewServerTLSFromFile(
|
|
path.Join(getTestingCertsPath(), "localhost.crt"),
|
|
path.Join(getTestingCertsPath(), "localhost.key"),
|
|
)
|
|
require.NoError(s.T(), err, "failed reading server credentials for localhost.crt")
|
|
s.ServerOpts = append(s.ServerOpts, grpc.Creds(creds))
|
|
}
|
|
// This is the point where we hook up the interceptor
|
|
s.Server = grpc.NewServer(s.ServerOpts...)
|
|
// Crete a service of the instantiator hasn't provided one.
|
|
if s.TestService == nil {
|
|
s.TestService = &TestPingService{T: s.T()}
|
|
}
|
|
pb_testproto.RegisterTestServiceServer(s.Server, s.TestService)
|
|
|
|
go func() {
|
|
s.Server.Serve(s.ServerListener)
|
|
}()
|
|
s.Client = s.NewClient(s.ClientOpts...)
|
|
}
|
|
|
|
func (s *InterceptorTestSuite) NewClient(dialOpts ...grpc.DialOption) pb_testproto.TestServiceClient {
|
|
newDialOpts := append(dialOpts, grpc.WithBlock(), grpc.WithTimeout(2*time.Second))
|
|
if *flagTls {
|
|
creds, err := credentials.NewClientTLSFromFile(
|
|
path.Join(getTestingCertsPath(), "localhost.crt"), "localhost")
|
|
require.NoError(s.T(), err, "failed reading client credentials for localhost.crt")
|
|
newDialOpts = append(newDialOpts, grpc.WithTransportCredentials(creds))
|
|
} else {
|
|
newDialOpts = append(newDialOpts, grpc.WithInsecure())
|
|
}
|
|
clientConn, err := grpc.Dial(s.ServerAddr(), newDialOpts...)
|
|
require.NoError(s.T(), err, "must not error on client Dial")
|
|
return pb_testproto.NewTestServiceClient(clientConn)
|
|
}
|
|
|
|
func (s *InterceptorTestSuite) ServerAddr() string {
|
|
return s.ServerListener.Addr().String()
|
|
}
|
|
|
|
func (s *InterceptorTestSuite) SimpleCtx() context.Context {
|
|
ctx, _ := context.WithTimeout(context.TODO(), 2*time.Second)
|
|
return ctx
|
|
}
|
|
|
|
func (s *InterceptorTestSuite) DeadlineCtx(deadline time.Time) context.Context {
|
|
ctx, _ := context.WithDeadline(context.TODO(), deadline)
|
|
return ctx
|
|
}
|
|
|
|
func (s *InterceptorTestSuite) TearDownSuite() {
|
|
time.Sleep(10 * time.Millisecond)
|
|
if s.ServerListener != nil {
|
|
s.Server.GracefulStop()
|
|
s.T().Logf("stopped grpc.Server at: %v", s.ServerAddr())
|
|
s.ServerListener.Close()
|
|
|
|
}
|
|
if s.clientConn != nil {
|
|
s.clientConn.Close()
|
|
}
|
|
}
|