MPP: Heap test for JS

Author: elizarov

core/kotlinx-coroutines-core/src/test/kotlin/kotlinx/coroutines/experimental/internal/ThreadSafeHeapTest.kt

@@ -16,10 +16,7 @@
 
 package kotlinx.coroutines.experimental.internal
 
-import org.hamcrest.core.IsEqual
-import org.hamcrest.core.IsNull
-import org.junit.Assert.assertThat
-import org.junit.Test
+import kotlin.test.*
 import java.util.*
 
 class ThreadSafeHeapTest {
@@ -34,33 +31,33 @@ class ThreadSafeHeapTest {
     @Test
     fun testBasic() {
         val h = ThreadSafeHeap<Node>()
-        assertThat(h.peek(), IsNull())
+        assertEquals(null, h.peek())
         val n1 = Node(1)
         h.addLast(n1)
-        assertThat(h.peek(), IsEqual(n1))
+        assertEquals(n1, h.peek())
         val n2 = Node(2)
         h.addLast(n2)
-        assertThat(h.peek(), IsEqual(n1))
+        assertEquals(n1, h.peek())
         val n3 = Node(3)
         h.addLast(n3)
-        assertThat(h.peek(), IsEqual(n1))
+        assertEquals(n1, h.peek())
         val n4 = Node(4)
         h.addLast(n4)
-        assertThat(h.peek(), IsEqual(n1))
+        assertEquals(n1, h.peek())
         val n5 = Node(5)
         h.addLast(n5)
-        assertThat(h.peek(), IsEqual(n1))
-        assertThat(h.removeFirstOrNull(), IsEqual(n1))
-        assertThat(n1.index, IsEqual(-1))
-        assertThat(h.peek(), IsEqual(n2))
+        assertEquals(n1, h.peek())
+        assertEquals(n1, h.removeFirstOrNull())
+        assertEquals(-1, n1.index)
+        assertEquals(n2, h.peek())
         h.remove(n2)
-        assertThat(h.peek(), IsEqual(n3))
+        assertEquals(n3, h.peek())
         h.remove(n4)
-        assertThat(h.peek(), IsEqual(n3))
+        assertEquals(n3, h.peek())
         h.remove(n3)
-        assertThat(h.peek(), IsEqual(n5))
+        assertEquals(n5, h.peek())
         h.remove(n5)
-        assertThat(h.peek(), IsNull())
+        assertEquals(null, h.peek())
     }
 
     @Test
@@ -71,7 +68,7 @@ class ThreadSafeHeapTest {
         val a = IntArray(n) { r.nextInt() }
         repeat(n) { h.addLast(Node(a[it])) }
         a.sort()
-        repeat(n) { assertThat(h.removeFirstOrNull(), IsEqual(Node(a[it]))) }
-        assertThat(h.peek(), IsNull())
+        repeat(n) { assertEquals(Node(a[it]), h.removeFirstOrNull()) }
+        assertEquals(null, h.peek())
     }
 }

js/kotlinx-coroutines-core-js/src/test/kotlin/kotlinx/coroutines/experimental/internal/HeapTest.kt

@@ -0,0 +1,57 @@
+package kotlinx.coroutines.experimental.internal
+
+import kotlin.js.Math
+import kotlin.test.*
+
+class HeapTest {
+    class Node(val value: Int) : HeapNode, Comparable<Node> {
+        override var index = -1
+        override fun compareTo(other: Node): Int = value.compareTo(other.value)
+        override fun equals(other: Any?): Boolean = other is Node && other.value == value
+        override fun hashCode(): Int = value
+        override fun toString(): String = "$value"
+    }
+
+    @Test
+    fun testBasic() {
+        val h = Heap<Node>()
+        assertEquals(null, h.peek())
+        val n1 = Node(1)
+        h.addLast(n1)
+        assertEquals(n1, h.peek())
+        val n2 = Node(2)
+        h.addLast(n2)
+        assertEquals(n1, h.peek())
+        val n3 = Node(3)
+        h.addLast(n3)
+        assertEquals(n1, h.peek())
+        val n4 = Node(4)
+        h.addLast(n4)
+        assertEquals(n1, h.peek())
+        val n5 = Node(5)
+        h.addLast(n5)
+        assertEquals(n1, h.peek())
+        assertEquals(n1, h.removeFirstOrNull())
+        assertEquals(-1, n1.index)
+        assertEquals(n2, h.peek())
+        h.remove(n2)
+        assertEquals(n3, h.peek())
+        h.remove(n4)
+        assertEquals(n3, h.peek())
+        h.remove(n3)
+        assertEquals(n5, h.peek())
+        h.remove(n5)
+        assertEquals(null, h.peek())
+    }
+
+    @Test
+    fun testRandomSort() {
+        val n = 1000
+        val h = Heap<Node>()
+        val a = IntArray(n) { (Math.random() * 10000).toInt() }
+        repeat(n) { h.addLast(Node(a[it])) }
+        a.sort()
+        repeat(n) { assertEquals(Node(a[it]), h.removeFirstOrNull()) }
+        assertEquals(null, h.peek())
+    }
+}