Fix test timeout while running Dev environment integration tests (#3705)

Author: rli

jetbrains-gateway/it/software/aws/toolkits/jetbrains/gateway/DevEnvConnectTest.kt

@@ -40,6 +40,8 @@ import software.aws.toolkits.core.utils.tryOrNull
 import software.aws.toolkits.jetbrains.core.AwsClientManager
 import software.aws.toolkits.jetbrains.core.MockClientManager
 import software.aws.toolkits.jetbrains.core.credentials.ManagedBearerSsoConnection
+import software.aws.toolkits.jetbrains.core.credentials.pinning.CodeCatalystConnection
+import software.aws.toolkits.jetbrains.core.credentials.pinning.ConnectionPinningManager
 import software.aws.toolkits.jetbrains.core.credentials.sono.SONO_REGION
 import software.aws.toolkits.jetbrains.core.credentials.sono.SONO_URL
 import software.aws.toolkits.jetbrains.core.credentials.sso.bearer.BearerTokenProvider
@@ -166,6 +168,8 @@ class DevEnvConnectTest : AfterAllCallback {
         // can probably abstract this out as an extension
         // force auth to complete now
         connection = ManagedBearerSsoConnection(SONO_URL, SONO_REGION, listOf("codecatalyst:read_write"))
+        // pin connection to avoid dialog prompt
+        ConnectionPinningManager.getInstance().setPinnedConnection(CodeCatalystConnection.getInstance(), connection)
         (connection.getConnectionSettings().tokenProvider.delegate as BearerTokenProvider).reauthenticate()
 
         (service<JetBrainsClientDownloaderConfigurationProvider>() as TestJetBrainsClientDownloaderConfigurationProvider).apply {

jetbrains-gateway/src-222-231/software/aws/toolkits/jetbrains/gateway/CawsGatewayConnectionHandle.kt

@@ -13,8 +13,11 @@ class CawsGatewayConnectionHandle(
     private val envId: String,
     private val componentProvider: (GatewayConnectionHandle) -> JComponent
 ) : GatewayConnectionHandle(lifetime) {
+    // framework does not call componentProvider in test mode
+    private val component = componentProvider(this)
+
     override fun customComponentProvider() = DefaultCustomConnectionFrameComponentProvider(getTitle()) {
-        componentProvider(this)
+        component
     }
 
     override fun getTitle() = message("caws.connection_progress_panel_title", envId)

jetbrains-gateway/src-232+/software/aws/toolkits/jetbrains/gateway/CawsGatewayConnectionHandle.kt

@@ -13,8 +13,11 @@ class CawsGatewayConnectionHandle(
     private val envId: String,
     private val componentProvider: (GatewayConnectionHandle) -> JComponent
 ) : GatewayConnectionHandle(lifetime) {
+    // framework does not call componentProvider in test mode
+    private val component = componentProvider(this)
+
     override fun customComponentProvider(lifetime: Lifetime) = DefaultCustomConnectionFrameComponentProvider(getTitle()) {
-        componentProvider(this)
+        component
     }
 
     override fun getTitle() = message("caws.connection_progress_panel_title", envId)