rpcserver_test.go

package xrpc

import (
	"context"
	"github.com/grpc-ecosystem/grpc-gateway/v2/runtime"
	pb "github.com/mix-go/xrpc/api"
	"google.golang.org/grpc"
	"log"
	"os"
	"testing"
)

type service struct {
	pb.UnimplementedAppMessagesServer
}

func (t *service) Send(ctx context.Context, in *pb.SendRequest) (*pb.SendResponse, error) {
	log.Printf("%+v", in)
	return &pb.SendResponse{
		MessageId: 1,
	}, nil
}

func TestRPCServer_Serve(t *testing.T) {
	s := &RpcServer{
		GrpcServer: &GrpcServer{
			Addr: "0.0.0.0:50000",
			Registrar: func(s *grpc.Server) {
				pb.RegisterAppMessagesServer(s, &service{})
			},
		},
		GatewayServer: &GatewayServer{ // Optional
			Addr: "0.0.0.0:50001",
			Registrar: func(mux *runtime.ServeMux, conn *grpc.ClientConn) {
				pb.RegisterAppMessagesHandler(context.Background(), mux, conn)
			},
		},
		Logger: nil,
	}
	s.Serve()
	// s.Shutdown()
}

func TestRpcServerTLS_Serve(t *testing.T) {
	dir, _ := os.Getwd()
	tlsConf, err := LoadServerTLSConfig(dir+"/certificates/ca.pem", dir+"/certificates/server.pem", dir+"/certificates/server.key")
	if err != nil {
		log.Fatal(err)
	}
	tlsClientConf, err := LoadClientTLSConfig(dir+"/certificates/ca.pem", dir+"/certificates/client.pem", dir+"/certificates/client.key")
	if err != nil {
		log.Fatal(err)
	}
	s := &RpcServer{
		GrpcServer: &GrpcServer{
			Addr: "0.0.0.0:50000",
			Registrar: func(s *grpc.Server) {
				pb.RegisterAppMessagesServer(s, &service{})
			},
		},
		GatewayServer: &GatewayServer{ // Optional
			Addr: "0.0.0.0:50001",
			Registrar: func(mux *runtime.ServeMux, conn *grpc.ClientConn) {
				pb.RegisterAppMessagesHandler(context.Background(), mux, conn)
			},
		},
		Logger:          nil,
		TLSConfig:       tlsConf,
		TLSClientConfig: tlsClientConf,
	}
	s.Serve()
}