From 057729834413d4028bf9cb482ecd12ee7d221945 Mon Sep 17 00:00:00 2001
From: Himself65 <himself65@outlook.com>
Date: Thu, 6 Apr 2023 09:01:44 -0500
Subject: [PATCH] test: compare public page json (#1793)

---
 .../src/affine/__tests__/api.spec.ts          | 22 +++++++++++++++----
 1 file changed, 18 insertions(+), 4 deletions(-)

diff --git a/packages/workspace/src/affine/__tests__/api.spec.ts b/packages/workspace/src/affine/__tests__/api.spec.ts
index e67db582ef..ac40ade4d1 100644
--- a/packages/workspace/src/affine/__tests__/api.spec.ts
+++ b/packages/workspace/src/affine/__tests__/api.spec.ts
@@ -308,6 +308,8 @@ describe('api', () => {
       id,
       public: true,
     });
+    let originalPage0Json: any = null;
+    let originalPage1Json: any = null;
     {
       const binary = await workspaceApis.downloadWorkspace(id, false);
       const workspace = new Workspace({
@@ -316,8 +318,12 @@ describe('api', () => {
         .register(AffineSchemas)
         .register(__unstableSchemas);
       applyUpdate(workspace.doc, new Uint8Array(binary));
-      expect(workspace.getPage('page0')).not.toBeUndefined();
-      expect(workspace.getPage('page1')).not.toBeUndefined();
+      const page0 = workspace.getPage('page0') as Page;
+      const page1 = workspace.getPage('page1') as Page;
+      expect(page0).not.toBeUndefined();
+      expect(page1).not.toBeUndefined();
+      originalPage0Json = workspace.doc.share.get(page0.prefixedId)?.toJSON();
+      originalPage1Json = workspace.doc.share.get(page1.prefixedId)?.toJSON();
     }
     {
       const workspace = new Workspace({
@@ -330,8 +336,12 @@ describe('api', () => {
         'page0'
       );
       applyUpdate(workspace.doc, new Uint8Array(binary));
-      expect(workspace.getPage('page0')).not.toBeNull();
+      const page0 = workspace.getPage('page0') as Page;
+      expect(page0).not.toBeNull();
       expect(workspace.getPage('page1')).toBeNull();
+      expect(workspace.doc.share.get(page0.prefixedId)?.toJSON()).toEqual(
+        originalPage0Json
+      );
     }
     {
       const workspace = new Workspace({
@@ -344,8 +354,12 @@ describe('api', () => {
         'page1'
       );
       applyUpdate(workspace.doc, new Uint8Array(binary));
+      const page1 = workspace.getPage('page1') as Page;
       expect(workspace.getPage('page0')).toBeNull();
-      expect(workspace.getPage('page1')).not.toBeNull();
+      expect(page1).not.toBeNull();
+      expect(workspace.doc.share.get(page1.prefixedId)?.toJSON()).toEqual(
+        originalPage1Json
+      );
     }
   });