feat: fix compile errors

polarismesh · Dec 26, 2021 · f669d33 · f669d33
1 parent b21a6a5
commit f669d33
Show file tree

Hide file tree

Showing 2 changed files with 18 additions and 6 deletions.
diff --git a/test/client_test.go b/test/client_test.go
@@ -54,15 +54,20 @@ func (s *clientTestingSuite) TearDownSuite(c *check.C) {
 }
 
 func (s *clientTestingSuite) TestClientCall(c *check.C) {
-	srv := polaris.NewServer(
-		polaris.WithServerApplication(serverSvc), polaris.WithServerNamespace(serverNamespace), polaris.WithTTL(2))
-	hello.RegisterHelloServer(srv.GRPCServer(), &helloServer{})
+	srv := grpc.NewServer()
+	hello.RegisterHelloServer(srv, &helloServer{})
 	listen, err := net.Listen("tcp", "0.0.0.0:0")
 	if err != nil {
 		log.Fatalf("Failed to listen: %v", err)
 	}
 	log.Printf("success to listen on %s\n", listen.Addr())
 	defer listen.Close()
+	pSrv, err := polaris.Register(srv, listen,
+		polaris.WithServerApplication(serverSvc), polaris.WithServerNamespace(serverNamespace), polaris.WithTTL(2))
+	if nil != err {
+		log.Fatal(err)
+	}
+	defer pSrv.Deregister()
 	go func() {
 		err = srv.Serve(listen)
 		if nil != err {

diff --git a/test/server_test.go b/test/server_test.go
@@ -24,6 +24,8 @@ import (
 	"net"
 	"time"
 
+	"google.golang.org/grpc"
+
 	polaris "github.com/polarismesh/grpc-go-polaris"
 	"github.com/polarismesh/grpc-go-polaris/test/hello"
 	"github.com/polarismesh/grpc-go-polaris/test/mock"
@@ -67,16 +69,21 @@ func (t *helloServer) SayHello(ctx context.Context, request *hello.HelloRequest)
 }
 
 func (s *serverTestingSuite) TestRegister(c *check.C) {
-	srv := polaris.NewServer(
-		polaris.WithServerApplication(serverSvc), polaris.WithServerNamespace(serverNamespace), polaris.WithTTL(2))
-	hello.RegisterHelloServer(srv.GRPCServer(), &helloServer{})
+	srv := grpc.NewServer()
+	hello.RegisterHelloServer(srv, &helloServer{})
 	// 监听端口
 	address := "127.0.0.1:8988"
 	listen, err := net.Listen("tcp", address)
 	if err != nil {
 		log.Fatalf("Failed to addr %s: %v", address, err)
 	}
 	defer listen.Close()
+	pSrv, err := polaris.Register(srv, listen,
+		polaris.WithServerApplication(serverSvc), polaris.WithServerNamespace(serverNamespace), polaris.WithTTL(2))
+	if nil != err {
+		log.Fatal(err)
+	}
+	defer pSrv.Deregister()
 	go func() {
 		err = srv.Serve(listen)
 		if nil != err {