From b51192cb7a41454f1b4af34bdf53a1898c2c5349 Mon Sep 17 00:00:00 2001
From: GreLI <grelimail@gmail.com>
Date: Wed, 25 Oct 2017 22:25:57 +0300
Subject: [PATCH] Fixed addClassesToSVGElement and added test

---
 plugins/addClassesToSVGElement.js          | 17 +----------------
 test/plugins/addClassesToSVGElement.01.svg | 13 +++++++++++++
 2 files changed, 14 insertions(+), 16 deletions(-)
 create mode 100644 test/plugins/addClassesToSVGElement.01.svg

diff --git a/plugins/addClassesToSVGElement.js b/plugins/addClassesToSVGElement.js
index 2f740c29..ef215fc9 100644
--- 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
new file mode 100644
index 00000000..496864d9
--- /dev/null
+++ b/test/plugins/addClassesToSVGElement.01.svg
@@ -0,0 +1,13 @@
+<svg xmlns="http://www.w3.org/2000/svg">
+    test
+</svg>
+
+@@@
+
+<svg xmlns="http://www.w3.org/2000/svg" class="mySvg size-big">
+    test
+</svg>
+
+@@@
+
+{"classNames":["mySvg","size-big"]}