fix

Signed-off-by: Ryan Leung <[email protected]>
tikv · Oct 9, 2024 · 41384b9 · 41384b9
1 parent 34b0620
commit 41384b9
Showing 1 changed file with 10 additions and 9 deletions.
diff --git a/tests/server/server_test.go b/tests/server/server_test.go
@@ -175,11 +175,12 @@ func TestGRPCRateLimit(t *testing.T) {
 			Header:    &pdpb.RequestHeader{ClusterId: leaderServer.GetClusterID()},
 			RegionKey: []byte(""),
 		})
-		re.NoError(err)
+		re.Empty(resp.GetHeader().GetError())
 		if i == 0 {
-			re.Empty(resp.GetHeader().GetError())
+			re.NoError(err)
 		} else {
-			re.Contains(resp.GetHeader().GetError().GetMessage(), "rate limit exceeded")
+			re.Error(err)
+			re.Contains(err.Error(), "rate limit exceeded")
 		}
 	}
 
@@ -219,9 +220,9 @@ func TestGRPCRateLimit(t *testing.T) {
 			Header:    &pdpb.RequestHeader{ClusterId: leaderServer.GetClusterID()},
 			RegionKey: []byte(""),
 		})
-		re.NoError(err)
-		if resp.GetHeader().GetError() != nil {
-			errCh <- resp.GetHeader().GetError().GetMessage()
+		re.Empty(resp.GetHeader().GetError())
+		if err != nil {
+			errCh <- err.Error()
 		} else {
 			okCh <- struct{}{}
 		}
@@ -234,9 +235,9 @@ func TestGRPCRateLimit(t *testing.T) {
 			Header:    &pdpb.RequestHeader{ClusterId: leaderServer.GetClusterID()},
 			RegionKey: []byte(""),
 		})
-		re.NoError(err)
-		if resp.GetHeader().GetError() != nil {
-			errCh <- resp.GetHeader().GetError().GetMessage()
+		re.Empty(resp.GetHeader().GetError())
+		if err != nil {
+			errCh <- err.Error()
 		} else {
 			okCh <- struct{}{}
 		}