From ac81ce66ff4cc1d75ea33a2eff6675d833da6ac8 Mon Sep 17 00:00:00 2001 From: Kharec Date: Thu, 13 Nov 2025 08:15:21 +0100 Subject: [PATCH] refactor: use assertCookie helper for auth token check --- .../page_handler_integration_test.go | 17 +++-------------- 1 file changed, 3 insertions(+), 14 deletions(-) diff --git a/internal/integration/page_handler_integration_test.go b/internal/integration/page_handler_integration_test.go index 7f9fbb5..788a6a7 100644 --- a/internal/integration/page_handler_integration_test.go +++ b/internal/integration/page_handler_integration_test.go @@ -62,18 +62,7 @@ func TestIntegration_PageHandler(t *testing.T) { router.ServeHTTP(rec, req) - cookies := rec.Result().Cookies() - csrfFound := false - for _, cookie := range cookies { - if cookie.Name == "csrf_token" { - csrfFound = true - break - } - } - - if !csrfFound { - t.Error("Expected CSRF token cookie to be set") - } + assertCookie(t, rec, "csrf_token", "") }) t.Run("PageHandler_Form_Submission", func(t *testing.T) { @@ -114,7 +103,7 @@ func TestIntegration_PageHandler(t *testing.T) { t.Run("PageHandler_Authenticated_Access", func(t *testing.T) { ctx.Suite.EmailSender.Reset() - user := createAuthenticatedUser(t, ctx.AuthService, ctx.Suite.UserRepo, "page_auth_user", "page_auth@example.com") + user := createUserWithCleanup(t, ctx, "page_auth_user", "page_auth@example.com") req := httptest.NewRequest("GET", "/settings", nil) req.AddCookie(&http.Cookie{Name: "auth_token", Value: user.Token}) @@ -127,7 +116,7 @@ func TestIntegration_PageHandler(t *testing.T) { t.Run("PageHandler_Post_Display", func(t *testing.T) { ctx.Suite.EmailSender.Reset() - user := createAuthenticatedUser(t, ctx.AuthService, ctx.Suite.UserRepo, "page_post_user", "page_post@example.com") + user := createUserWithCleanup(t, ctx, "page_post_user", "page_post@example.com") post := testutils.CreatePostWithRepo(t, ctx.Suite.PostRepo, user.User.ID, "Page Test Post", "https://example.com/page-test")