fixed the all records printing test

Author: mfaisalkhatri

src/test/java/io/github/mfaisalkhatri/pages/theinternet/DataTablePage.java

@@ -1,5 +1,8 @@
 package io.github.mfaisalkhatri.pages.theinternet;
 
+import java.util.ArrayList;
+import java.util.List;
+
 import com.microsoft.playwright.Locator;
 import com.microsoft.playwright.Page;
 import com.microsoft.playwright.options.AriaRole;
@@ -23,42 +26,17 @@ public Locator tableTwo () {
     }
 
     public Locator getTotalRowsInTable (final Locator table) {
-        return table.getByRole (AriaRole.ROW);
-    }
-
-    //    public int getTotalRowsInTable (final int tableNumber) {
-    //        final Locator table = switch (tableNumber) {
-    //            case 1 -> tableOne ();
-    //            case 2 -> tableTwo ();
-    //            default -> throw new IllegalStateException ("Invalid Table Number: " + tableNumber);
-    //        };
-    //
-    //    }
-
-    public void tableRecords (final int tableNumber) {
-        final Locator table = switch (tableNumber) {
-            case 1 -> tableOne ();
-            case 2 -> tableTwo ();
-            default -> throw new IllegalStateException ("Invalid Table Number: " + tableNumber);
-        };
-
-        final Locator rows = table.getByRole (AriaRole.ROW);
-        for (int i = 1; i < rows.count (); i++) {
-            System.out.println (rows.nth (i)
-                .innerText ());
-        }
+        return table.locator ("tbody")
+            .getByRole (AriaRole.ROW);
     }
 
-    public void getColumnsOfTableOne () {
-        final Locator columnHeader = tableOne ().getByRole (AriaRole.COLUMNHEADER);
-        for (int i = 0; i < columnHeader.count (); i++) {
-            columnHeader.nth (i)
-                .innerText ();
-        }
+    public List<String> getAllRecordsFromTable (final Locator table) {
+        return table.locator ("tbody")
+            .getByRole (AriaRole.ROW)
+            .allInnerTexts ();
     }
 
-    private static Locator getRows (final Locator rows) {
-        return rows;
+    public Locator getColumnHeadersOfTable (final Locator table) {
+        return table.getByRole (AriaRole.COLUMNHEADER);
     }
-
 }

src/test/java/io/github/mfaisalkhatri/tests/DataTableTests.java

@@ -31,26 +31,30 @@ public void setup () {
     @Test
     public void testRowNumber () {
         this.page.navigate ("https://the-internet.herokuapp.com/tables");
-        assertThat (this.dataTablePage.getTotalRowsInTable (this.dataTablePage.tableOne ())).hasCount (5);
-        assertThat (this.dataTablePage.getTotalRowsInTable (this.dataTablePage.tableTwo ())).hasCount (5);
+        assertThat (this.dataTablePage.getTotalRowsInTable (this.dataTablePage.tableOne ())).hasCount (4);
+        assertThat (this.dataTablePage.getTotalRowsInTable (this.dataTablePage.tableTwo ())).hasCount (4);
     }
 
     @Test
     public void testPrintTableRecords () {
         this.page.navigate ("https://the-internet.herokuapp.com/tables");
         System.out.println ("Printing records of Table 1");
-        this.dataTablePage.tableRecords (1);
+        this.dataTablePage.getAllRecordsFromTable (this.dataTablePage.tableOne ())
+            .forEach (System.out::println);
         System.out.println ("Printing records of Table 2");
-        this.dataTablePage.tableRecords (2);
+        this.dataTablePage.getAllRecordsFromTable (this.dataTablePage.tableTwo ())
+            .forEach (System.out::println);
     }
 
-//    @Test
-//    public void testTableOneColumnHeaders () {
-//        this.page.navigate ("https://the-internet.herokuapp.com/tables");
-//        assertEquals (this.dataTablePage.getColumnsOfTableOne (),
-//            "Last Name\tFirst Name\tEmail\tDue\tWeb Site\tAction");
-//
-//    }
+    @Test
+    public void testTableOneColumnHeaders () {
+        this.page.navigate ("https://the-internet.herokuapp.com/tables");
+        final String[] expectedColumnHeaders = { "Last Name", "First Name", "Email", "Due", "Web Site", "Action" };
+        assertThat (this.dataTablePage.getColumnHeadersOfTable (this.dataTablePage.tableOne ())).hasText (
+            expectedColumnHeaders);
+    }
+
+
 
     @AfterMethod
     public void tearDown () {