From f5574c68fedefd018d5048a88eeffba5cf355c96 Mon Sep 17 00:00:00 2001
From: Himself65 <himself65@outlook.com>
Date: Sun, 2 Apr 2023 02:16:53 -0500
Subject: [PATCH] test: improve coverage (#1780)

---
 .../src/local/__tests__/crud.spec.ts          | 68 +++++++++++++++++++
 1 file changed, 68 insertions(+)
 create mode 100644 packages/workspace/src/local/__tests__/crud.spec.ts

diff --git a/packages/workspace/src/local/__tests__/crud.spec.ts b/packages/workspace/src/local/__tests__/crud.spec.ts
new file mode 100644
index 0000000000..0b7b3adbd9
--- /dev/null
+++ b/packages/workspace/src/local/__tests__/crud.spec.ts
@@ -0,0 +1,68 @@
+/**
+ * @vitest-environment happy-dom
+ */
+import 'fake-indexeddb/auto';
+
+import type { LocalWorkspace, WorkspaceCRUD } from '@affine/workspace/type';
+import { WorkspaceFlavour } from '@affine/workspace/type';
+import { __unstableSchemas, AffineSchemas } from '@blocksuite/blocks/models';
+import { Workspace } from '@blocksuite/store';
+import { afterEach, assertType, describe, expect, test } from 'vitest';
+
+import { CRUD } from '../crud';
+
+afterEach(() => {
+  localStorage.clear();
+});
+
+describe('crud', () => {
+  test('type', () => {
+    assertType<WorkspaceCRUD<WorkspaceFlavour.LOCAL>>(CRUD);
+  });
+
+  test('basic', async () => {
+    const workspace = await CRUD.get('not_exist');
+    expect(workspace).toBeNull();
+
+    expect(await CRUD.list()).toEqual([]);
+  });
+
+  test('delete not exist', async () => {
+    expect(async () =>
+      CRUD.delete({
+        id: 'not_exist',
+        flavour: WorkspaceFlavour.LOCAL,
+        blockSuiteWorkspace: new Workspace({ id: 'test' }),
+        providers: [],
+      })
+    ).rejects.toThrowError();
+  });
+
+  test('create & delete', async () => {
+    const workspace = new Workspace({ id: 'test' })
+      .register(AffineSchemas)
+      .register(__unstableSchemas);
+    const page = workspace.createPage('test');
+    const pageBlockId = page.addBlock('affine:page', {
+      title: new page.Text(''),
+    });
+    page.addBlock('affine:surface', {}, null);
+    const frameId = page.addBlock('affine:frame', {}, pageBlockId);
+    page.addBlock('affine:paragraph', {}, frameId);
+
+    const id = await CRUD.create(workspace);
+    const list = await CRUD.list();
+    expect(list.length).toBe(1);
+    expect(list[0].id).toBe(id);
+    const localWorkspace = list.at(0) as LocalWorkspace;
+    expect(localWorkspace.id).toBe(id);
+    expect(localWorkspace.flavour).toBe(WorkspaceFlavour.LOCAL);
+    expect(
+      Workspace.Y.encodeStateAsUpdate(localWorkspace.blockSuiteWorkspace.doc)
+    ).toEqual(Workspace.Y.encodeStateAsUpdate(workspace.doc));
+
+    await CRUD.delete(localWorkspace);
+    expect(await CRUD.get(id)).toBeNull();
+    expect(await CRUD.list()).toEqual([]);
+  });
+});