lowcode-engine/packages/designer/tests/utils-ut/slot.test.ts

// @ts-nocheck
import { includeSlot, removeSlot } from '../../src/utils/slot';

const genGetExtraProp = (val: string) => () => {
  return {
    getAsString() {
      return val;
    },
  };
};

const remove = () => {};

const mockNode = {
  slots: [{
    getExtraProp: genGetExtraProp('haha'),
    remove,
  }, {
    getExtraProp: genGetExtraProp('heihei'),
    remove,
  }]
};

// 没有 slots
const mockNode2 = {};

it('includeSlot', () => {
  expect(includeSlot(mockNode, 'haha')).toBeTruthy();
  expect(includeSlot(mockNode, 'heihei')).toBeTruthy();
  expect(includeSlot(mockNode, 'xixi')).toBeFalsy();
  expect(includeSlot(mockNode2, 'xixi')).toBeFalsy();
});

it('removeSlot', () => {
  expect(removeSlot(mockNode, 'xixi')).toBeFalsy();
  expect(mockNode.slots).toHaveLength(2);
  expect(removeSlot(mockNode, 'haha')).toBeTruthy();
  expect(mockNode.slots).toHaveLength(1);
  expect(removeSlot(mockNode, 'heihei')).toBeTruthy();
  expect(mockNode.slots).toHaveLength(0);

  expect(removeSlot(mockNode2, 'xixi')).toBeFalsy();
});