diff options
Diffstat (limited to 'internal/db/mock_DB.go')
-rw-r--r-- | internal/db/mock_DB.go | 363 |
1 files changed, 363 insertions, 0 deletions
diff --git a/internal/db/mock_DB.go b/internal/db/mock_DB.go new file mode 100644 index 000000000..d4c25bb79 --- /dev/null +++ b/internal/db/mock_DB.go @@ -0,0 +1,363 @@ +// 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 +} |