Fix unit test

Author: opieter-aws

packages/amazonq/test/unit/amazonq/lsp/chat/messages.test.ts

@@ -10,6 +10,7 @@ import { AuthUtil } from 'aws-core-vscode/codewhisperer'
 import { registerMessageListeners } from '../../../../../src/lsp/chat/messages'
 import { AmazonQChatViewProvider } from '../../../../../src/lsp/chat/webviewProvider'
 import { secondaryAuth, authConnection, AuthFollowUpType } from 'aws-core-vscode/amazonq'
+import { messages } from 'aws-core-vscode/shared'
 
 describe('registerMessageListeners', () => {
     let languageClient: LanguageClient
@@ -129,41 +130,33 @@ describe('registerMessageListeners', () => {
     })
 
     describe('COPY_TO_CLIPBOARD', () => {
+        let copyToClipboardStub: sinon.SinonStub
         const testCode = 'test'
         const copyMessage = {
             command: 'copyToClipboard',
             params: {
                 code: testCode,
             },
         }
-        let writeTextStub: sinon.SinonStub
 
         beforeEach(() => {
-            writeTextStub = sinon.stub()
-            sandbox.stub(vscode.env, 'clipboard').value({
-                writeText: writeTextStub,
-            })
+            copyToClipboardStub = sandbox.stub().resolves()
+            sandbox.stub(messages, 'copyToClipboard').get(() => copyToClipboardStub)
         })
 
         it('successfully copies code to clipboard', async () => {
-            writeTextStub.resolves()
-
             await messageHandler(copyMessage)
 
-            sinon.assert.calledWith(writeTextStub, testCode)
+            sinon.assert.calledWith(copyToClipboardStub, testCode)
         })
 
         it('handles clipboard copy failure', async () => {
             const errorMessage = 'Failed to copy'
-            writeTextStub.rejects(new Error(errorMessage))
+            copyToClipboardStub.rejects(new Error(errorMessage))
 
             await messageHandler(copyMessage)
 
-            sinon.assert.calledOnce(errorStub)
-            sinon.assert.calledWith(
-                errorStub,
-                sinon.match(`[VSCode Client] Failed to copy to clipboard: ${errorMessage}`)
-            )
+            sinon.assert.calledWith(errorStub, `[VSCode Client] Failed to copy to clipboard: ${errorMessage}`)
         })
     })
 })