From fd869bdf21aeeff29204707fdcc92e4ddb320f72 Mon Sep 17 00:00:00 2001
From: wuhuizuo <wuhuizuo@126.com>
Date: Thu, 20 Jan 2022 14:00:23 +0000
Subject: [PATCH] feat: add method MockApiVersionsResponse.SetApiKeys

- testers can create a broker with custom supported api versions.
---
 mockresponses.go | 29 ++++++++++++++++++-----------
 1 file changed, 18 insertions(+), 11 deletions(-)

diff --git a/mockresponses.go b/mockresponses.go
index 4c9cac809..d0349e13a 100644
--- a/mockresponses.go
+++ b/mockresponses.go
@@ -1347,31 +1347,38 @@ func (m *MockDescribeLogDirsResponse) For(reqBody versionedDecoder) encoderWithH
 }
 
 type MockApiVersionsResponse struct {
-	t TestReporter
+	t       TestReporter
+	apiKeys []ApiVersionsResponseKey
 }
 
 func NewMockApiVersionsResponse(t TestReporter) *MockApiVersionsResponse {
-	return &MockApiVersionsResponse{t: t}
-}
-
-func (mr *MockApiVersionsResponse) For(reqBody versionedDecoder) encoderWithHeader {
-	req := reqBody.(*ApiVersionsRequest)
-	res := &ApiVersionsResponse{
-		Version: req.Version,
-		ApiKeys: []ApiVersionsResponseKey{
+	return &MockApiVersionsResponse{
+		t: t,
+		apiKeys: []ApiVersionsResponseKey{
 			{
-				Version:    req.Version,
 				ApiKey:     0,
 				MinVersion: 5,
 				MaxVersion: 8,
 			},
 			{
-				Version:    req.Version,
 				ApiKey:     1,
 				MinVersion: 7,
 				MaxVersion: 11,
 			},
 		},
 	}
+}
+
+func (m *MockApiVersionsResponse) SetApiKeys(apiKeys []ApiVersionsResponseKey) *MockApiVersionsResponse {
+	m.apiKeys = apiKeys
+	return m
+}
+
+func (m *MockApiVersionsResponse) For(reqBody versionedDecoder) encoderWithHeader {
+	req := reqBody.(*ApiVersionsRequest)
+	res := &ApiVersionsResponse{
+		Version: req.Version,
+		ApiKeys: m.apiKeys,
+	}
 	return res
 }