🐛 Fix loosing permissions when updating the user

systemli · Nov 26, 2023 · 6660198 · 6660198
1 parent 3c105ed
commit 6660198
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/internal/api/middleware/me/me.go b/internal/api/middleware/me/me.go
@@ -8,15 +8,15 @@ import (
 	"github.com/systemli/ticker/internal/storage"
 )
 
-func MeMiddleware(storage storage.Storage) gin.HandlerFunc {
+func MeMiddleware(store storage.Storage) gin.HandlerFunc {
 	return func(c *gin.Context) {
 		userID, exists := c.Get("id")
 		if !exists {
 			c.AbortWithStatusJSON(http.StatusBadRequest, response.ErrorResponse(response.CodeDefault, response.UserIdentifierMissing))
 			return
 		}
 
-		user, err := storage.FindUserByID(int(userID.(float64)))
+		user, err := store.FindUserByID(int(userID.(float64)), storage.WithTickers())
 		if err != nil {
 			c.AbortWithStatusJSON(http.StatusBadRequest, response.ErrorResponse(response.CodeDefault, response.UserNotFound))
 			return

diff --git a/internal/api/middleware/me/me_test.go b/internal/api/middleware/me/me_test.go
@@ -35,7 +35,7 @@ func (s *MeTestSuite) TestMeMiddleware() {
 	s.Run("when id is present", func() {
 		s.Run("when user is not found", func() {
 			mockStorage := &storage.MockStorage{}
-			mockStorage.On("FindUserByID", mock.Anything).Return(storage.User{}, errors.New("not found"))
+			mockStorage.On("FindUserByID", mock.Anything, mock.Anything).Return(storage.User{}, errors.New("not found"))
 			mw := MeMiddleware(mockStorage)
 			w := httptest.NewRecorder()
 			c, _ := gin.CreateTestContext(w)
@@ -48,7 +48,7 @@ func (s *MeTestSuite) TestMeMiddleware() {
 
 		s.Run("when user is found", func() {
 			mockStorage := &storage.MockStorage{}
-			mockStorage.On("FindUserByID", mock.Anything).Return(storage.User{ID: 1}, nil)
+			mockStorage.On("FindUserByID", mock.Anything, mock.Anything).Return(storage.User{ID: 1}, nil)
 			mw := MeMiddleware(mockStorage)
 			w := httptest.NewRecorder()
 			c, _ := gin.CreateTestContext(w)