update

Signed-off-by: husharp <jinhao.hu@pingcap.com>
HuSharp · Jul 13, 2023 · 2db9f9a · 2db9f9a
1 parent 242132e
commit 2db9f9a
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 19 deletions.
diff --git a/client/resource_group/controller/model.go b/client/resource_group/controller/model.go
@@ -18,7 +18,7 @@ import (
 	"os"
 	"time"
 
-	"github.com/cloudfoundry/gosigar"
+	sigar "github.com/cloudfoundry/gosigar"
 	"go.uber.org/zap"
 
 	rmpb "github.com/pingcap/kvproto/pkg/resource_manager"
@@ -37,7 +37,6 @@ type RequestInfo interface {
 	WriteBytes() uint64
 	ReplicaNumber() int64
 	StoreID() uint64
-	RequestSource() string
 }
 
 // ResponseInfo is the interface of the response information provider. A response should be

diff --git a/client/resource_group/controller/testutil.go b/client/resource_group/controller/testutil.go
@@ -57,16 +57,6 @@ func (tri *TestRequestInfo) ReplicaNumber() int64 {
 	return 1
 }
 
-// RequestSource implements the RequestInfo interface.
-func (tri *TestRequestInfo) RequestSource() string {
-	return tri.requestSource
-}
-
-// SetRequestSource set the request source.
-func (tri *TestRequestInfo) SetRequestSource(requestSource string) {
-	tri.requestSource = requestSource
-}
-
 // TestResponseInfo is used to test the response info interface.
 type TestResponseInfo struct {
 	readBytes uint64

diff --git a/tests/integrations/mcs/resourcemanager/resource_manager_test.go b/tests/integrations/mcs/resourcemanager/resource_manager_test.go
@@ -1165,15 +1165,11 @@ func (suite *resourceManagerClientTestSuite) TestSkipConsumptionForBackgroundJob
 	c.Start(suite.ctx)
 
 	resourceGroupName := suite.initGroups[1].Name
-	req := controller.NewTestRequestInfo(false, 0, 1)
-	re.False(c.IsBackgroundRequest(suite.ctx, resourceGroupName, req.RequestSource()))
+	re.False(c.IsBackgroundRequest(suite.ctx, resourceGroupName, "internal_default"))
 
 	resourceGroupName = "background_job"
-	req.SetRequestSource("internal_br")
-	re.True(c.IsBackgroundRequest(suite.ctx, resourceGroupName, req.RequestSource()))
-
-	req.SetRequestSource("internal_lightning")
-	re.True(c.IsBackgroundRequest(suite.ctx, resourceGroupName, req.RequestSource()))
+	re.True(c.IsBackgroundRequest(suite.ctx, resourceGroupName, "internal_br"))
+	re.True(c.IsBackgroundRequest(suite.ctx, resourceGroupName, "internal_lightning"))
 
 	c.Stop()
 }