WIP input-association

bpmn-io · Nov 27, 2018 · 205003d · 205003d
1 parent f2d4870
commit 205003d
Show file tree

Hide file tree

Showing 2 changed files with 17 additions and 27 deletions.
diff --git a/lib/features/modeling/behavior/DataInputAssociationBehavior.js b/lib/features/modeling/behavior/DataInputAssociationBehavior.js
@@ -57,7 +57,7 @@ export default function DataInputAssociationBehavior(eventBus, bpmnFactory) {
   function createDataInput(element) {
     var ioSpecification = element.get('ioSpecification');
 
-    var inputSet;
+    var inputSet, outputSet;
 
     if (!ioSpecification) {
       ioSpecification = bpmnFactory.create('bpmn:InputOutputSpecification', {
@@ -75,6 +75,15 @@ export default function DataInputAssociationBehavior(eventBus, bpmnFactory) {
       inputSet.$parent = ioSpecification;
 
       collectionAdd(ioSpecification.get('inputSets'), inputSet);
+
+      outputSet = bpmnFactory.create('bpmn:OutputSet', {
+        dataOutputRefs: [],
+        name: 'Outputs'
+      });
+
+      outputSet.$parent = ioSpecification;
+
+      collectionAdd(ioSpecification.get('outputSets'), outputSet);
     }
 
     var dataInput = bpmnFactory.create('bpmn:DataInput');
@@ -244,34 +253,10 @@ function cleanUpIoSpecification(element) {
       delete ioSpecification.dataOutputs;
     }
 
-    if (inputSets && inputSets.length) {
-      forEach(inputSets, function(inputSet) {
-        if (!inputSet.dataInputRefs.length) {
-          collectionRemove(inputSets, inputSet);
-        }
-      });
-
-      if (!inputSets.length) {
-        delete ioSpecification.inputSets;
-      }
-    }
-
-    if (outputSets && outputSets.length) {
-      forEach(outputSets, function(outputSet) {
-        if (!outputSet.dataOutputRefs.length) {
-          collectionRemove(outputSets, outputSet);
-        }
-      });
-
-      if (!outputSets.length) {
-        delete ioSpecification.outputSets;
-      }
-    }
-
     if ((!dataInputs || !dataInputs.length) &&
         (!dataOutputs || !dataOutputs.length) &&
-        (!inputSets || !inputSets.length) &&
-        (!outputSets || !outputSets.length)) {
+        !inputSets[0].dataInputRefs.length &&
+        !outputSets[0].dataOutputRefs.length) {
 
       delete element.ioSpecification;
     }

diff --git a/test/spec/features/modeling/behavior/DataInputAssociationBehaviorSpec.js b/test/spec/features/modeling/behavior/DataInputAssociationBehaviorSpec.js
@@ -34,6 +34,11 @@ describe('modeling/behavior - DataInputAssociationBehavior', function() {
     var dataInputAssociationBo = dataInputAssociation.businessObject;
 
     expect(taskBo.ioSpecification).to.exist;
+    expect(taskBo.ioSpecification.inputSets).to.exist;
+    expect(taskBo.ioSpecification.inputSets).to.have.length(1);
+    expect(taskBo.ioSpecification.outputSets).to.exist;
+    expect(taskBo.ioSpecification.outputSets).to.have.length(1);
+
     expect(dataInputAssociationBo.targetRef).to.exist;
     expect(dataInputAssociationBo.targetRef).to.eql(getDataInput(taskBo, dataInputAssociationBo.targetRef));
   }));