tst

Author: rli

plugins/amazonq/shared/jetbrains-community/tst/software/aws/toolkits/jetbrains/services/amazonq/lsp/auth/DefaultAuthCredentialsServiceTest.kt

@@ -34,6 +34,7 @@ import software.aws.toolkits.jetbrains.core.credentials.sso.bearer.InteractiveBe
 import software.aws.toolkits.jetbrains.services.amazonq.lsp.AmazonQLanguageServer
 import software.aws.toolkits.jetbrains.services.amazonq.lsp.AmazonQLspService
 import software.aws.toolkits.jetbrains.services.amazonq.lsp.encryption.JwtEncryptionManager
+import software.aws.toolkits.jetbrains.services.amazonq.lsp.model.aws.LspServerConfigurations
 import software.aws.toolkits.jetbrains.services.amazonq.lsp.model.aws.credentials.ConnectionMetadata
 import software.aws.toolkits.jetbrains.services.amazonq.lsp.model.aws.credentials.SsoProfileData
 import software.aws.toolkits.jetbrains.services.amazonq.lsp.model.aws.credentials.UpdateCredentialsPayload
@@ -88,6 +89,10 @@ class DefaultAuthCredentialsServiceTest {
             mockLanguageServer.deleteTokenCredentials()
         } returns Unit
 
+        every {
+            mockLanguageServer.updateConfiguration(any())
+        } returns CompletableFuture.completedFuture(LspServerConfigurations(emptyList()))
+
         every { project.getService(AmazonQLspService::class.java) } returns mockLspService
         every { project.serviceIfCreated<AmazonQLspService>() } returns mockLspService
     }

plugins/amazonq/shared/jetbrains-community/tst/software/aws/toolkits/jetbrains/services/amazonq/lsp/textdocument/TextDocumentServiceHandlerTest.kt

@@ -86,7 +86,7 @@ class TextDocumentServiceHandlerTest {
 
         // Mock the LSP service's executeSync method as a suspend function
         coEvery {
-            mockLspService.execute<CompletableFuture<ResponseMessage>>(any())
+            mockLspService.executeIfRunning<CompletableFuture<ResponseMessage>>(any())
         } coAnswers {
             val func = firstArg<suspend AmazonQLspService.(AmazonQLanguageServer) -> CompletableFuture<ResponseMessage>>()
             func.invoke(mockLspService, mockLanguageServer)