fix(amazonq): Fix Generate unit tests from right click context menu

plugins/amazonq/chat/jetbrains-community/src/software/aws/toolkits/jetbrains/services/amazonq/apps/AmazonQAppInitContext.kt

@@ -5,6 +5,7 @@
 
 import com.intellij.openapi.project.Project
 import software.aws.toolkits.jetbrains.services.amazonq.commands.MessageTypeRegistry
+import software.aws.toolkits.jetbrains.services.amazonq.messages.MessageConnector
 import software.aws.toolkits.jetbrains.services.amazonq.messages.MessageListener
 import software.aws.toolkits.jetbrains.services.amazonq.messages.MessagePublisher
 import software.aws.toolkits.jetbrains.services.amazonq.webview.FqnWebviewAdapter
@@ -19,3 +20,7 @@
     val messageTypeRegistry: MessageTypeRegistry,
     val fqnWebviewAdapter: FqnWebviewAdapter,
 )
+
+
+
+

plugins/amazonq/chat/jetbrains-community/src/software/aws/toolkits/jetbrains/services/amazonq/toolwindow/AmazonQPanel.kt

@@ -201,6 +201,17 @@ class AmazonQPanel(val project: Project, private val scope: CoroutineScope) : Di
         }
     }
 
+    fun getDefaultAppInitContext() : AmazonQAppInitContext {
+
+        return AmazonQAppInitContext(
+            project = project,
+            messagesFromAppToUi = MessageConnector(),
+            messagesFromUiToApp = MessageConnector(),
+            messageTypeRegistry = MessageTypeRegistry(),
+            fqnWebviewAdapter = FqnWebviewAdapter(jcefBrowser = browser.get().jcefBrowser, browserConnector),
+        )
+    }
+
     override fun dispose() {
     }
 }

plugins/amazonq/chat/jetbrains-community/src/software/aws/toolkits/jetbrains/services/amazonq/toolwindow/AmazonQToolWindow.kt

@@ -11,9 +11,11 @@
 import com.intellij.openapi.util.Disposer
 import com.intellij.openapi.wm.ToolWindowManager
 import kotlinx.coroutines.CoroutineScope
+import kotlinx.coroutines.runBlocking
 import software.aws.toolkits.jetbrains.services.amazonq.onboarding.OnboardingPageInteraction
 import software.aws.toolkits.jetbrains.services.amazonq.onboarding.OnboardingPageInteractionType
 import software.aws.toolkits.jetbrains.services.amazonqCodeScan.runCodeScanMessage
+import software.aws.toolkits.jetbrains.services.cwc.controller.TestCommandMessage
 
 @Service(Service.Level.PROJECT)
 class AmazonQToolWindow private constructor(
@@ -52,6 +54,14 @@
             val window = getInstance(project)
             window.chatPanel.sendMessageAppToUi(runCodeScanMessage, tabType = "codescan")
         }
+
+        fun sendTestMessage(project: Project) {
+            runBlocking {
+                val a = getInstance(project).chatPanel.getDefaultAppInitContext()
+                val b = a.messagesFromAppToUi.publish(TestCommandMessage())
+            }
+
+        }
     }
 
     override fun dispose() {

plugins/amazonq/chat/jetbrains-community/src/software/aws/toolkits/jetbrains/services/cwc/commands/ActionRegistrar.kt

@@ -6,7 +6,15 @@
 import com.intellij.openapi.project.Project
 import kotlinx.coroutines.flow.MutableSharedFlow
 import kotlinx.coroutines.flow.asSharedFlow
+import kotlinx.coroutines.runBlocking
+import software.aws.toolkits.jetbrains.services.amazonq.apps.AmazonQAppInitContext
+import software.aws.toolkits.jetbrains.services.amazonq.commands.MessageTypeRegistry
+import software.aws.toolkits.jetbrains.services.amazonq.lsp.flareChat.AsyncChatUiListener
 import software.aws.toolkits.jetbrains.services.amazonq.messages.AmazonQMessage
+import software.aws.toolkits.jetbrains.services.amazonq.messages.MessageConnector
+import software.aws.toolkits.jetbrains.services.amazonq.toolwindow.AmazonQToolWindow
+import software.aws.toolkits.jetbrains.services.amazonq.webview.FqnWebviewAdapter
+import software.aws.toolkits.jetbrains.services.cwc.controller.TestCommandMessage
 
 // Register Editor Actions in the Editor Context Menu
 class ActionRegistrar {
@@ -15,7 +23,14 @@
     val flow = _messages.asSharedFlow()
 
     fun reportMessageClick(command: EditorContextCommand, project: Project) {
-        _messages.tryEmit(ContextMenuActionMessage(command, project))
+        // language=JSON
+        AmazonQToolWindow.sendTestMessage(project)
+        //AsyncChatUiListener.notifyPartialMessageUpdate(a)
+       // _messages.tryEmit(ContextMenuActionMessage(command, project))
+//        runBlocking {
+//            MessageConnector().publish(messageToPublish)
+//        }
+
     }
 
     fun reportMessageClick(command: EditorContextCommand, issue: MutableMap<String, String>, project: Project) {
@@ -27,3 +42,10 @@
         val instance = ActionRegistrar()
     }
 }
+//fun getContext(project: Project) = AmazonQAppInitContext(
+//    project,
+//    MessageConnector(),
+//    MessageConnector(),
+//    MessageTypeRegistry(),
+//    FqnWebviewAdapter(project)
+//)

plugins/amazonq/chat/jetbrains-community/src/software/aws/toolkits/jetbrains/services/cwc/controller/ChatController.kt

@@ -77,9 +77,9 @@ import software.aws.toolkits.telemetry.CwsprChatCommandType
 import java.util.UUID
 
 data class TestCommandMessage(
-    val sender: String = "codetest",
+    val sender: String = "testChat",
     val command: String = "test",
-    val type: String = "addAnswer",
+    val type: String = "chatMessage",
 ) : AmazonQMessage
 
 class ChatController private constructor(