go.chromium.org/luci@v0.0.0-20240309015107-7cdc2e660f33/server/tq/internal/testutil/db.mock.go (about) 1 // Code generated by MockGen. DO NOT EDIT. 2 // Source: ../db/db.go 3 4 package testutil 5 6 import ( 7 context "context" 8 reflect "reflect" 9 10 gomock "github.com/golang/mock/gomock" 11 reminder "go.chromium.org/luci/server/tq/internal/reminder" 12 ) 13 14 // MockDB is a mock of DB interface. 15 type MockDB struct { 16 ctrl *gomock.Controller 17 recorder *MockDBMockRecorder 18 } 19 20 // MockDBMockRecorder is the mock recorder for MockDB. 21 type MockDBMockRecorder struct { 22 mock *MockDB 23 } 24 25 // NewMockDB creates a new mock instance. 26 func NewMockDB(ctrl *gomock.Controller) *MockDB { 27 mock := &MockDB{ctrl: ctrl} 28 mock.recorder = &MockDBMockRecorder{mock} 29 return mock 30 } 31 32 // EXPECT returns an object that allows the caller to indicate expected use. 33 func (m *MockDB) EXPECT() *MockDBMockRecorder { 34 return m.recorder 35 } 36 37 // Defer mocks base method. 38 func (m *MockDB) Defer(arg0 context.Context, arg1 func(context.Context)) { 39 m.ctrl.T.Helper() 40 m.ctrl.Call(m, "Defer", arg0, arg1) 41 } 42 43 // Defer indicates an expected call of Defer. 44 func (mr *MockDBMockRecorder) Defer(arg0, arg1 interface{}) *gomock.Call { 45 mr.mock.ctrl.T.Helper() 46 return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Defer", reflect.TypeOf((*MockDB)(nil).Defer), arg0, arg1) 47 } 48 49 // DeleteReminder mocks base method. 50 func (m *MockDB) DeleteReminder(arg0 context.Context, arg1 *reminder.Reminder) error { 51 m.ctrl.T.Helper() 52 ret := m.ctrl.Call(m, "DeleteReminder", arg0, arg1) 53 ret0, _ := ret[0].(error) 54 return ret0 55 } 56 57 // DeleteReminder indicates an expected call of DeleteReminder. 58 func (mr *MockDBMockRecorder) DeleteReminder(arg0, arg1 interface{}) *gomock.Call { 59 mr.mock.ctrl.T.Helper() 60 return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "DeleteReminder", reflect.TypeOf((*MockDB)(nil).DeleteReminder), arg0, arg1) 61 } 62 63 // FetchReminderRawPayloads mocks base method. 64 func (m *MockDB) FetchReminderRawPayloads(arg0 context.Context, arg1 []*reminder.Reminder) ([]*reminder.Reminder, error) { 65 m.ctrl.T.Helper() 66 ret := m.ctrl.Call(m, "FetchReminderRawPayloads", arg0, arg1) 67 ret0, _ := ret[0].([]*reminder.Reminder) 68 ret1, _ := ret[1].(error) 69 return ret0, ret1 70 } 71 72 // FetchReminderRawPayloads indicates an expected call of FetchReminderRawPayloads. 73 func (mr *MockDBMockRecorder) FetchReminderRawPayloads(arg0, arg1 interface{}) *gomock.Call { 74 mr.mock.ctrl.T.Helper() 75 return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "FetchReminderRawPayloads", reflect.TypeOf((*MockDB)(nil).FetchReminderRawPayloads), arg0, arg1) 76 } 77 78 // FetchRemindersMeta mocks base method. 79 func (m *MockDB) FetchRemindersMeta(ctx context.Context, low, high string, limit int) ([]*reminder.Reminder, error) { 80 m.ctrl.T.Helper() 81 ret := m.ctrl.Call(m, "FetchRemindersMeta", ctx, low, high, limit) 82 ret0, _ := ret[0].([]*reminder.Reminder) 83 ret1, _ := ret[1].(error) 84 return ret0, ret1 85 } 86 87 // FetchRemindersMeta indicates an expected call of FetchRemindersMeta. 88 func (mr *MockDBMockRecorder) FetchRemindersMeta(ctx, low, high, limit interface{}) *gomock.Call { 89 mr.mock.ctrl.T.Helper() 90 return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "FetchRemindersMeta", reflect.TypeOf((*MockDB)(nil).FetchRemindersMeta), ctx, low, high, limit) 91 } 92 93 // Kind mocks base method. 94 func (m *MockDB) Kind() string { 95 m.ctrl.T.Helper() 96 ret := m.ctrl.Call(m, "Kind") 97 ret0, _ := ret[0].(string) 98 return ret0 99 } 100 101 // Kind indicates an expected call of Kind. 102 func (mr *MockDBMockRecorder) Kind() *gomock.Call { 103 mr.mock.ctrl.T.Helper() 104 return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Kind", reflect.TypeOf((*MockDB)(nil).Kind)) 105 } 106 107 // SaveReminder mocks base method. 108 func (m *MockDB) SaveReminder(arg0 context.Context, arg1 *reminder.Reminder) error { 109 m.ctrl.T.Helper() 110 ret := m.ctrl.Call(m, "SaveReminder", arg0, arg1) 111 ret0, _ := ret[0].(error) 112 return ret0 113 } 114 115 // SaveReminder indicates an expected call of SaveReminder. 116 func (mr *MockDBMockRecorder) SaveReminder(arg0, arg1 interface{}) *gomock.Call { 117 mr.mock.ctrl.T.Helper() 118 return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "SaveReminder", reflect.TypeOf((*MockDB)(nil).SaveReminder), arg0, arg1) 119 }