test with auth

Author: manodnyab

plugins/core/jetbrains-community/tstFixtures/abc/AbcTest.kt

@@ -0,0 +1,64 @@
+// Copyright 2025 Amazon.com, Inc. or its affiliates. All Rights Reserved.
+// SPDX-License-Identifier: Apache-2.0
+
+package abc
+
+import com.intellij.openapi.Disposable
+import com.intellij.openapi.util.Disposer
+import com.intellij.testFramework.ApplicationExtension
+import com.intellij.testFramework.junit5.TestDisposable
+import org.junit.Rule
+import org.junit.jupiter.api.AfterEach
+import org.junit.jupiter.api.BeforeEach
+import org.junit.jupiter.api.Test
+import org.junit.jupiter.api.extension.ExtendWith
+import software.aws.toolkits.core.rules.SystemPropertyHelper
+import software.aws.toolkits.jetbrains.core.credentials.LegacyManagedBearerSsoConnection
+import software.aws.toolkits.jetbrains.core.credentials.ManagedBearerSsoConnection
+import software.aws.toolkits.jetbrains.core.credentials.sono.Q_SCOPES
+import software.aws.toolkits.jetbrains.core.credentials.sso.bearer.BearerTokenProvider
+import software.aws.toolkits.jetbrains.utils.extensions.SsoLogin
+import software.aws.toolkits.jetbrains.utils.extensions.SsoLoginExtension
+
+
+@ExtendWith(ApplicationExtension::class, SsoLoginExtension::class)
+@SsoLogin("amazonq-test-account")
+//@DisabledIfEnvironmentVariable(named = "IS_PROD", matches = "false")
+class AbcTest {
+
+
+    @TestDisposable
+    private lateinit var disposable: Disposable
+
+
+    @Rule
+    @JvmField
+    val systemPropertyHelper = SystemPropertyHelper()
+
+    private lateinit var connection: ManagedBearerSsoConnection
+
+
+
+    @BeforeEach
+    fun setUp() {
+        // Setup test environment
+        System.setProperty("AWS_PROFILE","act_cred")
+
+    }
+
+
+    @Test
+    fun `can set up Connection`() {
+        println("This ran")
+        connection = LegacyManagedBearerSsoConnection("start_url", "us-east-1", Q_SCOPES)
+        //ConnectionPinningManager.getInstance().setPinnedConnection(QConnection.getInstance(), connection)
+        (connection.getConnectionSettings().tokenProvider.delegate as BearerTokenProvider).reauthenticate()
+
+        //Disposer.register(disposable, connection)
+        // MockClientManager.useRealImplementations(disposableExtension.disposable)
+
+    }
+
+
+
+}

ui-tests-starter/build.gradle.kts

@@ -13,6 +13,11 @@ plugins {
 }
 
 val ideProfile = IdeVersions.ideProfile(project)
+val testPlugins by configurations.registering
+val testPrep by sourceSets.creating {
+    java.srcDirs(findFolders(project, "tst-prep", ideProfile))
+}
+
 
 // Add our source sets per IDE profile version (i.e. src-211)
 sourceSets {
@@ -27,7 +32,7 @@ intellijPlatform {
     instrumentCode = false
 }
 
-val testPlugins by configurations.registering
+
 
 dependencies {
     //testImplementation(platform("com.jetbrains.intellij.tools:ide-starter"))
@@ -39,15 +44,18 @@ dependencies {
     testImplementation(project(":plugin-core:jetbrains-community"))
     testImplementation(project(":plugin-core:core"))
     testImplementation(testFixtures(project(":plugin-core:jetbrains-community")))
+    val testPrepImplementation by configurations.getting
 
+    testPrepImplementation(testFixtures(project(":plugin-core:jetbrains-community")))
 
 
 
 
     intellijPlatform {
-        intellijIdeaCommunity(IdeVersions.ideProfile(providers).map { it.name })
-        intellijIdeaCommunity(ideProfile.community.sdkVersion)
-
+        //intellijIdeaCommunity(IdeVersions.ideProfile(providers).map { it.name })
+        //intellijIdeaCommunity(ideProfile.community.sdkVersion)
+        val version = ideProfile.community.sdkVersion
+        intellijIdeaCommunity(version, !version.contains("SNAPSHOT"))
         testFramework(TestFrameworkType.JUnit5)
         testFramework(TestFrameworkType.Starter)
         testFramework(TestFrameworkType.Bundled)
@@ -57,9 +65,22 @@ dependencies {
     testPlugins(project(":plugin-core", "pluginZip"))
 }
 
+val prepareAmazonQTest = tasks.register<Test>("prepareAmazonQTest") {
+//    dependsOn(":plugin-core:jetbrains-community:test:abc")
+//    filter {
+//        setIncludePatterns("abc.*")
+//        //includeTestsMatching("abc.AbcTest")
+//        //excludeTestsMatching("software.aws.toolkits.jetbrains.*")
+//    }
+    testClassesDirs = testPrep.output.classesDirs
+    classpath = testPrep.runtimeClasspath
+
+    useJUnitPlatform()// Assuming your test task is named amazonQTest
+}
+
 tasks.test {
     dependsOn(testPlugins)
-
+    dependsOn(prepareAmazonQTest)
     useJUnitPlatform()
 
     systemProperty("ui.test.plugins", testPlugins.get().asPath)

ui-tests-starter/tst/software/aws/toolkits/jetbrains/uitests/AmazonQChatTest.kt

@@ -64,20 +64,20 @@ class AmazonQChatTest {
 
 
 
-        Paths.get(System.getProperty("user.home"), ".aws", "sso", "cache", "ee1d2538cb8d358377d7661466c866af747a8a3f.json")
-            .createParentDirectories()
-            .writeText(
-                """
-                  paste your client reg here
-                """.trimIndent()
-            )
-
-        Paths.get(System.getProperty("user.home"), ".aws", "sso", "cache", "d3b447f809607422aac1470dd17fbb32e358cdb3.json")
-            .writeText(
-                """
-               paste your access token here
-                """.trimIndent()
-            )
+//        Paths.get(System.getProperty("user.home"), ".aws", "sso", "cache", "ee1d2538cb8d358377d7661466c866af747a8a3f.json")
+//            .createParentDirectories()
+//            .writeText(
+//                """
+//                  paste your client reg here
+//                """.trimIndent()
+//            )
+//
+//        Paths.get(System.getProperty("user.home"), ".aws", "sso", "cache", "d3b447f809607422aac1470dd17fbb32e358cdb3.json")
+//            .writeText(
+//                """
+//               paste your access token here
+//                """.trimIndent()
+//            )
 
         Starter.newContext(CurrentTestMethod.hyphenateWithClass(), testCase).apply {
             System.getProperty("ui.test.plugins").split(File.pathSeparator).forEach { path ->