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

import '../../../src/behavior'
import Behavior from '../../../src/behavior/behavior'
import { IBehavior } from '../../../types';

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('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')
    const firstInstance = new FirstInstance()
    const secondBehavior = new SecondBehavior()
    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();
    expect(config1.style).toBe(undefined);

    const events1 = firstInstance.getEvents()
    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');
    const drag = secondBehavior.onDrag()
    expect(drag).toEqual('drag')
  })
})

describe('Default Behavior', () => {
  it('drag-canvas', () => {
    const DragCanvas = Behavior.getBehavior('drag-canvas')
    expect(DragCanvas).not.toBe(undefined)

    const dragCanvas: IBehavior = new DragCanvas()
    const config = dragCanvas.getDefaultCfg()
    expect(config).toEqual({ direction: 'both' })

    const events = dragCanvas.getEvents()
    const keys = Object.keys(events)
    expect(keys.length).toBe(7)
  })
})