Return table headers for aria tables.

LayoutTests/accessibility/table-headers.html

Index: LayoutTests/accessibility/table-headers.html
diff --git a/LayoutTests/accessibility/table-headers.html b/LayoutTests/accessibility/table-headers.html
index 0c3672de35468d7d33be3ac06ee39576dd9b4861..eb2e03630683a3ba85f00996ab0514c01b125c44 100644
--- a/LayoutTests/accessibility/table-headers.html
+++ b/LayoutTests/accessibility/table-headers.html
@@ -132,7 +132,23 @@
     <th>row head</td>
   </tr>
 </table>
-
+<div id="table7" role="grid">
+  <div role="row">
+    <div role="columnheader">col head</div>
+    <div role="columnheader">col head</div>
+    <div role="columnheader">col head</div>
+  </div>
+  <div role="row">
+    <div role="rowheader">row head</div>
+    <div role="gridcell">data</div>
+    <div role="rowheader">row head</div>
+  </div>
+  <div role="row">
+    <div role="rowheader">row head</div>
+    <div role="rowheader">row head</div>
+    <div role="columnheader">col head</div>
+  </div>
+</div>
 <p id="description"></p>
 <div id="console"></div>
 
@@ -210,6 +226,21 @@
         shouldBeTrue("table6.columnHeaderAtIndex(3).isEqual(table6.cellForColumnAndRow(2, 0))");
         shouldBeTrue("table6.columnHeaderAtIndex(4).isEqual(table6.cellForColumnAndRow(2, 2))");
         shouldBe("table6.columnHeadersCount", "5");
+
+        //Aria tables
+        var table7 = accessibilityController.accessibleElementById("table7");
+        // Row headers
+        shouldBeTrue("table7.rowHeaderAtIndex(0).isEqual(table7.cellForColumnAndRow(0, 1))");
+        shouldBeTrue("table7.rowHeaderAtIndex(1).isEqual(table7.cellForColumnAndRow(2, 1))");
+        shouldBeTrue("table7.rowHeaderAtIndex(2).isEqual(table7.cellForColumnAndRow(0, 2))");
+        shouldBeTrue("table7.rowHeaderAtIndex(3).isEqual(table7.cellForColumnAndRow(1, 2))");
+        shouldBe("table7.rowHeadersCount", "4");
+        // Column headers
+        shouldBeTrue("table7.columnHeaderAtIndex(0).isEqual(table7.cellForColumnAndRow(0, 0))");
+        shouldBeTrue("table7.columnHeaderAtIndex(1).isEqual(table7.cellForColumnAndRow(1, 0))");
+        shouldBeTrue("table7.columnHeaderAtIndex(2).isEqual(table7.cellForColumnAndRow(2, 0))");
+        shouldBeTrue("table7.columnHeaderAtIndex(3).isEqual(table7.cellForColumnAndRow(2, 2))");
+        shouldBe("table7.columnHeadersCount", "4");
     }
 
 </script>