@@ -10,45 +10,48 @@ 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- 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 (excludeUserID * uuid.UUID ) ([]types.User , error )
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 )
44- CleanUpChallengesForUserFunc func (userID uuid.UUID ) error
45- AddFriendFunc func (userID , friendID uuid.UUID ) error
46- GetFriendsFunc func (userID uuid.UUID ) ([]types.User , error )
47- GetFriendsRankedByPointsFunc func (userID uuid.UUID ) ([]types.User , error )
48- DeleteFriendFunc func (userID , friendID uuid.UUID ) error
49- SaveRunFunc func (userID uuid.UUID , route string , duration string , distance float64 ) error
50- GetAllRunsByUserFunc func (userID uuid.UUID ) ([]types.RunDTO , error )
51- DeleteRunFunc func (runID uuid.UUID ) 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 (excludeUserID * uuid.UUID ) ([]types.User , error )
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 )
44+ CleanUpChallengesForUserFunc func (userID uuid.UUID ) error
45+ AddFriendFunc func (userID , friendID uuid.UUID ) error
46+ GetFriendsFunc func (userID uuid.UUID ) ([]types.User , error )
47+ GetFriendsRankedByPointsFunc func (userID uuid.UUID ) ([]types.User , error )
48+ DeleteFriendFunc func (userID , friendID uuid.UUID ) error
49+ SaveRunFunc func (userID uuid.UUID , route string , duration string , distance float64 ) error
50+ GetAllRunsByUserFunc func (userID uuid.UUID ) ([]types.RunDTO , error )
51+ DeleteRunFunc func (runID uuid.UUID ) error
52+ GetChallengeByIDFunc func (challengeID uuid.UUID ) (* types.Challenge , error )
53+ SaveActivityFunc func (userID uuid.UUID , message string ) error
54+ GetActivitiesForUserAndFriendsFunc func (userID uuid.UUID ) ([]types.ActivityWithUser , error )
5255}
5356
5457func (m * MockDB ) ExecuteRawSQL (query string ) (sql.Result , error ) {
@@ -303,6 +306,27 @@ func (m *MockDB) GetAllUsers(userId *uuid.UUID) ([]types.User, error) {
303306 return nil , nil
304307}
305308
309+ func (m * MockDB ) GetChallengeByID (challengeID uuid.UUID ) (* types.Challenge , error ) {
310+ if m .GetChallengeByIDFunc != nil {
311+ return m .GetChallengeByIDFunc (challengeID )
312+ }
313+ return nil , nil
314+ }
315+
316+ func (m * MockDB ) SaveActivity (userID uuid.UUID , message string ) error {
317+ if m .SaveActivityFunc != nil {
318+ return m .SaveActivityFunc (userID , message )
319+ }
320+ return nil
321+ }
322+
323+ func (m * MockDB ) GetActivitiesForUserAndFriends (userID uuid.UUID ) ([]types.ActivityWithUser , error ) {
324+ if m .GetActivitiesForUserAndFriendsFunc != nil {
325+ return m .GetActivitiesForUserAndFriendsFunc (userID )
326+ }
327+ return nil , nil
328+ }
329+
306330func (m * MockDB ) SaveRun (userID uuid.UUID , route string , duration string , distance float64 ) error {
307331 if m .SaveRunFunc != nil {
308332 return m .SaveRun (userID , route , duration , distance )
0 commit comments