tmagic-editor/packages/data-source/tests/DataSource.spec.ts

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

import App from '@tmagic/core';

import { DataSource } from '@data-source/index';
import { DeepObservedData } from '@data-source/observed-data';

describe('DataSource', () => {
  test('instance', () => {
    const ds = new DataSource({
      schema: {
        type: 'base',
        id: '1',
        fields: [{ name: 'name' }],
        methods: [],
        events: [],
      },
      app: new App({}),
    });

    expect(ds).toBeInstanceOf(DataSource);
    expect(ds.data).toHaveProperty('name');
  });

  test('init', () => {
    const ds = new DataSource({
      schema: {
        type: 'base',
        id: '1',
        fields: [{ name: 'name' }],
        methods: [],
        events: [],
      },
      app: new App({}),
    });

    ds.init();

    expect(ds.isInit).toBeTruthy();
  });
});

describe('DataSource setData', () => {
  test('setData', () => {
    const ds = new DataSource({
      schema: {
        type: 'base',
        id: '1',
        fields: [{ name: 'name', defaultValue: 'name' }],
        methods: [],
        events: [],
      },
      app: new App({}),
    });

    ds.init();

    expect(ds.data.name).toBe('name');

    ds.setData({ name: 'name2' });

    expect(ds.data.name).toBe('name2');

    ds.setData('name3', 'name');

    expect(ds.data.name).toBe('name3');
  });

  test('setDataByPath', () => {
    const ds = new DataSource({
      schema: {
        type: 'base',
        id: '1',
        fields: [
          { name: 'name', defaultValue: 'name' },
          {
            name: 'obj',
            type: 'object',
            fields: [{ name: 'a' }, { name: 'b', type: 'array', fields: [{ name: 'c' }] }],
          },
        ],
        methods: [],
        events: [],
      },
      app: new App({}),
    });

    ds.init();

    expect(ds.data.name).toBe('name');
    expect(ds.data.obj.b).toHaveLength(0);

    ds.setData({
      name: 'name',
      obj: {
        a: 'a',
        b: [
          {
            c: 'c',
          },
        ],
      },
    });

    expect(ds.data.obj.b).toHaveLength(1);
    expect(ds.data.obj.b[0].c).toBe('c');

    ds.setData('c1', 'obj.b.0.c');
    expect(ds.data.obj.b[0].c).toBe('c1');
    expect(ds.data.obj.a).toBe('a');
    ds.setData('a1', 'obj.a');
    expect(ds.data.obj.a).toBe('a1');
  });
});

describe('DataSource lifecycle / mock', () => {
  test('编辑器中使用 mock 数据', () => {
    const app = new App({}) as any;
    app.platform = 'editor';
    const ds = new DataSource({
      app,
      schema: {
        type: 'base',
        id: '1',
        fields: [{ name: 'name' }],
        methods: [],
        events: [],
        mocks: [{ useInEditor: true, data: { name: 'mock' }, enable: true }],
      } as any,
    });
    expect(ds.data.name).toBe('mock');
  });

  test('useMock=true 在运行时使用 mock', () => {
    const ds = new DataSource({
      app: new App({}),
      useMock: true,
      schema: {
        type: 'base',
        id: '1',
        fields: [{ name: 'name' }],
        methods: [],
        events: [],
        mocks: [{ enable: true, data: { name: 'enabled-mock' } }],
      } as any,
    });
    expect(ds.data.name).toBe('enabled-mock');
  });

  test('initialData 优先时设置 isInit', () => {
    const ds = new DataSource({
      app: new App({}),
      initialData: { name: 'preset' },
      schema: {
        type: 'base',
        id: '1',
        fields: [{ name: 'name' }],
        methods: [],
        events: [],
      },
    });
    expect(ds.isInit).toBe(true);
    expect(ds.data.name).toBe('preset');
  });

  test('支持自定义 ObservedDataClass', () => {
    const ds = new DataSource({
      app: new App({}),
      ObservedDataClass: DeepObservedData,
      schema: {
        type: 'base',
        id: '1',
        fields: [{ name: 'name' }],
        methods: [],
        events: [],
      },
    });
    const cb = vi.fn();
    ds.onDataChange('name', cb);
    ds.setData('next', 'name');
    expect(cb).toHaveBeenCalled();
    ds.offDataChange('name', cb);
  });

  test('setValue 等价于按 path 的 setData 并发出 change', () => {
    const ds = new DataSource({
      app: new App({}),
      schema: {
        type: 'base',
        id: '1',
        fields: [{ name: 'name' }],
        methods: [],
        events: [],
      },
    });
    const change = vi.fn();
    ds.on('change', change);
    ds.setValue('name', 'V');
    expect(ds.data.name).toBe('V');
    expect(change).toHaveBeenCalledWith({ updateData: 'V', path: 'name' });
  });

  test('setFields / setMethods / DATA_SOURCE_SET_DATA_METHOD_NAME 自动注入', () => {
    const ds = new DataSource({
      app: new App({}),
      schema: {
        type: 'base',
        id: '1',
        fields: [{ name: 'name' }],
        methods: [],
        events: [],
      },
    });
    ds.setFields([{ name: 'foo' }] as any);
    expect(ds.fields[0].name).toBe('foo');
    ds.setMethods([{ name: 'doIt' } as any]);
    expect(ds.methods[0].name).toBe('doIt');

    (ds as any).setDataFromEvent({ params: { field: ['name'], data: 'X' } });
    expect(ds.data.name).toBe('X');
  });

  test('destroy 清理 fields 与监听', () => {
    const ds = new DataSource({
      app: new App({}),
      schema: {
        type: 'base',
        id: '1',
        fields: [{ name: 'name' }],
        methods: [],
        events: [],
      },
    });
    const handler = vi.fn();
    ds.on('change', handler);
    ds.destroy();
    expect(ds.fields).toHaveLength(0);
    ds.emit('change', {});
    expect(handler).not.toHaveBeenCalled();
  });
});