add static mock to lsp connection test case

aws · Jan 24, 2025 · bd8d40c · bd8d40c
1 parent 87a617e
commit bd8d40c
Show file tree

Hide file tree

Showing 2 changed files with 19 additions and 1 deletion.
diff --git a/plugin/src/software/aws/toolkits/eclipse/amazonq/lsp/manager/LspStatusManager.java b/plugin/src/software/aws/toolkits/eclipse/amazonq/lsp/manager/LspStatusManager.java
@@ -8,7 +8,7 @@
 public final class LspStatusManager {
 
     private LspStatusManager() {
-        //prevent
+        //prevent instantiation
     }
 
     private static LspState lspState = LspState.PENDING;

diff --git a/.../tst/software/aws/toolkits/eclipse/amazonq/lsp/connection/QLspConnectionProviderTest.java b/.../tst/software/aws/toolkits/eclipse/amazonq/lsp/connection/QLspConnectionProviderTest.java
@@ -3,16 +3,21 @@
 
 package software.aws.toolkits.eclipse.amazonq.lsp.connection;
 
+import org.junit.jupiter.api.AfterEach;
+import org.junit.jupiter.api.BeforeEach;
 import org.junit.jupiter.api.Test;
 import org.junit.jupiter.api.extension.RegisterExtension;
 import org.mockito.MockedStatic;
 import org.mockito.Mockito;
+
 import software.aws.toolkits.eclipse.amazonq.extensions.implementation.ActivatorStaticMockExtension;
 import software.aws.toolkits.eclipse.amazonq.extensions.implementation.DefaultLspEncryptionManagerStaticMockExtension;
 import software.aws.toolkits.eclipse.amazonq.extensions.implementation.LspManagerProviderStaticMockExtension;
 import software.aws.toolkits.eclipse.amazonq.extensions.implementation.ProxyUtilsStaticMockExtension;
 import software.aws.toolkits.eclipse.amazonq.lsp.encryption.LspEncryptionManager;
 import software.aws.toolkits.eclipse.amazonq.lsp.manager.LspInstallResult;
+import software.aws.toolkits.eclipse.amazonq.lsp.manager.LspStatusManager;
+
 import org.eclipse.lsp4e.server.ProcessStreamConnectionProvider;
 import software.aws.toolkits.eclipse.amazonq.util.LoggingService;
 import software.aws.toolkits.eclipse.amazonq.util.ProxyUtil;
@@ -30,6 +35,7 @@
 import static org.mockito.Mockito.doNothing;
 import static org.mockito.Mockito.doReturn;
 import static org.mockito.Mockito.doThrow;
+import static org.mockito.Mockito.mockStatic;
 import static org.mockito.Mockito.verify;
 
 public final class QLspConnectionProviderTest {
@@ -48,6 +54,18 @@ public final class QLspConnectionProviderTest {
     @RegisterExtension
     private static ProxyUtilsStaticMockExtension proxyUtilsStaticMockExtension = new ProxyUtilsStaticMockExtension();
 
+    private MockedStatic<LspStatusManager> mockLspStatusManager;
+
+    @BeforeEach
+    void setupBeforeEach() {
+        mockLspStatusManager = mockStatic(LspStatusManager.class);
+    }
+
+    @AfterEach
+    void tearDown() {
+        mockLspStatusManager.close();
+    }
+
     private static final class TestProcessConnectionProvider extends ProcessStreamConnectionProvider {
 
         TestProcessConnectionProvider(final List<String> commands, final String workingDirectory) {