From 66359894ee920aae7b9b12bdeaf4446d877ecc65 Mon Sep 17 00:00:00 2001
From: Andrey Breslav <andrey.breslav@jetbrains.com>
Date: Thu, 27 Sep 2012 14:47:01 +0400
Subject: [PATCH] Additional tests for new behavior of getOrElse() and
 getOrPut()

---
 libraries/stdlib/test/MapTest.kt | 8 ++++++++
 1 file changed, 8 insertions(+)

diff --git a/libraries/stdlib/test/MapTest.kt b/libraries/stdlib/test/MapTest.kt
index f0ea77fdf6e..19a9fbbb998 100644
--- a/libraries/stdlib/test/MapTest.kt
+++ b/libraries/stdlib/test/MapTest.kt
@@ -15,6 +15,10 @@ class MapTest {
         val b = data.getOrElse("foo"){3}
         assertEquals(3, b)
         assertEquals(0, data.size())
+
+        val empty = HashMap<String, Int?>()
+        val c = empty.getOrElse("") {null}
+        assertEquals(null, c)
     }
 
     test fun getOrPut() {
@@ -26,6 +30,10 @@ class MapTest {
         assertEquals(2, b)
 
         assertEquals(1, data.size())
+
+        val empty = HashMap<String, Int?>()
+        val c = empty.getOrPut("") {null}
+        assertEquals(null, c)
     }
 
     test fun sizeAndEmpty() {