diff --git a/src/diagrams/flowchart/parser/flow.spec.js b/src/diagrams/flowchart/parser/flow.spec.js index 66e12318e..f6a7517ea 100644 --- a/src/diagrams/flowchart/parser/flow.spec.js +++ b/src/diagrams/flowchart/parser/flow.spec.js @@ -1712,6 +1712,23 @@ describe('when parsing ', function () { expect(classes['exClass'].styles[0]).toBe('background:#bbb') expect(classes['exClass'].styles[1]).toBe('border:1px solid red') }) + it('should be possible to apply a class to a vertex directly : usecase A[text].class-->B[test2] ', function () { + let statement = '' + + statement = statement + 'graph TD;' + '\n' + statement = statement + 'classDef exClass background:#bbb,border:1px solid red;' + '\n' + statement = statement + 'A[test].exClass-->B[test2];' + '\n' + + const res = flow.parser.parse(statement) + const vertices = flow.parser.yy.getVertices() + const classes = flow.parser.yy.getClasses() + + expect(classes['exClass'].styles.length).toBe(2) + expect(vertices['A'].classes[0]).toBe('exClass') + expect(classes['exClass'].styles[0]).toBe('background:#bbb') + expect(classes['exClass'].styles[1]).toBe('border:1px solid red') + }) + expect(classes['exClass'].styles.length).toBe(2) expect(vertices['b'].classes[0]).toBe('exClass') expect(classes['exClass'].styles[0]).toBe('background:#bbb')