add e2e test for reject case

Author: Randall-Jiang

ui-tests-starter/tst-243+/software/aws/toolkits/jetbrains/uitests/testTests/QTestGenerationChatTest.kt

@@ -180,6 +180,40 @@ class QTestGenerationChatTest {
             }
     }
 
+    @Test
+    fun `test reject path from the chat`() {
+        val testCase = TestCase(
+            IdeProductProvider.IC,
+            LocalProjectInfo(
+                Paths.get("tstData", "qTestGenerationTestProject/")
+            )
+        ).useRelease(System.getProperty("org.gradle.project.ideProfileName"))
+
+        // inject connection
+        useExistingConnectionForTest()
+
+        Starter.newContext(CurrentTestMethod.hyphenateWithClass(), testCase).apply {
+            System.getProperty("ui.test.plugins").split(File.pathSeparator).forEach { path ->
+                pluginConfigurator.installPluginFromPath(
+                    Path.of(path)
+                )
+            }
+
+            copyExistingConfig(Paths.get("tstData", "configAmazonQTests"))
+            updateGeneralSettings()
+        }.runIdeWithDriver()
+            .useDriverAndCloseIde {
+                waitForProjectOpen()
+                openFile(Paths.get("testModule2", "UnSupportedLanguage.kt").toString())
+                Thread.sleep(30000)
+                val result = executePuppeteerScript(testRejectPathScript)
+                assertTrue(result.contains("new tab opened"))
+                assertTrue(result.contains("View Diff opened"))
+                assertTrue(result.contains("Result Reject"))
+                assertTrue(result.contains("Unit test generation completed."))
+            }
+    }
+
     companion object {
         @JvmStatic
         @AfterAll

ui-tests-starter/tst-243+/software/aws/toolkits/jetbrains/uitests/testTests/QTestGenerationChatTestScripts.kt

@@ -327,3 +327,101 @@ val unsupportedLanguagePath = """
     }
     testNavigation().catch(console.error);
 """.trimIndent()
+
+val testRejectPathScript = """
+        const puppeteer = require('puppeteer');
+        async function testNavigation() {
+            const browser = await puppeteer.connect({
+                browserURL: "http://localhost:9222"
+            })
+            try {
+                const pages = await browser.pages()
+                for(const page of pages) {
+                    const contents = await page.evaluate(el => el.innerHTML, await page.${'$'}(':root'));
+                    const element = await page.${'$'}('.mynah-chat-prompt-input')
+                    if(element) {
+                        const elements = await page.${'$'}${'$'}('.mynah-chat-command-selector-command');
+                        const attr = await Promise.all(
+                            elements.map(elem => elem.evaluate(el => el.getAttribute('command')))
+                        );
+                        await page.type('.mynah-chat-prompt-input', '/test')
+                        await page.keyboard.press('Enter');
+                        await page.keyboard.press('Enter');
+                        try {
+                            await waitForElementWithText(page, "Q - Test")
+                            console.log("new tab opened")
+                            await page.waitForFunction(
+                                () => {
+                                    const button = document.querySelector('button[action-id="utg_view_diff"]');
+                                    return button && button.isEnabled !== false && button.disabled !== true;
+                                },
+                                { timeout: 300000 }
+                            );
+                            await page.evaluate(() => {
+                                const button = document.querySelector('button[action-id="utg_view_diff"]');
+                                if (button) {
+                                    button.click();
+                                } else {
+                                    throw new Error('Button not found after waiting');
+                                }
+                            });
+                            console.log("View Diff opened")
+                            await page.waitForFunction(
+                                () => {
+                                    const button = document.querySelector('button[action-id="utg_reject"]');
+                                    return button && button.isEnabled !== false && button.disabled !== true;
+                                },
+                                { timeout: 300000 }
+                            );
+                            await page.evaluate(() => {
+                                const button = document.querySelector('button[action-id="utg_reject"]');
+                                if (button) {
+                                    button.click();
+                                } else {
+                                    throw new Error('Accept button not found after waiting');
+                                }
+                            });
+                            console.log("Result Reject")
+                            await waitForElementWithText(page, "Unit test generation completed.")
+                            console.log("Unit test generation completed.")
+                        } catch (e) {
+                            console.log("Element with text not found")
+                            console.log(e)
+                            throw e
+                        }
+
+                    }
+                }
+            } finally {
+                await browser.close();
+            }
+        }
+
+        async function waitForElementWithText(page, text) {
+            await page.waitForFunction(
+                (expectedText) => {
+                    const elements = document.querySelectorAll('*');
+                    return Array.from(elements).find(element =>
+                        element.textContent?.trim() === expectedText
+                    );
+                },
+                {},
+                text
+            );
+        }
+
+        async function waitAndGetElementByText(page, text) {
+            const element = await page.waitForFunction(
+                (expectedText) => {
+                    const elements = document.querySelectorAll('*');
+                    return Array.from(elements).find(element =>
+                        element.textContent?.trim() === expectedText
+                    );
+                },
+                {},
+                text
+            );
+            return element;
+        }
+        testNavigation().catch(console.error); 
+""".trimIndent()