Adds check to prevent auto trigger from firing when user is logged out (

#307)
aws · Dec 12, 2024 · e37834e · e37834e
1 parent d1bb152
commit e37834e
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 0 deletions.
diff --git a/plugin/src/software/aws/toolkits/eclipse/amazonq/util/AutoTriggerDocumentListener.java b/plugin/src/software/aws/toolkits/eclipse/amazonq/util/AutoTriggerDocumentListener.java
@@ -9,6 +9,8 @@
 import org.eclipse.ui.PlatformUI;
 import org.eclipse.ui.commands.ICommandService;
 
+import software.aws.toolkits.eclipse.amazonq.plugin.Activator;
+
 import static software.aws.toolkits.eclipse.amazonq.util.QEclipseEditorUtils.getActiveTextEditor;
 
 import java.util.concurrent.ExecutionException;
@@ -45,6 +47,10 @@ public synchronized void documentChanged(final DocumentEvent e) {
     }
 
     private boolean shouldSendQuery(final DocumentEvent e, final QInvocationSession session) {
+        if (!Activator.getLoginService().getAuthState().isLoggedIn()) {
+            return false;
+        }
+
         if (e.getText().length() <= 0) {
             return false;
         }

diff --git a/plugin/tst/software/aws/toolkits/eclipse/amazonq/util/AutoTriggerDocumentListenerTest.java b/plugin/tst/software/aws/toolkits/eclipse/amazonq/util/AutoTriggerDocumentListenerTest.java
@@ -27,12 +27,15 @@
 import org.mockito.invocation.InvocationOnMock;
 import org.mockito.stubbing.Answer;
 
+import software.aws.toolkits.eclipse.amazonq.plugin.Activator;
+
 public class AutoTriggerDocumentListenerTest {
     private static final String TEXT_STUB = "some stuff";
 
     private static MockedStatic<QInvocationSession> sessionMockStatic;
     private static MockedStatic<QEclipseEditorUtils> editorUtilsMockStatic;
     private static MockedStatic<PlatformUI> platformUIMockStatic;
+    private static MockedStatic<Activator> activatorMockStatic;
 
     private static QInvocationSession sessionMock;
     private static IExecutionListener executionListenerMock;
@@ -51,13 +54,17 @@ public static void setUp() {
         ICommandService commandServiceMock = mock(ICommandService.class);
         platformUIMockStatic.when(() -> PlatformUI.getWorkbench().getService(ICommandService.class))
                 .thenReturn(commandServiceMock);
+
+        activatorMockStatic = mockStatic(Activator.class, RETURNS_DEEP_STUBS);
+        activatorMockStatic.when(() -> Activator.getLoginService().getAuthState().isLoggedIn()).thenReturn(true);
     }
 
     @AfterAll
     public static void tearDown() {
         sessionMockStatic.close();
         editorUtilsMockStatic.close();
         platformUIMockStatic.close();
+        activatorMockStatic.close();
     }
 
     @Test