fix: revert change to removeAllKeys

oleg-py · May 4, 2019 · bb472bd · bb472bd
1 parent 6736cd7
commit bb472bd
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/shared/src/main/scala/com/olegpy/stm/internal/Monitor.scala b/shared/src/main/scala/com/olegpy/stm/internal/Monitor.scala
@@ -39,7 +39,13 @@ class Monitor private[stm] () {
 
     def removeAllKeys(): Unit = {
       val j = store.current()
-      j.read(this).asInstanceOf[Set[AnyRef @unchecked]].foreach(removeFromSet(_, this))
+      j.read(this) match {
+        case set: Set[AnyRef @unchecked] => set.foreach(removeFromSet(_, this))
+        // This might not be hit in a single test run, avoid fluctuating coverage
+        // $COVERAGE-OFF$
+        case _ =>
+        // $COVERAGE-ON$
+      }
       j.update(this, null)
     }
   }