Update tests for changes in gRPC

Signed-off-by: Sebastiaan van Stijn <github@gone.nl>
moby · May 11, 2018 · 2a14d4a · 2a14d4a
1 parent 19605fa
commit 2a14d4a
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/manager/state/raft/raft_test.go b/manager/state/raft/raft_test.go
@@ -983,9 +983,9 @@ func TestStreamRaftMessage(t *testing.T) {
 	assert.Error(t, err, "Received unexpected error EOF")
 
 	_, err = stream.CloseAndRecv()
-	errStr := fmt.Sprintf("grpc: received message length %d exceeding the max size %d", raftMsg.Size(), transport.GRPCMaxMsgSize)
+	errStr := fmt.Sprintf("grpc: received message larger than max (%d vs. %d)", raftMsg.Size(), transport.GRPCMaxMsgSize)
 	s, _ := status.FromError(err)
-	assert.Equal(t, codes.Internal, s.Code())
+	assert.Equal(t, codes.ResourceExhausted, s.Code())
 	assert.Equal(t, errStr, s.Message())
 
 	// Sending multiple snap messages with different indexes