github.com/masterhung0112/hk_server/v5@v5.0.0-20220302090640-ec71aef15e1c/store/storetest/mocks/TokenStore.go (about)

     1  // Code generated by mockery v1.0.0. DO NOT EDIT.
     2  
     3  // 'Regenerate
     4  
     5  package mocks
     6  
     7  import (
     8  	model "github.com/masterhung0112/hk_server/v5/model"
     9  	mock "github.com/stretchr/testify/mock"
    10  )
    11  
    12  // TokenStore is an autogenerated mock type for the TokenStore type
    13  type TokenStore struct {
    14  	mock.Mock
    15  }
    16  
    17  // Cleanup provides a mock function with given fields:
    18  func (_m *TokenStore) Cleanup() {
    19  	_m.Called()
    20  }
    21  
    22  // Delete provides a mock function with given fields: token
    23  func (_m *TokenStore) Delete(token string) error {
    24  	ret := _m.Called(token)
    25  
    26  	var r0 error
    27  	if rf, ok := ret.Get(0).(func(string) error); ok {
    28  		r0 = rf(token)
    29  	} else {
    30  		r0 = ret.Error(0)
    31  	}
    32  
    33  	return r0
    34  }
    35  
    36  // GetAllTokensByType provides a mock function with given fields: tokenType
    37  func (_m *TokenStore) GetAllTokensByType(tokenType string) ([]*model.Token, error) {
    38  	ret := _m.Called(tokenType)
    39  
    40  	var r0 []*model.Token
    41  	if rf, ok := ret.Get(0).(func(string) []*model.Token); ok {
    42  		r0 = rf(tokenType)
    43  	} else {
    44  		if ret.Get(0) != nil {
    45  			r0 = ret.Get(0).([]*model.Token)
    46  		}
    47  	}
    48  
    49  	var r1 error
    50  	if rf, ok := ret.Get(1).(func(string) error); ok {
    51  		r1 = rf(tokenType)
    52  	} else {
    53  		r1 = ret.Error(1)
    54  	}
    55  
    56  	return r0, r1
    57  }
    58  
    59  // GetByToken provides a mock function with given fields: token
    60  func (_m *TokenStore) GetByToken(token string) (*model.Token, error) {
    61  	ret := _m.Called(token)
    62  
    63  	var r0 *model.Token
    64  	if rf, ok := ret.Get(0).(func(string) *model.Token); ok {
    65  		r0 = rf(token)
    66  	} else {
    67  		if ret.Get(0) != nil {
    68  			r0 = ret.Get(0).(*model.Token)
    69  		}
    70  	}
    71  
    72  	var r1 error
    73  	if rf, ok := ret.Get(1).(func(string) error); ok {
    74  		r1 = rf(token)
    75  	} else {
    76  		r1 = ret.Error(1)
    77  	}
    78  
    79  	return r0, r1
    80  }
    81  
    82  // RemoveAllTokensByType provides a mock function with given fields: tokenType
    83  func (_m *TokenStore) RemoveAllTokensByType(tokenType string) error {
    84  	ret := _m.Called(tokenType)
    85  
    86  	var r0 error
    87  	if rf, ok := ret.Get(0).(func(string) error); ok {
    88  		r0 = rf(tokenType)
    89  	} else {
    90  		r0 = ret.Error(0)
    91  	}
    92  
    93  	return r0
    94  }
    95  
    96  // Save provides a mock function with given fields: recovery
    97  func (_m *TokenStore) Save(recovery *model.Token) error {
    98  	ret := _m.Called(recovery)
    99  
   100  	var r0 error
   101  	if rf, ok := ret.Get(0).(func(*model.Token) error); ok {
   102  		r0 = rf(recovery)
   103  	} else {
   104  		r0 = ret.Error(0)
   105  	}
   106  
   107  	return r0
   108  }