fix(Amazon Q): fix for test generation target file missing from payload.

.changes/next-release/bugfix-f47212d1-8e2b-4c82-9e0d-450ce896e9bc.json

@@ -0,0 +1,4 @@
+{
+  "type" : "bugfix",
+  "description" : "Amazon Q /test: Fix for test generation payload creation to not filter out target file."
+}

plugins/amazonq/chat/jetbrains-community/src/software/aws/toolkits/jetbrains/services/amazonqCodeTest/controller/CodeTestChatController.kt

@@ -277,7 +277,7 @@ class CodeTestChatController(
 
             val requestData = ChatRequestData(
                 tabId = session.tabId,
-                message = "Generate unit tests for the following part of my code: ${message.prompt}",
+                message = "Generate unit tests for the following part of my code: ${message.prompt.ifBlank { fileInfo.fileName }}",
                 activeFileContext = activeFileContext,
                 userIntent = UserIntent.GENERATE_UNIT_TESTS,
                 triggerType = TriggerType.ContextMenu,

plugins/amazonq/chat/jetbrains-community/tst/software/aws/toolkits/jetbrains/services/amazonqFeatureDev/FeatureDevSessionContextTest.kt

@@ -3,6 +3,7 @@
 
 import com.intellij.openapi.vfs.VirtualFile
 import com.intellij.testFramework.RuleChain
+import org.junit.Assert.assertEquals
 import org.junit.Assert.assertFalse
 import org.junit.Assert.assertTrue
 import org.junit.Before
@@ -75,6 +76,7 @@ class FeatureDevSessionContextTest : FeatureDevTestBase(HeavyJavaCodeInsightTest
     @Test
     fun testZipProject() {
         addFilesToProjectModule(
+            ".gitignore",
             ".gradle/cached.jar",
             "src/MyClass.java",
             "gradlew",
@@ -83,6 +85,19 @@ class FeatureDevSessionContextTest : FeatureDevTestBase(HeavyJavaCodeInsightTest
             "settings.gradle",
             "build.gradle",
             "gradle/wrapper/gradle-wrapper.properties",
+            "builder/GetTestBuilder.java", //check for false positives
+            ".aws-sam/build/function1",
+            ".gem/specs.rb",
+            "archive.zip",
+            "output.bin",
+            "images/logo.png",
+            "assets/header.jpg",
+            "icons/menu.svg",
+            "license.txt",
+            "License.md",
+            "node_modules/express",
+            "build/outputs",
+            "dist/bundle.js"
         )
 
         val zipResult = featureDevSessionContext.getProjectZip()
@@ -102,11 +117,73 @@ class FeatureDevSessionContextTest : FeatureDevTestBase(HeavyJavaCodeInsightTest
             "gradlew",
             "gradlew.bat",
             "README.md",
-            "settings.gradle",
-            "build.gradle",
             "gradle/wrapper/gradle-wrapper.properties",
+            "builder/GetTestBuilder.java"
         )
 
         assertTrue(zippedFiles == expectedFiles)
     }
+
+    @Test
+    fun `test basic pattern conversion`() {
+        val input = "*.txt"
+        val expected = "(?:^|.*/?)[^/]*[^/]\\.txt(?:/.*)?\$"
+        assertEquals(expected, featureDevSessionContext.convertGitIgnorePatternToRegex(input))
+    }
+
+    @Test
+    fun `test pattern with special characters`() {
+        val input = "test[abc].txt"
+        val expected = "(?:^|.*/?)test\\[abc\\]\\.txt(?:/.*)?$"
+        assertEquals(expected, featureDevSessionContext.convertGitIgnorePatternToRegex(input))
+    }
+
+    @Test
+    fun `test pattern with double asterisk`() {
+        val input = "**/build"
+        val expected = "(?:^|.*/?).[^/]*[^/][^/]/build(?:/.*)?\$"
+        assertEquals(expected, featureDevSessionContext.convertGitIgnorePatternToRegex(input))
+    }
+
+    @Test
+    fun `test pattern starting with slash`() {
+        val input = "/root/file.txt"
+        val expected = "^root/file\\.txt(?:/.*)?$"
+        assertEquals(expected, featureDevSessionContext.convertGitIgnorePatternToRegex(input))
+    }
+
+    @Test
+    fun `test pattern ending with slash`() {
+        val input = "build/"
+        val expected = "(?:^|.*/?)build/.*"
+        assertEquals(expected, featureDevSessionContext.convertGitIgnorePatternToRegex(input))
+    }
+
+    @Test
+    fun `test pattern with question mark`() {
+        val input = "file?.txt"
+        val expected = "(?:^|.*/?)file[^/]\\.txt(?:/.*)?$"
+        assertEquals(expected, featureDevSessionContext.convertGitIgnorePatternToRegex(input))
+    }
+
+    @Test
+    fun `test complex pattern with multiple special characters`() {
+        val input = "**/test-[0-9]*.{java,kt}"
+        val expected = "(?:^|.*/?).[^/]*[^/][^/]/test-\\[0-9\\][^/]*[^/]\\.\\{java\\,kt\\}(?:/.*)?\$"
+        assertEquals(expected, featureDevSessionContext.convertGitIgnorePatternToRegex(input))
+    }
+
+    @Test
+    fun `test empty pattern`() {
+        val input = ""
+        val expected = "(?:^|.*/?)(?:/.*)?$"
+        assertEquals(expected, featureDevSessionContext.convertGitIgnorePatternToRegex(input))
+    }
+
+    @Test
+    fun `test pattern with all special regex characters`() {
+        val input = ".$+()[]{}^|"
+        val expected = "(?:^|.*/?)\\.\\\$\\+\\(\\)\\[\\]\\{\\}\\^\\|(?:/.*)?$"
+        assertEquals(expected, featureDevSessionContext.convertGitIgnorePatternToRegex(input))
+    }
 }

plugins/amazonq/codewhisperer/jetbrains-community/src/software/aws/toolkits/jetbrains/services/codewhisperer/codetest/sessionconfig/CodeTestSessionConfig.kt

@@ -141,7 +141,7 @@ class CodeTestSessionConfig(
         }
 
         // 2. Add the "utgRequiredArtifactsDir" directory
-        val utgDir = "utgRequiredArtifactsDir"
+        val utgDir = "utgRequiredArtifactsDir/" // Note the trailing slash which adds it as a directory and not a file
         LOG.debug { "Adding directory to ZIP: $utgDir" }
         val utgEntry = ZipEntry(utgDir)
         it.putNextEntry(utgEntry)
@@ -150,7 +150,7 @@ class CodeTestSessionConfig(
         val buildAndExecuteLogDir = "buildAndExecuteLogDir"
         val subDirs = listOf(buildAndExecuteLogDir, "repoMapData", "testCoverageDir")
         subDirs.forEach { subDir ->
-            val subDirPathString = Path.of(utgDir, subDir).name
+            val subDirPathString = Path.of(utgDir, subDir).toString() + "/" // Added trailing slash similar to utgRequiredArtifactsDir
             LOG.debug { "Adding empty directory to ZIP: $subDirPathString" }
             val zipEntry = ZipEntry(subDirPathString)
             it.putNextEntry(zipEntry)
@@ -168,6 +168,18 @@ class CodeTestSessionConfig(
         var currentTotalLines = 0L
         val languageCounts = mutableMapOf<CodeWhispererProgrammingLanguage, Int>()
 
+        // Adding Target File to make sure target file doesn't get filtered out.
+        selectedFile?.let { selected ->
+            files.add(selected.path)
+            currentTotalFileSize += selected.length
+            currentTotalLines += countLinesInVirtualFile(selected)
+            selected.programmingLanguage().let { language ->
+                if (language !is CodeWhispererUnknownLanguage) {
+                    languageCounts[language] = (languageCounts[language] ?: 0) + 1
+                }
+            }
+        }
+
         moduleLoop@ for (module in project.modules) {
             val changeListManager = ChangeListManager.getInstance(module.project)
             if (module.guessModuleDir() != null) {
@@ -176,7 +188,8 @@ class CodeTestSessionConfig(
                     val current = stack.pop()
 
                     if (!current.isDirectory) {
-                        if (current.isFile && !changeListManager.isIgnoredFile(current) &&
+                        if (current.isFile && current.path != selectedFile?.path &&
+                            !changeListManager.isIgnoredFile(current) &&
                             runBlocking { !featureDevSessionContext.ignoreFile(current) } &&
                             runReadAction { !fileIndex.isInLibrarySource(current) }
                         ) {

plugins/core/jetbrains-community/src/software/aws/toolkits/jetbrains/services/amazonq/FeatureDevSessionContext.kt

@@ -252,10 +252,40 @@ class FeatureDevSessionContext(val project: Project, val maxProjectSizeBytes: Lo
     }
 
     // gitignore patterns are not regex, method update needed.
-    private fun convertGitIgnorePatternToRegex(pattern: String): String = pattern
+    fun convertGitIgnorePatternToRegex(pattern: String): String = pattern
+        // Escape special regex characters except * and ?
         .replace(".", "\\.")
-        .replace("*", ".*")
-        .let { if (it.endsWith("/")) "$it.*" else "$it/.*" } // Add a trailing /* to all patterns. (we add a trailing / to all files when matching)
+        .replace("+", "\\+")
+        .replace("(", "\\(")
+        .replace(")", "\\)")
+        .replace("[", "\\[")
+        .replace("]", "\\]")
+        .replace("{", "\\{")
+        .replace("}", "\\}")
+        .replace(",", "\\,")
+        .replace("^", "\\^")
+        .replace("$", "\\$")
+        .replace("|", "\\|")
+        // Convert gitignore glob patterns to regex
+        .replace("**", ".*?") // Match any directory depth
+        .replace("*", "[^/]*?") // Match any character except path separator
+        .replace("?", "[^/]") // Match single character except path separator
+        .let { pattern ->
+            when {
+                // If pattern starts with '/', anchor it to the start of the path
+                pattern.startsWith("/") -> "^${pattern.substring(1)}"
+                // If pattern doesn't start with '/', it can match anywhere in the path
+                else -> "(?:^|.*/?)$pattern"
+            }
+        }
+        .let { pattern ->
+            when {
+                // If pattern ends with '/', it should match directories
+                pattern.endsWith("/") -> "$pattern.*"
+                // Otherwise match exactly or with a trailing slash for directories
+                else -> "$pattern(?:/.*)?$"
+            }
+        }
 
     var selectedSourceFolder: VirtualFile
         set(newRoot) {