package grpc import ( "context" "net" "testing" "time" "google.golang.org/grpc" pgrpc "google.golang.org/grpc" pb "google.golang.org/grpc/examples/helloworld/helloworld" ) func testPool(t *testing.T, size int, ttl time.Duration) { // setup server l, err := net.Listen("tcp", ":0") if err != nil { t.Fatalf("failed to listen: %v", err) } defer l.Close() s := pgrpc.NewServer() pb.RegisterGreeterServer(s, &greeterServer{}) go s.Serve(l) defer s.Stop() // zero pool p := newPool(size, ttl) for i := 0; i < 10; i++ { // get a conn cc, err := p.getConn(l.Addr().String(), grpc.WithInsecure()) if err != nil { t.Fatal(err) } rsp := pb.HelloReply{} err = cc.Invoke(context.TODO(), "/helloworld.Greeter/SayHello", &pb.HelloRequest{Name: "John"}, &rsp) if err != nil { t.Fatal(err) } if rsp.Message != "Hello John" { t.Fatalf("Got unexpected response %v", rsp.Message) } // release the conn p.release(l.Addr().String(), cc, nil) p.Lock() if i := len(p.conns[l.Addr().String()]); i > size { p.Unlock() t.Fatalf("pool size %d is greater than expected %d", i, size) } p.Unlock() } } func TestGRPCPool(t *testing.T) { testPool(t, 0, time.Minute) testPool(t, 2, time.Minute) }