Add tests to verify the new tables

Author: zsxwing

streaming/src/main/scala/org/apache/spark/streaming/ui/AllBatchesTable.scala

@@ -22,7 +22,7 @@ import scala.xml.Node
 import org.apache.spark.streaming.scheduler.BatchInfo
 import org.apache.spark.ui.UIUtils
 
-private[ui] class BatchTableBase {
+private[ui] class BatchTableBase(tableId: String) {
 
   protected def columns: Seq[Node] = {
     <th>Batch Time</th>
@@ -53,7 +53,7 @@ private[ui] class BatchTableBase {
   }
 
   private def batchTable: Seq[Node] = {
-    <table class="table table-bordered table-striped table-condensed sortable">
+    <table id={tableId} class="table table-bordered table-striped table-condensed sortable">
       <thead>
         {columns}
       </thead>
@@ -74,7 +74,7 @@ private[ui] class BatchTableBase {
 }
 
 private[ui] class ActiveBatchTable(runningBatches: Seq[BatchInfo], waitingBatches: Seq[BatchInfo])
-  extends BatchTableBase {
+  extends BatchTableBase("active-batches-table") {
 
   override protected def columns: Seq[Node] = super.columns ++ <th>Status</th>
 
@@ -94,7 +94,8 @@ private[ui] class ActiveBatchTable(runningBatches: Seq[BatchInfo], waitingBatche
   }
 }
 
-private[ui] class CompletedBatchTable(batches: Seq[BatchInfo]) extends BatchTableBase {
+private[ui] class CompletedBatchTable(batches: Seq[BatchInfo])
+  extends BatchTableBase("completed-batches-table") {
 
   override protected def columns: Seq[Node] = super.columns ++ <th>Total Delay</th>
 

streaming/src/test/scala/org/apache/spark/streaming/UISeleniumSuite.scala

@@ -75,6 +75,17 @@ class UISeleniumSuite
         val statisticText = findAll(cssSelector("li strong")).map(_.text).toSeq
         statisticText should contain("Network receivers:")
         statisticText should contain("Batch interval:")
+
+        val h4Text = findAll(cssSelector("h4")).map(_.text).toSeq
+        h4Text should contain("Active Batches (0)")
+        h4Text should contain("Completed Batches (0)")
+
+        findAll(cssSelector("""#active-batches-table th""")).map(_.text).toSeq should be {
+          List("Batch Time", "Input Size", "Scheduling Delay", "Processing Time", "Status")
+        }
+        findAll(cssSelector("""#completed-batches-table th""")).map(_.text).toSeq should be {
+          List("Batch Time", "Input Size", "Scheduling Delay", "Processing Time", "Total Delay")
+        }
       }
 
       ssc.stop(false)