diff --git a/lib/import/Importer.js b/lib/import/Importer.js index 6c3de08c..9b2de31d 100644 --- a/lib/import/Importer.js +++ b/lib/import/Importer.js @@ -12,7 +12,8 @@ function hasLabel(element) { element.$instanceOf('bpmn:Gateway') || element.$instanceOf('bpmn:DataStoreReference') || element.$instanceOf('bpmn:DataObjectReference') || - element.$instanceOf('bpmn:SequenceFlow'); + element.$instanceOf('bpmn:SequenceFlow') || + element.$instanceOf('bpmn:MessageFlow'); } diff --git a/test/spec/browser/import/elements/LabelImportSpec.js b/test/spec/browser/import/elements/LabelImportSpec.js index 2c68cfaa..2a07ee26 100644 --- a/test/spec/browser/import/elements/LabelImportSpec.js +++ b/test/spec/browser/import/elements/LabelImportSpec.js @@ -44,6 +44,18 @@ describe('import - labels', function() { }); }); + + it('on message flows', function(done) { + + var xml = fs.readFileSync('test/fixtures/bpmn/import/labels/collaboration-message-flows.bpmn', 'utf8'); + + var renderer = new Viewer(container); + + renderer.importXML(xml, function(err) { + done(err); + }); + }); + });