From 35cb21c40927bf23990eb3c5a4018eefa6d9a987 Mon Sep 17 00:00:00 2001
From: Edoardo Luppi <lp.edoardo@gmail.com>
Date: Mon, 26 Aug 2024 10:31:33 +0200
Subject: [PATCH] test: verify different equal objects as keys

---
 .../antlrkotlin/runtime/IdentityHashMapTests.kt  | 16 ++++++++++++++++
 1 file changed, 16 insertions(+)

diff --git a/antlr-kotlin-runtime/src/jsTest/kotlin/com/strumenta/antlrkotlin/runtime/IdentityHashMapTests.kt b/antlr-kotlin-runtime/src/jsTest/kotlin/com/strumenta/antlrkotlin/runtime/IdentityHashMapTests.kt
index 8dccf1b7..12751c21 100644
--- a/antlr-kotlin-runtime/src/jsTest/kotlin/com/strumenta/antlrkotlin/runtime/IdentityHashMapTests.kt
+++ b/antlr-kotlin-runtime/src/jsTest/kotlin/com/strumenta/antlrkotlin/runtime/IdentityHashMapTests.kt
@@ -33,6 +33,22 @@ class IdentityHashMapTests {
 
     prev = imap.remove(a)
     assertEquals("two", prev)
+    assertTrue(imap.isEmpty())
+
+    val one = A("same")
+    val two = A("same")
+    val three = B("same")
+    assertEquals(one, two)
+    assertEquals<Any>(two, three)
+
+    imap[one] = "one"
+    imap[two] = "two"
+    imap[three] = "three"
+
+    assertEquals(3, imap.size)
+    assertEquals("one", imap[one])
+    assertEquals("two", imap[two])
+    assertEquals("three", imap[three])
   }
 
   @Test