github.com/letsencrypt/trillian@v1.1.2-0.20180615153820-ae375a99d36a/quota/mock_quota.go (about)

     1  // Code generated by MockGen. DO NOT EDIT.
     2  // Source: github.com/google/trillian/quota (interfaces: Manager)
     3  
     4  // Package quota is a generated GoMock package.
     5  package quota
     6  
     7  import (
     8  	context "context"
     9  	gomock "github.com/golang/mock/gomock"
    10  	reflect "reflect"
    11  )
    12  
    13  // MockManager is a mock of Manager interface
    14  type MockManager struct {
    15  	ctrl     *gomock.Controller
    16  	recorder *MockManagerMockRecorder
    17  }
    18  
    19  // MockManagerMockRecorder is the mock recorder for MockManager
    20  type MockManagerMockRecorder struct {
    21  	mock *MockManager
    22  }
    23  
    24  // NewMockManager creates a new mock instance
    25  func NewMockManager(ctrl *gomock.Controller) *MockManager {
    26  	mock := &MockManager{ctrl: ctrl}
    27  	mock.recorder = &MockManagerMockRecorder{mock}
    28  	return mock
    29  }
    30  
    31  // EXPECT returns an object that allows the caller to indicate expected use
    32  func (m *MockManager) EXPECT() *MockManagerMockRecorder {
    33  	return m.recorder
    34  }
    35  
    36  // GetTokens mocks base method
    37  func (m *MockManager) GetTokens(arg0 context.Context, arg1 int, arg2 []Spec) error {
    38  	ret := m.ctrl.Call(m, "GetTokens", arg0, arg1, arg2)
    39  	ret0, _ := ret[0].(error)
    40  	return ret0
    41  }
    42  
    43  // GetTokens indicates an expected call of GetTokens
    44  func (mr *MockManagerMockRecorder) GetTokens(arg0, arg1, arg2 interface{}) *gomock.Call {
    45  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetTokens", reflect.TypeOf((*MockManager)(nil).GetTokens), arg0, arg1, arg2)
    46  }
    47  
    48  // GetUser mocks base method
    49  func (m *MockManager) GetUser(arg0 context.Context, arg1 interface{}) string {
    50  	ret := m.ctrl.Call(m, "GetUser", arg0, arg1)
    51  	ret0, _ := ret[0].(string)
    52  	return ret0
    53  }
    54  
    55  // GetUser indicates an expected call of GetUser
    56  func (mr *MockManagerMockRecorder) GetUser(arg0, arg1 interface{}) *gomock.Call {
    57  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetUser", reflect.TypeOf((*MockManager)(nil).GetUser), arg0, arg1)
    58  }
    59  
    60  // PeekTokens mocks base method
    61  func (m *MockManager) PeekTokens(arg0 context.Context, arg1 []Spec) (map[Spec]int, error) {
    62  	ret := m.ctrl.Call(m, "PeekTokens", arg0, arg1)
    63  	ret0, _ := ret[0].(map[Spec]int)
    64  	ret1, _ := ret[1].(error)
    65  	return ret0, ret1
    66  }
    67  
    68  // PeekTokens indicates an expected call of PeekTokens
    69  func (mr *MockManagerMockRecorder) PeekTokens(arg0, arg1 interface{}) *gomock.Call {
    70  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "PeekTokens", reflect.TypeOf((*MockManager)(nil).PeekTokens), arg0, arg1)
    71  }
    72  
    73  // PutTokens mocks base method
    74  func (m *MockManager) PutTokens(arg0 context.Context, arg1 int, arg2 []Spec) error {
    75  	ret := m.ctrl.Call(m, "PutTokens", arg0, arg1, arg2)
    76  	ret0, _ := ret[0].(error)
    77  	return ret0
    78  }
    79  
    80  // PutTokens indicates an expected call of PutTokens
    81  func (mr *MockManagerMockRecorder) PutTokens(arg0, arg1, arg2 interface{}) *gomock.Call {
    82  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "PutTokens", reflect.TypeOf((*MockManager)(nil).PutTokens), arg0, arg1, arg2)
    83  }
    84  
    85  // ResetQuota mocks base method
    86  func (m *MockManager) ResetQuota(arg0 context.Context, arg1 []Spec) error {
    87  	ret := m.ctrl.Call(m, "ResetQuota", arg0, arg1)
    88  	ret0, _ := ret[0].(error)
    89  	return ret0
    90  }
    91  
    92  // ResetQuota indicates an expected call of ResetQuota
    93  func (mr *MockManagerMockRecorder) ResetQuota(arg0, arg1 interface{}) *gomock.Call {
    94  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ResetQuota", reflect.TypeOf((*MockManager)(nil).ResetQuota), arg0, arg1)
    95  }