Harden server draft sync (Save & Exit + post-login transfer)

tests/components/AuthModalContext.test.tsx

@@ -1,4 +1,4 @@
-import React, { Suspense } from "react";
+import { Suspense } from "react";
 import { describe, it, expect, vi, beforeEach } from "vitest";
 import { screen, waitFor } from "@testing-library/react";
 import userEvent from "@testing-library/user-event";

tests/components/Login.test.tsx

@@ -1,4 +1,3 @@
-import React from "react";
 import { describe, it, expect, vi, beforeEach } from "vitest";
 import { screen, fireEvent, waitFor } from "@testing-library/react";
 import "@testing-library/jest-dom/vitest";

tests/components/LoginForm.test.tsx

@@ -1,4 +1,4 @@
-import React, { Suspense } from "react";
+import { Suspense } from "react";
 import { describe, it, expect, vi, beforeEach } from "vitest";
 import { screen, waitFor } from "@testing-library/react";
 import userEvent from "@testing-library/user-event";

tests/unit/draftHydrationUtils.test.ts

@@ -0,0 +1,13 @@
+import { describe, it, expect } from "vitest";
+import { createFlowStateHasKeys } from "../../lib/create/draftHydrationUtils";
+
+describe("createFlowStateHasKeys", () => {
+  it("returns false for empty object", () => {
+    expect(createFlowStateHasKeys({})).toBe(false);
+  });
+
+  it("returns true when any key is present", () => {
+    expect(createFlowStateHasKeys({ title: "x" })).toBe(true);
+    expect(createFlowStateHasKeys({ currentStep: "text" })).toBe(true);
+  });
+});

tests/unit/saveDraftToServer.test.ts

@@ -0,0 +1,104 @@
+import { describe, it, expect, vi, beforeEach, afterEach } from "vitest";
+import { saveDraftToServer } from "../../lib/create/api";
+import type { CreateFlowState } from "../../app/create/types";
+
+const minimalState: CreateFlowState = {};
+
+describe("saveDraftToServer", () => {
+  const originalFetch = globalThis.fetch;
+
+  beforeEach(() => {
+    vi.resetAllMocks();
+  });
+
+  afterEach(() => {
+    globalThis.fetch = originalFetch;
+  });
+
+  it("returns ok true on 200", async () => {
+    globalThis.fetch = vi.fn().mockResolvedValue(
+      new Response(JSON.stringify({ draft: { payload: {}, updatedAt: "" } }), {
+        status: 200,
+        headers: { "Content-Type": "application/json" },
+      }),
+    );
+    const result = await saveDraftToServer(minimalState);
+    expect(result).toEqual({ ok: true });
+  });
+
+  it("returns message from validation error body", async () => {
+    globalThis.fetch = vi.fn().mockResolvedValue(
+      new Response(
+        JSON.stringify({
+          error: { code: "validation_error", message: "Payload invalid" },
+        }),
+        { status: 400, headers: { "Content-Type": "application/json" } },
+      ),
+    );
+    const result = await saveDraftToServer(minimalState);
+    expect(result).toEqual({
+      ok: false,
+      message: "Payload invalid",
+      status: 400,
+    });
+  });
+
+  it("returns message from 413 payload_too_large", async () => {
+    globalThis.fetch = vi.fn().mockResolvedValue(
+      new Response(
+        JSON.stringify({
+          error: {
+            code: "payload_too_large",
+            message: "Request body must be at most 524288 bytes",
+          },
+        }),
+        { status: 413, headers: { "Content-Type": "application/json" } },
+      ),
+    );
+    const result = await saveDraftToServer(minimalState);
+    expect(result.ok).toBe(false);
+    if (result.ok === false) {
+      expect(result.message).toContain("524288");
+      expect(result.status).toBe(413);
+    }
+  });
+
+  it("returns Unauthorized string from 401 legacy shape", async () => {
+    globalThis.fetch = vi.fn().mockResolvedValue(
+      new Response(JSON.stringify({ error: "Unauthorized" }), {
+        status: 401,
+        headers: { "Content-Type": "application/json" },
+      }),
+    );
+    const result = await saveDraftToServer(minimalState);
+    expect(result).toEqual({
+      ok: false,
+      message: "Unauthorized",
+      status: 401,
+    });
+  });
+
+  it("falls back when error body is not JSON", async () => {
+    globalThis.fetch = vi.fn().mockResolvedValue(
+      new Response("not json", {
+        status: 500,
+        statusText: "Internal Server Error",
+      }),
+    );
+    const result = await saveDraftToServer(minimalState);
+    expect(result).toEqual({
+      ok: false,
+      message: "Internal Server Error",
+      status: 500,
+    });
+  });
+
+  it("returns network message when fetch rejects", async () => {
+    globalThis.fetch = vi.fn().mockRejectedValue(new Error("offline"));
+    const result = await saveDraftToServer(minimalState);
+    expect(result).toEqual({
+      ok: false,
+      message: "Something went wrong. Check your connection and try again.",
+    });
+  });
+});