Automated cherry pick of #34247 (#34257)

Automatic Merge

server/channels/store/storetest/mocks/TokenStore.go

@@ -19,9 +19,9 @@ func (_m *TokenStore) Cleanup(expiryTime int64) {
 	_m.Called(expiryTime)
 }
 
-// ConsumeOnce provides a mock function with given fields: tokenStr
-func (_m *TokenStore) ConsumeOnce(tokenStr string) (*model.Token, error) {
-	ret := _m.Called(tokenStr)
+// ConsumeOnce provides a mock function with given fields: tokenType, tokenStr
+func (_m *TokenStore) ConsumeOnce(tokenType string, tokenStr string) (*model.Token, error) {
+	ret := _m.Called(tokenType, tokenStr)
 
 	if len(ret) == 0 {
 		panic("no return value specified for ConsumeOnce")
@@ -29,19 +29,19 @@ func (_m *TokenStore) ConsumeOnce(tokenStr string) (*model.Token, error) {
 
 	var r0 *model.Token
 	var r1 error
-	if rf, ok := ret.Get(0).(func(string) (*model.Token, error)); ok {
-		return rf(tokenStr)
+	if rf, ok := ret.Get(0).(func(string, string) (*model.Token, error)); ok {
+		return rf(tokenType, tokenStr)
 	}
-	if rf, ok := ret.Get(0).(func(string) *model.Token); ok {
-		r0 = rf(tokenStr)
+	if rf, ok := ret.Get(0).(func(string, string) *model.Token); ok {
+		r0 = rf(tokenType, tokenStr)
 	} else {
 		if ret.Get(0) != nil {
 			r0 = ret.Get(0).(*model.Token)
 		}
 	}
 
-	if rf, ok := ret.Get(1).(func(string) error); ok {
-		r1 = rf(tokenStr)
+	if rf, ok := ret.Get(1).(func(string, string) error); ok {
+		r1 = rf(tokenType, tokenStr)
 	} else {
 		r1 = ret.Error(1)
 	}