From 972ebd5fec2c4e426fa2cb9b5b50b9422f0049a6 Mon Sep 17 00:00:00 2001
From: Dmitry Jemerov <yole@jetbrains.com>
Date: Wed, 6 Jul 2016 19:22:26 +0200
Subject: [PATCH] check that main() returns Unit  #KT-12760 Fixed

---
 .../src/org/jetbrains/kotlin/idea/MainFunctionDetector.kt      | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/compiler/frontend/src/org/jetbrains/kotlin/idea/MainFunctionDetector.kt b/compiler/frontend/src/org/jetbrains/kotlin/idea/MainFunctionDetector.kt
index c9e60295128..df899dab8a0 100644
--- a/compiler/frontend/src/org/jetbrains/kotlin/idea/MainFunctionDetector.kt
+++ b/compiler/frontend/src/org/jetbrains/kotlin/idea/MainFunctionDetector.kt
@@ -101,6 +101,9 @@ class MainFunctionDetector {
                 return false
             }
 
+            val returnType = descriptor.returnType
+            if (returnType == null ||  !KotlinBuiltIns.isUnit(returnType)) return false
+
             if (DescriptorUtils.isTopLevelDeclaration(descriptor)) return true
 
             val containingDeclaration = descriptor.containingDeclaration