fixed the row count test assertions

Author: mfaisalkhatri

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

@@ -13,21 +13,52 @@ public DataTablePage (final Page page) {
     }
 
     public Locator tableOne () {
-        return this.page.getByRole (AriaRole.TABLE).first ();
+        return this.page.getByRole (AriaRole.TABLE)
+            .first ();
     }
 
     public Locator tableTwo () {
-        return this.page.getByRole (AriaRole.TABLE).nth (1);
+        return this.page.getByRole (AriaRole.TABLE)
+            .nth (1);
     }
 
-    public int getTotalRowsInTable (final int tableNumber) {
+    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);
         };
-        return table.getByRole(AriaRole.ROW)
-            .count ();
+
+        final Locator rows = table.getByRole (AriaRole.ROW);
+        for (int i = 1; i < rows.count (); i++) {
+            System.out.println (rows.nth (i)
+                .innerText ());
+        }
+    }
+
+    public void getColumnsOfTableOne () {
+        final Locator columnHeader = tableOne ().getByRole (AriaRole.COLUMNHEADER);
+        for (int i = 0; i < columnHeader.count (); i++) {
+            columnHeader.nth (i)
+                .innerText ();
+        }
+    }
+
+    private static Locator getRows (final Locator rows) {
+        return rows;
     }
 
 }

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

@@ -14,8 +14,8 @@
 
 public class DataTableTests {
 
-    private Playwright playwright;
-    private Page       page;
+    private Playwright    playwright;
+    private Page          page;
     private DataTablePage dataTablePage;
 
     @BeforeMethod
@@ -27,13 +27,31 @@ public void setup () {
         this.page = browser.newPage ();
         this.dataTablePage = new DataTablePage (this.page);
     }
+
     @Test
-    public void testRowNumber() {
-        this.page.navigate("https://the-internet.herokuapp.com/tables");
-        assertEquals(this.dataTablePage.getTotalRowsInTable (1),5);
-        assertEquals(this.dataTablePage.getTotalRowsInTable (2),5);
+    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);
     }
 
+    @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);
+        System.out.println ("Printing records of Table 2");
+        this.dataTablePage.tableRecords (2);
+    }
+
+//    @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");
+//
+//    }
+
     @AfterMethod
     public void tearDown () {
         this.page.close ();