fix: svg color change

sirPixieJerry · Feb 7, 2024 · 02adf9c · 02adf9c
1 parent 3b58b58
commit 02adf9c
Show file tree

Hide file tree

Showing 3 changed files with 11 additions and 5 deletions.
diff --git a/src/modules/colorReflection.test.ts b/src/modules/colorReflection.test.ts
@@ -23,7 +23,10 @@ describe('colorReflection', () => {
 
     colorReflection('red');
 
-    expect(reflexOne!.style.backgroundColor).to.equal('red');
-    expect(reflexTwo!.style.backgroundColor).to.equal('red');
+    if (!reflexOne) throw new Error('reflexOne not found');
+    if (!reflexTwo) throw new Error('reflexTwo not found');
+
+    expect(reflexOne.style.backgroundColor).to.equal('red');
+    expect(reflexTwo.style.backgroundColor).to.equal('red');
   });
 });
diff --git a/src/modules/colorSVG.test.ts b/src/modules/colorSVG.test.ts
@@ -23,7 +23,10 @@ describe('colorSVG', () => {
 
     colorSVG('red');
 
-    expect(box!.style.backgroundColor).to.equal('red');
-    expect(path!.style.fill).to.equal('red');
+    if (!box) throw new Error('box not found');
+    if (!path) throw new Error('path not found');
+
+    expect(box.style.backgroundColor).to.equal('red');
+    expect(path.getAttribute('fill')).to.equal('red');
   });
 });
diff --git a/src/modules/colorSVG.ts b/src/modules/colorSVG.ts
@@ -5,5 +5,5 @@ export const colorSVG = (color: string): void => {
   if (box === null || path === null) return;
 
   box.style.backgroundColor = color;
-  path.style.fill = color;
+  path.setAttribute('fill', color);
 };