add copycode test

DataDog · Apr 16, 2024 · 85f08ab · 85f08ab
1 parent f6b8d07
commit 85f08ab
Show file tree

Hide file tree

Showing 2 changed files with 25 additions and 1 deletion.
diff --git a/assets/scripts/components/copy-code.js b/assets/scripts/components/copy-code.js
@@ -83,4 +83,4 @@ function updateCopyBtnText(btn){
 initCopyCode()
 
 // Export the functions for testing
-module.exports = {initCopyCode, getCode, addCopyButton, updateCopyBtnText}
+module.exports = {initCopyCode, getCode, copyCode, addCopyButton, updateCopyBtnText}
diff --git a/assets/scripts/tests/copy-code.test.js b/assets/scripts/tests/copy-code.test.js
@@ -98,4 +98,28 @@ describe('initCopyCode', () => {
         cc.updateCopyBtnText(btn);
         expect(btn.textContent).toBe('Copied!');
     })
+
+
+    it('should copy code snippet text to clipboard with the writeText method', () => {
+        document.body.innerHTML = `
+            <div class="highlight code-snippet">
+                <pre><code>some code</code></pre>
+                <button class="js-copy-button">Copy</button>
+            </div>
+        `;
+        const btn = document.querySelector('.js-copy-button');
+
+        const clipboardMock = {
+            writeText: jest.fn(() => Promise.resolve())
+        };
+        Object.defineProperty(document, 'createRange', {
+            value: jest.fn(() => ({
+                selectNode: jest.fn(),
+            }))
+        })
+        navigator.clipboard = clipboardMock;
+
+        cc.copyCode(btn);
+        expect(clipboardMock.writeText).toHaveBeenCalledTimes(1);
+    })
 })