From bff6c721074bda50e2d5a23d8aa96bf1f939f62a Mon Sep 17 00:00:00 2001
From: Robert Jaros <rjaros@finn.pl>
Date: Thu, 27 Feb 2020 09:22:19 +0100
Subject: Fix tests

---
 .../kotlin/test/pl/treksoft/kvision/core/WidgetSpec.kt  | 17 +++++++++++++++--
 1 file changed, 15 insertions(+), 2 deletions(-)

(limited to 'src/test/kotlin')

diff --git a/src/test/kotlin/test/pl/treksoft/kvision/core/WidgetSpec.kt b/src/test/kotlin/test/pl/treksoft/kvision/core/WidgetSpec.kt
index 42a07ba8..159df178 100644
--- a/src/test/kotlin/test/pl/treksoft/kvision/core/WidgetSpec.kt
+++ b/src/test/kotlin/test/pl/treksoft/kvision/core/WidgetSpec.kt
@@ -96,16 +96,29 @@ class WidgetSpec : WSpec {
     fun setEventListener() {
         runW { widget, _ ->
             widget.setEventListener<Widget> { click = { } }
-            assertTrue("Element should have an event listener") { widget.listenersMap.size == 1 }
+            assertTrue("Element should have one event listener") { widget.listenersMap["click"]?.size == 1 }
+            widget.setEventListener<Widget> { click = { } }
+            assertTrue("Element should have two event listeners") { widget.listenersMap["click"]?.size == 2 }
         }
     }
 
     @Test
     fun removeEventListener() {
         runW { widget, _ ->
+            val id = widget.setEventListener<Widget> { click = { } }
+            widget.setEventListener<Widget> { click = { } }
+            widget.removeEventListener(id)
+            assertTrue("One of event listeners is removed") { widget.listenersMap["click"]?.size == 1 }
+        }
+    }
+
+    @Test
+    fun removeEventListeners() {
+        runW { widget, _ ->
+            widget.setEventListener<Widget> { click = { } }
             widget.setEventListener<Widget> { click = { } }
             widget.removeEventListeners()
-            assertTrue("Element should not have any event listener") { widget.listenersMap.size == 0 }
+            assertTrue("Element should not have any event listeners") { widget.listenersMap.isEmpty() }
         }
     }
 
-- 
cgit