// Code generated by MockGen. DO NOT EDIT. // Source: github.com/rancher/steve/pkg/sqlcache/db (interfaces: Rows,Client) // // Generated by this command: // // mockgen --build_flags=--mod=mod -package store -destination ./db_mocks_test.go github.com/rancher/steve/pkg/sqlcache/db Rows,Client // // Package store is a generated GoMock package. package store import ( context "context" sql "database/sql" reflect "reflect" db "github.com/rancher/steve/pkg/sqlcache/db" transaction "github.com/rancher/steve/pkg/sqlcache/db/transaction" gomock "go.uber.org/mock/gomock" ) // MockRows is a mock of Rows interface. type MockRows struct { ctrl *gomock.Controller recorder *MockRowsMockRecorder isgomock struct{} } // MockRowsMockRecorder is the mock recorder for MockRows. type MockRowsMockRecorder struct { mock *MockRows } // NewMockRows creates a new mock instance. func NewMockRows(ctrl *gomock.Controller) *MockRows { mock := &MockRows{ctrl: ctrl} mock.recorder = &MockRowsMockRecorder{mock} return mock } // EXPECT returns an object that allows the caller to indicate expected use. func (m *MockRows) EXPECT() *MockRowsMockRecorder { return m.recorder } // Close mocks base method. func (m *MockRows) Close() error { m.ctrl.T.Helper() ret := m.ctrl.Call(m, "Close") ret0, _ := ret[0].(error) return ret0 } // Close indicates an expected call of Close. func (mr *MockRowsMockRecorder) Close() *gomock.Call { mr.mock.ctrl.T.Helper() return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Close", reflect.TypeOf((*MockRows)(nil).Close)) } // Err mocks base method. func (m *MockRows) Err() error { m.ctrl.T.Helper() ret := m.ctrl.Call(m, "Err") ret0, _ := ret[0].(error) return ret0 } // Err indicates an expected call of Err. func (mr *MockRowsMockRecorder) Err() *gomock.Call { mr.mock.ctrl.T.Helper() return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Err", reflect.TypeOf((*MockRows)(nil).Err)) } // Next mocks base method. func (m *MockRows) Next() bool { m.ctrl.T.Helper() ret := m.ctrl.Call(m, "Next") ret0, _ := ret[0].(bool) return ret0 } // Next indicates an expected call of Next. func (mr *MockRowsMockRecorder) Next() *gomock.Call { mr.mock.ctrl.T.Helper() return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Next", reflect.TypeOf((*MockRows)(nil).Next)) } // Scan mocks base method. func (m *MockRows) Scan(dest ...any) error { m.ctrl.T.Helper() varargs := []any{} for _, a := range dest { varargs = append(varargs, a) } ret := m.ctrl.Call(m, "Scan", varargs...) ret0, _ := ret[0].(error) return ret0 } // Scan indicates an expected call of Scan. func (mr *MockRowsMockRecorder) Scan(dest ...any) *gomock.Call { mr.mock.ctrl.T.Helper() return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Scan", reflect.TypeOf((*MockRows)(nil).Scan), dest...) } // MockClient is a mock of Client interface. type MockClient struct { ctrl *gomock.Controller recorder *MockClientMockRecorder isgomock struct{} } // MockClientMockRecorder is the mock recorder for MockClient. type MockClientMockRecorder struct { mock *MockClient } // NewMockClient creates a new mock instance. func NewMockClient(ctrl *gomock.Controller) *MockClient { mock := &MockClient{ctrl: ctrl} mock.recorder = &MockClientMockRecorder{mock} return mock } // EXPECT returns an object that allows the caller to indicate expected use. func (m *MockClient) EXPECT() *MockClientMockRecorder { return m.recorder } // CloseStmt mocks base method. func (m *MockClient) CloseStmt(closable db.Closable) error { m.ctrl.T.Helper() ret := m.ctrl.Call(m, "CloseStmt", closable) ret0, _ := ret[0].(error) return ret0 } // CloseStmt indicates an expected call of CloseStmt. func (mr *MockClientMockRecorder) CloseStmt(closable any) *gomock.Call { mr.mock.ctrl.T.Helper() return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "CloseStmt", reflect.TypeOf((*MockClient)(nil).CloseStmt), closable) } // NewConnection mocks base method. func (m *MockClient) NewConnection(isTemp bool) (string, error) { m.ctrl.T.Helper() ret := m.ctrl.Call(m, "NewConnection", isTemp) ret0, _ := ret[0].(string) ret1, _ := ret[1].(error) return ret0, ret1 } // NewConnection indicates an expected call of NewConnection. func (mr *MockClientMockRecorder) NewConnection(isTemp any) *gomock.Call { mr.mock.ctrl.T.Helper() return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "NewConnection", reflect.TypeOf((*MockClient)(nil).NewConnection), isTemp) } // Prepare mocks base method. func (m *MockClient) Prepare(stmt string) *sql.Stmt { m.ctrl.T.Helper() ret := m.ctrl.Call(m, "Prepare", stmt) ret0, _ := ret[0].(*sql.Stmt) return ret0 } // Prepare indicates an expected call of Prepare. func (mr *MockClientMockRecorder) Prepare(stmt any) *gomock.Call { mr.mock.ctrl.T.Helper() return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Prepare", reflect.TypeOf((*MockClient)(nil).Prepare), stmt) } // QueryForRows mocks base method. func (m *MockClient) QueryForRows(ctx context.Context, stmt transaction.Stmt, params ...any) (*sql.Rows, error) { m.ctrl.T.Helper() varargs := []any{ctx, stmt} for _, a := range params { varargs = append(varargs, a) } ret := m.ctrl.Call(m, "QueryForRows", varargs...) ret0, _ := ret[0].(*sql.Rows) ret1, _ := ret[1].(error) return ret0, ret1 } // QueryForRows indicates an expected call of QueryForRows. func (mr *MockClientMockRecorder) QueryForRows(ctx, stmt any, params ...any) *gomock.Call { mr.mock.ctrl.T.Helper() varargs := append([]any{ctx, stmt}, params...) return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "QueryForRows", reflect.TypeOf((*MockClient)(nil).QueryForRows), varargs...) } // ReadInt mocks base method. func (m *MockClient) ReadInt(rows db.Rows) (int, error) { m.ctrl.T.Helper() ret := m.ctrl.Call(m, "ReadInt", rows) ret0, _ := ret[0].(int) ret1, _ := ret[1].(error) return ret0, ret1 } // ReadInt indicates an expected call of ReadInt. func (mr *MockClientMockRecorder) ReadInt(rows any) *gomock.Call { mr.mock.ctrl.T.Helper() return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ReadInt", reflect.TypeOf((*MockClient)(nil).ReadInt), rows) } // ReadObjects mocks base method. func (m *MockClient) ReadObjects(rows db.Rows, typ reflect.Type, shouldDecrypt bool) ([]any, error) { m.ctrl.T.Helper() ret := m.ctrl.Call(m, "ReadObjects", rows, typ, shouldDecrypt) ret0, _ := ret[0].([]any) ret1, _ := ret[1].(error) return ret0, ret1 } // ReadObjects indicates an expected call of ReadObjects. func (mr *MockClientMockRecorder) ReadObjects(rows, typ, shouldDecrypt any) *gomock.Call { mr.mock.ctrl.T.Helper() return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ReadObjects", reflect.TypeOf((*MockClient)(nil).ReadObjects), rows, typ, shouldDecrypt) } // ReadStrings mocks base method. func (m *MockClient) ReadStrings(rows db.Rows) ([]string, error) { m.ctrl.T.Helper() ret := m.ctrl.Call(m, "ReadStrings", rows) ret0, _ := ret[0].([]string) ret1, _ := ret[1].(error) return ret0, ret1 } // ReadStrings indicates an expected call of ReadStrings. func (mr *MockClientMockRecorder) ReadStrings(rows any) *gomock.Call { mr.mock.ctrl.T.Helper() return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ReadStrings", reflect.TypeOf((*MockClient)(nil).ReadStrings), rows) } // ReadStrings2 mocks base method. func (m *MockClient) ReadStrings2(rows db.Rows) ([][]string, error) { m.ctrl.T.Helper() ret := m.ctrl.Call(m, "ReadStrings2", rows) ret0, _ := ret[0].([][]string) ret1, _ := ret[1].(error) return ret0, ret1 } // ReadStrings2 indicates an expected call of ReadStrings2. func (mr *MockClientMockRecorder) ReadStrings2(rows any) *gomock.Call { mr.mock.ctrl.T.Helper() return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ReadStrings2", reflect.TypeOf((*MockClient)(nil).ReadStrings2), rows) } // Upsert mocks base method. func (m *MockClient) Upsert(tx transaction.Client, stmt *sql.Stmt, key string, obj any, shouldEncrypt bool) error { m.ctrl.T.Helper() ret := m.ctrl.Call(m, "Upsert", tx, stmt, key, obj, shouldEncrypt) ret0, _ := ret[0].(error) return ret0 } // Upsert indicates an expected call of Upsert. func (mr *MockClientMockRecorder) Upsert(tx, stmt, key, obj, shouldEncrypt any) *gomock.Call { mr.mock.ctrl.T.Helper() return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Upsert", reflect.TypeOf((*MockClient)(nil).Upsert), tx, stmt, key, obj, shouldEncrypt) } // WithTransaction mocks base method. func (m *MockClient) WithTransaction(ctx context.Context, forWriting bool, f db.WithTransactionFunction) error { m.ctrl.T.Helper() ret := m.ctrl.Call(m, "WithTransaction", ctx, forWriting, f) ret0, _ := ret[0].(error) return ret0 } // WithTransaction indicates an expected call of WithTransaction. func (mr *MockClientMockRecorder) WithTransaction(ctx, forWriting, f any) *gomock.Call { mr.mock.ctrl.T.Helper() return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "WithTransaction", reflect.TypeOf((*MockClient)(nil).WithTransaction), ctx, forWriting, f) }