mirror of
https://github.com/toeverything/AFFiNE.git
synced 2024-12-25 20:12:13 +03:00
test: remove deprecated test (#2880)
This commit is contained in:
Alex Yang
GitHub
parent
78b74d5b15
commit
26ac56e163
@ -1,484 +0,0 @@
|
||||
/**
|
||||
* @vitest-environment happy-dom
|
||||
*/
|
||||
import 'fake-indexeddb/auto';
|
||||
|
||||
import { readFile } from 'node:fs/promises';
|
||||
|
||||
import { MessageCode } from '@affine/env/constant';
|
||||
import { WorkspaceFlavour } from '@affine/env/workspace';
|
||||
import {
|
||||
createWorkspaceResponseSchema,
|
||||
usageResponseSchema,
|
||||
} from '@affine/env/workspace/legacy-cloud';
|
||||
import user1 from '@affine-test/fixtures/built-in-user1.json';
|
||||
import user2 from '@affine-test/fixtures/built-in-user2.json';
|
||||
import { __unstableSchemas, AffineSchemas } from '@blocksuite/blocks/models';
|
||||
import { assertExists } from '@blocksuite/global/utils';
|
||||
import type { Page, PageMeta } from '@blocksuite/store';
|
||||
import { Workspace } from '@blocksuite/store';
|
||||
import { faker } from '@faker-js/faker';
|
||||
import { beforeEach, describe, expect, test, vi } from 'vitest';
|
||||
import { WebSocket } from 'ws';
|
||||
import { applyUpdate } from 'yjs';
|
||||
|
||||
import { createEmptyBlockSuiteWorkspace } from '../../utils';
|
||||
import { createUserApis, createWorkspaceApis, RequestError } from '../api';
|
||||
import { createStatusApis } from '../api/status';
|
||||
import { KeckProvider } from '../keck';
|
||||
import {
|
||||
createAffineAuth,
|
||||
getLoginStorage,
|
||||
loginResponseSchema,
|
||||
setLoginStorage,
|
||||
} from '../login';
|
||||
|
||||
declare module '@blocksuite/store' {
|
||||
interface PageMeta {
|
||||
isPublic?: boolean;
|
||||
}
|
||||
}
|
||||
|
||||
// @ts-expect-error
|
||||
globalThis.WebSocket = WebSocket;
|
||||
|
||||
let workspaceApis: ReturnType<typeof createWorkspaceApis>;
|
||||
let userApis: ReturnType<typeof createUserApis>;
|
||||
let affineAuth: ReturnType<typeof createAffineAuth>;
|
||||
let statusApis: ReturnType<typeof createStatusApis>;
|
||||
|
||||
async function initPage(page: Page) {
|
||||
await page.waitForLoaded();
|
||||
// Add page block and surface block at root level
|
||||
const pageBlockId = page.addBlock('affine:page', {
|
||||
title: new page.Text(''),
|
||||
});
|
||||
page.addBlock('affine:surface', {}, pageBlockId);
|
||||
const frameId = page.addBlock('affine:note', {}, pageBlockId);
|
||||
page.addBlock('affine:paragraph', {}, frameId);
|
||||
page.resetHistory();
|
||||
return {
|
||||
pageBlockId,
|
||||
frameId,
|
||||
} as const;
|
||||
}
|
||||
|
||||
const mockUser = {
|
||||
name: faker.name.fullName(),
|
||||
email: faker.internet.email(),
|
||||
password: faker.internet.password(),
|
||||
};
|
||||
|
||||
async function waitForConnected(provider: KeckProvider) {
|
||||
return new Promise<void>(resolve => {
|
||||
provider.once('status', ({ status }: any) => {
|
||||
expect(status).toBe('connected');
|
||||
resolve();
|
||||
});
|
||||
});
|
||||
}
|
||||
|
||||
beforeEach(() => {
|
||||
// create a new user for each test, so that each test can be run independently
|
||||
mockUser.name = faker.name.fullName();
|
||||
mockUser.email = faker.internet.email();
|
||||
mockUser.password = faker.internet.password();
|
||||
});
|
||||
|
||||
beforeEach(() => {
|
||||
affineAuth = createAffineAuth('http://127.0.0.1:3000/');
|
||||
userApis = createUserApis('http://127.0.0.1:3000/');
|
||||
workspaceApis = createWorkspaceApis('http://127.0.0.1:3000/');
|
||||
statusApis = createStatusApis('http://127.0.0.1:3000/');
|
||||
});
|
||||
|
||||
beforeEach(async () => {
|
||||
expect(await statusApis.healthz(), 'health check').toBe(true);
|
||||
});
|
||||
|
||||
beforeEach(async () => {
|
||||
const data = await fetch('http://127.0.0.1:3000/api/user/token', {
|
||||
method: 'POST',
|
||||
headers: {
|
||||
'Content-Type': 'application/json',
|
||||
},
|
||||
body: JSON.stringify({
|
||||
type: 'DebugCreateUser',
|
||||
...mockUser,
|
||||
}),
|
||||
}).then(r => r.json());
|
||||
setLoginStorage(data);
|
||||
loginResponseSchema.parse(data);
|
||||
});
|
||||
|
||||
declare global {
|
||||
interface DocumentEventMap {
|
||||
'affine-error': CustomEvent<{
|
||||
code: (typeof MessageCode)[keyof typeof MessageCode];
|
||||
}>;
|
||||
}
|
||||
}
|
||||
|
||||
async function createWorkspace(
|
||||
workspaceApi: typeof workspaceApis,
|
||||
callback?: (workspace: Workspace) => Promise<void>
|
||||
): Promise<string> {
|
||||
const workspace = createEmptyBlockSuiteWorkspace(
|
||||
faker.datatype.uuid(),
|
||||
WorkspaceFlavour.LOCAL
|
||||
);
|
||||
if (callback) {
|
||||
await callback(workspace);
|
||||
}
|
||||
const binary = Workspace.Y.encodeStateAsUpdate(workspace.doc);
|
||||
const data = await workspaceApi.createWorkspace(binary);
|
||||
createWorkspaceResponseSchema.parse(data);
|
||||
return data.id;
|
||||
}
|
||||
|
||||
describe('api', () => {
|
||||
test('built-in mock user', async () => {
|
||||
const data = await fetch('http://127.0.0.1:3000/api/user/token', {
|
||||
method: 'POST',
|
||||
headers: {
|
||||
'Content-Type': 'application/json',
|
||||
},
|
||||
body: JSON.stringify({
|
||||
type: 'DebugLoginUser',
|
||||
email: user1.email,
|
||||
password: user1.password,
|
||||
}),
|
||||
}).then(r => r.json());
|
||||
loginResponseSchema.parse(data);
|
||||
const data2 = await fetch('http://127.0.0.1:3000/api/user/token', {
|
||||
method: 'POST',
|
||||
headers: {
|
||||
'Content-Type': 'application/json',
|
||||
},
|
||||
body: JSON.stringify({
|
||||
type: 'DebugLoginUser',
|
||||
email: user2.email,
|
||||
password: user2.password,
|
||||
}),
|
||||
}).then(r => r.json());
|
||||
loginResponseSchema.parse(data2);
|
||||
});
|
||||
|
||||
test('failed', async () => {
|
||||
workspaceApis = createWorkspaceApis('http://127.0.0.1:10086/404/');
|
||||
const listener = vi.fn(
|
||||
(
|
||||
e: CustomEvent<{
|
||||
code: (typeof MessageCode)[keyof typeof MessageCode];
|
||||
}>
|
||||
) => {
|
||||
expect(e.detail.code).toBe(MessageCode.loadListFailed);
|
||||
}
|
||||
);
|
||||
|
||||
document.addEventListener('affine-error', listener);
|
||||
expect(listener).toBeCalledTimes(0);
|
||||
await workspaceApis.getWorkspaces().catch(e => {
|
||||
expect(e).toBeInstanceOf(RequestError);
|
||||
});
|
||||
expect(listener).toBeCalledTimes(1);
|
||||
document.removeEventListener('affine-error', listener);
|
||||
});
|
||||
|
||||
test('no permission', async () => {
|
||||
await workspaceApis.downloadWorkspace('not-exist').catch(e => {
|
||||
expect(e).toBeInstanceOf(RequestError);
|
||||
expect(e.code).toBe(MessageCode.noPermission);
|
||||
});
|
||||
});
|
||||
|
||||
test('blob too large', async () => {
|
||||
await workspaceApis
|
||||
.uploadBlob('test', new ArrayBuffer(1024 * 1024 * 1024 + 1), 'image/png')
|
||||
.catch(e => {
|
||||
expect(e).toBeInstanceOf(RequestError);
|
||||
expect(e.code).toBe(MessageCode.blobTooLarge);
|
||||
});
|
||||
});
|
||||
|
||||
test('refresh token', async () => {
|
||||
const storage = getLoginStorage();
|
||||
assertExists(storage);
|
||||
loginResponseSchema.parse(await affineAuth.refreshToken(storage));
|
||||
});
|
||||
|
||||
test(
|
||||
'create workspace',
|
||||
async () => {
|
||||
const id = await createWorkspace(workspaceApis);
|
||||
expect(id).toBeTypeOf('string');
|
||||
},
|
||||
{
|
||||
timeout: 30000,
|
||||
}
|
||||
);
|
||||
|
||||
test(
|
||||
'delete workspace',
|
||||
async () => {
|
||||
const id = await createWorkspace(workspaceApis);
|
||||
const response = await workspaceApis.deleteWorkspace({
|
||||
id,
|
||||
});
|
||||
expect(response).toBe(true);
|
||||
},
|
||||
{
|
||||
timeout: 30000,
|
||||
}
|
||||
);
|
||||
|
||||
test('get workspaces', async () => {
|
||||
const id = await createWorkspace(workspaceApis);
|
||||
const response = await workspaceApis.getWorkspaces();
|
||||
expect(response).toBeInstanceOf(Array);
|
||||
expect(response.length).toBe(1);
|
||||
expect(response[0].id).toBe(id);
|
||||
});
|
||||
|
||||
test(
|
||||
'blob',
|
||||
async () => {
|
||||
const workspace = new Workspace({
|
||||
id: 'test',
|
||||
});
|
||||
const path = require.resolve('@affine-test/fixtures/smile.png');
|
||||
const imageBuffer = await readFile(path);
|
||||
const binary = Workspace.Y.encodeStateAsUpdate(workspace.doc);
|
||||
const data = await workspaceApis.createWorkspace(binary);
|
||||
createWorkspaceResponseSchema.parse(data);
|
||||
const workspaceId = data.id;
|
||||
const blobId = await workspaceApis.uploadBlob(
|
||||
workspaceId,
|
||||
imageBuffer,
|
||||
'image/png'
|
||||
);
|
||||
expect(blobId).toBeTypeOf('string');
|
||||
const arrayBuffer = await workspaceApis.getBlob(workspaceId, blobId);
|
||||
expect(arrayBuffer).toBeInstanceOf(ArrayBuffer);
|
||||
expect(arrayBuffer.byteLength).toEqual(imageBuffer.byteLength);
|
||||
expect(Buffer.from(arrayBuffer)).toEqual(imageBuffer);
|
||||
},
|
||||
{
|
||||
timeout: 30000,
|
||||
}
|
||||
);
|
||||
|
||||
test(
|
||||
'workspace binary',
|
||||
async () => {
|
||||
const id = await createWorkspace(workspaceApis);
|
||||
await workspaceApis.updateWorkspace({
|
||||
id,
|
||||
public: true,
|
||||
});
|
||||
const binary = await workspaceApis.downloadWorkspace(id, false);
|
||||
const publicBinary = await workspaceApis.downloadWorkspace(id, true);
|
||||
expect(binary).toBeInstanceOf(ArrayBuffer);
|
||||
expect(publicBinary).toBeInstanceOf(ArrayBuffer);
|
||||
expect(binary).toEqual(publicBinary);
|
||||
expect(binary.byteLength).toEqual(publicBinary.byteLength);
|
||||
},
|
||||
{
|
||||
timeout: 30000,
|
||||
}
|
||||
);
|
||||
|
||||
test.fails('workspace page binary', async () => {
|
||||
const id = await createWorkspace(workspaceApis, async workspace => {
|
||||
{
|
||||
const page = workspace.createPage('page0');
|
||||
const { frameId } = await initPage(page);
|
||||
page.addBlock(
|
||||
'affine:paragraph',
|
||||
{
|
||||
text: new page.Text('This is page0'),
|
||||
},
|
||||
frameId
|
||||
);
|
||||
}
|
||||
{
|
||||
const page = workspace.createPage('page1');
|
||||
const { frameId } = await initPage(page);
|
||||
page.addBlock(
|
||||
'affine:paragraph',
|
||||
{
|
||||
text: new page.Text('This is page1'),
|
||||
},
|
||||
frameId
|
||||
);
|
||||
}
|
||||
});
|
||||
await workspaceApis.updateWorkspace({
|
||||
id,
|
||||
public: true,
|
||||
});
|
||||
let originalPage0Json: any = null;
|
||||
let originalPage1Json: any = null;
|
||||
{
|
||||
const binary = await workspaceApis.downloadWorkspace(id, false);
|
||||
const workspace = new Workspace({
|
||||
id: faker.datatype.uuid(),
|
||||
})
|
||||
.register(AffineSchemas)
|
||||
.register(__unstableSchemas);
|
||||
applyUpdate(workspace.doc, new Uint8Array(binary));
|
||||
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({
|
||||
id: faker.datatype.uuid(),
|
||||
})
|
||||
.register(AffineSchemas)
|
||||
.register(__unstableSchemas);
|
||||
const binary = await workspaceApis.downloadPublicWorkspacePage(
|
||||
id,
|
||||
'page0'
|
||||
);
|
||||
applyUpdate(workspace.doc, new Uint8Array(binary));
|
||||
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({
|
||||
id: faker.datatype.uuid(),
|
||||
})
|
||||
.register(AffineSchemas)
|
||||
.register(__unstableSchemas);
|
||||
const binary = await workspaceApis.downloadPublicWorkspacePage(
|
||||
id,
|
||||
'page1'
|
||||
);
|
||||
applyUpdate(workspace.doc, new Uint8Array(binary));
|
||||
const page1 = workspace.getPage('page1') as Page;
|
||||
expect(workspace.getPage('page0')).toBeNull();
|
||||
expect(page1).not.toBeNull();
|
||||
expect(workspace.doc.share.get(page1.prefixedId)?.toJSON()).toEqual(
|
||||
originalPage1Json
|
||||
);
|
||||
}
|
||||
});
|
||||
|
||||
test(
|
||||
'usage',
|
||||
async () => {
|
||||
const usageResponse = await userApis.getUsage();
|
||||
usageResponseSchema.parse(usageResponse);
|
||||
const id = await createWorkspace(workspaceApis);
|
||||
const path = require.resolve('@affine-test/fixtures/smile.png');
|
||||
const imageBuffer = await readFile(path);
|
||||
const blobId = await workspaceApis.uploadBlob(
|
||||
id,
|
||||
imageBuffer,
|
||||
'image/png'
|
||||
);
|
||||
const buffer = await workspaceApis.getBlob(id, blobId);
|
||||
expect(buffer.byteLength).toEqual(imageBuffer.byteLength);
|
||||
const newUsageResponse = await userApis.getUsage();
|
||||
usageResponseSchema.parse(newUsageResponse);
|
||||
expect(usageResponse.blob_usage.usage).not.equals(
|
||||
newUsageResponse.blob_usage.usage
|
||||
);
|
||||
expect(newUsageResponse.blob_usage.usage).equals(96);
|
||||
},
|
||||
{
|
||||
timeout: 30000,
|
||||
}
|
||||
);
|
||||
|
||||
test.fails(
|
||||
'public page',
|
||||
async () => {
|
||||
const id = await createWorkspace(workspaceApis, async workspace => {
|
||||
const page = workspace.createPage('page0');
|
||||
const { frameId } = await initPage(page);
|
||||
page.addBlock(
|
||||
'affine:paragraph',
|
||||
{
|
||||
text: new page.Text('This is page0'),
|
||||
},
|
||||
frameId
|
||||
);
|
||||
});
|
||||
const binary = await workspaceApis.downloadWorkspace(id, false);
|
||||
const workspace = createEmptyBlockSuiteWorkspace(
|
||||
id,
|
||||
WorkspaceFlavour.LOCAL
|
||||
);
|
||||
Workspace.Y.applyUpdate(workspace.doc, new Uint8Array(binary));
|
||||
const workspace2 = createEmptyBlockSuiteWorkspace(
|
||||
id,
|
||||
WorkspaceFlavour.LOCAL
|
||||
);
|
||||
{
|
||||
const wsUrl = `ws://127.0.0.1:3000/api/sync/`;
|
||||
const provider = new KeckProvider(wsUrl, workspace.id, workspace.doc, {
|
||||
params: { token: getLoginStorage()?.token },
|
||||
awareness: workspace.awarenessStore.awareness,
|
||||
connect: false,
|
||||
});
|
||||
const provider2 = new KeckProvider(
|
||||
wsUrl,
|
||||
workspace2.id,
|
||||
workspace2.doc,
|
||||
{
|
||||
params: { token: getLoginStorage()?.token },
|
||||
awareness: workspace2.awarenessStore.awareness,
|
||||
connect: false,
|
||||
}
|
||||
);
|
||||
|
||||
provider.connect();
|
||||
provider2.connect();
|
||||
|
||||
await Promise.all([
|
||||
await waitForConnected(provider),
|
||||
await waitForConnected(provider2),
|
||||
]);
|
||||
const pageId = 'page0';
|
||||
const page = workspace.getPage(pageId) as Page;
|
||||
expect(page).not.toBeNull();
|
||||
expect(page).not.toBeUndefined();
|
||||
workspace.setPageMeta(pageId, {
|
||||
isPublic: true,
|
||||
});
|
||||
|
||||
await new Promise(resolve => setTimeout(resolve, 1000));
|
||||
|
||||
const page2 = workspace2.getPage(pageId) as Page;
|
||||
expect(page2).not.toBeNull();
|
||||
const meta = workspace2.meta.getPageMeta(pageId) as PageMeta;
|
||||
expect(meta.isPublic).toBe(true);
|
||||
|
||||
const binary = await workspaceApis.downloadPublicWorkspacePage(
|
||||
id,
|
||||
pageId
|
||||
);
|
||||
const publicWorkspace = createEmptyBlockSuiteWorkspace(
|
||||
id,
|
||||
WorkspaceFlavour.LOCAL
|
||||
);
|
||||
Workspace.Y.applyUpdate(publicWorkspace.doc, new Uint8Array(binary));
|
||||
const publicPage = publicWorkspace.getPage(pageId) as Page;
|
||||
expect(publicPage).not.toBeNull();
|
||||
}
|
||||
},
|
||||
{
|
||||
timeout: 30000,
|
||||
retry: 3,
|
||||
}
|
||||
);
|
||||
});
|
||||
@ -1,20 +0,0 @@
|
||||
import { getDefaultStore } from 'jotai';
|
||||
import { describe, expect, test } from 'vitest';
|
||||
|
||||
import { currentAffineUserAtom } from '../atom';
|
||||
|
||||
describe('atom', () => {
|
||||
test('currentAffineUserAtom', () => {
|
||||
const store = getDefaultStore();
|
||||
const mock = {
|
||||
created_at: 0,
|
||||
exp: 0,
|
||||
email: '',
|
||||
id: '',
|
||||
name: '',
|
||||
avatar_url: '',
|
||||
};
|
||||
store.set(currentAffineUserAtom, mock);
|
||||
expect(store.get(currentAffineUserAtom)).toEqual(mock);
|
||||
});
|
||||
});
|
||||
@ -1,45 +0,0 @@
|
||||
/**
|
||||
* @vitest-environment happy-dom
|
||||
*/
|
||||
import { describe, expect, test } from 'vitest';
|
||||
|
||||
import type { AccessTokenMessage } from '../login';
|
||||
import {
|
||||
getLoginStorage,
|
||||
isExpired,
|
||||
setLoginStorage,
|
||||
STORAGE_KEY,
|
||||
} from '../login';
|
||||
|
||||
describe('storage', () => {
|
||||
test('should work', () => {
|
||||
setLoginStorage({
|
||||
token: '1',
|
||||
refresh: '2',
|
||||
});
|
||||
const data = localStorage.getItem(STORAGE_KEY);
|
||||
expect(data).toBe('{"token":"1","refresh":"2"}');
|
||||
const login = getLoginStorage();
|
||||
expect(login).toEqual({
|
||||
token: '1',
|
||||
refresh: '2',
|
||||
});
|
||||
});
|
||||
});
|
||||
|
||||
describe('utils', () => {
|
||||
test('isExpired', async () => {
|
||||
const now = Math.floor(Date.now() / 1000);
|
||||
expect(isExpired({ exp: now + 1 } as AccessTokenMessage, 0)).toBeFalsy();
|
||||
const promise = new Promise<void>(resolve => {
|
||||
setTimeout(() => {
|
||||
expect(
|
||||
isExpired({ exp: now + 1 } as AccessTokenMessage, 0)
|
||||
).toBeTruthy();
|
||||
resolve();
|
||||
}, 2000);
|
||||
});
|
||||
expect(isExpired({ exp: now - 1 } as AccessTokenMessage, 0)).toBeTruthy();
|
||||
await promise;
|
||||
});
|
||||
});
|
||||
@ -1,184 +0,0 @@
|
||||
/**
|
||||
* @vitest-environment happy-dom
|
||||
*/
|
||||
import 'fake-indexeddb/auto';
|
||||
|
||||
import { WorkspaceFlavour } from '@affine/env/workspace';
|
||||
import type { Workspace } from '@affine/env/workspace/legacy-cloud';
|
||||
import { PermissionType } from '@affine/env/workspace/legacy-cloud';
|
||||
import user1 from '@affine-test/fixtures/built-in-user1.json';
|
||||
import user2 from '@affine-test/fixtures/built-in-user2.json';
|
||||
import type { ParagraphBlockModel } from '@blocksuite/blocks/models';
|
||||
import type { Page, Text } from '@blocksuite/store';
|
||||
import { uuidv4, Workspace as BlockSuiteWorkspace } from '@blocksuite/store';
|
||||
import { beforeEach, describe, expect, test, vi } from 'vitest';
|
||||
import { WebSocket } from 'ws';
|
||||
|
||||
import type { LoginResponse } from '../../affine/login';
|
||||
import { loginResponseSchema } from '../../affine/login';
|
||||
import { createEmptyBlockSuiteWorkspace } from '../../utils';
|
||||
import { createWorkspaceApis } from '../api';
|
||||
import { KeckProvider } from '../keck';
|
||||
|
||||
declare module '@blocksuite/store' {
|
||||
interface PageMeta {
|
||||
foo: string;
|
||||
}
|
||||
}
|
||||
|
||||
// @ts-expect-error
|
||||
globalThis.WebSocket = WebSocket;
|
||||
|
||||
const currentTokenRef = {
|
||||
current: null as LoginResponse | null,
|
||||
};
|
||||
|
||||
vi.stubGlobal('localStorage', {
|
||||
getItem: () => JSON.stringify(currentTokenRef.current),
|
||||
setItem: () => null,
|
||||
});
|
||||
|
||||
let workspaceApis: ReturnType<typeof createWorkspaceApis>;
|
||||
|
||||
let user1Token: LoginResponse;
|
||||
let user2Token: LoginResponse;
|
||||
|
||||
beforeEach(() => {
|
||||
workspaceApis = createWorkspaceApis('http://127.0.0.1:3000/');
|
||||
});
|
||||
|
||||
beforeEach(async () => {
|
||||
const data = await fetch('http://127.0.0.1:3000/api/user/token', {
|
||||
method: 'POST',
|
||||
headers: {
|
||||
'Content-Type': 'application/json',
|
||||
},
|
||||
body: JSON.stringify({
|
||||
type: 'DebugLoginUser',
|
||||
email: user1.email,
|
||||
password: user1.password,
|
||||
}),
|
||||
}).then(r => r.json());
|
||||
loginResponseSchema.parse(data);
|
||||
user1Token = data;
|
||||
const data2 = await fetch('http://127.0.0.1:3000/api/user/token', {
|
||||
method: 'POST',
|
||||
headers: {
|
||||
'Content-Type': 'application/json',
|
||||
},
|
||||
body: JSON.stringify({
|
||||
type: 'DebugLoginUser',
|
||||
email: user2.email,
|
||||
password: user2.password,
|
||||
}),
|
||||
}).then(r => r.json());
|
||||
loginResponseSchema.parse(data2);
|
||||
user2Token = data2;
|
||||
});
|
||||
|
||||
const wsUrl = `ws://127.0.0.1:3000/api/sync/`;
|
||||
|
||||
describe('ydoc sync', () => {
|
||||
test(
|
||||
'page',
|
||||
async () => {
|
||||
currentTokenRef.current = user1Token;
|
||||
const list = await workspaceApis.getWorkspaces();
|
||||
const root = list.find(
|
||||
workspace => workspace.permission === PermissionType.Owner
|
||||
) as Workspace;
|
||||
expect(root).toBeDefined();
|
||||
const binary = await workspaceApis.downloadWorkspace(root.id);
|
||||
const workspace1 = createEmptyBlockSuiteWorkspace(
|
||||
root.id,
|
||||
WorkspaceFlavour.AFFINE,
|
||||
{
|
||||
workspaceApis,
|
||||
}
|
||||
);
|
||||
const workspace2 = createEmptyBlockSuiteWorkspace(
|
||||
root.id,
|
||||
WorkspaceFlavour.AFFINE,
|
||||
{
|
||||
workspaceApis,
|
||||
}
|
||||
);
|
||||
BlockSuiteWorkspace.Y.applyUpdate(workspace1.doc, new Uint8Array(binary));
|
||||
BlockSuiteWorkspace.Y.applyUpdate(workspace2.doc, new Uint8Array(binary));
|
||||
const provider1 = new KeckProvider(wsUrl, workspace1.id, workspace1.doc, {
|
||||
params: { token: user1Token.token },
|
||||
awareness: workspace1.awarenessStore.awareness,
|
||||
// @ts-expect-error
|
||||
disableBc: true,
|
||||
connect: false,
|
||||
});
|
||||
|
||||
const provider2 = new KeckProvider(wsUrl, workspace2.id, workspace2.doc, {
|
||||
params: { token: user2Token.token },
|
||||
awareness: workspace2.awarenessStore.awareness,
|
||||
// @ts-expect-error
|
||||
disableBc: true,
|
||||
connect: false,
|
||||
});
|
||||
|
||||
provider1.connect();
|
||||
provider2.connect();
|
||||
|
||||
function waitForConnected(provider: KeckProvider) {
|
||||
return new Promise<void>(resolve => {
|
||||
provider.once('status', ({ status }: any) => {
|
||||
expect(status).toBe('connected');
|
||||
resolve();
|
||||
});
|
||||
});
|
||||
}
|
||||
|
||||
await Promise.all([
|
||||
waitForConnected(provider1),
|
||||
waitForConnected(provider2),
|
||||
]);
|
||||
|
||||
const pageId = uuidv4();
|
||||
const page1 = workspace1.createPage({ id: pageId });
|
||||
await page1.waitForLoaded();
|
||||
const pageBlockId = page1.addBlock('affine:page', {
|
||||
title: new page1.Text(''),
|
||||
});
|
||||
page1.addBlock('affine:surface', {}, pageBlockId);
|
||||
const frameId = page1.addBlock('affine:note', {}, pageBlockId);
|
||||
const paragraphId = page1.addBlock('affine:paragraph', {}, frameId);
|
||||
await new Promise(resolve => setTimeout(resolve, 1000));
|
||||
expect(workspace2.getPage(pageId)).toBeDefined();
|
||||
expect(workspace2.doc.getMap(`space:${pageId}`).toJSON()).toEqual(
|
||||
workspace1.doc.getMap(`space:${pageId}`).toJSON()
|
||||
);
|
||||
const page2 = workspace2.getPage(pageId) as Page;
|
||||
await page2.waitForLoaded();
|
||||
page1.updateBlock(
|
||||
page1.getBlockById(paragraphId) as ParagraphBlockModel,
|
||||
{
|
||||
text: new page1.Text('hello world'),
|
||||
}
|
||||
);
|
||||
workspace1.meta.setPageMeta(pageId, {
|
||||
foo: 'bar',
|
||||
});
|
||||
await new Promise(resolve => setTimeout(resolve, 1000));
|
||||
const pageMeta = workspace2.meta.getPageMeta(pageId);
|
||||
expect(pageMeta).toBeDefined();
|
||||
expect(pageMeta?.foo).toBe('bar');
|
||||
const paragraph2 = page2.getBlockById(paragraphId) as ParagraphBlockModel;
|
||||
const text = paragraph2.text as Text;
|
||||
expect(text.toString()).toEqual(
|
||||
page1.getBlockById(paragraphId)?.text?.toString()
|
||||
);
|
||||
|
||||
provider1.disconnect();
|
||||
provider2.disconnect();
|
||||
},
|
||||
{
|
||||
timeout: 30000,
|
||||
retry: 3,
|
||||
}
|
||||
);
|
||||
});
|
||||
Loading…
Reference in New Issue
Block a user