bpmn-js/test/spec/features/modeling/DeleteShapeSpec.js

import {
  bootstrapModeler,
  inject
} from 'test/TestHelper';

import modelingModule from 'lib/features/modeling';
import coreModule from 'lib/core';


var testModules = [ coreModule, modelingModule ];


describe('features/modeling - #removeShape', function() {

  var diagramXML = require('../../../fixtures/bpmn/sequence-flows.bpmn');

  beforeEach(bootstrapModeler(diagramXML, { modules: testModules }));

  describe('shape handling', function() {

    it('should execute', inject(function(elementRegistry, modeling) {

      // given
      var taskShape = elementRegistry.get('Task_1'),
          task = taskShape.businessObject;

      // when
      modeling.removeShape(taskShape);

      // then
      expect(task.$parent).to.be.null;
    }));
  });


  describe('undo support', function() {

    it('should undo', inject(function(elementRegistry, modeling, commandStack) {

      // given
      var taskShape = elementRegistry.get('Task_1'),
          task = taskShape.businessObject,
          parent = task.$parent;

      // when
      modeling.removeShape(taskShape);
      commandStack.undo();

      // then
      expect(task.$parent).to.eql(parent);
    }));
  });


  describe('redo support', function() {

    it('redo', inject(function(elementRegistry, modeling, commandStack) {

      // given
      var taskShape = elementRegistry.get('Task_1'),
          task = taskShape.businessObject;

      // when
      modeling.removeShape(taskShape);
      commandStack.undo();
      commandStack.redo();

      // then
      expect(task.$parent).to.be.null;
    }));
  });

});


describe('features/modeling - #removeShape', function() {

  var diagramXML = require('./DeleteShape.cropping.bpmn');

  beforeEach(bootstrapModeler(diagramXML, { modules: testModules }));


  it('should crop waypoints on undo/redo', inject(
    function(elementRegistry, commandStack, modeling) {

      // given
      var taskShape = elementRegistry.get('Task_A'),
          sequenceFlowConnection = elementRegistry.get('SequenceFlow_1'),
          incomingFlow = taskShape.incoming[0],
          outgoingFlow = taskShape.outgoing[0],
          expectedStart = incomingFlow.waypoints[0],
          expectedEnd = outgoingFlow.waypoints[1];

      // when
      modeling.removeShape(taskShape);
      commandStack.undo();
      commandStack.redo();

      // then
      expect(sequenceFlowConnection).to.have.waypoints([ expectedStart, expectedEnd ]);
    }
  ));

});