Debugging parallel hash tables.

No review.

1 files changed, 23 insertions, 1 deletions

diff --git a/test/files/scalacheck/parallel-collections/ParallelHashMapCheck.scala b/test/files/scalacheck/parallel-collections/ParallelHashMapCheck.scala
index 5278d28f0f..d53c0ba9d6 100644
--- a/test/files/scalacheck/parallel-collections/ParallelHashMapCheck.scala
+++ b/test/files/scalacheck/parallel-collections/ParallelHashMapCheck.scala
@@ -58,10 +58,32 @@ with PairValues[Int, Int]
 
   override def printDataStructureDebugInfo(ds: AnyRef) = ds match {
     case pm: ParHashMap[k, v] =>
-      println("Mutable parallel hash map")
+      println("Mutable parallel hash map\n" + pm.hashTableContents.debugInformation)
     case _ =>
       println("could not match data structure type: " + ds.getClass)
   }
+
+  override def checkDataStructureInvariants(orig: Traversable[(Int, Int)], ds: AnyRef) = ds match {
+    case pm: ParHashMap[k, v] =>
+      val invs = pm.brokenInvariants
+
+      val containsall = (for ((k, v) <- orig) yield {
+        if (pm.asInstanceOf[ParHashMap[Int, Int]].get(k) == Some(v)) true
+        else {
+          println("Does not contain original element: " + (k, v))
+          false
+        }
+      }).foldLeft(true)(_ && _)
+
+
+      if (invs.isEmpty) containsall
+      else {
+        println("Invariants broken:\n" + invs.mkString("\n"))
+        false
+      }
+    case _ => true
+  }
+
 }