// Test gRPC interface using a vtworker and vtworkerclient. func TestVtworkerServer(t *testing.T) { wi := vtworkerclient.CreateWorkerInstance(t) // Listen on a random port. listener, err := net.Listen("tcp", ":0") if err != nil { t.Fatalf("Cannot listen: %v", err) } port := listener.Addr().(*net.TCPAddr).Port // Create a gRPC server and listen on the port. server := grpc.NewServer() vtworkerservicepb.RegisterVtworkerServer(server, grpcvtworkerserver.NewVtworkerServer(wi)) go server.Serve(listener) // Create a VtworkerClient gRPC client to talk to the vtworker. client, err := gRPCVtworkerClientFactory(fmt.Sprintf("localhost:%v", port), 30*time.Second) if err != nil { t.Fatalf("Cannot create client: %v", err) } defer client.Close() vtworkerclient.TestSuite(t, wi, client) }
// StartServer registers the VtworkerServer for RPCs func StartServer(s *grpc.Server, wi *worker.Instance) { pbs.RegisterVtworkerServer(s, NewVtworkerServer(wi)) }
// StartServer registers the VtworkerServer for RPCs func StartServer(s *grpc.Server, wi *worker.Instance) { vtworkerservicepb.RegisterVtworkerServer(s, NewVtworkerServer(wi)) }