Fix testing app

izumin5210 · Apr 12, 2018 · 447abde · 447abde
1 parent 43681e3
commit 447abde
Show file tree

Hide file tree

Showing 2 changed files with 20 additions and 12 deletions.
diff --git a/testing/app/server/library_server.go b/testing/app/server/library_server.go
@@ -4,9 +4,7 @@ import (
 	"context"
 
 	"github.com/golang/protobuf/ptypes/empty"
-	"github.com/grpc-ecosystem/grpc-gateway/runtime"
 	"github.com/izumin5210/grapi/pkg/grapiserver"
-	"google.golang.org/grpc"
 	"google.golang.org/grpc/codes"
 	"google.golang.org/grpc/status"
 
@@ -24,16 +22,6 @@ func NewLibraryServiceServer() interface {
 type libraryServiceServerImpl struct {
 }
 
-// RegisterWithServer implements grapiserver.Server.RegisterWithServer.
-func (s *libraryServiceServerImpl) RegisterWithServer(grpcSvr *grpc.Server) {
-	api_pb.RegisterLibraryServiceServer(grpcSvr, s)
-}
-
-// RegisterWithHandler implements grapiserver.Server.RegisterWithHandler.
-func (s *libraryServiceServerImpl) RegisterWithHandler(ctx context.Context, mux *runtime.ServeMux, conn *grpc.ClientConn) error {
-	return api_pb.RegisterLibraryServiceHandler(ctx, mux, conn)
-}
-
 func (s *libraryServiceServerImpl) ListBooks(ctx context.Context, req *api_pb.ListBooksRequest) (*api_pb.ListBooksResponse, error) {
 	return &api_pb.ListBooksResponse{
 		Books: []*api_pb.Book{

diff --git a/testing/app/server/library_server_register_funcs.go b/testing/app/server/library_server_register_funcs.go
@@ -0,0 +1,20 @@
+package server
+
+import (
+	"context"
+
+	"github.com/grpc-ecosystem/grpc-gateway/runtime"
+	"google.golang.org/grpc"
+
+	api_pb "github.com/izumin5210/grapi/testing/api"
+)
+
+// RegisterWithServer implements grapiserver.Server.RegisterWithServer.
+func (s *libraryServiceServerImpl) RegisterWithServer(grpcSvr *grpc.Server) {
+	api_pb.RegisterLibraryServiceServer(grpcSvr, s)
+}
+
+// RegisterWithHandler implements grapiserver.Server.RegisterWithHandler.
+func (s *libraryServiceServerImpl) RegisterWithHandler(ctx context.Context, mux *runtime.ServeMux, conn *grpc.ClientConn) error {
+	return api_pb.RegisterLibraryServiceHandler(ctx, mux, conn)
+}