initial commit

tests/integration/auth_desktop_handoff_code_normalization_test.go

@@ -0,0 +1,88 @@
+/*
+ * Copyright (C) 2026 Fluxer Contributors
+ *
+ * This file is part of Fluxer.
+ *
+ * Fluxer is free software: you can redistribute it and/or modify
+ * it under the terms of the GNU Affero General Public License as published by
+ * the Free Software Foundation, either version 3 of the License, or
+ * (at your option) any later version.
+ *
+ * Fluxer is distributed in the hope that it will be useful,
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
+ * GNU Affero General Public License for more details.
+ *
+ * You should have received a copy of the GNU Affero General Public License
+ * along with Fluxer. If not, see <https://www.gnu.org/licenses/>.
+ */
+
+package integration
+
+import (
+	"fmt"
+	"net/http"
+	"strings"
+	"testing"
+)
+
+// TestAuthDesktopHandoffCodeNormalization tests that codes work with or without dashes
+// and are case-insensitive
+func TestAuthDesktopHandoffCodeNormalization(t *testing.T) {
+	client := newTestClient(t)
+	account := createTestAccount(t, client)
+	login := loginTestUser(t, client, account.Email, account.Password)
+
+	resp, err := client.postJSON("/auth/handoff/initiate", nil)
+	if err != nil {
+		t.Fatalf("failed to initiate desktop handoff: %v", err)
+	}
+	assertStatus(t, resp, http.StatusOK)
+	var initResp handoffInitiateResponse
+	decodeJSONResponse(t, resp, &initResp)
+
+	if !validateHandoffCodeFormat(initResp.Code) {
+		t.Fatalf("expected code in XXXX-XXXX format, got %s", initResp.Code)
+	}
+
+	codeWithoutDash := strings.ReplaceAll(initResp.Code, "-", "")
+	statusURL := fmt.Sprintf("/auth/handoff/%s/status", codeWithoutDash)
+	resp, err = client.get(statusURL)
+	if err != nil {
+		t.Fatalf("failed to poll status with code without dash: %v", err)
+	}
+	assertStatus(t, resp, http.StatusOK)
+	var status handoffStatusResponse
+	decodeJSONResponse(t, resp, &status)
+	if status.Status != "pending" {
+		t.Fatalf("expected pending status, got %s", status.Status)
+	}
+
+	lowercaseCode := strings.ToLower(initResp.Code)
+	resp, err = client.postJSON("/auth/handoff/complete", map[string]string{
+		"code":    lowercaseCode,
+		"token":   login.Token,
+		"user_id": login.UserID,
+	})
+	if err != nil {
+		t.Fatalf("failed to complete handoff with lowercase code: %v", err)
+	}
+	assertStatus(t, resp, http.StatusNoContent)
+	resp.Body.Close()
+
+	resp, err = client.get(fmt.Sprintf("/auth/handoff/%s/status", initResp.Code))
+	if err != nil {
+		t.Fatalf("failed to poll completed handoff status: %v", err)
+	}
+	assertStatus(t, resp, http.StatusOK)
+	decodeJSONResponse(t, resp, &status)
+	if status.Status != "completed" {
+		t.Fatalf("expected completed status, got %s", status.Status)
+	}
+	if status.Token == "" {
+		t.Fatalf("expected token in completed status")
+	}
+	if status.Token == login.Token {
+		t.Fatalf("expected handoff token to be a new session token")
+	}
+}