kt343 fixed (Exceptions while function literal ends with no unit and 'Unit' is expected)

idea/testData/codegen/regressions/kt343.jet

@@ -0,0 +1,16 @@
+import java.util.ArrayList
+
+fun box(): String {
+    val list = ArrayList<Int>()
+    val foo : fun() : Unit = {
+        list.add(2)  //first exception
+    }
+    foo()
+
+    val bar = {
+        val x = 1   //second exception
+    }
+    bar()
+
+    return if (list.get(0) == 2) "OK" else "fail"
+}

idea/tests/org/jetbrains/jet/codegen/ArrayGenTest.java

@@ -58,4 +58,46 @@ public class ArrayGenTest extends CodegenTestCase {
         System.out.println(invoke.getClass());
         assertTrue(invoke instanceof Integer);
     }
+
+    public void testIterator () throws Exception {
+        loadText("fun box() { val x = Array<Int>(5, { it } ).iterator(); while(x.hasNext()) { java.lang.System.out?.println(x.next()) } }");
+        System.out.println(generateToText());
+        Method foo = generateFunction();
+        foo.invoke(null);
+    }
+
+    public void testPrimitiveIterator () throws Exception {
+        loadText("fun box() { val x = ByteArray(5).iterator(); while(x.hasNext()) { java.lang.System.out?.println(x.next()) } }");
+        System.out.println(generateToText());
+        Method foo = generateFunction();
+        foo.invoke(null);
+    }
+
+    public void testLongIterator () throws Exception {
+        loadText("fun box() { val x = LongArray(5).iterator(); while(x.hasNext()) { java.lang.System.out?.println(x.next()) } }");
+        System.out.println(generateToText());
+        Method foo = generateFunction();
+        foo.invoke(null);
+    }
+
+    public void testCharIterator () throws Exception {
+        loadText("fun box() { val x = CharArray(5).iterator(); while(x.hasNext()) { java.lang.System.out?.println(x.next()) } }");
+        System.out.println(generateToText());
+        Method foo = generateFunction();
+        foo.invoke(null);
+    }
+
+    public void testArrayIndices () throws Exception {
+        loadText("fun box() { val x = Array<Int>(5, {it}).indices.iterator(); while(x.hasNext()) { java.lang.System.out?.println(x.next()) } }");
+        System.out.println(generateToText());
+        Method foo = generateFunction();
+        foo.invoke(null);
+    }
+
+    public void testCharIndices () throws Exception {
+        loadText("fun box() { val x = CharArray(5).indices.iterator(); while(x.hasNext()) { java.lang.System.out?.println(x.next()) } }");
+        System.out.println(generateToText());
+        Method foo = generateFunction();
+        foo.invoke(null);
+    }
 }

idea/tests/org/jetbrains/jet/codegen/ClassGenTest.java

@@ -185,4 +185,9 @@ public class ClassGenTest extends CodegenTestCase {
         assertEquals(method.getReturnType().getName(), "java.lang.Object");
         System.out.println(generateToText());
     }
+
+    public void testKt343 () throws Exception {
+        blackBoxFile("regressions/kt343.jet");
+        System.out.println(generateToText());
+    }
 }