Changing tests to improve coverage

topfreegames · Jul 21, 2018 · eb9ed2c · eb9ed2c
1 parent 6c2547c
commit eb9ed2c
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/rpc_test.go b/rpc_test.go
@@ -83,7 +83,7 @@ func TestDoSendRPC(t *testing.T) {
 				mockSD.EXPECT().GetServer("myserver").Return(&cluster.Server{}, nil)
 				mockRPCClient.EXPECT().Call(ctx, protos.RPCType_User, gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any()).Return(&protos.Response{Data: b}, nil)
 			}
-			err := doSendRPC(ctx, "myserver", table.routeStr, table.reply, table.arg)
+			err := RPCTo(ctx, "myserver", table.routeStr, table.reply, table.arg)
 			assert.Equal(t, table.err, err)
 		})
 	}

diff --git a/service/remote_test.go b/service/remote_test.go
@@ -385,7 +385,7 @@ func TestRemoteServiceRemoteProcess(t *testing.T) {
 
 			expectedMsg := &message.Message{
 				ID:    uint(rand.Int()),
-				Type:  message.Request,
+				Type:  table.msgType,
 				Route: rt.Short(),
 				Data:  []byte("ok"),
 			}