refactor: provide context to repository to apply timeout

2 files changed, 4 insertions, 4 deletions

diff --git a/app/middleware/user.go b/app/middleware/user.go
index 03c04d6..c0181f9 100644
--- a/app/middleware/user.go
+++ b/app/middleware/user.go
@@ -20,10 +20,10 @@ func User(db domain.Repository[*domain.User]) func(http.Handler) http.Handler {
 				return
 			}
 
-			user := db.Find(domain.ID(idToken.Subject))
+			user := db.Find(r.Context(), domain.ID(idToken.Subject))
 			if !x.IsPresent(user) {
 				user = mapper.MapFrom[*oidc.IDToken, *domain.User](idToken)
-				if err := db.Save(user); err != nil {
+				if err := db.Save(r.Context(), user); err != nil {
 					pls.LogError(r.Context(), err)
 					next.ServeHTTP(w, r)
 					return
diff --git a/app/middleware/user_test.go b/app/middleware/user_test.go
index e1bbcf9..e6ba09d 100644
--- a/app/middleware/user_test.go
+++ b/app/middleware/user_test.go
@@ -19,7 +19,7 @@ func TestUser(t *testing.T) {
 	middleware := User(repository)
 
 	knownUser := &domain.User{ID: domain.ID(pls.GenerateULID())}
-	require.NoError(t, repository.Save(knownUser))
+	require.NoError(t, repository.Save(t.Context(), knownUser))
 
 	t.Run("when ID Token is provided", func(t *testing.T) {
 		t.Run("when user is known", func(t *testing.T) {
@@ -56,7 +56,7 @@ func TestUser(t *testing.T) {
 			server.ServeHTTP(w, r)
 
 			assert.Equal(t, http.StatusTeapot, w.Code)
-			require.NotNil(t, repository.Find(domain.ID(unknownID)))
+			require.NotNil(t, repository.Find(t.Context(), domain.ID(unknownID)))
 		})
 	})