g6/tests/unit/behavior/index-spec.ts

import Behavior from '../../../src/behavior/behavior'

describe('Behavior', () => {
  it('register signle behavior', () => {
    Behavior.registerBehavior('first-behavior', {
      getEvents() {
        return {
          click: 'onClick'
        }
      },
      onClick() {
        console.log('trigger click')
      },
      shouldBegin() {
        return false
      }
    });

    expect(Behavior.hasBehavior('first-behavior')).toBe(true)
    expect(Behavior.hasBehavior('test')).toBe(false)
    const BehaviorInstance = Behavior.getBehavior('first-behavior')
    const instance = new BehaviorInstance()
    const events = instance.getEvents()
    expect(Object.keys(events)).toEqual(['click'])
    expect(instance.shouldBegin()).toEqual(false);
  })

  it.only('register multiple behavior', () => {
    Behavior.registerBehavior('first', {
      getEvents() {
        return {
          click: 'onClick',
          "edge:click": 'onEdgeClick',
          contextmenu: 'onContextMenu'
        }
      },
      onClick() {
        return 'onclick'
      },
      onContextMenu() {
        return 'onContextMenu'
      }
    });

    Behavior.registerBehavior('second', {
      getDefaultCfg() {
        return {
          style: {
            fill: 'red'
          }
        }
      },
      getEvents() {
        return {
          drag: 'onDrag'
        }
      },
      onDrag() {
        return 'drag'
      }
    })

    const firstInstance = Behavior.getBehavior('first')
    const secondBehavior = Behavior.getBehavior('second')
    expect(firstInstance).not.toBe(undefined);
    expect(secondBehavior).not.toBe(undefined);
    expect(Behavior.getBehavior('three')).toBe(undefined);
    expect(Behavior.hasBehavior('first')).toBe(true);
    expect(Behavior.hasBehavior('three')).toBe(false);

    const config1 = firstInstance.getDefaultCfg();
    console.log('config1.style', config1)
    expect(config1.style).toBe(undefined);

    const events1 = firstInstance.getEvents()
    console.log(events1, Object.keys(events1), events1)
    expect(Object.keys(events1).length).toEqual(3);
    expect(Object.keys(events1)).toEqual(['click', 'edge:click', 'contextmenu'])

    const config = secondBehavior.getDefaultCfg();
    expect(config.style.fill).toEqual('red');
    expect(config.style.fill).not.toEqual('blue');
    console.log('config.style', config.style)

    const drag = secondBehavior.onDrag()
    expect(drag).toEqual('drag')
  })
})