fix: correctly display watch/blacklist button text

userscripters · Jul 11, 2024 · b0bec6c · b0bec6c
1 parent 842b38d
commit b0bec6c
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 2 deletions.
diff --git a/src/dom_utils.ts b/src/dom_utils.ts
@@ -36,7 +36,7 @@ function getButton(action: 'watch' | 'blacklist'): HTMLAnchorElement {
     const button = document.createElement('a');
     button.classList.add(`fire-extra-${action}`);
     button.style.display = 'none';
-    button.innerHTML = '!!/${action}';
+    button.innerHTML = `!!/${action}`;
 
     return button;
 }

diff --git a/test/dom_utils.spec.ts b/test/dom_utils.spec.ts
@@ -3,7 +3,8 @@ import {
     getTag,
     getResultsContainer,
     updateMsCounts,
-    updateSeCount
+    updateSeCount,
+    getWatchBlacklistButtons
 } from '../src/dom_utils';
 import { helpers } from '../src/index';
 
@@ -27,6 +28,17 @@ describe('DOM utils', () => {
         expect(resultsContainer).not.to.be.null;
     });
 
+    it('should correctly return the container of !!/watch and !!/blacklist buttons', () => {
+        const container = getWatchBlacklistButtons();
+        const [watch, blacklist] = [...container.children];
+
+        expect(watch.innerHTML).to.equal('!!/watch');
+        expect(watch.className).to.equal('fire-extra-watch');
+
+        expect(blacklist.innerHTML).to.equal('!!/blacklist');
+        expect(blacklist.className).to.equal('fire-extra-blacklist');
+    });
+
     it('should get a tag in metasmoke\'s style', () => {
         ['shortener', 'ip', 'whitelisted', 'stuff-up'].forEach(tagName => {
             const tag = getTag(tagName);