fix #289 test & locators

Author: Naseem77

e2e/logic/POM/codeGraph.ts

@@ -38,10 +38,14 @@ export default class CodeGraph extends BasePage {
         return this.page.locator("//button[@role='combobox']")
     }
 
-    private get selectGraphInComboBox(): (graph: string) => Locator {
+    private get selectGraphInComboBoxByName(): (graph: string) => Locator {
         return (graph: string) => this.page.locator(`//div[@role='presentation']//div//span[contains(text(), '${graph}')]`);
     }
 
+    private get selectGraphInComboBoxById(): (graph: string) => Locator {
+        return (graph: string) => this.page.locator(`//div[@role='presentation']//div[${graph}]`);
+    }
+
     private get lastElementInChat(): Locator {
         return this.page.locator("//main[@data-name='main-chat']/*[last()]/span");
     }
@@ -307,10 +311,15 @@ export default class CodeGraph extends BasePage {
     }
 
     /* CodeGraph functionality */
-    async selectGraph(graph: string): Promise<void> {
+    async selectGraph(graph: string | number): Promise<void> {
         await this.comboBoxbtn.click();
-        await this.selectGraphInComboBox(graph).waitFor({ state : 'visible'})
-        await this.selectGraphInComboBox(graph).click();
+        if(typeof graph === 'number'){
+            await this.selectGraphInComboBoxById(graph.toString()).waitFor({ state : 'visible'})
+            await this.selectGraphInComboBoxById(graph.toString()).click();
+        } else {
+            await this.selectGraphInComboBoxByName(graph).waitFor({ state : 'visible'})
+            await this.selectGraphInComboBoxByName(graph).click();
+        }
     }
 
     async createProject(url : string): Promise<void> {

e2e/tests/codeGraph.spec.ts

@@ -164,11 +164,11 @@ test.describe("Code graph tests", () => {
     expect(initialNodeCount).toBe(finalNodeCount);
   });
 
-  for (let index = 0; index < 3; index++) {
+  for (let index = 0; index < 2; index++) {
     const checkboxIndex = index + 1;
     test(`Verify selecting different graphs displays nodes in canvas - Iteration ${index + 1}`, async () => {
       const codeGraph = await browser.createNewPage(CodeGraph, urls.baseUrl);
-      await codeGraph.selectGraph(checkboxIndex.toString());
+      await codeGraph.selectGraph(checkboxIndex);
       const result = await codeGraph.getCanvasAnalysis();
       const nodesLength = result.green.length + result.yellow.length + result.red.length;
       expect(nodesLength).toBeGreaterThan(1);