AFFiNE-Mirror/packages/data-center/src/workspace-unit-collection.spec.ts

import { describe, expect, test } from 'vitest';

import { WorkspaceUnit } from './workspace-unit';
import type { WorkspaceUnitCollectionChangeEvent } from './workspace-unit-collection';
import { WorkspaceUnitCollection } from './workspace-unit-collection';

describe('workspace meta collection observable', () => {
  const workspaceUnitCollection = new WorkspaceUnitCollection();

  const scope = workspaceUnitCollection.createScope();

  test('add workspace', () => {
    workspaceUnitCollection.once(
      'change',
      (event: WorkspaceUnitCollectionChangeEvent) => {
        expect(event.added?.[0]?.id).toEqual('123');
      }
    );
    const workspaceUnit = new WorkspaceUnit({
      id: '123',
      name: 'test',
      avatar: undefined,
      memberCount: 1,
      provider: '',
      syncMode: 'core',
    });
    scope.add(workspaceUnit);
  });

  test('list workspace', () => {
    const list = scope.list();
    expect(list.length).toEqual(1);
    expect(list[0].id).toEqual('123');
  });

  test('get workspace', () => {
    expect(scope.get('123')?.id).toEqual('123');
  });

  test('update workspace', () => {
    workspaceUnitCollection.once(
      'change',
      (event: WorkspaceUnitCollectionChangeEvent) => {
        expect(event.updated?.name).toEqual('demo');
      }
    );
    scope.update('123', { name: 'demo' });
  });

  test('get workspace form other scope', () => {
    const scope1 = workspaceUnitCollection.createScope();
    expect(scope1.get('123')).toBeFalsy();
  });

  test('delete workspace', () => {
    workspaceUnitCollection.once(
      'change',
      (event: WorkspaceUnitCollectionChangeEvent) => {
        expect(event.deleted?.[0]?.id).toEqual('123');
      }
    );
    scope.remove('123');
  });
});