// Code generated by mockery v2.7.4. DO NOT EDIT. package db import ( context "context" mock "github.com/stretchr/testify/mock" gtsmodel "github.com/superseriousbusiness/gotosocial/internal/db/gtsmodel" net "net" pub "github.com/go-fed/activity/pub" ) // MockDB is an autogenerated mock type for the DB type type MockDB struct { mock.Mock } // Blocked provides a mock function with given fields: account1, account2 func (_m *MockDB) Blocked(account1 string, account2 string) (bool, error) { ret := _m.Called(account1, account2) var r0 bool if rf, ok := ret.Get(0).(func(string, string) bool); ok { r0 = rf(account1, account2) } else { r0 = ret.Get(0).(bool) } var r1 error if rf, ok := ret.Get(1).(func(string, string) error); ok { r1 = rf(account1, account2) } 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 } // EmojiStringsToEmojis provides a mock function with given fields: emojis, originAccountID, statusID func (_m *MockDB) EmojiStringsToEmojis(emojis []string, originAccountID string, statusID string) ([]*gtsmodel.Emoji, error) { ret := _m.Called(emojis, originAccountID, statusID) var r0 []*gtsmodel.Emoji if rf, ok := ret.Get(0).(func([]string, string, string) []*gtsmodel.Emoji); ok { r0 = rf(emojis, originAccountID, statusID) } else { if ret.Get(0) != nil { r0 = ret.Get(0).([]*gtsmodel.Emoji) } } var r1 error if rf, ok := ret.Get(1).(func([]string, string, string) error); ok { r1 = rf(emojis, originAccountID, statusID) } else { r1 = ret.Error(1) } return r0, r1 } // 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 *gtsmodel.Account) error { ret := _m.Called(userID, account) var r0 error if rf, ok := ret.Get(0).(func(string, *gtsmodel.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 } // GetAvatarForAccountID provides a mock function with given fields: avatar, accountID func (_m *MockDB) GetAvatarForAccountID(avatar *gtsmodel.MediaAttachment, accountID string) error { ret := _m.Called(avatar, accountID) var r0 error if rf, ok := ret.Get(0).(func(*gtsmodel.MediaAttachment, string) error); ok { r0 = rf(avatar, accountID) } 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 *[]gtsmodel.FollowRequest) error { ret := _m.Called(accountID, followRequests) var r0 error if rf, ok := ret.Get(0).(func(string, *[]gtsmodel.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 *[]gtsmodel.Follow) error { ret := _m.Called(accountID, followers) var r0 error if rf, ok := ret.Get(0).(func(string, *[]gtsmodel.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 *[]gtsmodel.Follow) error { ret := _m.Called(accountID, following) var r0 error if rf, ok := ret.Get(0).(func(string, *[]gtsmodel.Follow) error); ok { r0 = rf(accountID, following) } else { r0 = ret.Error(0) } return r0 } // GetHeaderForAccountID provides a mock function with given fields: header, accountID func (_m *MockDB) GetHeaderForAccountID(header *gtsmodel.MediaAttachment, accountID string) error { ret := _m.Called(header, accountID) var r0 error if rf, ok := ret.Get(0).(func(*gtsmodel.MediaAttachment, string) error); ok { r0 = rf(header, accountID) } else { r0 = ret.Error(0) } return r0 } // GetLastStatusForAccountID provides a mock function with given fields: accountID, status func (_m *MockDB) GetLastStatusForAccountID(accountID string, status *gtsmodel.Status) error { ret := _m.Called(accountID, status) var r0 error if rf, ok := ret.Get(0).(func(string, *gtsmodel.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 *[]gtsmodel.Status) error { ret := _m.Called(accountID, statuses) var r0 error if rf, ok := ret.Get(0).(func(string, *[]gtsmodel.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 *[]gtsmodel.Status, limit int) error { ret := _m.Called(accountID, statuses, limit) var r0 error if rf, ok := ret.Get(0).(func(string, *[]gtsmodel.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 } // MentionStringsToMentions provides a mock function with given fields: targetAccounts, originAccountID, statusID func (_m *MockDB) MentionStringsToMentions(targetAccounts []string, originAccountID string, statusID string) ([]*gtsmodel.Mention, error) { ret := _m.Called(targetAccounts, originAccountID, statusID) var r0 []*gtsmodel.Mention if rf, ok := ret.Get(0).(func([]string, string, string) []*gtsmodel.Mention); ok { r0 = rf(targetAccounts, originAccountID, statusID) } else { if ret.Get(0) != nil { r0 = ret.Get(0).([]*gtsmodel.Mention) } } var r1 error if rf, ok := ret.Get(1).(func([]string, string, string) error); ok { r1 = rf(targetAccounts, originAccountID, statusID) } else { r1 = ret.Error(1) } return r0, r1 } // 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) (*gtsmodel.User, error) { ret := _m.Called(username, reason, requireApproval, email, password, signUpIP, locale, appID) var r0 *gtsmodel.User if rf, ok := ret.Get(0).(func(string, string, bool, string, string, net.IP, string, string) *gtsmodel.User); ok { r0 = rf(username, reason, requireApproval, email, password, signUpIP, locale, appID) } else { if ret.Get(0) != nil { r0 = ret.Get(0).(*gtsmodel.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 } // SetHeaderOrAvatarForAccountID provides a mock function with given fields: mediaAttachment, accountID func (_m *MockDB) SetHeaderOrAvatarForAccountID(mediaAttachment *gtsmodel.MediaAttachment, accountID string) error { ret := _m.Called(mediaAttachment, accountID) var r0 error if rf, ok := ret.Get(0).(func(*gtsmodel.MediaAttachment, string) error); ok { r0 = rf(mediaAttachment, accountID) } 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 } // TagStringsToTags provides a mock function with given fields: tags, originAccountID, statusID func (_m *MockDB) TagStringsToTags(tags []string, originAccountID string, statusID string) ([]*gtsmodel.Tag, error) { ret := _m.Called(tags, originAccountID, statusID) var r0 []*gtsmodel.Tag if rf, ok := ret.Get(0).(func([]string, string, string) []*gtsmodel.Tag); ok { r0 = rf(tags, originAccountID, statusID) } else { if ret.Get(0) != nil { r0 = ret.Get(0).([]*gtsmodel.Tag) } } var r1 error if rf, ok := ret.Get(1).(func([]string, string, string) error); ok { r1 = rf(tags, originAccountID, statusID) } else { r1 = ret.Error(1) } return r0, r1 } // 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 } // UpdateOneByID provides a mock function with given fields: id, key, value, i func (_m *MockDB) UpdateOneByID(id string, key string, value interface{}, i interface{}) error { ret := _m.Called(id, key, value, i) var r0 error if rf, ok := ret.Get(0).(func(string, string, interface{}, interface{}) error); ok { r0 = rf(id, key, value, i) } else { r0 = ret.Error(0) } return r0 }