@@ -10,39 +10,40 @@ import (
1010)
1111
1212type MockDB struct {
13- ExecuteRawSQLFunc func (query string ) (sql.Result , error )
14- QueryRowFunc func (query string , args ... interface {}) * sql.Row
15- HealthFunc func () map [string ]string
16- CloseFunc func () error
17- SaveCredentialsFunc func (creds types.Credentials ) error
18- GetUserByEmailFunc func (email string ) (types.Credentials , error )
19- CheckEmailFunc func (email string ) error
20- SaveUserProfileFunc func (user types.User ) error
21- GetUserByIDFunc func (userID uuid.UUID ) (types.User , error )
22- UpdateRocketPointsFunc func (userID uuid.UUID , rocketPoints int ) error
23- GetUserIDByNameFunc func (name string ) (uuid.UUID , error )
24- GetTopUsersFunc func (limit int ) ([]types.User , error )
25- GetAllUsersFunc func () ([]types.User , error ) // Missing method
26- UpdateDailyStepsFunc func (userID uuid.UUID , steps int ) error
27- GetUserStatisticsFunc func (userID uuid.UUID ) ([]types.StepStatistic , error )
28- GetSettingsByUserIDFunc func (userID uuid.UUID ) (* types.Settings , error )
29- CreateSettingsFunc func (settings types.Settings ) error
30- UpdateSettingsStepGoalFunc func (userID uuid.UUID , stepGoal int ) error
31- UpdateSettingsImageFunc func (userID uuid.UUID , imageID uuid.UUID ) error
32- UpdateStepGoalFunc func (userID uuid.UUID , stepGoal int ) error
33- UpdateImageFunc func (userID uuid.UUID , imageID uuid.UUID ) error
34- SaveImageFunc func (filename string , data []byte ) (uuid.UUID , error )
35- GetUserImageFunc func (userID uuid.UUID ) (* types.UserImage , error )
36- GetAllChallengesFunc func () ([]types.Challenge , error )
37- AssignChallengesToUserFunc func (userID uuid.UUID , challenges []types.Challenge ) error
38- GetUserDailyChallengesFunc func (userID uuid.UUID ) ([]types.Challenge , error )
39- ResetDailyChallengesFunc func () error
40- InsertChallengeFunc func (challenge types.Challenge ) error
41- CompleteChallengeFunc func (userID uuid.UUID , dto types.CompleteChallengesDTO ) error
42- IsNewDayForUserFunc func (userID uuid.UUID ) (bool , error )
13+ ExecuteRawSQLFunc func (query string ) (sql.Result , error )
14+ QueryRowFunc func (query string , args ... interface {}) * sql.Row
15+ HealthFunc func () map [string ]string
16+ CloseFunc func () error
17+ SaveCredentialsFunc func (creds types.Credentials ) error
18+ GetUserByEmailFunc func (email string ) (types.Credentials , error )
19+ CheckEmailFunc func (email string ) error
20+ SaveUserProfileFunc func (user types.User ) error
21+ GetUserByIDFunc func (userID uuid.UUID ) (types.User , error )
22+ UpdateRocketPointsFunc func (userID uuid.UUID , rocketPoints int ) error
23+ GetRocketPointsByUserIDFunc func (userID uuid.UUID ) (int , error )
24+ GetUserIDByNameFunc func (name string ) (uuid.UUID , error )
25+ GetTopUsersFunc func (limit int ) ([]types.User , error )
26+ GetAllUsersFunc func () ([]types.User , error ) // Missing method
27+ UpdateDailyStepsFunc func (userID uuid.UUID , steps int ) error
28+ GetUserStatisticsFunc func (userID uuid.UUID ) ([]types.StepStatistic , error )
29+ GetSettingsByUserIDFunc func (userID uuid.UUID ) (* types.Settings , error )
30+ CreateSettingsFunc func (settings types.Settings ) error
31+ UpdateSettingsStepGoalFunc func (userID uuid.UUID , stepGoal int ) error
32+ UpdateSettingsImageFunc func (userID uuid.UUID , imageID uuid.UUID ) error
33+ UpdateStepGoalFunc func (userID uuid.UUID , stepGoal int ) error
34+ UpdateImageFunc func (userID uuid.UUID , imageID uuid.UUID ) error
35+ SaveImageFunc func (filename string , data []byte ) (uuid.UUID , error )
36+ GetUserImageFunc func (userID uuid.UUID ) (* types.UserImage , error )
37+ GetAllChallengesFunc func () ([]types.Challenge , error )
38+ AssignChallengesToUserFunc func (userID uuid.UUID , challenges []types.Challenge ) error
39+ GetUserDailyChallengesFunc func (userID uuid.UUID ) ([]types.Challenge , error )
40+ ResetDailyChallengesFunc func () error
41+ InsertChallengeFunc func (challenge types.Challenge ) error
42+ CompleteChallengeFunc func (userID uuid.UUID , dto types.CompleteChallengesDTO ) error
43+ IsNewDayForUserFunc func (userID uuid.UUID ) (bool , error )
4344 CleanUpChallengesForUserFunc func (userID uuid.UUID ) error
44- AddFriendFunc func (userID , friendID uuid.UUID ) error
45- GetFriendsFunc func (userID uuid.UUID ) ([]types.User , error )
45+ AddFriendFunc func (userID , friendID uuid.UUID ) error
46+ GetFriendsFunc func (userID uuid.UUID ) ([]types.User , error )
4647 GetFriendsRankedByPointsFunc func (userID uuid.UUID ) ([]types.User , error )
4748 DeleteFriendFunc func (userID , friendID uuid.UUID ) error
4849}
@@ -117,6 +118,13 @@ func (m *MockDB) UpdateRocketPoints(userID uuid.UUID, rocketPoints int) error {
117118 return nil
118119}
119120
121+ func (m * MockDB ) GetRocketPointsByUserID (userID uuid.UUID ) (int , error ) {
122+ if m .GetRocketPointsByUserIDFunc != nil {
123+ return m .GetRocketPointsByUserIDFunc (userID )
124+ }
125+ return 0 , nil
126+ }
127+
120128func (m * MockDB ) GetUserIDByName (name string ) (uuid.UUID , error ) {
121129 if m .GetUserIDByNameFunc != nil {
122130 return m .GetUserIDByNameFunc (name )
0 commit comments