TRKJ/EdgeAPI
2
0
Fork 0
mirror of https://github.com/TeaOSLab/EdgeAPI.git synced 2025-11-14 16:30:26 +08:00
Code Issues Packages Projects Releases Wiki Activity

阶段性提交

Browse Source
This commit is contained in:
刘祥超
2020-07-24 09:17:48 +08:00
parent 23cca702b4
commit 943742b836
53 changed files with 2962 additions and 1696 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
internal/tests/grpc_test.go
View File

@@ -3,8 +3,8 @@ package tests
import (
"context"
"encoding/json"
"github.com/TeaOSLab/EdgeAPI/internal/rpc/admin"
nodepb "github.com/TeaOSLab/EdgeAPI/internal/rpc/node"
pb2 "github.com/TeaOSLab/EdgeAPI/internal/rpc/pb"
"github.com/TeaOSLab/EdgeAPI/internal/rpc/services"
pb "github.com/TeaOSLab/EdgeAPI/internal/tests/helloworld"
"google.golang.org/grpc"
"google.golang.org/grpc/credentials"
@@ -42,8 +42,7 @@ func TestTCPServer(t *testing.T) {
s := grpc.NewServer()
pb.RegisterGreeterServer(s, &server{})
nodepb.RegisterServiceServer(s, &nodepb.Service{})
admin.RegisterServiceServer(s, &admin.Service{})
pb2.RegisterNodeServiceServer(s, &services.NodeService{})
err = s.Serve(listener)
if err != nil {
@@ -85,13 +84,13 @@ func TestTCPClient_Node(t *testing.T) {
_ = conn.Close()
}()
c := nodepb.NewServiceClient(conn)
c := pb2.NewNodeServiceClient(conn)
before := time.Now()
ctx := context.Background()
ctx = metadata.AppendToOutgoingContext(ctx, "name", "liu", "age", "20")
reply, err := c.Config(ctx, &nodepb.ConfigRequest{
reply, err := c.Config(ctx, &pb2.ConfigRequest{
})
if err != nil {
t.Fatal(err)