Fixed addClassesToSVGElement and added test

Penll · Oct 25, 2017 · b51192c · b51192c
1 parent eceb95c
commit b51192c
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 16 deletions.
diff --git a/plugins/addClassesToSVGElement.js b/plugins/addClassesToSVGElement.js
@@ -42,22 +42,7 @@ exports.fn = function(data, params) {
         svg = data.content[0];
 
     if (svg.isElem('svg')) {
-        if (svg.hasAttr('class')) {
-            var classes = svg.attr('class').value.split(' ');
-            classNames.forEach(function(className){
-                if (classes.indexOf(className) < 0) {
-                    classes.push(className);
-                }
-            });
-            svg.attr('class').value = classes.join(' ');
-        } else {
-            svg.addAttr({
-                name: 'class',
-                value: classNames.join(' '),
-                prefix: '',
-                local: 'class'
-            });
-        }
+        svg.class.add.apply(svg.class, classNames);
     }
 
     return data;

diff --git a/test/plugins/addClassesToSVGElement.01.svg b/test/plugins/addClassesToSVGElement.01.svg