fix: set mocked gin request context manually

go-vela · Aug 28, 2023 · eb51458 · eb51458
1 parent bdfdff3
commit eb51458
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/internal/token/refresh_test.go b/internal/token/refresh_test.go
@@ -6,6 +6,7 @@ package token
 
 import (
 	"context"
+	"net/http"
 	"net/http/httptest"
 	"testing"
 	"time"
@@ -64,6 +65,8 @@ func TestTokenManager_Refresh(t *testing.T) {
 	resp := httptest.NewRecorder()
 	context, _ := gin.CreateTestContext(resp)
 	context.Set("database", db)
+	req, _ := http.NewRequestWithContext(context, "", "", nil)
+	context.Request = req
 
 	// run tests
 	got, err := tm.Refresh(context, rt)
@@ -123,6 +126,8 @@ func TestTokenManager_Refresh_Expired(t *testing.T) {
 	resp := httptest.NewRecorder()
 	context, _ := gin.CreateTestContext(resp)
 	context.Set("database", db)
+	req, _ := http.NewRequestWithContext(context, "", "", nil)
+	context.Request = req
 
 	// run tests
 	_, err = tm.Refresh(context, rt)