webeditor3d/tests/serialization/local-draft-storage.test.js

import { describe, expect, it } from "vitest";
import { createBoxBrush } from "../../src/document/brushes";
import { SCENE_DOCUMENT_VERSION, createEmptySceneDocument } from "../../src/document/scene-document";
import { serializeSceneDocument } from "../../src/serialization/scene-document-json";
import { DEFAULT_SCENE_DRAFT_STORAGE_KEY, getBrowserStorageAccess, loadOrCreateSceneDocument, loadSceneDocumentDraft, saveSceneDocumentDraft } from "../../src/serialization/local-draft-storage";
import { createDefaultViewportLayoutState } from "../../src/viewport-three/viewport-layout";
class MemoryStorage {
    values = new Map();
    getItem(key) {
        return this.values.get(key) ?? null;
    }
    setItem(key, value) {
        this.values.set(key, value);
    }
    removeItem(key) {
        this.values.delete(key);
    }
}
class ThrowingStorage {
    options;
    constructor(options = {}) {
        this.options = options;
    }
    getItem() {
        if (this.options.onGetItem !== undefined) {
            throw this.options.onGetItem;
        }
        return null;
    }
    setItem() {
        if (this.options.onSetItem !== undefined) {
            throw this.options.onSetItem;
        }
    }
    removeItem() {
        if (this.options.onRemoveItem !== undefined) {
            throw this.options.onRemoveItem;
        }
    }
}
describe("local draft storage", () => {
    it("falls back to a fresh document when stored draft JSON is invalid", () => {
        const storage = new MemoryStorage();
        storage.setItem(DEFAULT_SCENE_DRAFT_STORAGE_KEY, "{invalid-json");
        const result = loadOrCreateSceneDocument(storage);
        expect(result.document.version).toBe(SCENE_DOCUMENT_VERSION);
        expect(result.document).toEqual(createEmptySceneDocument());
        expect(result.diagnostic).toContain("Stored local draft could not be loaded.");
        expect(result.diagnostic).toContain("Starting with a fresh empty document.");
    });
    it("reports browser storage access failures without throwing", () => {
        const originalDescriptor = Object.getOwnPropertyDescriptor(window, "localStorage");
        Object.defineProperty(window, "localStorage", {
            configurable: true,
            get() {
                throw new Error("access denied");
            }
        });
        try {
            const result = getBrowserStorageAccess();
            expect(result.storage).toBeNull();
            expect(result.diagnostic).toContain("Browser local storage is unavailable.");
            expect(result.diagnostic).toContain("access denied");
        }
        finally {
            if (originalDescriptor !== undefined) {
                Object.defineProperty(window, "localStorage", originalDescriptor);
            }
            else {
                Reflect.deleteProperty(window, "localStorage");
            }
        }
    });
    it("returns an error result when reading from storage throws", () => {
        const result = loadSceneDocumentDraft(new ThrowingStorage({
            onGetItem: new Error("blocked read")
        }));
        expect(result.status).toBe("error");
        expect(result.message).toContain("blocked read");
    });
    it("returns an error result when saving to storage throws", () => {
        const result = saveSceneDocumentDraft(new ThrowingStorage({
            onSetItem: new Error("quota exceeded")
        }), createEmptySceneDocument());
        expect(result.status).toBe("error");
        expect(result.message).toContain("quota exceeded");
    });
    it("stores and restores editor viewport layout state alongside the document draft", () => {
        const storage = new MemoryStorage();
        const viewportLayoutState = createDefaultViewportLayoutState();
        viewportLayoutState.layoutMode = "quad";
        viewportLayoutState.activePanelId = "bottomRight";
        viewportLayoutState.panels.topLeft.displayMode = "wireframe";
        viewportLayoutState.panels.topLeft.cameraState.target = {
            x: 8,
            y: 3,
            z: -5
        };
        viewportLayoutState.panels.topLeft.cameraState.perspectiveOrbit.theta = 1.25;
        viewportLayoutState.panels.topLeft.cameraState.orthographicZoom = 2.5;
        expect(saveSceneDocumentDraft(storage, createEmptySceneDocument({ name: "Viewport Draft" }), viewportLayoutState)).toEqual({
            status: "saved",
            message: "Local draft saved."
        });
        const result = loadSceneDocumentDraft(storage);
        expect(result).toMatchObject({
            status: "loaded",
            document: {
                name: "Viewport Draft"
            },
            viewportLayoutState: {
                layoutMode: "quad",
                activePanelId: "bottomRight",
                panels: {
                    topLeft: {
                        displayMode: "wireframe",
                        cameraState: {
                            target: {
                                x: 8,
                                y: 3,
                                z: -5
                            },
                            perspectiveOrbit: {
                                theta: 1.25
                            },
                            orthographicZoom: 2.5
                        }
                    }
                }
            }
        });
    });
    it("loads older raw scene-document drafts without requiring viewport layout state", () => {
        const storage = new MemoryStorage();
        storage.setItem(DEFAULT_SCENE_DRAFT_STORAGE_KEY, serializeSceneDocument(createEmptySceneDocument({ name: "Legacy Draft" })));
        const result = loadSceneDocumentDraft(storage);
        expect(result).toMatchObject({
            status: "loaded",
            document: {
                name: "Legacy Draft"
            },
            viewportLayoutState: null
        });
    });
    it("refuses to save an invalid scene document draft", () => {
        const invalidBrush = createBoxBrush({
            id: "brush-invalid"
        });
        invalidBrush.faces.posX.materialId = "missing-material";
        const result = saveSceneDocumentDraft(new MemoryStorage(), {
            ...createEmptySceneDocument(),
            brushes: {
                [invalidBrush.id]: invalidBrush
            }
        });
        expect(result.status).toBe("error");
        expect(result.message).toContain("validation error");
    });
});