Fix the test

Signed-off-by: JmPotato <ghzpotato@gmail.com>
tikv · Feb 8, 2023 · 7eb528b · 7eb528b
1 parent 78082cf
commit 7eb528b
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/tests/mcs/resource_manager/resource_manager_test.go b/tests/mcs/resource_manager/resource_manager_test.go
@@ -233,6 +233,7 @@ func (ti *testRequestInfo) WriteBytes() uint64 {
 type testResponseInfo struct {
 	cpuMs     uint64
 	readBytes uint64
+	succeed   bool
 }
 
 func (tri *testResponseInfo) ReadBytes() uint64 {
@@ -243,6 +244,10 @@ func (tri *testResponseInfo) KVCPUMs() uint64 {
 	return tri.cpuMs
 }
 
+func (tri *testResponseInfo) Succeed() bool {
+	return tri.succeed
+}
+
 type tokenConsumptionPerSecond struct {
 	rruTokensAtATime float64
 	wruTokensAtATime float64
@@ -275,6 +280,7 @@ func (t tokenConsumptionPerSecond) makeWriteResponse() *testResponseInfo {
 	return &testResponseInfo{
 		readBytes: 0,
 		cpuMs:     0,
+		succeed:   true,
 	}
 }