From 41384b908b869a39a0e85a574d92e9950d2d4985 Mon Sep 17 00:00:00 2001 From: Ryan Leung Date: Wed, 9 Oct 2024 17:39:06 +0800 Subject: [PATCH] fix Signed-off-by: Ryan Leung --- tests/server/server_test.go | 19 ++++++++++--------- 1 file changed, 10 insertions(+), 9 deletions(-) diff --git a/tests/server/server_test.go b/tests/server/server_test.go index 68c4d5da65e..775c4ec8237 100644 --- 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{}{} }