refactor: clean variable names and modernize code
This commit is contained in:
@@ -21,60 +21,60 @@ func TestIntegration_SessionManagement(t *testing.T) {
|
|||||||
ctx.Suite.EmailSender.Reset()
|
ctx.Suite.EmailSender.Reset()
|
||||||
user := createAuthenticatedUser(t, ctx.AuthService, ctx.Suite.UserRepo, "session_pass_user", "session_pass@example.com")
|
user := createAuthenticatedUser(t, ctx.AuthService, ctx.Suite.UserRepo, "session_pass_user", "session_pass@example.com")
|
||||||
|
|
||||||
req1 := httptest.NewRequest("GET", "/api/auth/me", nil)
|
firstRequest := httptest.NewRequest("GET", "/api/auth/me", nil)
|
||||||
req1.Header.Set("Authorization", "Bearer "+user.Token)
|
firstRequest.Header.Set("Authorization", "Bearer "+user.Token)
|
||||||
req1 = testutils.WithUserContext(req1, middleware.UserIDKey, user.User.ID)
|
firstRequest = testutils.WithUserContext(firstRequest, middleware.UserIDKey, user.User.ID)
|
||||||
rec1 := httptest.NewRecorder()
|
firstRecorder := httptest.NewRecorder()
|
||||||
router.ServeHTTP(rec1, req1)
|
router.ServeHTTP(firstRecorder, firstRequest)
|
||||||
|
|
||||||
assertStatus(t, rec1, http.StatusOK)
|
assertStatus(t, firstRecorder, http.StatusOK)
|
||||||
|
|
||||||
reqBody := map[string]string{
|
requestBody := map[string]string{
|
||||||
"current_password": "SecurePass123!",
|
"current_password": "SecurePass123!",
|
||||||
"new_password": "NewSecurePass123!",
|
"new_password": "NewSecurePass123!",
|
||||||
}
|
}
|
||||||
body, _ := json.Marshal(reqBody)
|
body, _ := json.Marshal(requestBody)
|
||||||
req2 := httptest.NewRequest("PUT", "/api/auth/password", bytes.NewBuffer(body))
|
secondRequest := httptest.NewRequest("PUT", "/api/auth/password", bytes.NewBuffer(body))
|
||||||
req2.Header.Set("Content-Type", "application/json")
|
secondRequest.Header.Set("Content-Type", "application/json")
|
||||||
req2.Header.Set("Authorization", "Bearer "+user.Token)
|
secondRequest.Header.Set("Authorization", "Bearer "+user.Token)
|
||||||
req2 = testutils.WithUserContext(req2, middleware.UserIDKey, user.User.ID)
|
secondRequest = testutils.WithUserContext(secondRequest, middleware.UserIDKey, user.User.ID)
|
||||||
rec2 := httptest.NewRecorder()
|
secondRecorder := httptest.NewRecorder()
|
||||||
router.ServeHTTP(rec2, req2)
|
router.ServeHTTP(secondRecorder, secondRequest)
|
||||||
|
|
||||||
assertStatus(t, rec2, http.StatusOK)
|
assertStatus(t, secondRecorder, http.StatusOK)
|
||||||
|
|
||||||
req3 := httptest.NewRequest("GET", "/api/auth/me", nil)
|
thirdRequest := httptest.NewRequest("GET", "/api/auth/me", nil)
|
||||||
req3.Header.Set("Authorization", "Bearer "+user.Token)
|
thirdRequest.Header.Set("Authorization", "Bearer "+user.Token)
|
||||||
req3 = testutils.WithUserContext(req3, middleware.UserIDKey, user.User.ID)
|
thirdRequest = testutils.WithUserContext(thirdRequest, middleware.UserIDKey, user.User.ID)
|
||||||
rec3 := httptest.NewRecorder()
|
thirdRecorder := httptest.NewRecorder()
|
||||||
router.ServeHTTP(rec3, req3)
|
router.ServeHTTP(thirdRecorder, thirdRequest)
|
||||||
|
|
||||||
assertErrorResponse(t, rec3, http.StatusUnauthorized)
|
assertErrorResponse(t, thirdRecorder, http.StatusUnauthorized)
|
||||||
})
|
})
|
||||||
|
|
||||||
t.Run("Session_Invalidation_On_Account_Lock", func(t *testing.T) {
|
t.Run("Session_Invalidation_On_Account_Lock", func(t *testing.T) {
|
||||||
ctx.Suite.EmailSender.Reset()
|
ctx.Suite.EmailSender.Reset()
|
||||||
user := createAuthenticatedUser(t, ctx.AuthService, ctx.Suite.UserRepo, "session_lock_user", "session_lock@example.com")
|
user := createAuthenticatedUser(t, ctx.AuthService, ctx.Suite.UserRepo, "session_lock_user", "session_lock@example.com")
|
||||||
|
|
||||||
req1 := httptest.NewRequest("GET", "/api/auth/me", nil)
|
firstRequest := httptest.NewRequest("GET", "/api/auth/me", nil)
|
||||||
req1.Header.Set("Authorization", "Bearer "+user.Token)
|
firstRequest.Header.Set("Authorization", "Bearer "+user.Token)
|
||||||
req1 = testutils.WithUserContext(req1, middleware.UserIDKey, user.User.ID)
|
firstRequest = testutils.WithUserContext(firstRequest, middleware.UserIDKey, user.User.ID)
|
||||||
rec1 := httptest.NewRecorder()
|
firstRecorder := httptest.NewRecorder()
|
||||||
router.ServeHTTP(rec1, req1)
|
router.ServeHTTP(firstRecorder, firstRequest)
|
||||||
|
|
||||||
assertStatus(t, rec1, http.StatusOK)
|
assertStatus(t, firstRecorder, http.StatusOK)
|
||||||
|
|
||||||
if err := ctx.Suite.UserRepo.Lock(user.User.ID); err != nil {
|
if err := ctx.Suite.UserRepo.Lock(user.User.ID); err != nil {
|
||||||
t.Fatalf("Failed to lock user: %v", err)
|
t.Fatalf("Failed to lock user: %v", err)
|
||||||
}
|
}
|
||||||
|
|
||||||
req2 := httptest.NewRequest("GET", "/api/auth/me", nil)
|
secondRequest := httptest.NewRequest("GET", "/api/auth/me", nil)
|
||||||
req2.Header.Set("Authorization", "Bearer "+user.Token)
|
secondRequest.Header.Set("Authorization", "Bearer "+user.Token)
|
||||||
req2 = testutils.WithUserContext(req2, middleware.UserIDKey, user.User.ID)
|
secondRequest = testutils.WithUserContext(secondRequest, middleware.UserIDKey, user.User.ID)
|
||||||
rec2 := httptest.NewRecorder()
|
secondRecorder := httptest.NewRecorder()
|
||||||
router.ServeHTTP(rec2, req2)
|
router.ServeHTTP(secondRecorder, secondRequest)
|
||||||
|
|
||||||
assertErrorResponse(t, rec2, http.StatusUnauthorized)
|
assertErrorResponse(t, secondRecorder, http.StatusUnauthorized)
|
||||||
})
|
})
|
||||||
|
|
||||||
t.Run("Refresh_Token_Revocation", func(t *testing.T) {
|
t.Run("Refresh_Token_Revocation", func(t *testing.T) {
|
||||||
@@ -90,48 +90,48 @@ func TestIntegration_SessionManagement(t *testing.T) {
|
|||||||
t.Fatal("Expected refresh token")
|
t.Fatal("Expected refresh token")
|
||||||
}
|
}
|
||||||
|
|
||||||
reqBody := map[string]string{
|
requestBody := map[string]string{
|
||||||
"refresh_token": loginResult.RefreshToken,
|
"refresh_token": loginResult.RefreshToken,
|
||||||
}
|
}
|
||||||
body, _ := json.Marshal(reqBody)
|
body, _ := json.Marshal(requestBody)
|
||||||
req := httptest.NewRequest("POST", "/api/auth/refresh", bytes.NewBuffer(body))
|
request := httptest.NewRequest("POST", "/api/auth/refresh", bytes.NewBuffer(body))
|
||||||
req.Header.Set("Content-Type", "application/json")
|
request.Header.Set("Content-Type", "application/json")
|
||||||
rec := httptest.NewRecorder()
|
recorder := httptest.NewRecorder()
|
||||||
router.ServeHTTP(rec, req)
|
router.ServeHTTP(recorder, request)
|
||||||
|
|
||||||
assertStatus(t, rec, http.StatusOK)
|
assertStatus(t, recorder, http.StatusOK)
|
||||||
|
|
||||||
if err := ctx.AuthService.RevokeRefreshToken(loginResult.RefreshToken); err != nil {
|
if err := ctx.AuthService.RevokeRefreshToken(loginResult.RefreshToken); err != nil {
|
||||||
t.Fatalf("Failed to revoke token: %v", err)
|
t.Fatalf("Failed to revoke token: %v", err)
|
||||||
}
|
}
|
||||||
|
|
||||||
req2 := httptest.NewRequest("POST", "/api/auth/refresh", bytes.NewBuffer(body))
|
secondRequest := httptest.NewRequest("POST", "/api/auth/refresh", bytes.NewBuffer(body))
|
||||||
req2.Header.Set("Content-Type", "application/json")
|
secondRequest.Header.Set("Content-Type", "application/json")
|
||||||
rec2 := httptest.NewRecorder()
|
secondRecorder := httptest.NewRecorder()
|
||||||
router.ServeHTTP(rec2, req2)
|
router.ServeHTTP(secondRecorder, secondRequest)
|
||||||
|
|
||||||
assertErrorResponse(t, rec2, http.StatusUnauthorized)
|
assertErrorResponse(t, secondRecorder, http.StatusUnauthorized)
|
||||||
})
|
})
|
||||||
|
|
||||||
t.Run("Multiple_Sessions_Independent", func(t *testing.T) {
|
t.Run("Multiple_Sessions_Independent", func(t *testing.T) {
|
||||||
ctx.Suite.EmailSender.Reset()
|
ctx.Suite.EmailSender.Reset()
|
||||||
user1 := createAuthenticatedUser(t, ctx.AuthService, ctx.Suite.UserRepo, "multi_session_user1", "multi_session1@example.com")
|
firstUser := createAuthenticatedUser(t, ctx.AuthService, ctx.Suite.UserRepo, "multi_session_user1", "multi_session1@example.com")
|
||||||
user2 := createAuthenticatedUser(t, ctx.AuthService, ctx.Suite.UserRepo, "multi_session_user2", "multi_session2@example.com")
|
secondUser := createAuthenticatedUser(t, ctx.AuthService, ctx.Suite.UserRepo, "multi_session_user2", "multi_session2@example.com")
|
||||||
|
|
||||||
req1 := httptest.NewRequest("GET", "/api/auth/me", nil)
|
firstRequest := httptest.NewRequest("GET", "/api/auth/me", nil)
|
||||||
req1.Header.Set("Authorization", "Bearer "+user1.Token)
|
firstRequest.Header.Set("Authorization", "Bearer "+firstUser.Token)
|
||||||
req1 = testutils.WithUserContext(req1, middleware.UserIDKey, user1.User.ID)
|
firstRequest = testutils.WithUserContext(firstRequest, middleware.UserIDKey, firstUser.User.ID)
|
||||||
rec1 := httptest.NewRecorder()
|
firstRecorder := httptest.NewRecorder()
|
||||||
router.ServeHTTP(rec1, req1)
|
router.ServeHTTP(firstRecorder, firstRequest)
|
||||||
|
|
||||||
req2 := httptest.NewRequest("GET", "/api/auth/me", nil)
|
secondRequest := httptest.NewRequest("GET", "/api/auth/me", nil)
|
||||||
req2.Header.Set("Authorization", "Bearer "+user2.Token)
|
secondRequest.Header.Set("Authorization", "Bearer "+secondUser.Token)
|
||||||
req2 = testutils.WithUserContext(req2, middleware.UserIDKey, user2.User.ID)
|
secondRequest = testutils.WithUserContext(secondRequest, middleware.UserIDKey, secondUser.User.ID)
|
||||||
rec2 := httptest.NewRecorder()
|
secondRecorder := httptest.NewRecorder()
|
||||||
router.ServeHTTP(rec2, req2)
|
router.ServeHTTP(secondRecorder, secondRequest)
|
||||||
|
|
||||||
assertStatus(t, rec1, http.StatusOK)
|
assertStatus(t, firstRecorder, http.StatusOK)
|
||||||
assertStatus(t, rec2, http.StatusOK)
|
assertStatus(t, secondRecorder, http.StatusOK)
|
||||||
})
|
})
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -144,17 +144,17 @@ func TestIntegration_AccountDeletion(t *testing.T) {
|
|||||||
user := createAuthenticatedUser(t, ctx.AuthService, ctx.Suite.UserRepo, "del_flow_user", "del_flow@example.com")
|
user := createAuthenticatedUser(t, ctx.AuthService, ctx.Suite.UserRepo, "del_flow_user", "del_flow@example.com")
|
||||||
post := testutils.CreatePostWithRepo(t, ctx.Suite.PostRepo, user.User.ID, "Test Post", "https://example.com")
|
post := testutils.CreatePostWithRepo(t, ctx.Suite.PostRepo, user.User.ID, "Test Post", "https://example.com")
|
||||||
|
|
||||||
reqBody := map[string]string{}
|
requestBody := map[string]string{}
|
||||||
body, _ := json.Marshal(reqBody)
|
body, _ := json.Marshal(requestBody)
|
||||||
req := httptest.NewRequest("DELETE", "/api/auth/account", bytes.NewBuffer(body))
|
request := httptest.NewRequest("DELETE", "/api/auth/account", bytes.NewBuffer(body))
|
||||||
req.Header.Set("Content-Type", "application/json")
|
request.Header.Set("Content-Type", "application/json")
|
||||||
req.Header.Set("Authorization", "Bearer "+user.Token)
|
request.Header.Set("Authorization", "Bearer "+user.Token)
|
||||||
req = testutils.WithUserContext(req, middleware.UserIDKey, user.User.ID)
|
request = testutils.WithUserContext(request, middleware.UserIDKey, user.User.ID)
|
||||||
rec := httptest.NewRecorder()
|
recorder := httptest.NewRecorder()
|
||||||
|
|
||||||
router.ServeHTTP(rec, req)
|
router.ServeHTTP(recorder, request)
|
||||||
|
|
||||||
response := assertJSONResponse(t, rec, http.StatusOK)
|
response := assertJSONResponse(t, recorder, http.StatusOK)
|
||||||
if response == nil {
|
if response == nil {
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
@@ -171,13 +171,13 @@ func TestIntegration_AccountDeletion(t *testing.T) {
|
|||||||
"token": deletionToken,
|
"token": deletionToken,
|
||||||
}
|
}
|
||||||
confirmBodyBytes, _ := json.Marshal(confirmBody)
|
confirmBodyBytes, _ := json.Marshal(confirmBody)
|
||||||
confirmReq := httptest.NewRequest("POST", "/api/auth/account/confirm", bytes.NewBuffer(confirmBodyBytes))
|
confirmRequest := httptest.NewRequest("POST", "/api/auth/account/confirm", bytes.NewBuffer(confirmBodyBytes))
|
||||||
confirmReq.Header.Set("Content-Type", "application/json")
|
confirmRequest.Header.Set("Content-Type", "application/json")
|
||||||
confirmRec := httptest.NewRecorder()
|
confirmRecorder := httptest.NewRecorder()
|
||||||
|
|
||||||
router.ServeHTTP(confirmRec, confirmReq)
|
router.ServeHTTP(confirmRecorder, confirmRequest)
|
||||||
|
|
||||||
confirmResponse := assertJSONResponse(t, confirmRec, http.StatusOK)
|
confirmResponse := assertJSONResponse(t, confirmRecorder, http.StatusOK)
|
||||||
if confirmResponse == nil {
|
if confirmResponse == nil {
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
@@ -209,17 +209,17 @@ func TestIntegration_AccountDeletion(t *testing.T) {
|
|||||||
user := createAuthenticatedUser(t, ctx.AuthService, ctx.Suite.UserRepo, "del_posts_user", "del_posts@example.com")
|
user := createAuthenticatedUser(t, ctx.AuthService, ctx.Suite.UserRepo, "del_posts_user", "del_posts@example.com")
|
||||||
post := testutils.CreatePostWithRepo(t, ctx.Suite.PostRepo, user.User.ID, "Deletion Post", "https://example.com/deletion")
|
post := testutils.CreatePostWithRepo(t, ctx.Suite.PostRepo, user.User.ID, "Deletion Post", "https://example.com/deletion")
|
||||||
|
|
||||||
reqBody := map[string]string{}
|
requestBody := map[string]string{}
|
||||||
body, _ := json.Marshal(reqBody)
|
body, _ := json.Marshal(requestBody)
|
||||||
req := httptest.NewRequest("DELETE", "/api/auth/account", bytes.NewBuffer(body))
|
request := httptest.NewRequest("DELETE", "/api/auth/account", bytes.NewBuffer(body))
|
||||||
req.Header.Set("Content-Type", "application/json")
|
request.Header.Set("Content-Type", "application/json")
|
||||||
req.Header.Set("Authorization", "Bearer "+user.Token)
|
request.Header.Set("Authorization", "Bearer "+user.Token)
|
||||||
req = testutils.WithUserContext(req, middleware.UserIDKey, user.User.ID)
|
request = testutils.WithUserContext(request, middleware.UserIDKey, user.User.ID)
|
||||||
rec := httptest.NewRecorder()
|
recorder := httptest.NewRecorder()
|
||||||
|
|
||||||
router.ServeHTTP(rec, req)
|
router.ServeHTTP(recorder, request)
|
||||||
|
|
||||||
response := assertJSONResponse(t, rec, http.StatusOK)
|
response := assertJSONResponse(t, recorder, http.StatusOK)
|
||||||
if response == nil {
|
if response == nil {
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
@@ -237,13 +237,13 @@ func TestIntegration_AccountDeletion(t *testing.T) {
|
|||||||
"delete_posts": true,
|
"delete_posts": true,
|
||||||
}
|
}
|
||||||
confirmBodyBytes, _ := json.Marshal(confirmBody)
|
confirmBodyBytes, _ := json.Marshal(confirmBody)
|
||||||
confirmReq := httptest.NewRequest("POST", "/api/auth/account/confirm", bytes.NewBuffer(confirmBodyBytes))
|
confirmRequest := httptest.NewRequest("POST", "/api/auth/account/confirm", bytes.NewBuffer(confirmBodyBytes))
|
||||||
confirmReq.Header.Set("Content-Type", "application/json")
|
confirmRequest.Header.Set("Content-Type", "application/json")
|
||||||
confirmRec := httptest.NewRecorder()
|
confirmRecorder := httptest.NewRecorder()
|
||||||
|
|
||||||
router.ServeHTTP(confirmRec, confirmReq)
|
router.ServeHTTP(confirmRecorder, confirmRequest)
|
||||||
|
|
||||||
confirmResponse := assertJSONResponse(t, confirmRec, http.StatusOK)
|
confirmResponse := assertJSONResponse(t, confirmRecorder, http.StatusOK)
|
||||||
if confirmResponse == nil {
|
if confirmResponse == nil {
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
@@ -275,12 +275,12 @@ func TestIntegration_MetricsCollection(t *testing.T) {
|
|||||||
router := ctx.Router
|
router := ctx.Router
|
||||||
|
|
||||||
t.Run("Metrics_Endpoint_Returns_Data", func(t *testing.T) {
|
t.Run("Metrics_Endpoint_Returns_Data", func(t *testing.T) {
|
||||||
req := httptest.NewRequest("GET", "/metrics", nil)
|
request := httptest.NewRequest("GET", "/metrics", nil)
|
||||||
rec := httptest.NewRecorder()
|
recorder := httptest.NewRecorder()
|
||||||
|
|
||||||
router.ServeHTTP(rec, req)
|
router.ServeHTTP(recorder, request)
|
||||||
|
|
||||||
response := assertJSONResponse(t, rec, http.StatusOK)
|
response := assertJSONResponse(t, recorder, http.StatusOK)
|
||||||
if response != nil {
|
if response != nil {
|
||||||
if data, ok := response["data"].(map[string]any); ok {
|
if data, ok := response["data"].(map[string]any); ok {
|
||||||
if _, exists := data["database"]; !exists {
|
if _, exists := data["database"]; !exists {
|
||||||
@@ -294,13 +294,13 @@ func TestIntegration_MetricsCollection(t *testing.T) {
|
|||||||
ctx.Suite.EmailSender.Reset()
|
ctx.Suite.EmailSender.Reset()
|
||||||
createAuthenticatedUser(t, ctx.AuthService, ctx.Suite.UserRepo, "metrics_user", "metrics@example.com")
|
createAuthenticatedUser(t, ctx.AuthService, ctx.Suite.UserRepo, "metrics_user", "metrics@example.com")
|
||||||
|
|
||||||
req := httptest.NewRequest("GET", "/metrics", nil)
|
request := httptest.NewRequest("GET", "/metrics", nil)
|
||||||
rec := httptest.NewRecorder()
|
recorder := httptest.NewRecorder()
|
||||||
|
|
||||||
router.ServeHTTP(rec, req)
|
router.ServeHTTP(recorder, request)
|
||||||
|
|
||||||
var response map[string]any
|
var response map[string]any
|
||||||
if err := json.NewDecoder(rec.Body).Decode(&response); err == nil {
|
if err := json.NewDecoder(recorder.Body).Decode(&response); err == nil {
|
||||||
if data, ok := response["data"].(map[string]any); ok {
|
if data, ok := response["data"].(map[string]any); ok {
|
||||||
if dbData, exists := data["database"].(map[string]any); exists {
|
if dbData, exists := data["database"].(map[string]any); exists {
|
||||||
if _, hasQueries := dbData["total_queries"]; !hasQueries {
|
if _, hasQueries := dbData["total_queries"]; !hasQueries {
|
||||||
@@ -323,7 +323,7 @@ func TestIntegration_ConcurrentRequests(t *testing.T) {
|
|||||||
var wg sync.WaitGroup
|
var wg sync.WaitGroup
|
||||||
errors := make(chan error, 10)
|
errors := make(chan error, 10)
|
||||||
|
|
||||||
for i := 0; i < 10; i++ {
|
for idx := range 10 {
|
||||||
wg.Add(1)
|
wg.Add(1)
|
||||||
go func(index int) {
|
go func(index int) {
|
||||||
defer wg.Done()
|
defer wg.Done()
|
||||||
@@ -334,18 +334,18 @@ func TestIntegration_ConcurrentRequests(t *testing.T) {
|
|||||||
"content": "Concurrent test content",
|
"content": "Concurrent test content",
|
||||||
}
|
}
|
||||||
body, _ := json.Marshal(postBody)
|
body, _ := json.Marshal(postBody)
|
||||||
req := httptest.NewRequest("POST", "/api/posts", bytes.NewBuffer(body))
|
request := httptest.NewRequest("POST", "/api/posts", bytes.NewBuffer(body))
|
||||||
req.Header.Set("Content-Type", "application/json")
|
request.Header.Set("Content-Type", "application/json")
|
||||||
req.Header.Set("Authorization", "Bearer "+user.Token)
|
request.Header.Set("Authorization", "Bearer "+user.Token)
|
||||||
req = testutils.WithUserContext(req, middleware.UserIDKey, user.User.ID)
|
request = testutils.WithUserContext(request, middleware.UserIDKey, user.User.ID)
|
||||||
rec := httptest.NewRecorder()
|
recorder := httptest.NewRecorder()
|
||||||
|
|
||||||
router.ServeHTTP(rec, req)
|
router.ServeHTTP(recorder, request)
|
||||||
|
|
||||||
if rec.Code != http.StatusCreated {
|
if recorder.Code != http.StatusCreated {
|
||||||
errors <- fmt.Errorf("Post %d failed with status %d", index, rec.Code)
|
errors <- fmt.Errorf("Post %d failed with status %d", index, recorder.Code)
|
||||||
}
|
}
|
||||||
}(i)
|
}(idx)
|
||||||
}
|
}
|
||||||
|
|
||||||
wg.Wait()
|
wg.Wait()
|
||||||
@@ -370,28 +370,26 @@ func TestIntegration_ConcurrentRequests(t *testing.T) {
|
|||||||
var wg sync.WaitGroup
|
var wg sync.WaitGroup
|
||||||
errors := make(chan error, 5)
|
errors := make(chan error, 5)
|
||||||
|
|
||||||
for i := 0; i < 5; i++ {
|
for range 5 {
|
||||||
wg.Add(1)
|
wg.Go(func() {
|
||||||
go func() {
|
|
||||||
defer wg.Done()
|
|
||||||
|
|
||||||
voteBody := map[string]string{
|
voteBody := map[string]string{
|
||||||
"type": "up",
|
"type": "up",
|
||||||
}
|
}
|
||||||
body, _ := json.Marshal(voteBody)
|
body, _ := json.Marshal(voteBody)
|
||||||
req := httptest.NewRequest("POST", fmt.Sprintf("/api/posts/%d/vote", post.ID), bytes.NewBuffer(body))
|
request := httptest.NewRequest("POST", fmt.Sprintf("/api/posts/%d/vote", post.ID), bytes.NewBuffer(body))
|
||||||
req.Header.Set("Content-Type", "application/json")
|
request.Header.Set("Content-Type", "application/json")
|
||||||
req.Header.Set("Authorization", "Bearer "+user.Token)
|
request.Header.Set("Authorization", "Bearer "+user.Token)
|
||||||
req = testutils.WithUserContext(req, middleware.UserIDKey, user.User.ID)
|
request = testutils.WithUserContext(request, middleware.UserIDKey, user.User.ID)
|
||||||
req = testutils.WithURLParams(req, map[string]string{"id": fmt.Sprintf("%d", post.ID)})
|
request = testutils.WithURLParams(request, map[string]string{"id": fmt.Sprintf("%d", post.ID)})
|
||||||
rec := httptest.NewRecorder()
|
recorder := httptest.NewRecorder()
|
||||||
|
|
||||||
router.ServeHTTP(rec, req)
|
router.ServeHTTP(recorder, request)
|
||||||
|
|
||||||
if rec.Code != http.StatusOK {
|
if recorder.Code != http.StatusOK {
|
||||||
errors <- fmt.Errorf("Vote failed with status %d", rec.Code)
|
errors <- fmt.Errorf("Vote failed with status %d", recorder.Code)
|
||||||
}
|
}
|
||||||
}()
|
})
|
||||||
}
|
}
|
||||||
|
|
||||||
wg.Wait()
|
wg.Wait()
|
||||||
@@ -411,20 +409,18 @@ func TestIntegration_ConcurrentRequests(t *testing.T) {
|
|||||||
var wg sync.WaitGroup
|
var wg sync.WaitGroup
|
||||||
errors := make(chan error, 20)
|
errors := make(chan error, 20)
|
||||||
|
|
||||||
for i := 0; i < 20; i++ {
|
for range 20 {
|
||||||
wg.Add(1)
|
wg.Go(func() {
|
||||||
go func() {
|
|
||||||
defer wg.Done()
|
|
||||||
|
|
||||||
req := httptest.NewRequest("GET", "/api/posts", nil)
|
request := httptest.NewRequest("GET", "/api/posts", nil)
|
||||||
rec := httptest.NewRecorder()
|
recorder := httptest.NewRecorder()
|
||||||
|
|
||||||
router.ServeHTTP(rec, req)
|
router.ServeHTTP(recorder, request)
|
||||||
|
|
||||||
if rec.Code != http.StatusOK {
|
if recorder.Code != http.StatusOK {
|
||||||
errors <- fmt.Errorf("Read failed with status %d", rec.Code)
|
errors <- fmt.Errorf("Read failed with status %d", recorder.Code)
|
||||||
}
|
}
|
||||||
}()
|
})
|
||||||
}
|
}
|
||||||
|
|
||||||
wg.Wait()
|
wg.Wait()
|
||||||
|
|||||||
Reference in New Issue
Block a user