@@ -52,6 +52,11 @@ type MockDB struct {
5252 GetChallengeByIDFunc func (challengeID uuid.UUID ) (* types.Challenge , error )
5353 SaveActivityFunc func (userID uuid.UUID , message string ) error
5454 GetActivitiesForUserAndFriendsFunc func (userID uuid.UUID ) ([]types.ActivityWithUser , error )
55+ SaveChatMessageFunc func (userID uuid.UUID , message string , timestamp string ) (uuid.UUID , error )
56+ GetChatMessagesFunc func (userID uuid.UUID ) ([]types.ChatMessage , error )
57+ AddReactionToChatMessageFunc func (userID uuid.UUID , messageID uuid.UUID ) error
58+ CountReactionsForMessageFunc func (messageID uuid.UUID ) (int , error )
59+ GetIDByMessageIDFunc func (messageID uuid.UUID ) (uuid.UUID , error )
5560}
5661
5762func (m * MockDB ) ExecuteRawSQL (query string ) (sql.Result , error ) {
@@ -347,3 +352,38 @@ func (m *MockDB) DeleteRun(runID uuid.UUID) error {
347352 }
348353 return nil
349354}
355+
356+ func (m * MockDB ) SaveChatMessage (userID uuid.UUID , message string , timestamp string ) (uuid.UUID , error ) {
357+ if m .SaveChatMessageFunc != nil {
358+ return m .SaveChatMessageFunc (userID , message , timestamp )
359+ }
360+ return uuid .Nil , nil
361+ }
362+
363+ func (m * MockDB ) GetChatMessages (userID uuid.UUID ) ([]types.ChatMessage , error ) {
364+ if m .GetChatMessagesFunc != nil {
365+ return m .GetChatMessagesFunc (userID )
366+ }
367+ return nil , nil
368+ }
369+
370+ func (m * MockDB ) AddReactionToChatMessage (userID uuid.UUID , messageID uuid.UUID ) error {
371+ if m .AddReactionToChatMessageFunc != nil {
372+ return m .AddReactionToChatMessageFunc (userID , messageID )
373+ }
374+ return nil
375+ }
376+
377+ func (m * MockDB ) CountReactionsForMessage (messageID uuid.UUID ) (int , error ) {
378+ if m .CountReactionsForMessageFunc != nil {
379+ return m .CountReactionsForMessageFunc (messageID )
380+ }
381+ return 0 , nil
382+ }
383+
384+ func (m * MockDB ) GetIDByMessageID (messageID uuid.UUID ) (uuid.UUID , error ) {
385+ if m .GetIDByMessageIDFunc != nil {
386+ return m .GetIDByMessageIDFunc (messageID )
387+ }
388+ return uuid .Nil , nil
389+ }
0 commit comments