// Code generated by mockery v2.7.4. DO NOT EDIT. package db import ( context "context" mock "github.com/stretchr/testify/mock" mastotypes "github.com/superseriousbusiness/gotosocial/pkg/mastotypes" model "github.com/superseriousbusiness/gotosocial/internal/db/model" net "net" pub "github.com/go-fed/activity/pub" ) // MockDB is an autogenerated mock type for the DB type type MockDB struct { mock.Mock } // AccountToMastoSensitive provides a mock function with given fields: account func (_m *MockDB) AccountToMastoSensitive(account *model.Account) (*mastotypes.Account, error) { ret := _m.Called(account) var r0 *mastotypes.Account if rf, ok := ret.Get(0).(func(*model.Account) *mastotypes.Account); ok { r0 = rf(account) } else { if ret.Get(0) != nil { r0 = ret.Get(0).(*mastotypes.Account) } } var r1 error if rf, ok := ret.Get(1).(func(*model.Account) error); ok { r1 = rf(account) } else { r1 = ret.Error(1) } return r0, r1 } // CreateTable provides a mock function with given fields: i func (_m *MockDB) CreateTable(i interface{}) error { ret := _m.Called(i) var r0 error if rf, ok := ret.Get(0).(func(interface{}) error); ok { r0 = rf(i) } else { r0 = ret.Error(0) } return r0 } // DeleteByID provides a mock function with given fields: id, i func (_m *MockDB) DeleteByID(id string, i interface{}) error { ret := _m.Called(id, i) var r0 error if rf, ok := ret.Get(0).(func(string, interface{}) error); ok { r0 = rf(id, i) } else { r0 = ret.Error(0) } return r0 } // DeleteWhere provides a mock function with given fields: key, value, i func (_m *MockDB) DeleteWhere(key string, value interface{}, i interface{}) error { ret := _m.Called(key, value, i) var r0 error if rf, ok := ret.Get(0).(func(string, interface{}, interface{}) error); ok { r0 = rf(key, value, i) } else { r0 = ret.Error(0) } return r0 } // DropTable provides a mock function with given fields: i func (_m *MockDB) DropTable(i interface{}) error { ret := _m.Called(i) var r0 error if rf, ok := ret.Get(0).(func(interface{}) error); ok { r0 = rf(i) } else { r0 = ret.Error(0) } return r0 } // Federation provides a mock function with given fields: func (_m *MockDB) Federation() pub.Database { ret := _m.Called() var r0 pub.Database if rf, ok := ret.Get(0).(func() pub.Database); ok { r0 = rf() } else { if ret.Get(0) != nil { r0 = ret.Get(0).(pub.Database) } } return r0 } // GetAccountByUserID provides a mock function with given fields: userID, account func (_m *MockDB) GetAccountByUserID(userID string, account *model.Account) error { ret := _m.Called(userID, account) var r0 error if rf, ok := ret.Get(0).(func(string, *model.Account) error); ok { r0 = rf(userID, account) } else { r0 = ret.Error(0) } return r0 } // GetAll provides a mock function with given fields: i func (_m *MockDB) GetAll(i interface{}) error { ret := _m.Called(i) var r0 error if rf, ok := ret.Get(0).(func(interface{}) error); ok { r0 = rf(i) } else { r0 = ret.Error(0) } return r0 } // GetByID provides a mock function with given fields: id, i func (_m *MockDB) GetByID(id string, i interface{}) error { ret := _m.Called(id, i) var r0 error if rf, ok := ret.Get(0).(func(string, interface{}) error); ok { r0 = rf(id, i) } else { r0 = ret.Error(0) } return r0 } // GetFollowRequestsForAccountID provides a mock function with given fields: accountID, followRequests func (_m *MockDB) GetFollowRequestsForAccountID(accountID string, followRequests *[]model.FollowRequest) error { ret := _m.Called(accountID, followRequests) var r0 error if rf, ok := ret.Get(0).(func(string, *[]model.FollowRequest) error); ok { r0 = rf(accountID, followRequests) } else { r0 = ret.Error(0) } return r0 } // GetFollowersByAccountID provides a mock function with given fields: accountID, followers func (_m *MockDB) GetFollowersByAccountID(accountID string, followers *[]model.Follow) error { ret := _m.Called(accountID, followers) var r0 error if rf, ok := ret.Get(0).(func(string, *[]model.Follow) error); ok { r0 = rf(accountID, followers) } else { r0 = ret.Error(0) } return r0 } // GetFollowingByAccountID provides a mock function with given fields: accountID, following func (_m *MockDB) GetFollowingByAccountID(accountID string, following *[]model.Follow) error { ret := _m.Called(accountID, following) var r0 error if rf, ok := ret.Get(0).(func(string, *[]model.Follow) error); ok { r0 = rf(accountID, following) } else { r0 = ret.Error(0) } return r0 } // GetLastStatusForAccountID provides a mock function with given fields: accountID, status func (_m *MockDB) GetLastStatusForAccountID(accountID string, status *model.Status) error { ret := _m.Called(accountID, status) var r0 error if rf, ok := ret.Get(0).(func(string, *model.Status) error); ok { r0 = rf(accountID, status) } else { r0 = ret.Error(0) } return r0 } // GetStatusesByAccountID provides a mock function with given fields: accountID, statuses func (_m *MockDB) GetStatusesByAccountID(accountID string, statuses *[]model.Status) error { ret := _m.Called(accountID, statuses) var r0 error if rf, ok := ret.Get(0).(func(string, *[]model.Status) error); ok { r0 = rf(accountID, statuses) } else { r0 = ret.Error(0) } return r0 } // GetStatusesByTimeDescending provides a mock function with given fields: accountID, statuses, limit func (_m *MockDB) GetStatusesByTimeDescending(accountID string, statuses *[]model.Status, limit int) error { ret := _m.Called(accountID, statuses, limit) var r0 error if rf, ok := ret.Get(0).(func(string, *[]model.Status, int) error); ok { r0 = rf(accountID, statuses, limit) } else { r0 = ret.Error(0) } return r0 } // GetWhere provides a mock function with given fields: key, value, i func (_m *MockDB) GetWhere(key string, value interface{}, i interface{}) error { ret := _m.Called(key, value, i) var r0 error if rf, ok := ret.Get(0).(func(string, interface{}, interface{}) error); ok { r0 = rf(key, value, i) } else { r0 = ret.Error(0) } return r0 } // IsEmailAvailable provides a mock function with given fields: email func (_m *MockDB) IsEmailAvailable(email string) error { ret := _m.Called(email) var r0 error if rf, ok := ret.Get(0).(func(string) error); ok { r0 = rf(email) } else { r0 = ret.Error(0) } return r0 } // IsHealthy provides a mock function with given fields: ctx func (_m *MockDB) IsHealthy(ctx context.Context) error { ret := _m.Called(ctx) var r0 error if rf, ok := ret.Get(0).(func(context.Context) error); ok { r0 = rf(ctx) } else { r0 = ret.Error(0) } return r0 } // IsUsernameAvailable provides a mock function with given fields: username func (_m *MockDB) IsUsernameAvailable(username string) error { ret := _m.Called(username) var r0 error if rf, ok := ret.Get(0).(func(string) error); ok { r0 = rf(username) } else { r0 = ret.Error(0) } return r0 } // NewSignup provides a mock function with given fields: username, reason, requireApproval, email, password, signUpIP, locale, appID func (_m *MockDB) NewSignup(username string, reason string, requireApproval bool, email string, password string, signUpIP net.IP, locale string, appID string) (*model.User, error) { ret := _m.Called(username, reason, requireApproval, email, password, signUpIP, locale, appID) var r0 *model.User if rf, ok := ret.Get(0).(func(string, string, bool, string, string, net.IP, string, string) *model.User); ok { r0 = rf(username, reason, requireApproval, email, password, signUpIP, locale, appID) } else { if ret.Get(0) != nil { r0 = ret.Get(0).(*model.User) } } var r1 error if rf, ok := ret.Get(1).(func(string, string, bool, string, string, net.IP, string, string) error); ok { r1 = rf(username, reason, requireApproval, email, password, signUpIP, locale, appID) } else { r1 = ret.Error(1) } return r0, r1 } // Put provides a mock function with given fields: i func (_m *MockDB) Put(i interface{}) error { ret := _m.Called(i) var r0 error if rf, ok := ret.Get(0).(func(interface{}) error); ok { r0 = rf(i) } else { r0 = ret.Error(0) } return r0 } // Stop provides a mock function with given fields: ctx func (_m *MockDB) Stop(ctx context.Context) error { ret := _m.Called(ctx) var r0 error if rf, ok := ret.Get(0).(func(context.Context) error); ok { r0 = rf(ctx) } else { r0 = ret.Error(0) } return r0 } // UpdateByID provides a mock function with given fields: id, i func (_m *MockDB) UpdateByID(id string, i interface{}) error { ret := _m.Called(id, i) var r0 error if rf, ok := ret.Get(0).(func(string, interface{}) error); ok { r0 = rf(id, i) } else { r0 = ret.Error(0) } return r0 }