// Code generated by MockGen. DO NOT EDIT. // Source: database.go // // Generated by this command: // // mockgen -destination databasemock/database_mock.go --package databasemock -source database.go // // Package databasemock is a generated GoMock package. package databasemock import ( context "context" reflect "reflect" database "github.com/coze-dev/coze-studio/backend/api/model/crossdomain/database" gomock "go.uber.org/mock/gomock" ) // MockDatabase is a mock of Database interface. type MockDatabase struct { ctrl *gomock.Controller recorder *MockDatabaseMockRecorder isgomock struct{} } // MockDatabaseMockRecorder is the mock recorder for MockDatabase. type MockDatabaseMockRecorder struct { mock *MockDatabase } // NewMockDatabase creates a new mock instance. func NewMockDatabase(ctrl *gomock.Controller) *MockDatabase { mock := &MockDatabase{ctrl: ctrl} mock.recorder = &MockDatabaseMockRecorder{mock} return mock } // EXPECT returns an object that allows the caller to indicate expected use. func (m *MockDatabase) EXPECT() *MockDatabaseMockRecorder { return m.recorder } // BindDatabase mocks base method. func (m *MockDatabase) BindDatabase(ctx context.Context, req *database.BindDatabaseToAgentRequest) error { m.ctrl.T.Helper() ret := m.ctrl.Call(m, "BindDatabase", ctx, req) ret0, _ := ret[0].(error) return ret0 } // BindDatabase indicates an expected call of BindDatabase. func (mr *MockDatabaseMockRecorder) BindDatabase(ctx, req any) *gomock.Call { mr.mock.ctrl.T.Helper() return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "BindDatabase", reflect.TypeOf((*MockDatabase)(nil).BindDatabase), ctx, req) } // Delete mocks base method. func (m *MockDatabase) Delete(arg0 context.Context, arg1 *database.DeleteRequest) (*database.Response, error) { m.ctrl.T.Helper() ret := m.ctrl.Call(m, "Delete", arg0, arg1) ret0, _ := ret[0].(*database.Response) ret1, _ := ret[1].(error) return ret0, ret1 } // Delete indicates an expected call of Delete. func (mr *MockDatabaseMockRecorder) Delete(arg0, arg1 any) *gomock.Call { mr.mock.ctrl.T.Helper() return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Delete", reflect.TypeOf((*MockDatabase)(nil).Delete), arg0, arg1) } // DeleteDatabase mocks base method. func (m *MockDatabase) DeleteDatabase(ctx context.Context, req *database.DeleteDatabaseRequest) error { m.ctrl.T.Helper() ret := m.ctrl.Call(m, "DeleteDatabase", ctx, req) ret0, _ := ret[0].(error) return ret0 } // DeleteDatabase indicates an expected call of DeleteDatabase. func (mr *MockDatabaseMockRecorder) DeleteDatabase(ctx, req any) *gomock.Call { mr.mock.ctrl.T.Helper() return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "DeleteDatabase", reflect.TypeOf((*MockDatabase)(nil).DeleteDatabase), ctx, req) } // Execute mocks base method. func (m *MockDatabase) Execute(ctx context.Context, request *database.CustomSQLRequest) (*database.Response, error) { m.ctrl.T.Helper() ret := m.ctrl.Call(m, "Execute", ctx, request) ret0, _ := ret[0].(*database.Response) ret1, _ := ret[1].(error) return ret0, ret1 } // Execute indicates an expected call of Execute. func (mr *MockDatabaseMockRecorder) Execute(ctx, request any) *gomock.Call { mr.mock.ctrl.T.Helper() return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Execute", reflect.TypeOf((*MockDatabase)(nil).Execute), ctx, request) } // ExecuteSQL mocks base method. func (m *MockDatabase) ExecuteSQL(ctx context.Context, req *database.ExecuteSQLRequest) (*database.ExecuteSQLResponse, error) { m.ctrl.T.Helper() ret := m.ctrl.Call(m, "ExecuteSQL", ctx, req) ret0, _ := ret[0].(*database.ExecuteSQLResponse) ret1, _ := ret[1].(error) return ret0, ret1 } // ExecuteSQL indicates an expected call of ExecuteSQL. func (mr *MockDatabaseMockRecorder) ExecuteSQL(ctx, req any) *gomock.Call { mr.mock.ctrl.T.Helper() return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ExecuteSQL", reflect.TypeOf((*MockDatabase)(nil).ExecuteSQL), ctx, req) } // GetAllDatabaseByAppID mocks base method. func (m *MockDatabase) GetAllDatabaseByAppID(ctx context.Context, req *database.GetAllDatabaseByAppIDRequest) (*database.GetAllDatabaseByAppIDResponse, error) { m.ctrl.T.Helper() ret := m.ctrl.Call(m, "GetAllDatabaseByAppID", ctx, req) ret0, _ := ret[0].(*database.GetAllDatabaseByAppIDResponse) ret1, _ := ret[1].(error) return ret0, ret1 } // GetAllDatabaseByAppID indicates an expected call of GetAllDatabaseByAppID. func (mr *MockDatabaseMockRecorder) GetAllDatabaseByAppID(ctx, req any) *gomock.Call { mr.mock.ctrl.T.Helper() return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetAllDatabaseByAppID", reflect.TypeOf((*MockDatabase)(nil).GetAllDatabaseByAppID), ctx, req) } // Insert mocks base method. func (m *MockDatabase) Insert(ctx context.Context, request *database.InsertRequest) (*database.Response, error) { m.ctrl.T.Helper() ret := m.ctrl.Call(m, "Insert", ctx, request) ret0, _ := ret[0].(*database.Response) ret1, _ := ret[1].(error) return ret0, ret1 } // Insert indicates an expected call of Insert. func (mr *MockDatabaseMockRecorder) Insert(ctx, request any) *gomock.Call { mr.mock.ctrl.T.Helper() return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Insert", reflect.TypeOf((*MockDatabase)(nil).Insert), ctx, request) } // MGetDatabase mocks base method. func (m *MockDatabase) MGetDatabase(ctx context.Context, req *database.MGetDatabaseRequest) (*database.MGetDatabaseResponse, error) { m.ctrl.T.Helper() ret := m.ctrl.Call(m, "MGetDatabase", ctx, req) ret0, _ := ret[0].(*database.MGetDatabaseResponse) ret1, _ := ret[1].(error) return ret0, ret1 } // MGetDatabase indicates an expected call of MGetDatabase. func (mr *MockDatabaseMockRecorder) MGetDatabase(ctx, req any) *gomock.Call { mr.mock.ctrl.T.Helper() return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "MGetDatabase", reflect.TypeOf((*MockDatabase)(nil).MGetDatabase), ctx, req) } // PublishDatabase mocks base method. func (m *MockDatabase) PublishDatabase(ctx context.Context, req *database.PublishDatabaseRequest) (*database.PublishDatabaseResponse, error) { m.ctrl.T.Helper() ret := m.ctrl.Call(m, "PublishDatabase", ctx, req) ret0, _ := ret[0].(*database.PublishDatabaseResponse) ret1, _ := ret[1].(error) return ret0, ret1 } // PublishDatabase indicates an expected call of PublishDatabase. func (mr *MockDatabaseMockRecorder) PublishDatabase(ctx, req any) *gomock.Call { mr.mock.ctrl.T.Helper() return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "PublishDatabase", reflect.TypeOf((*MockDatabase)(nil).PublishDatabase), ctx, req) } // Query mocks base method. func (m *MockDatabase) Query(ctx context.Context, request *database.QueryRequest) (*database.Response, error) { m.ctrl.T.Helper() ret := m.ctrl.Call(m, "Query", ctx, request) ret0, _ := ret[0].(*database.Response) ret1, _ := ret[1].(error) return ret0, ret1 } // Query indicates an expected call of Query. func (mr *MockDatabaseMockRecorder) Query(ctx, request any) *gomock.Call { mr.mock.ctrl.T.Helper() return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Query", reflect.TypeOf((*MockDatabase)(nil).Query), ctx, request) } // UnBindDatabase mocks base method. func (m *MockDatabase) UnBindDatabase(ctx context.Context, req *database.UnBindDatabaseToAgentRequest) error { m.ctrl.T.Helper() ret := m.ctrl.Call(m, "UnBindDatabase", ctx, req) ret0, _ := ret[0].(error) return ret0 } // UnBindDatabase indicates an expected call of UnBindDatabase. func (mr *MockDatabaseMockRecorder) UnBindDatabase(ctx, req any) *gomock.Call { mr.mock.ctrl.T.Helper() return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UnBindDatabase", reflect.TypeOf((*MockDatabase)(nil).UnBindDatabase), ctx, req) } // Update mocks base method. func (m *MockDatabase) Update(arg0 context.Context, arg1 *database.UpdateRequest) (*database.Response, error) { m.ctrl.T.Helper() ret := m.ctrl.Call(m, "Update", arg0, arg1) ret0, _ := ret[0].(*database.Response) ret1, _ := ret[1].(error) return ret0, ret1 } // Update indicates an expected call of Update. func (mr *MockDatabaseMockRecorder) Update(arg0, arg1 any) *gomock.Call { mr.mock.ctrl.T.Helper() return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Update", reflect.TypeOf((*MockDatabase)(nil).Update), arg0, arg1) }